primitive 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

primitive/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2024-present Dylan Stein <dylan@steins.studio>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.1.4"
+__version__ = "0.1.6"

primitive/agent/actions.py

@@ -0,0 +1,59 @@
+from time import sleep
+from primitive.utils.actions import BaseAction
+from loguru import logger
+
+
+class Agent(BaseAction):
+    def execute(
+        self,
+    ):
+        logger.enable("primitive")
+        logger.debug("Starting Primitive Agent")
+        while True:
+            hardware = self.primitive.hardware.get_own_hardware_details()
+
+            active_reservation_id = None
+            if hardware.get("activeReservation"):
+                active_reservation_id = hardware["activeReservation"]["id"]
+            if not active_reservation_id:
+                logger.debug("No active reservation found")
+                sleep(5)
+                continue
+
+            job_runs_data = self.primitive.projects.get_job_runs(
+                status="pending", first=1, reservation_id=active_reservation_id
+            )
+
+            pending_job_runs = [
+                edge["node"] for edge in job_runs_data["jobRuns"]["edges"]
+            ]
+
+            for job_run in pending_job_runs:
+                logger.debug("Found pending Job Run")
+                logger.debug(f"Job Run ID: {job_run['id']}")
+                logger.debug(f"Job Name: {job_run['job']['name']}")
+
+                if job_run["job"]["slug"] == "lint":
+                    logger.debug("Executing Lint Job")
+
+                    self.primitive.projects.job_run_update(
+                        job_run["id"], status="request_completed"
+                    )
+
+                    result, message = self.primitive.lint.execute()
+                    if result:
+                        self.primitive.projects.job_run_update(
+                            job_run["id"],
+                            status="request_completed",
+                            conclusion="success",
+                            stdout=message,
+                        )
+                    else:
+                        self.primitive.projects.job_run_update(
+                            job_run["id"],
+                            status="request_completed",
+                            conclusion="failure",
+                            stdout=message,
+                        )
+
+            sleep(5)

primitive/agent/commands.py

