remotivelabs-cli 0.2.0a2__tar.gz → 0.2.2__tar.gz

{remotivelabs_cli-0.2.0a2 → remotivelabs_cli-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: remotivelabs-cli
-Version: 0.2.0a2
+Version: 0.2.2
 Summary: CLI for operating RemotiveCloud and RemotiveBroker
 Author: Johan Rask
 Author-email: johan.rask@remotivelabs.com
@@ -13,9 +13,11 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: click (<8.2.0)
 Requires-Dist: dacite (>=1.9.2,<2.0.0)
+Requires-Dist: email-validator (>=2.2.0,<3.0.0)
 Requires-Dist: grpc-stubs (>=1.53.0.5)
 Requires-Dist: mypy-protobuf (>=3.0.0)
 Requires-Dist: plotext (>=5.2,<6.0)
+Requires-Dist: pydantic (>=2.11.7,<3.0.0)
 Requires-Dist: pyjwt (>=2.6,<3.0)
 Requires-Dist: python-can (>=4.3.1)
 Requires-Dist: python-socketio (>=4.6.1)

{remotivelabs_cli-0.2.0a2 → remotivelabs_cli-0.2.2}/cli/cloud/auth/cmd.py

@@ -27,9 +27,6 @@ def login(browser: bool = typer.Option(default=True, help="Does not automaticall
     If not able to open a browser it will show fallback to headless login and show a link that
     users can copy into any browser when this is unsupported where running the cli - such as in docker,
     virtual machine or ssh sessions.
-
-    This will be used as the current access token in all subsequent requests. This would
-    be the same as activating a personal access key or service-account access key.
     """
     do_login(headless=not browser)
 

{remotivelabs_cli-0.2.0a2 → remotivelabs_cli-0.2.2}/cli/cloud/auth_tokens.py

@@ -29,10 +29,6 @@ def _prompt_choice(  # noqa: C901, PLR0912
     info_message: Optional[str] = None,
 ) -> Optional[TokenFile]:
     accounts = settings.get_cli_config().accounts
-    try:
-        active_account = settings.get_cli_config().get_active()
-    except TokenNotFoundError:
-        active_account = None
 
     table = Table("#", "Active", "Type", "Token", "Account", "Created", "Expires")
 
@@ -56,19 +52,27 @@ def _prompt_choice(  # noqa: C901, PLR0912
 
     included_tokens.sort(key=lambda token: token.created, reverse=True)
 
-    def get_token_or_none(account: Optional[Account]) -> Optional[TokenFile]:
-        if account is None:
-            return None
+    def get_active_account_or_none() -> Optional[Account]:
         try:
-            return settings.get_token_file(account.credentials_file)
+            return settings.get_cli_config().get_active()
         except TokenNotFoundError:
             return None
 
+    def get_active_token_or_none() -> Optional[TokenFile]:
+        try:
+            active_account = get_active_account_or_none()
+            if active_account is not None:
+                return settings.get_token_file(active_account.credentials_file)
+        except TokenNotFoundError:
+            pass
+        return None
+
+    active_token = get_active_token_or_none()
     active_token_index = None
     for idx, choice in enumerate(included_tokens, start=1):
-        active_token = get_token_or_none(active_account)
-        is_active = active_account is not None and active_token is not None and active_token.name == choice.name
+        is_active = active_token is not None and active_token.name == choice.name
         active_token_index = idx if is_active else active_token_index
+
         table.add_row(
             f"[yellow]{idx}",
             ":white_check_mark:" if is_active else "",
@@ -78,7 +82,6 @@ def _prompt_choice(  # noqa: C901, PLR0912
             str(choice.created),
             str(choice.expires),
         )
-    # console.print("It seems like you have access tokens from previous login, you can select one of these instead of logging in")
     console.print(table)
 
     if skip_prompt:
@@ -181,9 +184,7 @@ def select_personal_token(
     do_activate(token_name)
 
 
-def do_activate(
-    token_name: Optional[str],
-) -> Optional[TokenFile]:
+def do_activate(token_name: Optional[str]) -> Optional[TokenFile]:
     if token_name is not None:
         try:
             token_file = settings.get_token_file(token_name)
@@ -224,8 +225,6 @@ def list_and_select_personal_token(
         sa_tokens = settings.list_service_account_tokens()
         personal_tokens.extend(sa_tokens)
 
-    # merged = _merge_local_tokens_with_cloud(personal_tokens)
-
     selected_token = _prompt_choice(personal_tokens, skip_prompt=skip_prompt, info_message=info_message)
     if selected_token is not None:
         settings.activate_token(selected_token)

{remotivelabs_cli-0.2.0a2 → remotivelabs_cli-0.2.2}/cli/cloud/organisations.py

@@ -76,7 +76,7 @@ def do_select_default_org(organisation_uid: Optional[str] = None, get: bool = Fa
     """
     if get:
         default_organisation = settings.get_cli_config().get_active_default_organisation()
-        if default_organisation is not None:
+        if default_organisation:
             console.print(default_organisation)
         else:
             console.print("No default organization set")
@@ -84,7 +84,7 @@ def do_select_default_org(organisation_uid: Optional[str] = None, get: bool = Fa
         settings.set_default_organisation(organisation_uid)
     else:
         account = settings.get_cli_config().get_active()
-        if account is not None:
+        if account:
             token = settings.get_token_file(account.credentials_file)
             if token.type != "authorized_user":
                 ErrorPrinter.print_hint(

{remotivelabs_cli-0.2.0a2 → remotivelabs_cli-0.2.2}/cli/cloud/recordings.py

@@ -137,14 +137,13 @@ def mount(  # noqa: C901
         r = Rest.handle_get(url=f"/api/project/{project}/brokers/personal", return_response=True, allow_status_codes=[404])
 
         if r.status_code == 200:
-            broker = r.json()["shortName"]
-
+            broker_info = r.json()
+            broker = broker_info["shortName"]
         elif r.status_code == 404:
             r = do_start("personal", project, "", return_response=True)
             if r.status_code != 200:
                 print(r.text)
                 sys.exit(0)
-            broker = r.json()["shortName"]
         else:
             sys.stderr.write(f"Got http status code {r.status_code}")
             raise typer.Exit(0)
@@ -154,7 +153,6 @@ def mount(  # noqa: C901
         if r.status_code == 404:
             if ensure_broker_started:
                 r = do_start(broker, project, "", return_response=True)
-
                 if r.status_code != 200:
                     print(r.text)
                     sys.exit(1)
@@ -164,6 +162,9 @@ def mount(  # noqa: C901
         elif r.status_code != 200:
             sys.stderr.write(f"Got http status code {r.status_code}")
             raise typer.Exit(1)
+
+    broker_info = r.json()
+    broker = broker_info["shortName"]
     broker_config_query = ""
     if transformation_name != "default":
         broker_config_query = f"?brokerConfigName={transformation_name}"
@@ -174,7 +175,8 @@ def mount(  # noqa: C901
         return_response=True,
         progress_label="Preparing recording on broker...",
     )
-    print("Successfully mounted recording on broker")
+    err_console.print("Successfully mounted recording on broker")
+    print(json.dumps(broker_info))
 
 
 @app.command(help="Downloads the specified recording file to disk")

{remotivelabs_cli-0.2.0a2 → remotivelabs_cli-0.2.2}/cli/remotive.py

@@ -6,20 +6,30 @@ from importlib.metadata import version
 import typer
 from rich import print as rich_print
 from rich.console import Console
-from trogon import Trogon  # type: ignore
+from trogon import Trogon
 from typer.main import get_group
 
 from cli.broker.brokers import app as broker_app
 from cli.cloud.cloud_cli import app as cloud_app
 from cli.connect.connect import app as connect_app
 from cli.settings import settings
-from cli.settings.migrate_all_token_files import migrate_any_legacy_tokens
+from cli.settings.core import Settings
+from cli.settings.migration.migrate_all_token_files import migrate_any_legacy_tokens
+from cli.settings.migration.migrate_config_file import migrate_config_file
+from cli.settings.migration.migrate_legacy_dirs import migrate_legacy_settings_dirs
 from cli.tools.tools import app as tools_app
 from cli.topology.cmd import app as topology_app
 from cli.typer import typer_utils
+from cli.utils.version_check import check_for_update
 
 err_console = Console(stderr=True)
 
+
+def is_featue_flag_enabled(env_var: str) -> bool:
+    """Check if an environment variable indicates a feature is enabled."""
+    return os.getenv(env_var, "").lower() in ("true", "1", "yes", "on")
+
+
 if os.getenv("GRPC_VERBOSITY") is None:
     os.environ["GRPC_VERBOSITY"] = "NONE"
 
@@ -32,14 +42,11 @@ For documentation - https://docs.remotivelabs.com
 """,
 )
 
-# settings.set_default_config_as_env()
-
 
 def version_callback(value: bool) -> None:
     if value:
         my_version = version("remotivelabs-cli")
         typer.echo(my_version)
-        raise typer.Exit()
 
 
 def test_callback(value: int) -> None:
@@ -48,14 +55,30 @@ def test_callback(value: int) -> None:
         raise typer.Exit()
 
 
-def _migrate_old_tokens() -> None:
-    tokens = settings.list_personal_tokens()
-    tokens.extend(settings.list_service_account_tokens())
-    if migrate_any_legacy_tokens(tokens):
+def check_for_newer_version(settings: Settings) -> None:
+    check_for_update("remotivelabs-cli", version("remotivelabs-cli"), settings)
+
+
+def run_migrations(settings: Settings) -> None:
+    """
+    Run all migration scripts.
+
+    Each migration script is responsible for a particular migration, and order matters.
+    """
+    # 1. Migrate legacy settings dirs
+    migrate_legacy_settings_dirs(settings.config_dir)
+
+    # 2. Migrate any legacy tokens
+    has_migrated_tokens = migrate_any_legacy_tokens(settings)
+
+    # 3. Migrate legacy config file format
+    migrate_config_file(settings.config_file_path, settings)
+
+    if has_migrated_tokens:
         err_console.print("Migrated old credentials and configuration files, you may need to login again or activate correct credentials")
 
 
-def _set_default_org_as_env() -> None:
+def set_default_org_as_env(settings: Settings) -> None:
     """
     If not already set, take the default organisation from file and set as env
     This has to be done early before it is read
@@ -68,10 +91,17 @@ def _set_default_org_as_env() -> None:
 
 @app.callback()
 def main(
-    _the_version: bool = typer.Option(None, "--version", callback=version_callback, is_eager=False, help="Print current version"),
+    _the_version: bool = typer.Option(
+        None,
+        "--version",
+        callback=version_callback,
+        is_eager=True,
+        help="Print current version",
+    ),
 ) -> None:
-    _set_default_org_as_env()
-    _migrate_old_tokens()
+    run_migrations(settings)
+    check_for_newer_version(settings)
+    set_default_org_as_env(settings)
     # Do other global stuff, handle other global options here
 
 
@@ -90,14 +120,16 @@ app.add_typer(
     name="cloud",
     help="Manage resources in RemotiveCloud",
 )
-app.add_typer(
-    topology_app,
-    name="topology",
-    help="""
-RemotiveTopology actions
-
-Read more at https://docs.remotivelabs.com/docs/remotive-topology
-""",
-)
 app.add_typer(connect_app, name="connect", help="Integrations with other systems")
 app.add_typer(tools_app, name="tools")
+
+if is_featue_flag_enabled("REMOTIVE_TOPOLOGY_ENABLED"):
+    app.add_typer(
+        topology_app,
+        name="topology",
+        help="""
+    RemotiveTopology actions
+
+    Read more at https://docs.remotivelabs.com/docs/remotive-topology
+    """,
+    )

remotivelabs_cli-0.2.2/cli/settings/__init__.py

@@ -0,0 +1,21 @@
+from cli.settings.config_file import Account, ConfigFile
+from cli.settings.config_file import dumps as dumps_config_file
+from cli.settings.config_file import loads as loads_config_file
+from cli.settings.core import InvalidSettingsFilePathError, Settings, TokenNotFoundError, settings
+from cli.settings.token_file import TokenFile
+from cli.settings.token_file import dumps as dumps_token_file
+from cli.settings.token_file import loads as loads_token_file
+
+__all__ = [
+    "settings",
+    "TokenNotFoundError",
+    "InvalidSettingsFilePathError",
+    "Settings",
+    "TokenFile",
+    "ConfigFile",
+    "Account",
+    "dumps_config_file",
+    "loads_config_file",
+    "dumps_token_file",
+    "loads_token_file",
+]

remotivelabs_cli-0.2.2/cli/settings/config_file.py

@@ -0,0 +1,93 @@
+from __future__ import annotations
+
+import dataclasses
+import json
+from dataclasses import dataclass
+from typing import Any, Optional
+
+from dacite import from_dict
+
+from cli.settings.token_file import TokenFile
+
+
+def loads(data: str) -> ConfigFile:
+    d = json.loads(data)
+    return from_dict(ConfigFile, d)
+
+
+def dumps(config: ConfigFile) -> str:
+    return json.dumps(dataclasses.asdict(config), default=str)
+
+
+@dataclass
+class Account:
+    credentials_file: str
+    default_organization: Optional[str] = None
+    # Add project as well
+
+
+@dataclass
+class ConfigFile:
+    version: str = "1.0"
+    active: Optional[str] = None
+    accounts: dict[str, Account] = dataclasses.field(default_factory=dict)
+
+    def get_active_default_organisation(self) -> Optional[str]:
+        active_account = self.get_active()
+        return active_account.default_organization if active_account else None
+
+    def get_active(self) -> Optional[Account]:
+        if not self.active:
+            return None
+        account = self.get_account(self.active)
+        if not account:
+            raise KeyError(f"Activated account {self.active} is not a valid account")
+        return account
+
+    def activate(self, email: str) -> None:
+        account = self.get_account(email)
+        if not account:
+            raise KeyError(f"Account {email} does not exists")
+        self.active = email
+
+    def get_account(self, email: str) -> Optional[Account]:
+        if not self.accounts:
+            return None
+        return self.accounts.get(email, None)
+
+    def remove_account(self, email: str) -> None:
+        if self.accounts:
+            self.accounts.pop(email, None)
+
+    def init_account(self, email: str, token_file: TokenFile) -> None:
+        if self.accounts is None:
+            self.accounts = {}
+
+        account = self.get_account(email)
+        if not account:
+            account = Account(credentials_file=token_file.get_token_file_name())
+        else:
+            account.credentials_file = token_file.get_token_file_name()
+        self.accounts[email] = account
+
+    def set_account_field(self, email: str, default_organization: Optional[str] = None) -> ConfigFile:
+        if self.accounts is None:
+            self.accounts = {}
+
+        account = self.get_account(email)
+        if not account:
+            raise KeyError(f"Account with email {email} has not been initialized with token")
+
+        # Update only fields explicitly passed
+        if default_organization is not None:
+            account.default_organization = default_organization
+
+        return self
+
+    @staticmethod
+    def from_json_str(data: str) -> ConfigFile:
+        return loads(data)
+
+    @staticmethod
+    def from_dict(data: dict[str, Any]) -> ConfigFile:
+        return from_dict(ConfigFile, data)