remotivelabs-cli 0.5.0a1__py3-none-any.whl

remotivelabs/cli/cloud/configs.py

@@ -0,0 +1,109 @@
+import os.path
+import shutil
+from pathlib import Path
+
+import requests
+import typer
+from rich.progress import Progress, SpinnerColumn, TextColumn
+
+from remotivelabs.cli.typer import typer_utils
+from remotivelabs.cli.utils.console import print_generic_error, print_success
+from remotivelabs.cli.utils.rest_helper import RestHelper as Rest
+
+app = typer_utils.create_typer()
+
+
+@app.command("list")
+def list_signal_databases(project: str = typer.Option(..., help="Project ID", envvar="REMOTIVE_CLOUD_PROJECT")) -> None:
+    """
+    List available signal databases in project
+    """
+    Rest.handle_get(f"/api/project/{project}/files/config")
+
+
+@app.command("delete")
+def delete(
+    signal_db_file: str = typer.Argument("", help="Signal database file"),
+    project: str = typer.Option(..., help="Project ID", envvar="REMOTIVE_CLOUD_PROJECT"),
+) -> None:
+    """
+    Deletes the specified signal database
+    """
+    Rest.handle_delete(f"/api/project/{project}/files/config/{signal_db_file}")
+
+
+@app.command()
+def upload(
+    path: Path = typer.Argument(
+        ...,
+        exists=True,
+        file_okay=True,
+        dir_okay=False,
+        writable=False,
+        readable=True,
+        resolve_path=True,
+        help="Path to signal database file to upload",
+    ),
+    project: str = typer.Option(..., help="Project ID", envvar="REMOTIVE_CLOUD_PROJECT"),
+) -> None:
+    """
+    Uploads signal database to project
+    """
+    res_text = Rest.handle_put(url=f"/api/project/{project}/files/config/{os.path.basename(path)}/uploadfile", return_response=True)
+    if res_text is not None:
+        res_json = res_text.json()
+        Rest.upload_file_with_signed_url(
+            path=path,
+            url=res_json["url"],
+            upload_headers={"Content-Type": "application/octet-stream"},
+            progress_label=f"Uploading {path}...",
+        )
+
+
+@app.command()
+def describe(
+    signal_db_file: str = typer.Argument("", help="Signal database file"),
+    project: str = typer.Option(..., help="Project ID", envvar="REMOTIVE_CLOUD_PROJECT"),
+) -> None:
+    """
+    Shows all metadata related to this signal database
+    """
+    Rest.handle_get(f"/api/project/{project}/files/config/{signal_db_file}")
+
+
+@app.command()
+def download(
+    signal_db_file: str = typer.Argument("", help="Signal database file"),
+    project: str = typer.Option(..., help="Project ID", envvar="REMOTIVE_CLOUD_PROJECT"),
+) -> None:
+    """
+    Downloads the specified signal database to disk
+    """
+    with Progress(
+        SpinnerColumn(),
+        TextColumn("[progress.description]{task.description}"),
+        transient=True,
+    ) as progress:
+        Rest.ensure_auth_token()
+
+        progress.add_task(description=f"Downloading {signal_db_file}", total=None)
+
+        # First request the download url from cloud. This is a public signed url that is valid
+        # for a short period of time
+        get_signed_url_resp = requests.get(
+            f"{Rest.get_base_url()}/api/project/{project}/files/config/{signal_db_file}/download",
+            headers=Rest.get_headers(),
+            allow_redirects=True,
+            timeout=60,
+        )
+        if get_signed_url_resp.status_code == 200:
+            # Next download the actual file
+            download_resp = requests.get(url=get_signed_url_resp.text, stream=True, timeout=60)
+            if download_resp.status_code == 200:
+                with open(signal_db_file, "wb") as out_file:
+                    shutil.copyfileobj(download_resp.raw, out_file)
+                print_success(f"{signal_db_file} downloaded")
+            else:
+                print_generic_error(f"Got unexpected status {download_resp.status_code}")
+        else:
+            print_generic_error(f"Got unexpected status {get_signed_url_resp.status_code}\n")

remotivelabs/cli/cloud/licenses/cmd.py

@@ -0,0 +1,14 @@
+import typer
+
+from remotivelabs.cli.typer import typer_utils
+from remotivelabs.cli.utils.rest_helper import RestHelper
+
+app = typer_utils.create_typer()
+
+
+@app.command(help="List licenses for an organization")
+def licenses(
+    organization: str = typer.Option(..., help="Organization ID", envvar="REMOTIVE_CLOUD_ORGANIZATION"),
+    filter_option: str = typer.Option("all", help="all, valid, expired"),
+) -> None:
+    RestHelper.handle_get(f"/api/bu/{organization}/licenses", {"filter": filter_option})

remotivelabs/cli/cloud/organisations.py

