primitive 0.1.89__py3-none-any.whl → 0.1.91__py3-none-any.whl

primitive/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2024-present Dylan Stein <dylan@primitive.tech>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.1.89"
+__version__ = "0.1.91"

primitive/agent/actions.py

@@ -28,7 +28,12 @@ class Agent(BaseAction):
 
         # self.primitive.hardware.update_hardware_system_info()
         try:
-            self.primitive.hardware.check_in_http(is_available=True, is_online=True)
+            # hey stupid:
+            # do not set is_available to True here, it will mess up the reservation logic
+            # only set is_available after we've checked that no active reservation is present
+            # setting is_available of the parent also effects the children,
+            # which may have active reservations as well
+            self.primitive.hardware.check_in_http(is_online=True)
         except Exception as ex:
             logger.error(f"Error checking in hardware: {ex}")
             sys.exit(1)

primitive/agent/runner.py

@@ -193,7 +193,9 @@ class AgentRunner:
 
                 returncode = proc.wait()
 
-                logger.debug(f"Process {step['name']} finished with return code {returncode}")
+                logger.debug(
+                    f"Process {step['name']} finished with return code {returncode}"
+                )
 
                 if proc.errors > 0 and self.failure_level >= FailureLevel.ERROR:
                     fail_level_detected = True
@@ -212,7 +214,9 @@ class AgentRunner:
                 self.primitive.jobs.job_run_update(
                     self.job_id, status="request_completed", conclusion="failure"
                 )
-                logger.error(f"Step {step['name']} failed with return code {returncode}")
+                logger.error(
+                    f"Step {step['name']} failed with return code {returncode}"
+                )
                 return
 
         if fail_level_detected and self.parse_logs:

primitive/exec/actions.py

@@ -6,6 +6,7 @@ if typing.TYPE_CHECKING:
     pass
 
 
+from loguru import logger
 from paramiko import SSHClient
 
 from primitive.utils.actions import BaseAction
@@ -16,6 +17,15 @@ class Exec(BaseAction):
         super().__init__(*args, **kwargs)
 
     def execute_command(self, hardware_identifier: str, command: str) -> None:
+        child_hardware = None
+        if "." in hardware_identifier:
+            hardware_slugs = hardware_identifier.split(".")
+            hardware_identifier = hardware_slugs[0]
+            child_hardware_identifier = hardware_slugs[-1]
+            child_hardware = self.primitive.hardware.get_hardware_from_slug_or_id(
+                hardware_identifier=child_hardware_identifier
+            )
+
         hardware = self.primitive.hardware.get_hardware_from_slug_or_id(
             hardware_identifier=hardware_identifier
         )
@@ -37,14 +47,23 @@ class Exec(BaseAction):
             reservation_result = self.primitive.reservations.create_reservation(
                 requested_hardware_ids=[hardware["id"]],
                 reason="Executing command from Primitive CLI",
+                wait=False,
             )
             reservation = reservation_result.data["reservationCreate"]
             created_reservation_on_behalf_of_user = True
 
-        reservation = self.primitive.reservations.wait_for_reservation_status(
+        reservation_result = self.primitive.reservations.wait_for_reservation_status(
             reservation_id=reservation["id"], desired_status="in_progress"
         )
 
+        reservation = reservation_result.data["reservation"]
+        if reservation.get("status") != "in_progress":
+            logger.enable("primitive")
+            logger.info(
+                f"Reservation {reservation.get('id')} is in status {reservation.get('status')}, cannot execute command at this time."
+            )
+            return
+
         ssh_hostname = hardware["hostname"]
         ssh_username = hardware["sshUsername"]
 
@@ -57,11 +76,29 @@ class Exec(BaseAction):
         )
 
         if command:
-            formatted_command = " ".join(command)
+            if child_hardware:
+                # if the child hardware has ssh credentials, format the proxy command
+                if child_hardware.get("systemInfo").get("os_family"):
+                    formatted_command = (
+                        f"adb -s {child_hardware.get('slug')} shell {command}"
+                    )
+            else:
+                # happy path!
+                formatted_command = " ".join(command)
+
             stdin, stdout, stderr = ssh_client.exec_command(formatted_command)
-            print(stdout.read())
+
+            stdout_string = stdout.read().decode("utf-8").rstrip("\n")
+            stderr_string = stderr.read().decode("utf-8").rstrip("\n")
+            if stdout_string != b"":
+                print(stdout_string)
+            if stderr.read() != b"":
+                print(stderr_string)
+
             ssh_client.close()
         else:
+            # if the child hardware has ssh credentials, format the proxy jump
+
             channel = ssh_client.get_transport().open_session()
             channel.get_pty()
             channel.invoke_shell()

primitive/git/actions.py

@@ -38,9 +38,7 @@ class Git(BaseAction):
         source_dir = Path(destination).joinpath(git_repo_full_name.split("/")[-1])
 
         try:
