CM-26891 - Add cycode status command

Author: MarshalX

cycode/cli/commands/auth/auth_command.py

@@ -1,19 +1,15 @@
 import click
 
 from cycode.cli.commands.auth.auth_manager import AuthManager
+from cycode.cli.commands.auth_common import get_authorization_info
 from cycode.cli.exceptions.custom_exceptions import (
     KNOWN_USER_FRIENDLY_REQUEST_ERRORS,
     AuthProcessError,
-    HttpUnauthorizedError,
-    RequestHttpError,
 )
 from cycode.cli.models import CliError, CliErrors, CliResult
 from cycode.cli.printers import ConsolePrinter
 from cycode.cli.sentry import add_breadcrumb, capture_exception
-from cycode.cli.user_settings.credentials_manager import CredentialsManager
-from cycode.cli.utils.jwt_utils import get_user_and_tenant_ids_from_access_token
 from cycode.cyclient import logger
-from cycode.cyclient.cycode_token_based_client import CycodeTokenBasedClient
 
 
 @click.group(
@@ -49,35 +45,18 @@ def authorization_check(context: click.Context) -> None:
     add_breadcrumb('check')
 
     printer = ConsolePrinter(context)
-
-    failed_auth_check_res = CliResult(success=False, message='Cycode authentication failed')
-
-    client_id, client_secret = CredentialsManager().get_credentials()
-    if not client_id or not client_secret:
-        printer.print_result(failed_auth_check_res)
+    auth_info = get_authorization_info(context)
+    if auth_info is None:
+        printer.print_result(CliResult(success=False, message='Cycode authentication failed'))
         return
 
-    try:
-        access_token = CycodeTokenBasedClient(client_id, client_secret).get_access_token()
-        if not access_token:
-            printer.print_result(failed_auth_check_res)
-            return
-
-        user_id, tenant_id = get_user_and_tenant_ids_from_access_token(access_token)
-        printer.print_result(
-            CliResult(
-                success=True,
-                message='Cycode authentication verified',
-                data={'user_id': user_id, 'tenant_id': tenant_id},
-            )
+    printer.print_result(
+        CliResult(
+            success=True,
+            message='Cycode authentication verified',
+            data={'user_id': auth_info.user_id, 'tenant_id': auth_info.tenant_id},
         )
-
-        return
-    except (RequestHttpError, HttpUnauthorizedError):
-        ConsolePrinter(context).print_exception()
-
-        printer.print_result(failed_auth_check_res)
-        return
+    )
 
 
 def _handle_exception(context: click.Context, e: Exception) -> None:

cycode/cli/commands/auth_common.py

@@ -0,0 +1,33 @@
+from typing import NamedTuple, Optional
+
+import click
+
+from cycode.cli.exceptions.custom_exceptions import HttpUnauthorizedError, RequestHttpError
+from cycode.cli.printers import ConsolePrinter
+from cycode.cli.user_settings.credentials_manager import CredentialsManager
+from cycode.cli.utils.jwt_utils import get_user_and_tenant_ids_from_access_token
+from cycode.cyclient.cycode_token_based_client import CycodeTokenBasedClient
+
+
+class AuthInfo(NamedTuple):
+    user_id: str
+    tenant_id: str
+
+
+def get_authorization_info(context: Optional[click.Context] = None) -> Optional[AuthInfo]:
+    client_id, client_secret = CredentialsManager().get_credentials()
+    if not client_id or not client_secret:
+        return None
+
+    try:
+        access_token = CycodeTokenBasedClient(client_id, client_secret).get_access_token()
+        if not access_token:
+            return None
+
+        user_id, tenant_id = get_user_and_tenant_ids_from_access_token(access_token)
+        return AuthInfo(user_id=user_id, tenant_id=tenant_id)
+    except (RequestHttpError, HttpUnauthorizedError):
+        if context:
+            ConsolePrinter(context).print_exception()
+
+        return None

cycode/cli/commands/main_cli.py

@@ -8,6 +8,7 @@
 from cycode.cli.commands.ignore.ignore_command import ignore_command
 from cycode.cli.commands.report.report_command import report_command
 from cycode.cli.commands.scan.scan_command import scan_command
+from cycode.cli.commands.status.status_command import status_command
 from cycode.cli.commands.version.version_command import version_command
 from cycode.cli.consts import (
     CLI_CONTEXT_SETTINGS,
@@ -28,6 +29,7 @@
         'ignore': ignore_command,
         'auth': auth_command,
         'version': version_command,
+        'status': status_command,
     },
     context_settings=CLI_CONTEXT_SETTINGS,
 )

cycode/cli/commands/status/__init__.py

@@ -0,0 +1,72 @@
+import dataclasses
+import json
+import platform
+
+import click
+
+from cycode import __version__
+from cycode.cli.commands.auth_common import get_authorization_info
+from cycode.cli.consts import PROGRAM_NAME
+from cycode.cli.user_settings.configuration_manager import ConfigurationManager
+
+
+@dataclasses.dataclass
+class Status:
+    program: str
+    version: str
+    os: str
+    arch: str
+    python_version: str
+    installation_id: str
+    app_url: str
+    api_url: str
+    is_authenticated: bool
+    user_id: str = None
+    tenant_id: str = None
+
+    def as_text(self) -> str:
+        message_parts = []
+        for key, value in dataclasses.asdict(self).items():
+            human_readable_key = key.replace('_', ' ').capitalize()
+            message_parts.append(f'{human_readable_key}: {value}')
+
+        return '\n'.join(message_parts)
+
+    def as_json(self) -> str:
+        return json.dumps(dataclasses.asdict(self))
+
+
+def get_cli_status() -> Status:
+    configuration_manager = ConfigurationManager()
+    auth_info = get_authorization_info()
+
+    # TODO: Add supported modules; add AI status
+
+    return Status(
+        program=PROGRAM_NAME,
+        version=__version__,
+        os=platform.system(),
+        arch=platform.machine(),
+        python_version=platform.python_version(),
+        installation_id=configuration_manager.get_or_create_installation_id(),
+        app_url=configuration_manager.get_cycode_app_url(),
+        api_url=configuration_manager.get_cycode_api_url(),
+        is_authenticated=auth_info is not None,
+        user_id=auth_info.user_id if auth_info else None,
+        tenant_id=auth_info.tenant_id if auth_info else None,
+    )
+
+
+
+@click.command(short_help='Show the CLI status and exit.')
+@click.pass_context
+def status_command(context: click.Context) -> None:
+    output = context.obj['output']
+
+    status = get_cli_status()
+    message = status.as_text()
+    if output == 'json':
+        message = status.as_json()
+
+    click.echo(message, color=context.color)
+    context.exit()