@@ -0,0 +1,13 @@
+import click
+import typing
+
+if typing.TYPE_CHECKING:
+    from ..client import Primitive
+
+
+@click.command("agent")
+@click.pass_context
+def cli(context):
+    """agent"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    primitive.agent.execute()

primitive/cli.py

@@ -7,6 +7,7 @@ from .auth.commands import config_command, whoami_command
 from .files.commands import cli as file_commands
 from .hardware.commands import cli as hardware_commands
 from .lint.commands import cli as lint_commands
+from .agent.commands import cli as agent_commands
 
 
 @click.group()
@@ -55,6 +56,7 @@ cli.add_command(whoami_command, "whoami")
 cli.add_command(file_commands, "files")
 cli.add_command(hardware_commands, "hardware")
 cli.add_command(lint_commands, "lint")
+cli.add_command(agent_commands, "agent")
 
 if __name__ == "__main__":
     cli(obj={})

primitive/client.py

@@ -7,6 +7,7 @@ from .files.actions import Files
 from .simulations.actions import Simulations
 from .hardware.actions import Hardware
 from .lint.actions import Lint
+from .agent.actions import Agent
 
 from loguru import logger
 
@@ -57,6 +58,7 @@ class Primitive:
         self.simulations: Simulations = Simulations(self)
         self.hardware: Hardware = Hardware(self)
         self.lint: Lint = Lint(self)
+        self.agent: Agent = Agent(self)
 
     def get_host_config(self):
         self.full_config = read_config_file()

primitive/hardware/actions.py

@@ -4,7 +4,7 @@ import json
 import platform
 from shutil import which
 import subprocess
-from typing import Dict, List
+from typing import Dict, List, Optional
 import click
 from loguru import logger
 from primitive.utils.memory_size import MemorySize
@@ -445,3 +445,67 @@ class Hardware(BaseAction):
             message = " [*] Failed to check in! "
             logger.error(message)
             raise exception
+
+    def get_hardware_list(self, fingerprint: Optional[str] = None):
+        query = gql(
+            """
+fragment HardwareFragment on Hardware {
+  id
+  pk
+  name
+  slug
+  createdAt
+  updatedAt
+  capabilities {
+    id
+    pk
+  }
+  activeReservation {
+    id
+    pk
+  }
+}
+
+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
+      }
+    }
+  }
+}
+"""
+        )
+
+        filters = {}
+        if fingerprint:
+            filters["fingerprint"] = {"exact": fingerprint}
+
+        variables = {
+            "first": 1,
+            "filters": filters,
+        }
+        result = self.primitive.session.execute(query, variable_values=variables)
+        return result
+
+    def get_own_hardware_details(self):
+        hardware_list_data = self.get_hardware_list(
+            fingerprint=self.primitive.host_config.get("fingerprint")
+        )
+        hardware = hardware_list_data.get("hardwareList").get("edges")[0].get("node")
+        return hardware

primitive/projects/actions.py

@@ -6,6 +6,106 @@ from primitive.utils.actions import BaseAction
 
 
 class Projects(BaseAction):
+    def get_job_runs(
+        self,
+        organization_id: Optional[str] = None,
+        project_id: Optional[str] = None,
+        job_id: Optional[str] = None,
+        reservation_id: Optional[str] = None,
+        git_commit_id: Optional[str] = None,
+        status: Optional[str] = None,
+        conclusion: Optional[str] = None,
+        first: Optional[int] = 1,
+        last: Optional[int] = None,
+    ):
+        query = gql(
+            """
+fragment PageInfoFragment on PageInfo {
+  hasNextPage
+  hasPreviousPage
+  startCursor
+  endCursor
+}
+
+fragment JobRunFragment on JobRun {
+  id
+  pk
+  createdAt
+  updatedAt
+  completedAt
+  startedAt
+  status
+  conclusion
+  stdout
+  job {
+    id
+    pk
+    slug
+    name
+    createdAt
+    updatedAt
+  }
+}
+
+query jobRuns(
+  $before: String
+  $after: String
+  $first: Int
+  $last: Int
+  $filters: JobRunFilters
+  $order: JobRunOrder
+) {
+  jobRuns(
+    before: $before
+    after: $after
+    first: $first
+    last: $last
+    filters: $filters
+    order: $order
+  ) {
+    totalCount
+    pageInfo {
+      ...PageInfoFragment
+    }
+    edges {
+      cursor
+      node {
+        ...JobRunFragment
+      }
+    }
+  }
+}
+"""
+        )
+
+        filters = {}
+        if organization_id:
+            filters["organization"] = {"id": organization_id}
+        if project_id:
+            filters["project"] = {"id": project_id}
+        if job_id:
+            filters["job"] = {"id": job_id}
+        if reservation_id:
+            filters["reservation"] = {"id": reservation_id}
+        if git_commit_id:
+            filters["gitCommit"] = {"id": git_commit_id}
+        if status:
+            filters["status"] = {"exact": status}
+        if conclusion:
+            filters["conclusion"] = {"exact": status}
+
+        variables = {
+            "first": first,
+            "last": last,
+            "filters": filters,
+            "order": {
+                "createdAt": "DESC",
+            },
+        }
+
+        result = self.primitive.session.execute(query, variable_values=variables)
+        return result
+
     def get_job_run(self, id: str):
         query = gql(
             """
@@ -28,6 +128,7 @@ class Projects(BaseAction):
         id: str,
         status: str = None,
         conclusion: str = None,