@@ -0,0 +1,112 @@
+from __future__ import annotations
+
+import json
+from dataclasses import dataclass
+from typing import List, Optional
+
+import typer
+from rich.table import Table
+
+from remotivelabs.cli.settings import settings
+from remotivelabs.cli.typer import typer_utils
+from remotivelabs.cli.utils.console import (
+    print_generic_error,
+    print_generic_message,
+    print_hint,
+    print_newline,
+    print_unformatted_to_stderr,
+)
+from remotivelabs.cli.utils.rest_helper import RestHelper
+
+app = typer_utils.create_typer()
+
+
+@dataclass
+class Organisation:
+    display_name: str
+    uid: str
+
+
+def _prompt_choice(choices: List[Organisation]) -> Optional[Organisation]:
+    table = Table("#", "Name", "Uid", "Default")
+
+    account = settings.get_active_account()
+    current_default_org = account.default_organization if account else None
+
+    for idx, choice in enumerate(choices, start=1):
+        table.add_row(
+            f"[yellow]{idx}",
+            f"[bold]{choice.display_name}[/bold]",
+            choice.uid,
+            ":thumbsup:" if current_default_org is not None and current_default_org == choice.uid else "",
+        )
+    print_unformatted_to_stderr(table)
+
+    print_newline()
+    selection = typer.prompt(f"Enter the number(# 1-{len(choices)}) of the organization to select (or q to quit)")
+
+    if selection == "q":
+        return None
+    try:
+        index = int(selection) - 1
+        if 0 <= index < len(choices):
+            return choices[index]
+        raise ValueError
+    except ValueError:
+        print_generic_error("Invalid choice, please try again")
+        return _prompt_choice(choices)
+
+
+@app.command("default")
+def select_default_org(
+    organization_uid: str = typer.Argument(None, help="Organization uid or empty to select one"),
+    get: bool = typer.Option(False, help="Print current default organization"),
+) -> None:
+    do_select_default_org(organization_uid, get)
+
+
+def do_select_default_org(organisation_uid: Optional[str] = None, get: bool = False) -> None:
+    r"""
+    Set default organization for the currently activated user, empty to choose from available organizations or organization uid as argument
+
+    remotive cloud organizations default my_org \[set specific org uid]
+    remotive cloud organizations default \[select one from prompt]
+    remotive cloud organizations default --get \[print current default]
+
+    Note that service-accounts does Not have permission to list organizations and will get a 403 Forbidden response so you must
+    select the organization uid as argument
+    """
+    active_account = settings.get_active_account()
+    if get:
+        if active_account and active_account.default_organization:
+            print_unformatted_to_stderr(active_account.default_organization)
+        else:
+            print_unformatted_to_stderr("No default organization set")
+    elif organisation_uid is not None:
+        settings.set_default_organisation(organisation_uid)
+    else:
+        if active_account:
+            token = settings.get_token_file(active_account.credentials_file)
+            if token and token.type != "authorized_user":
+                print_hint(
+                    "You must supply the organization name as argument when using a service-account since the "
+                    "service-account is not allowed to list"
+                )
+                return
+
+        r = RestHelper.handle_get("/api/bu", return_response=True)
+        orgs = r.json()
+        orgs = [Organisation(display_name=o["organisation"]["displayName"], uid=o["organisation"]["uid"]) for o in orgs]
+
+        selected = _prompt_choice(orgs)
+
+        if selected is not None:
+            typer.echo(f"Default organisation: {selected.display_name} (uid: {selected.uid})")
+            settings.set_default_organisation(selected.uid)
+
+
+@app.command(name="list", help="List your available organizations")
+def list_orgs() -> None:
+    r = RestHelper.handle_get("/api/bu", return_response=True)
+    orgs = [{"uid": org["organisation"]["uid"], "displayName": org["organisation"]["displayName"]} for org in r.json()]
+    print_generic_message(json.dumps(orgs))

remotivelabs/cli/cloud/projects.py

@@ -0,0 +1,44 @@
+import json
+
+import typer
+
+from remotivelabs.cli.typer import typer_utils
+from remotivelabs.cli.utils.console import print_generic_error, print_generic_message
+from remotivelabs.cli.utils.rest_helper import RestHelper as Rest
+
+app = typer_utils.create_typer()
+
+
+@app.command(name="list", help="List your projects")
+def list_projects(organization: str = typer.Option(..., help="Organization ID", envvar="REMOTIVE_CLOUD_ORGANIZATION")) -> None:
+    r = Rest.handle_get(url=f"/api/bu/{organization}/me", return_response=True)
+    if r is None:
+        return
+
+    if r.status_code == 200:
+        # extract the project uid parts
+        projects = r.json()["projects"]
+        projects = map(lambda p: p["uid"], projects)
+        print_generic_message(json.dumps(list(projects)))
+    else:
+        print_generic_error(f"Got unexpected status {r.status_code}\n")
+
+
+@app.command(name="create")
+def create_project(
+    project_uid: str = typer.Argument(help="Project UID"),
+    organization: str = typer.Option(..., help="Organization ID", envvar="REMOTIVE_CLOUD_ORGANIZATION"),
+    project_display_name: str = typer.Option(default="", help="Project display name"),
+) -> None:
+    create_project_req = {
+        "uid": project_uid,
+        "displayName": project_display_name if project_display_name != "" else project_uid,
+        "description": "",
+    }
+
+    Rest.handle_post(url=f"/api/bu/{organization}/project", body=json.dumps(create_project_req))
+
+
+@app.command(name="delete")
+def delete(project: str = typer.Argument(help="Project ID", envvar="REMOTIVE_CLOUD_PROJECT")) -> None:
+    Rest.handle_delete(url=f"/api/project/{project}")