-            run(
-                ["git", "clone", url, source_dir, "--no-checkout"], check=True
-            )
+            run(["git", "clone", url, source_dir, "--no-checkout"], check=True)
         except CalledProcessError:
             raise Exception("Failed to download repository")
 

primitive/hardware/actions.py

@@ -20,15 +20,23 @@ from ..utils.config import update_config_file
 from .graphql.mutations import (
     hardware_checkin_mutation,
     hardware_update_mutation,
+    register_child_hardware_mutation,
     register_hardware_mutation,
+    unregister_hardware_mutation,
+)
+from .graphql.queries import (
+    hardware_details,
+    hardware_list,
+    nested_children_hardware_list,
 )
-from .graphql.queries import hardware_list
 
 if typing.TYPE_CHECKING:
     pass
 
 
+from primitive.hardware.android import AndroidDevice, list_devices
 from primitive.utils.actions import BaseAction
+from primitive.utils.shell import does_executable_exist
 
 
 class Hardware(BaseAction):
@@ -40,6 +48,7 @@ class Hardware(BaseAction):
             "isAvailable": False,
             "isOnline": False,
         }
+        self.children = []
 
     def _get_darwin_system_profiler_values(self) -> Dict[str, str]:
         system_profiler_hardware_data_type = subprocess.check_output(
@@ -276,10 +285,12 @@ class Hardware(BaseAction):
         return system_info
 
     @guard
-    def register(self):
+    def register(self, organization_id: Optional[str] = None):
         system_info = self.get_system_info()
         mutation = gql(register_hardware_mutation)
         input = {"systemInfo": system_info}
+        if organization_id:
+            input["organizationId"] = organization_id
         variables = {"input": input}
         result = self.primitive.session.execute(
             mutation, variable_values=variables, get_execution_result=True
@@ -303,6 +314,30 @@ class Hardware(BaseAction):
         # and headers are set correctly
         self.primitive.get_host_config()
         self.check_in_http(is_healthy=True)
+        for child in self._list_local_children():
+            self.register_child(child=child)
+        return result
+
+    @guard
+    def unregister(self, organization_id: Optional[str] = None):
+        mutation = gql(unregister_hardware_mutation)
+        input = {
+            "fingerprint": self.primitive.host_config.get("fingerprint"),
+        }
+        variables = {"input": input}
+        result = self.primitive.session.execute(
+            mutation, variable_values=variables, get_execution_result=True
+        )
+
+        if messages := result.data.get("unregisterHardware").get("messages"):
+            for message in messages:
+                logger.enable("primitive")
+                if message.get("kind") == "ERROR":
+                    logger.error(message.get("message"))
+                else:
+                    logger.debug(message.get("message"))
+            return False
+
         return result
 
     @guard
@@ -355,6 +390,7 @@ class Hardware(BaseAction):
         is_online: bool = True,
     ):
         fingerprint = self.primitive.host_config.get("fingerprint", None)
+
         if not fingerprint:
             message = (
                 "No fingerprint found. Please register: primitive hardware register"
@@ -417,15 +453,50 @@ class Hardware(BaseAction):
 
     @guard
     def get_hardware_list(
-        self, fingerprint: Optional[str] = None, slug: Optional[str] = None
+        self,
+        fingerprint: Optional[str] = None,
+        id: Optional[str] = None,
+        slug: Optional[str] = None,
+        nested_children: Optional[bool] = False,
     ):
-        query = gql(hardware_list)
+        query = gql(nested_children_hardware_list if nested_children else hardware_list)
+
+        filters = {
+            "isRegistered": {"exact": True},
+        }
+        if fingerprint is not None:
+            filters["fingerprint"] = {"exact": fingerprint}
+        if slug is not None:
+            filters["slug"] = {"exact": slug}
+        if id is not None:
+            filters["id"] = {"exact": id}
+        if nested_children is True:
+            filters["hasParent"] = {"exact": False}
+
+        variables = {
+            "filters": filters,
+        }
+        result = self.primitive.session.execute(
+            query, variable_values=variables, get_execution_result=True
+        )
+        return result
+
+    @guard
+    def get_hardware_details(
+        self,
+        fingerprint: Optional[str] = None,
+        id: Optional[str] = None,
+        slug: Optional[str] = None,
+    ):
+        query = gql(hardware_details)
 
         filters = {}
-        if fingerprint:
+        if fingerprint is not None:
             filters["fingerprint"] = {"exact": fingerprint}
-        if slug:
+        if slug is not None:
             filters["slug"] = {"exact": slug}
+        if id is not None:
+            filters["id"] = {"exact": id}
 
         variables = {
             "first": 1,
@@ -434,7 +505,10 @@ class Hardware(BaseAction):
         result = self.primitive.session.execute(
             query, variable_values=variables, get_execution_result=True
         )
-        return result
+        if edges := result.data["hardwareList"]["edges"]:
+            return edges[0]["node"]
+        else:
+            raise Exception(f"No hardware found with {filters}")
 
     def get_own_hardware_details(self):
         hardware_list_result = self.get_hardware_list(
@@ -448,6 +522,7 @@ class Hardware(BaseAction):
     def get_hardware_from_slug_or_id(self, hardware_identifier: str):
         is_id = False
         is_slug = False
+        id = None
         # first check if the hardware_identifier is a slug or ID
         try:
             type_name, id = from_base64(hardware_identifier)
@@ -466,16 +541,45 @@ class Hardware(BaseAction):
         hardware = None
 
         if is_id and is_hardware_type:
-            hardware_list_result = self.get_hardware_list(slug=hardware_identifier)
-            if edges := hardware_list_result.data["hardwareList"]["edges"]:
-                hardware = edges[0]["node"]
-            else:
-                raise Exception(f"No hardware found with slug {hardware_identifier}")
+            hardware = self.get_hardware_details(id=id)
         elif is_slug:
-            hardware_list_result = self.get_hardware_list(slug=hardware_identifier)
-            if edges := hardware_list_result.data["hardwareList"]["edges"]:
-                hardware = edges[0]["node"]
-            else:
-                raise Exception(f"No hardware found with slug {hardware_identifier}")
+            hardware = self.get_hardware_details(slug=hardware_identifier)
 
         return hardware
+
+    @guard
+    def register_child(self, child: AndroidDevice):
+        system_info = child.system_info
+        mutation = gql(register_child_hardware_mutation)
+        input = {"childSystemInfo": system_info}
+        variables = {"input": input}
+        result = self.primitive.session.execute(
+            mutation, variable_values=variables, get_execution_result=True
+        )
+
+        if messages := result.data.get("registerChildHardware").get("messages"):
+            for message in messages:
+                logger.enable("primitive")
+                if message.get("kind") == "ERROR":
+                    logger.error(message.get("message"))
+                else:
+                    logger.debug(message.get("message"))
+            return False
+        return result
+
+    def _list_local_children(self) -> List[AndroidDevice]:
+        if does_executable_exist("adb"):
+            self.children: List[AndroidDevice] = list_devices()
+        return self.children
+
+    @guard
+    def _remove_child(self):
+        pass
+
+    @guard
+    def _sync_children(self):
+        # get the existing children if any from the hardware details
+        # get the latest children from the node
+        # compare the two and update the node with the latest children
+        # remove any children from remote that are not in the latest children
+        pass

primitive/hardware/android.py

@@ -0,0 +1,75 @@
+from dataclasses import dataclass, field
+from subprocess import PIPE, Popen
+from typing import Dict
+
+
+@dataclass
+class AndroidDevice:
+    serial: str
+    usb: str
+    product: str
+    model: str
+    device: str
+    transport_id: str
+    system_info: Dict = field(default_factory=lambda: {})
+
+    def _get_android_values(self):
+        """Get the values of the Android device."""
+
+        # example line:
+        # System name | Network (domain) name | Kernel Release number | Kernel Version | Machine (hardware) name
+        # Linux localhost 6.1.75-android14-11-g48b922851ac5-ab12039954 #1 SMP PREEMPT Tue Jul  2 09:33:34 UTC 2024 aarch64 Toybox
+        uname_output = execute_command(serial=self.serial, command="uname -a")
+        self.system_info["name"] = self.serial
+        self.system_info["os_family"] = "Android"
+        self.system_info["os_release"] = uname_output[2]
+        self.system_info["os_version"] = uname_output[3]
+        self.system_info["platform"] = ""
+        self.system_info["processor"] = ""
+        self.system_info["machine"] = uname_output[13]
+        self.system_info["architecture"] = "64bit"
+        return self.system_info
+
+
+def list_devices():
+    """List all connected Android devices."""
+    devices = []
+
+    with Popen(["adb", "devices", "-l"], stdout=PIPE) as process:
+        for line in process.stdout.read().decode("utf-8").split("\n"):
+            if line == "":
+                continue
+
+            if "List of devices attached" in line:
+                continue
+
+            device_details_array = [
+                detail for detail in line.split(" ") if detail != ""
+            ]
+
+            android_device = AndroidDevice(
+                serial=device_details_array[0],
+                usb=device_details_array[1],
+                product=device_details_array[2],
+                model=device_details_array[3],
+                device=device_details_array[4],
+                transport_id=device_details_array[5],
+            )
+            android_device._get_android_values()
+            devices.append(android_device)
+
+    return devices
+
+
+def execute_command(serial: str, command: str):
+    """Execute a command on an Android device."""
+    with Popen(
+        ["adb", "-s", serial, "shell", *command.split(" ")], stdin=PIPE, stdout=PIPE
+    ) as process:
+        return process.stdout.read().decode("utf-8")
+
+
+def create_interactive_shell(serial: str, command: str):
+    """Create an interactive shell to an Android device."""
+    with Popen(["adb", "-s", serial, "shell"], stdin=PIPE, stdout=PIPE) as process:
+        return process.stdout.read().decode("utf-8")

primitive/hardware/commands.py

@@ -7,6 +7,9 @@ from ..utils.printer import print_result
 if typing.TYPE_CHECKING:
     from ..client import Primitive
 
+from rich.console import Console
+from rich.table import Table
+
 
 @click.group()
 @click.pass_context
@@ -40,6 +43,21 @@ def register_command(context):
     print_result(message=message, context=context, fg=color)
 
 
+@cli.command("unregister")
+@click.pass_context
+def unregister_command(context):
+    """Unregister Hardware with Primitive"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    result = primitive.hardware.unregister()
+    color = "green" if result else "red"
+    if not result:
+        message = "There was an error unregistering this device. Please review the above logs."
+        return
+    elif result.data.get("unregisterHardware"):
+        message = "Hardware unregistered successfully"
+    print_result(message=message, context=context, fg=color)
+
+
 @cli.command("checkin")
 @click.pass_context
 def checkin_command(context):
@@ -53,11 +71,85 @@ def checkin_command(context):
         print_result(message=message, context=context, fg="green")
 
 
+def hardware_status_string(hardware):
+    if activeReservation := hardware.get("activeReservation"):
+        if activeReservation.get("status", None) == "in_progress":
+            return "Reserved"
+    if hardware.get("isQuarantined"):
+        return "Quarantined"
+    if not hardware.get("isOnline"):
+        return "Offline"
+    if not hardware.get("isHealthy"):
+        return "Not healthy"
+    if not hardware.get("isAvailable"):
+        return "Not available"
+    else:
+        return "Available"
+
+
 @cli.command("list")
 @click.pass_context
 def list_command(context):
     """List Hardware"""
     primitive: Primitive = context.obj.get("PRIMITIVE")
-    get_hardware_list_result = primitive.hardware.get_hardware_list()
+    get_hardware_list_result = primitive.hardware.get_hardware_list(
+        nested_children=True
+    )
     message = get_hardware_list_result.data
-    print_result(message=message, context=context)
+
+    hardware_list = [
+        hardware.get("node")
+        for hardware in get_hardware_list_result.data.get("hardwareList").get("edges")
+    ]
+
+    message = hardware_list
+    if context.obj["JSON"]:
+        print_result(message=message, context=context)
+        return
+
+    console = Console()
+
+    table = Table(show_header=True, header_style="bold magenta")
+    table.add_column("Organization")
+    table.add_column("Name | Slug")
+    table.add_column("Status")
+    table.add_column("Reservation")
+
+    for hardware in hardware_list:
+        name = hardware.get("name")
+        slug = hardware.get("slug")
+        print_name = name
+        if name != slug:
+            print_name = f"{name} | {slug}"
+        child_table = Table(show_header=False, header_style="bold magenta")
+        child_table.add_column("Organization")
+        child_table.add_column("Name | Slug")
+        child_table.add_column("Status")
+        child_table.add_column("Reservation", justify="right")
+
+        table.add_row(
+            hardware.get("organization").get("name"),
+            print_name,
+            hardware_status_string(hardware),
+            f"{hardware.get('activeReservation').get('createdBy').get('username')} | {hardware.get('activeReservation').get('status')}"
+            if hardware.get("activeReservation", None)
+            else "",
+        )
+
+        if len(hardware.get("children", [])) > 0:
+            for child in hardware.get("children"):
+                name = child.get("name")
+                slug = child.get("slug")
+                print_name = name
+                if name != slug:
+                    print_name = f"└── {name} | {slug}"
+                table.add_row(
+                    hardware.get("organization").get("name"),
+                    print_name,
+                    hardware_status_string(hardware),
+                    f"{hardware.get('activeReservation').get('createdBy').get('username')} | {hardware.get('activeReservation').get('status')}"
+                    if hardware.get("activeReservation", None)
+                    else "",
+                )
+
+    console.print(table)

primitive/hardware/graphql/fragments.py

@@ -10,15 +10,22 @@ fragment HardwareFragment on Hardware {
   isOnline
   isQuarantined
   isHealthy
+  systemInfo
   hostname
   sshUsername
-  capabilities {
+  organization {
     id
     pk
+    name
+    slug
   }
   activeReservation {
     id
     pk
+    status
+    createdBy {
+      username
+    }
   }
 }
 """

primitive/hardware/graphql/mutations.py

@@ -14,6 +14,35 @@ mutation registerHardware($input: RegisterHardwareInput!) {
 """
 )
 
+register_child_hardware_mutation = (
+    operation_info_fragment
+    + """
+mutation registerChildHardware($input: RegisterChildHardwareInput!) {
+    registerChildHardware(input: $input) {
+        ... on Hardware {
+            fingerprint
+        }
+        ...OperationInfoFragment
+    }
+}
+"""
+)
+
+unregister_hardware_mutation = (
+    operation_info_fragment
+    + """
+mutation unregisterHardware($input: UnregisterHardwareInput!) {
+    unregisterHardware(input: $input) {
+        ... on Hardware {
+            fingerprint
+        }
+        ...OperationInfoFragment
+    }
+}
+"""
+)
+
+
 hardware_update_mutation = (
     operation_info_fragment
     + """

primitive/hardware/graphql/queries.py

@@ -29,3 +29,69 @@ query hardwareList(
 }
 """
 )
+
+nested_children_hardware_list = (
+    hardware_fragment
+    + """
+
+query hardwareList(
+  $before: String
+  $after: String
+  $first: Int
+  $last: Int
+  $filters: HardwareFilters
+) {
+  hardwareList(
+    before: $before
+    after: $after
+    first: $first
+    last: $last
+    filters: $filters
+  ) {
+    totalCount
+    edges {
+      cursor
+      node {
+        ...HardwareFragment
+        children {
+          ...HardwareFragment
+        }
+      }
+    }
+  }
+}
+"""
+)
+
+hardware_details = (
+    hardware_fragment
+    + """
+
+query hardwareDetails(
+  $before: String
+  $after: String
+  $first: Int
+  $last: Int
+  $filters: HardwareFilters
+) {
+  hardwareList(
+    before: $before
+    after: $after
+    first: $first
+    last: $last
+    filters: $filters
+  ) {
+    totalCount
+    edges {
+      cursor
+      node {
+        ...HardwareFragment
+        children {
+          ...HardwareFragment
+        }
+      }
+    }
+  }
+}
+"""
+)

primitive/organizations/actions.py

@@ -14,7 +14,7 @@ class Organizations(BaseAction):
         self,
         organization_id: Optional[str] = None,
         slug: Optional[str] = None,
-        first: Optional[int] = 1,
+        first: Optional[int] = None,
         last: Optional[int] = None,
     ):
         query = gql(organizations_query)

primitive/reservations/actions.py

@@ -1,14 +1,16 @@
 import typing
-from time import sleep
 
 from primitive.graphql.relay import from_base64
 
 if typing.TYPE_CHECKING:
     pass
 
+from datetime import datetime, timedelta
+from time import sleep
 from typing import List, Optional
 
 from gql import gql
+from loguru import logger
 
 from primitive.utils.actions import BaseAction
 
@@ -57,6 +59,7 @@ class Reservations(BaseAction):
     def create_reservation(
         self,
         reason: str,
+        wait: bool = True,
         requested_hardware_ids: Optional[List[str]] = None,
         organization_id: Optional[str] = None,
         hardware_identifier: Optional[str] = None,
@@ -84,6 +87,21 @@ class Reservations(BaseAction):
         result = self.primitive.session.execute(
             mutation, variable_values=variables, get_execution_result=True
         )
+        if messages := result.data.get("reservationCreate").get("messages"):
+            for message in messages:
+                logger.enable("primitive")
+                if message.get("kind") == "ERROR":
+                    logger.error(message.get("message"))
+                else:
+                    logger.debug(message.get("message"))
+            return False
+
+        if wait:
+            reservation = result.data["reservationCreate"]
+            result = self.wait_for_reservation_status(
+                reservation_id=reservation["id"], desired_status="in_progress"
+            )
+
         return result
 
     @guard
@@ -116,19 +134,47 @@ class Reservations(BaseAction):
         return result
 
     @guard
-    def wait_for_reservation_status(self, reservation_id: str, desired_status: str):
+    def wait_for_reservation_status(
+        self, reservation_id: str, desired_status: str, total_sleep_time: int = 30
+    ):
         reservation_result = self.get_reservation(reservation_id=reservation_id)
         reservation = reservation_result.data["reservation"]
         current_status = reservation["status"]
 
-        sleep_amount = 1
+        logger.enable("primitive")
+        logger.debug(
+            f"Waiting {total_sleep_time}s for reservation {reservation_id} to be in_progress."
+        )
+
+        now = datetime.now()
+        future_time = now + timedelta(seconds=total_sleep_time)
+
         while current_status != desired_status:
+            now = datetime.now()
+            logger.debug(
+                f"[{(future_time - now).seconds}s remaining] Waiting for reservation {reservation_id} to be {desired_status}. Current status: {current_status}"
+            )
+            if now > future_time:
+                logger.info(
+                    f"Reservation {reservation_id} did not reach {desired_status} status in {total_sleep_time}s."
+                )
+                break
+            if current_status == "completed":
+                logger.error(
+                    f"Reservation {reservation_id} concluded with {reservation['conclusionMessage']}. Reason: {reservation['conclusionMessage']}"
+                )
+                break
+
             reservation_result = self.get_reservation(reservation_id=reservation_id)
             reservation = reservation_result.data["reservation"]
             current_status = reservation["status"]
             if current_status == desired_status:
                 break
-            sleep(sleep_amount)
-            sleep_amount += 1
+            sleep(1)
+
+        if current_status == "waiting_for_hardware":
+            logger.info(
+                f"Reservation {reservation_id} is waiting for hardware to come online."
+            )
 
-        return reservation
+        return reservation_result

primitive/reservations/commands.py

@@ -44,11 +44,14 @@ def get(context, reservation_id: str):
 @click.pass_context
 @click.argument("hardware_identifier", type=str)
 @click.argument("reason", type=str)
-def create_reservation(context, hardware_identifier: str, reason: str):
+@click.option(
+    "--wait", default=True, type=bool, help="Wait for reservation to be in progress."
+)
+def create_reservation(context, hardware_identifier: str, reason: str, wait: bool):
     """Crate a reservation by a Hardware's ID or Slug"""
     primitive: Primitive = context.obj.get("PRIMITIVE")
     create_reservation_result = primitive.reservations.create_reservation(
-        hardware_identifier=hardware_identifier, reason=reason
+        hardware_identifier=hardware_identifier, reason=reason, wait=wait
     )
     message = create_reservation_result.data
     print_result(message=message, context=context)

primitive/utils/shell.py

@@ -1,5 +1,6 @@
-from pathlib import Path
 import subprocess
+from pathlib import Path
+from shutil import which
 from typing import Dict
 
 
@@ -56,3 +57,8 @@ def env_string_to_dict(env_str: str) -> Dict:
         env_dict[current_key] = "\n".join(current_value)
 
     return env_dict
+
+
+def does_executable_exist(executable_name: str) -> bool:
+    """Check if an executable is available on the path."""
+    return which(executable_name) is not None

{primitive-0.1.89.dist-info → primitive-0.1.91.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: primitive
-Version: 0.1.89
+Version: 0.1.91
 Project-URL: Documentation, https://github.com//primitivecorp/primitive-cli#readme
 Project-URL: Issues, https://github.com//primitivecorp/primitive-cli/issues
 Project-URL: Source, https://github.com//primitivecorp/primitive-cli
@@ -24,6 +24,7 @@ Requires-Dist: loguru
 Requires-Dist: paramiko[invoke]
 Requires-Dist: primitive-pal==0.1.4
 Requires-Dist: pyyaml
+Requires-Dist: rich>=13.9.4
 Requires-Dist: speedtest-cli
 Description-Content-Type: text/markdown
 

{primitive-0.1.89.dist-info → primitive-0.1.91.dist-info}/RECORD

@@ -1,13 +1,13 @@
-primitive/__about__.py,sha256=y9oBO8HmXAlJTY3fF293ERSrPhDBBLaBmiz1zVeWKa8,130
+primitive/__about__.py,sha256=Pcjvhbmmtb_HD3hZl5K0wZmVNyhcTeRZL0NrkJqP0yg,130
 primitive/__init__.py,sha256=bwKdgggKNVssJFVPfKSxqFMz4IxSr54WWbmiZqTMPNI,106
 primitive/cli.py,sha256=CiI60bG3UZyNFuLTpchr0KeJRG5SALj455Ob11CegGE,2412
 primitive/client.py,sha256=PPyIQRvKKSqCF9RRF5mJJ4Vqqolpzy1YXqffNLKIvAA,2390
 primitive/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/agent/actions.py,sha256=dkqSWsPY8-ipcMO3_rMKaC4WtQiQfUCDVToAnPUCE-Y,6493
+primitive/agent/actions.py,sha256=Ay76qUwMYNERndA0lAvwK8zjVOrR8nMmSi-NeSwIlA8,6817
 primitive/agent/commands.py,sha256=-dVDilELfkGfbZB7qfEPs77Dm1oT62qJj4tsIk4KoxI,254
 primitive/agent/process.py,sha256=32eoj0W1-LG-9xxeHia-jk9jTah1cnmjCYnvczgXYGU,3538
 primitive/agent/provision.py,sha256=rmwnro1K5F8mwtd45XAq7RVQmpDWnbBCQ8X_qgWhm3M,1546
-primitive/agent/runner.py,sha256=hzf_4KRdrInc351Fa5DleoKB1sT9pNYnnFvicDS062c,9281
+primitive/agent/runner.py,sha256=B_M0BOeCy7gO7Mp6nfxv-FJ-9yBbujGVNoZziAMcDjc,9357
 primitive/agent/uploader.py,sha256=OkgwXhWKoECOJnW_ZmpzmUS_cpb-orC_uebNcmf5byw,2948
 primitive/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/auth/actions.py,sha256=MPsG9LcKcOPwA7gZ9Ewk0PZJhTQvIrGfODdz4GxSzgA,999
@@ -20,7 +20,7 @@ primitive/daemons/commands.py,sha256=-Muh-6ib4uAVtPn_67AcMrDwuCwYlCnRQozCi2Xurmk
 primitive/daemons/launch_agents.py,sha256=qovt32gwpjGDd82z_SY5EGCUjaUyNA49pZFajZsw3eE,4796
 primitive/daemons/launch_service.py,sha256=FPB9qKEjhllRfEpct0ng2L9lpIaGJbQwn1JdFT8uBA8,5600
 primitive/exec/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/exec/actions.py,sha256=onBOsMwANCKjCf8aSmDhep9RTX7LhtS8nlcjZM6av1w,2632
+primitive/exec/actions.py,sha256=xw9Qyl3dJsMQQH3P_NUefx77PUFgWQXLDRtJmaSMZAY,4163
 primitive/exec/commands.py,sha256=66LO2kkJC-ynNZQpUCXv4Ol15QoacdSZAHblePDcmLo,510
 primitive/exec/interactive.py,sha256=TscY6s2ZysijidKPheq6y-fCErUVLS0zcdTW8XyFWGI,2435
 primitive/files/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -31,7 +31,7 @@ primitive/files/graphql/fragments.py,sha256=II6WHZjzSqX4IELwdiWokqHTKvDq6mMHF5gp
 primitive/files/graphql/mutations.py,sha256=Da_e6WSp-fsCYVE9A6SGkIQy9WDzjeQycNyHEn7vJqE,935
 primitive/files/graphql/queries.py,sha256=_ky-IRz928sKeSJuqaggTPxV4CGgmho3OyaAFu1z7nw,397
 primitive/git/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/git/actions.py,sha256=kXOJIfD9qDL2yNDIsJ7yZ91GnQiz0z95EeJvWl5Vnhc,1666
+primitive/git/actions.py,sha256=UKzPAK5zw9jKumY0LBcG6vcS1YCCX6tiL6hkSMqnxe0,1636
 primitive/git/commands.py,sha256=sCeSjkRgSEjCEsB5seXgB_h6xfk0KpvMvzMKoRfUbRA,1177
 primitive/git/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/git/graphql/queries.py,sha256=I1HGlqBb1lHIAWVSsC8tVY9JdsQ8DJVqs4nqSTcL30M,98
@@ -40,12 +40,13 @@ primitive/graphql/relay.py,sha256=bmij2AjdpURQ6GGVCxwWhauF-r_SxuAU2oJ4sDbLxpI,72
 primitive/graphql/sdk.py,sha256=DBFH8vw8FAGvRy8_FZc9WcjnwaQDlXmI8fiYmhCg-b0,1458
 primitive/graphql/utility_fragments.py,sha256=uIjwILC4QtWNyO5vu77VjQf_p0jvP3A9q_6zRq91zqs,303
 primitive/hardware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/hardware/actions.py,sha256=jtthNgRyeRD8txt4WqEZskPtsDWU2Yg2gJZLSrEx1io,18603
-primitive/hardware/commands.py,sha256=_HaWOdRQSkhnA1xZZHZWgadSQ9Gijxtnzg2vc_IDSMA,1854
+primitive/hardware/actions.py,sha256=6dnPUN_hjXy4DIVoKWneiqgdF6UWfwDN82GczAqRykw,22018
+primitive/hardware/android.py,sha256=NcvbDb9ZLl5PCeHII84yLxZo0V48Vn5K75AzkIsg9is,2623
+primitive/hardware/commands.py,sha256=sU9xES9UK3JO1RxZMeC0412pcWIwXWXwr8rmMzz1VGQ,5088
 primitive/hardware/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/hardware/graphql/fragments.py,sha256=PuqhW42fKUvRqOli5W7nOs2RfJ8FruSQC1gKBp3psBQ,271
-primitive/hardware/graphql/mutations.py,sha256=Zd6HxnIgTJ9mJQAfKJkdeDfstcPAal6Bj38pnKb_RuI,904
-primitive/hardware/graphql/queries.py,sha256=dhihQwr4O7zxDNRjeNWhkAXaSDOBsK-uqIczEGy1XLI,430
+primitive/hardware/graphql/fragments.py,sha256=kI6qnTNjaEaUr-C6eD55COphtueVYbYOWZwN5EW_3qw,350
+primitive/hardware/graphql/mutations.py,sha256=_4Hkbfik9Ron4T-meulu6T-9FR_BZjyPNwn745MPksU,1484
+primitive/hardware/graphql/queries.py,sha256=2k-ZMaSDDO8YsbYNV54pw0yHnsPco6Y9ZfKBOEnQCsw,1348
 primitive/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/jobs/actions.py,sha256=CtyO-Z9614TgIoXJJX1QGsoll0fgpBIjG9PJH5JwCQs,4901
 primitive/jobs/commands.py,sha256=MxPCkBEYW_eLNqgCRYeyj7ZcLOFAWfpVZlqDR2Y_S0o,830
@@ -54,7 +55,7 @@ primitive/jobs/graphql/fragments.py,sha256=GZ_rVc_pc3MfC8EhCY_X9rjeNUdiwEytdqrkn
 primitive/jobs/graphql/mutations.py,sha256=8ASvCmwQh7cMeeiykOdYaYVryG8FRIuVF6v_J8JJZuw,219
 primitive/jobs/graphql/queries.py,sha256=BrU_GnLjK0bTAmWsLSmGEUea7EM8MqTKxN1Qp6sSjwc,1597
 primitive/organizations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/organizations/actions.py,sha256=o7WnTONtH-WI0kzn71Uq-kF0CRjDS8Xb9YA7DIjYnwY,1085
+primitive/organizations/actions.py,sha256=Tgp_rox0jcvfhQ-LmcWc9vkPdeJu5Bk6U1rNuT9oDnw,1088
 primitive/organizations/commands.py,sha256=_dwgVEJCqMa5VgB_7P1wLPFc0AuT1p9dtyR9JRr4kpw,487
 primitive/organizations/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/organizations/graphql/fragments.py,sha256=a1qKq4FZB5qze0XTo1fOUeGAscIasjn_Ig4gA2_vStY,142
@@ -72,8 +73,8 @@ primitive/provisioning/actions.py,sha256=IYZYAbtomtZtlkqDaBxx4e7PFKGkRNqek_tABH6
 primitive/provisioning/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/provisioning/graphql/queries.py,sha256=cBtuKa6shoatYZfKSnQoPJP6B8g8y3QhFqJ_pkvMcG0,134
 primitive/reservations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/reservations/actions.py,sha256=XjjF0UJAgKryuSJqakLMAWshZIbuM-DkTmdU95cANs4,4434
-primitive/reservations/commands.py,sha256=OwWWE9DrvtrVBcBki0bKTOqCzCQk090c0rPIAt89JLY,2243
+primitive/reservations/actions.py,sha256=ELQRl9GEVqE_1c7lBxAFS3C9XJuhUEILgDVQkHv8Evc,6257
+primitive/reservations/commands.py,sha256=LFRoV59QGgWIjBdrGjJdffHugg8TLe0Fwlcyu_JaTkk,2369
 primitive/reservations/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/reservations/graphql/fragments.py,sha256=_TQfJeHky-Hh3WCHWobQ6-A1lpSvU-YkS0V9cqj2nOU,476
 primitive/reservations/graphql/mutations.py,sha256=IqzwQL7OclN7RpIcidrTQo9cGYofY7wqoBOdnY0pwN8,651
@@ -88,10 +89,10 @@ primitive/utils/files.py,sha256=QUa7c4t2PNvKOtyndLAxQMGvDM4cBftSeFh28xprVbM,752
 primitive/utils/git.py,sha256=1qNOu8X-33CavmrD580BmrFhD_WVO9PGWHUUboXJR_g,663
 primitive/utils/memory_size.py,sha256=4xfha21kW82nFvOTtDFx9Jk2ZQoEhkfXii-PGNTpIUk,3058
 primitive/utils/printer.py,sha256=f1XUpqi5dkTL3GWvYRUGlSwtj2IxU1q745T4Fxo7Tn4,370
-primitive/utils/shell.py,sha256=j7E1YwgNWw57dFHVfEbqRNVcPHX0xDefX2vFSNgeI_8,1648
+primitive/utils/shell.py,sha256=vpjr2Y7UQGYOvPGa6_RYXPPjqScfa9k7kT3tugF9h4Y,1837
 primitive/utils/verible.py,sha256=Zb5NUISvcaIgEvgCDBWr-GCoceMa79Tcwvr5Wl9lfnA,2252
-primitive-0.1.89.dist-info/METADATA,sha256=i5JNWnpMH7cIgrZX2mo9EheW1yJD92nspT2x-jD6Br0,3642
-primitive-0.1.89.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-primitive-0.1.89.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
-primitive-0.1.89.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
-primitive-0.1.89.dist-info/RECORD,,
+primitive-0.1.91.dist-info/METADATA,sha256=K3hzrLYG6ODSBpx-Vxz3gpKzzHSBa-VaoCYppm5NpQY,3670
+primitive-0.1.91.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+primitive-0.1.91.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
+primitive-0.1.91.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
+primitive-0.1.91.dist-info/RECORD,,