+        stdout: str = None,
         file_ids: Optional[List[str]] = [],
     ):
         mutation = gql(
@@ -50,6 +151,8 @@ class Projects(BaseAction):
             input["conclusion"] = conclusion
         if file_ids and len(file_ids) > 0:
             input["files"] = file_ids
+        if stdout:
+            input["stdout"] = stdout
         variables = {"input": input}
         result = self.primitive.session.execute(mutation, variable_values=variables)
         return result

primitive/utils/shell.py

@@ -3,61 +3,31 @@ import subprocess
 
 
 def add_path_to_shell(path: Path):
+    if not path.exists():
+        raise Exception(f"{path} does not exist.")
+
     try:
         subprocess.run(["fish_add_path", path], capture_output=True)
         return True
     except FileNotFoundError:
         pass
 
-    if Path.home() / ".bash_profile":
-        subprocess.run(
-            [
-                "echo",
-                f"'export PATH={path}:$PATH'",
-                ">>",
-                "~/.bash_profile",
-            ],
-            capture_output=True,
-        )
-    elif Path.home() / ".bashrc":
-        subprocess.run(
-            [
-                "echo",
-                f"'export PATH={path}:$PATH'",
-                ">>",
-                "~/.bashrc",
-            ],
-            capture_output=True,
-        )
-    elif Path.home() / ".zshrc":
-        subprocess.run(
-            [
-                "echo",
-                f"'export PATH={path}:$PATH'",
-                ">>",
-                "~/.zshrc",
-            ],
-            capture_output=True,
-        )
-    elif Path.home() / ".profile":
-        subprocess.run(
-            [
-                "echo",
-                f"'export PATH={path}:$PATH'",
-                ">>",
-                "~/.profile",
-            ],
-            capture_output=True,
-        )
-    elif Path.home() / ".bash_login":
-        subprocess.run(
-            [
-                "echo",
-                f"'export PATH={path}:$PATH'",
-                ">>",
-                "~/.bash_login",
-            ],
-            capture_output=True,
-        )
+    profile_path = None
+
+    if Path.home().joinpath(".bash_profile").exists():
+        profile_path = Path.home().joinpath(".bash_profile")
+    elif Path.home().joinpath(".bashrc").exists():
+        profile_path = Path.home().joinpath(".bashrc")
+    elif Path.home().joinpath(".zshrc").exists():
+        profile_path = Path.home().joinpath(".zshrc")
+    elif Path.home().joinpath(".profile").exists():
+        profile_path = Path.home().joinpath(".profile")
+    elif Path.home().joinpath(".bash_login").exists():
+        profile_path = Path.home().joinpath(".bash_login")
     else:
         raise Exception(f"Failed to add {path} to PATH")
+
+    with open(profile_path, "a") as file:
+        file.write(f"export PATH={path}:$PATH\n")
+
+    return True

primitive/utils/verible.py

@@ -18,9 +18,9 @@ def install_verible(system_info: dict) -> bool:
         url = VERIBLE_MAC_OS_LINK
     elif system_info.get("os_family") == "Windows":
         url = VERIBLE_WINDOWS_64_OS_LINK
-    elif system_info.processor == "x86_64":
+    elif system_info.get("processor") == "x86_64":
         url = VERIBLE_LINUX_X86_64_OS_LINK
-    elif system_info.processor == "arm":
+    elif system_info.get("processor") == "arm":
         url = VERIBLE_LINUX_X86_64_OS_LINK
 
     verible_dir_name = url.split("/")[-1].split(".tar.gz")[0]
@@ -43,7 +43,11 @@ def install_verible(system_info: dict) -> bool:
     logger.debug("Deleting tar.gz artifact")
     file_download_path.unlink()
 
-    verible_bin = Path.home() / verible_dir_name / "bin"
+    unpacked_verible_dir_name = verible_dir_name
+    if "linux" in unpacked_verible_dir_name:
+        unpacked_verible_dir_name = unpacked_verible_dir_name.split("-linux")[0]
+
+    verible_bin = Path.home().joinpath(unpacked_verible_dir_name).joinpath("bin")
 
     logger.debug("Adding verible to PATH")
     add_path_to_shell(verible_bin)

{primitive-0.1.4.dist-info → primitive-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: primitive
-Version: 0.1.4
+Version: 0.1.6
 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
@@ -36,6 +36,7 @@ Description-Content-Type: text/markdown
 **Table of Contents**
 
 - [Installation](#installation)
+- [Configuration](#configuration)
 - [License](#license)
 
 ## Installation
@@ -44,6 +45,20 @@ Description-Content-Type: text/markdown
 pip install primitive
 ```
 
+## Configuration
+
+### Authenticate
+
+```console
+primitive config
+```
+
+### Register your Hardware
+
+```console
+primitive hardware register
+```
+
 ## License
 
 `primitive` is distributed under the terms of the [MIT](https://spdx.org/licenses/MIT.html) license.

{primitive-0.1.4.dist-info → primitive-0.1.6.dist-info}/RECORD

@@ -1,7 +1,9 @@
-primitive/__about__.py,sha256=E2SXPkbuFZByKB4zIeOf_Me_bVBs-mUA8TpF7fO1FrM,128
+primitive/__about__.py,sha256=P5wASxX77-g_YRSjb4MnBDu4CWjITURFbU8iN8R3rlM,128
 primitive/__init__.py,sha256=bwKdgggKNVssJFVPfKSxqFMz4IxSr54WWbmiZqTMPNI,106
-primitive/cli.py,sha256=QC4jbu8s0uhNxFmaTO2B-X1o1gpyFX3xJVHU-j5CvQE,1705
-primitive/client.py,sha256=cZiJjzZ4HkCtYViDNOIoNuf7mcp772eSCSztsqh381E,2010
+primitive/cli.py,sha256=LXSpPtqQLxkOXsnsbs4UN3n7G_pHCz8AkBFCrk7Is4g,1796
+primitive/client.py,sha256=3RMRNY4ZiGT2OKnZtquWyrXjRSgDe5HlZvtFNKZErLQ,2083
+primitive/agent/actions.py,sha256=Fq6GEJqIwv-hWABFOl3YC8l0gBNME3byycNYqICpZG8,2144
+primitive/agent/commands.py,sha256=-dVDilELfkGfbZB7qfEPs77Dm1oT62qJj4tsIk4KoxI,254
 primitive/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/auth/actions.py,sha256=N2bGcwXNsB89pzs66gF9A5_WzUScY5fhfOyWixqo2y8,1054
 primitive/auth/commands.py,sha256=JahUq0E2e7Xa-FX1WEUv7TgM6ieDvNH4VwRRtxAW7HE,2340
@@ -9,12 +11,12 @@ primitive/files/actions.py,sha256=f4JN3QFB2WXw-0JpnE-4-movnqtvXIpCrGd_9pdkeW4,26
 primitive/files/commands.py,sha256=DDizo3xJnU3KLUBTMeeM72viVpnJinLwxs84tmqKhqo,810
 primitive/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/graphql/sdk.py,sha256=BhCGmDtc4sNnH8CxbQSJyFwOZ-ZSqMtjsxMB3JRBhPw,1456
-primitive/hardware/actions.py,sha256=BbFz17NKVDiI-9SiV1R1GPqsXRZwUm9oNj-Dc_bBFJ8,17133
+primitive/hardware/actions.py,sha256=UL2Wyh_XFXKVNEvlVCe1ltCZHpNaRSXPNwSR8hGJyR0,18343
 primitive/hardware/commands.py,sha256=QE7LLeFdfOqlvz3JwdwJJRZAY3fHI1zB9kYmmDajpq0,1477
 primitive/lint/actions.py,sha256=FVigksFrADPDD3zHS0g0riyVrge7eEtwWk5PG2aZv9A,2352
 primitive/lint/commands.py,sha256=3CZvkOEMpJspJWmaQzA5bpPKx0_VCijQIXA9l-eTnZE,487
 primitive/projects/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/projects/actions.py,sha256=Zi0uFCUG5KU7w7HNE8eGlhUPdsJyjVrakcdU1PR8-MQ,1481
+primitive/projects/actions.py,sha256=h1rgt3tWUPye0gnFvIZzZz7ibTbxyBS4q9RryjkVLzY,3600
 primitive/simulations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/simulations/actions.py,sha256=YR0oxxd7_kuUIH77BWZLUs9rLRiSJztPPhpgDJU2PbY,1267
 primitive/utils/actions.py,sha256=HOFrmM3-0A_A3NS84MqrZ6JmQEiiPSoDqEeuu6b_qfQ,196
@@ -23,10 +25,10 @@ primitive/utils/files.py,sha256=Ug5UqrS9eZoTd08EFtgyyp8flsJTfsG9CwXBWmI-yFA,606
 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=xGsz3-3mmUP5xRrFotNj0UpbHmRSsiliQnyPzB48xLI,1567
-primitive/utils/verible.py,sha256=o2NBbqteePgFY-28S7AaEivSEszZcePFkMRXpg_XhHg,1990
-primitive-0.1.4.dist-info/METADATA,sha256=bW4QfYYNW6TWBazUnmjK3s7see0nYczt4UHcyupynYw,1642
-primitive-0.1.4.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-primitive-0.1.4.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
-primitive-0.1.4.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
-primitive-0.1.4.dist-info/RECORD,,
+primitive/utils/shell.py,sha256=-7UjQaBqSGHzEEyX8pNjeYFFP0P3lVnDV0OkgPz1qHU,1050
+primitive/utils/verible.py,sha256=QYczN1IvxODfj4jeq0nqjFuF0Oi0Zdx-Q32ySOJgcw8,2205
+primitive-0.1.6.dist-info/METADATA,sha256=k2nIZBpBfzNRizaA-5a9BW7qtOGmdQTeNF2lOdvocPU,1817
+primitive-0.1.6.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+primitive-0.1.6.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
+primitive-0.1.6.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
+primitive-0.1.6.dist-info/RECORD,,