cycode-cli/cycode/cli/apps/configure/configure_command.py at 9dea2b9a72257e27a87a3e1272465e2bc2c74aea · cycodehq/cycode-cli · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
from typing import Optional

from cycode.cli.apps.configure.consts import CONFIGURATION_MANAGER, CREDENTIALS_MANAGER
from cycode.cli.apps.configure.messages import get_credentials_update_result_message, get_urls_update_result_message
from cycode.cli.apps.configure.prompts import (
    get_api_url_input,
    get_app_url_input,
    get_client_id_input,
    get_client_secret_input,
)
from cycode.cli.console import console
from cycode.cli.utils.sentry import add_breadcrumb


def _should_update_value(
    old_value: Optional[str],
    new_value: Optional[str],
) -> bool:
    if not new_value:
        return False

    return old_value != new_value


def configure_command() -> None:
    """Configure your CLI client authentication manually."""
    add_breadcrumb('configure')

    global_config_manager = CONFIGURATION_MANAGER.global_config_file_manager

    current_api_url = global_config_manager.get_api_url()
    current_app_url = global_config_manager.get_app_url()
    api_url = get_api_url_input(current_api_url)
    app_url = get_app_url_input(current_app_url)

    config_updated = False
    if _should_update_value(current_api_url, api_url):
        global_config_manager.update_api_base_url(api_url)
        config_updated = True
    if _should_update_value(current_app_url, app_url):
        global_config_manager.update_app_base_url(app_url)
        config_updated = True

    current_client_id, current_client_secret = CREDENTIALS_MANAGER.get_credentials_from_file()
    client_id = get_client_id_input(current_client_id)
    client_secret = get_client_secret_input(current_client_secret)

    credentials_updated = False
    if _should_update_value(current_client_id, client_id) or _should_update_value(current_client_secret, client_secret):
        credentials_updated = True
        CREDENTIALS_MANAGER.update_credentials(client_id, client_secret)

    if config_updated:
        console.print(get_urls_update_result_message())
    if credentials_updated:
        console.print(get_credentials_update_result_message())