primitive 0.1.11__py3-none-any.whl → 0.1.12__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.11"
+__version__ = "0.1.12"

primitive/agent/actions.py

@@ -45,28 +45,48 @@ class Agent(BaseAction):
                     logger.debug(f"Job Run ID: {job_run['id']}")
                     logger.debug(f"Job Name: {job_run['job']['name']}")
 
+                    git_repo_full_name = job_run["gitCommit"]["repoFullName"]
+                    git_ref = job_run["gitCommit"]["sha"]
+                    logger.debug(
+                        f"Downloading repository {git_repo_full_name} at ref {git_ref}"
+                    )
+
+                    github_access_token = (
+                        self.primitive.projects.github_access_token_for_job_run(
+                            job_run["id"]
+                        )
+                    )
+
+                    downloaded_git_repository_dir = (
+                        self.primitive.git.download_git_repository_at_ref(
+                            git_repo_full_name=git_repo_full_name,
+                            git_ref=git_ref,
+                            github_access_token=github_access_token,
+                        )
+                    )
+
                     if job_run["job"]["slug"] == "lint":
                         logger.debug("Executing Lint Job")
 
                         self.primitive.projects.job_run_update(
-                            job_run["id"], status="request_completed"
+                            job_run["id"], status="request_in_progress"
                         )
 
-                        result, message = self.primitive.lint.execute()
+                        result, message = self.primitive.lint.execute(
+                            source=downloaded_git_repository_dir
+                        )
                         if result:
-                            self.primitive.projects.job_run_update(
-                                job_run["id"],
-                                status="request_completed",
-                                conclusion="success",
-                                stdout=message,
-                            )
+                            conclusion = "success"
                         else:
-                            self.primitive.projects.job_run_update(
-                                job_run["id"],
-                                status="request_completed",
-                                conclusion="failure",
-                                stdout=message,
-                            )
+                            conclusion = "failure"
+                        self.primitive.projects.job_run_update(
+                            job_run["id"],
+                            status="request_completed",
+                            conclusion=conclusion,
+                            stdout=message,
+                        )
+
+                        logger.debug("Lint Job Completed")
 
                 sleep(5)
         except KeyboardInterrupt:

primitive/cli.py

@@ -8,6 +8,7 @@ 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
+from .git.commands import cli as git_commands
 
 
 @click.group()
@@ -57,6 +58,7 @@ cli.add_command(file_commands, "files")
 cli.add_command(hardware_commands, "hardware")
 cli.add_command(lint_commands, "lint")
 cli.add_command(agent_commands, "agent")
+cli.add_command(git_commands, "git")
 
 if __name__ == "__main__":
     cli(obj={})

primitive/client.py

@@ -8,6 +8,7 @@ from .simulations.actions import Simulations
 from .hardware.actions import Hardware
 from .lint.actions import Lint
 from .agent.actions import Agent
+from .git.actions import Git
 
 from loguru import logger
 
@@ -59,6 +60,7 @@ class Primitive:
         self.hardware: Hardware = Hardware(self)
         self.lint: Lint = Lint(self)
         self.agent: Agent = Agent(self)
+        self.git: Git = Git(self)
 
     def get_host_config(self):
         self.full_config = read_config_file()

primitive/git/actions.py

@@ -0,0 +1,42 @@
+from pathlib import Path
+from primitive.utils.actions import BaseAction
+from loguru import logger
+import os
+
+
+class Git(BaseAction):
+    def get_github_access_token(self) -> str:
+        query = """
+query githubAppToken{
+    githubAppToken {
+        token
+    }
+}
+"""
+
+        filters = {}
+        variables = {
+            "filters": filters,
+        }
+        result = self.primitive.session.execute(query, variable_values=variables)
+        return result
+
+    def download_git_repository_at_ref(
+        self,
+        git_repo_full_name: str,
+        github_access_token: str,
+        git_ref: str = "main",
+        destination: Path = Path.cwd(),
+    ) -> Path:
+        logger.debug(f"Downloading source code from {git_repo_full_name} {git_ref}")
+        # TODO: switch to subprocess.run or subprocess.Popen
+        url = f"https://api.github.com/repos/{git_repo_full_name}/tarball/{git_ref}"
+        untar_dir = Path(destination).joinpath(git_repo_full_name.split("/")[-1])
+        untar_dir.mkdir(parents=True, exist_ok=True)
+
+        result = os.system(
+            f"curl -s -L -H 'Accept: application/vnd.github+json' -H 'Authorization: Bearer {github_access_token}' -H 'X-GitHub-Api-Version: 2022-11-28' {url} | tar zx --strip-components 1 -C {untar_dir}"
+        )
+        if result != 0:
+            raise Exception("Failed to download repository")
+        return untar_dir

primitive/git/commands.py

@@ -0,0 +1,49 @@
+import click
+from pathlib import Path
+import typing
+from ..utils.printer import print_result
+import os
+
+if typing.TYPE_CHECKING:
+    from ..client import Primitive
+
+
+import typing
+
+if typing.TYPE_CHECKING:
+    from ..client import Primitive
+
+
+@click.group()
+@click.pass_context
+def cli(context):
+    """Git related commands"""
+    pass
+
+
+@cli.command("download-ref")
+@click.pass_context
+@click.argument("git_repo_full_name", type=str)
+@click.argument("git_ref", default="main", type=str)
+@click.option(
+    "--github-access-token",
+    type=str,
+    default=lambda: os.environ.get("GITHUB_ACCESS_TOKEN", ""),
+)
+@click.argument("destination", type=click.Path(exists=True), default=".")
+def download_ref_command(
+    context,
+    git_repo_full_name: str,
+    git_ref: str = "main",
+    github_access_token: str = "",
+    destination: Path = Path.cwd(),
+):
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    result, message = primitive.git.download_git_repository_at_ref(
+        git_repo_full_name=git_repo_full_name,
+        git_ref=git_ref,
+        github_access_token=github_access_token,
+        destination=destination,
+    )
+    fg = "red" if not result else "green"
+    print_result(message, context=context, fg=fg)

primitive/lint/actions.py

@@ -9,7 +9,7 @@ from ..utils.verible import install_verible
 
 class Lint(BaseAction):
     def execute(self, source: Path = Path.cwd()) -> Tuple[bool, str]:
-        logger.debug("Starting linter...")
+        logger.debug("Starting linter for source: {source}")
         files = find_files_for_extension(source, ".sv")
         if not files:
             message = "No files found to lint."
@@ -48,14 +48,11 @@ class Lint(BaseAction):
         logger.debug("Linting complete.")
 
         message = ""
-        if self.primitive.DEBUG:
-            if result.stderr:
-                logger.error("\n" + result.stderr)
-            if result.stdout:
-                logger.info("\n" + result.stdout)
-            message = "See above logs for linter output."
-        else:
-            message = result.stderr
+        if result.stderr:
+            logger.error("\n" + result.stderr)
+        if result.stdout:
+            logger.info("\n" + result.stdout)
+        message = "See above logs for linter output."
 
         if result.returncode != 0:
             if not self.primitive.DEBUG:

primitive/projects/actions.py

@@ -45,6 +45,11 @@ fragment JobRunFragment on JobRun {
     createdAt
     updatedAt
   }
+  gitCommit {
+    sha
+    branch
+    repoFullName
+  }
 }
 
 query jobRuns(
@@ -156,3 +161,15 @@ query jobRuns(
         variables = {"input": input}
         result = self.primitive.session.execute(mutation, variable_values=variables)
         return result
+
+    def github_access_token_for_job_run(self, job_run_id: str):
+        query = gql(
+            """
+query ghAppTokenForJobRun($jobRunId: GlobalID!) {
+    ghAppTokenForJobRun(jobRunId: $jobRunId)
+}
+"""
+        )
+        variables = {"jobRunId": job_run_id}
+        result = self.primitive.session.execute(query, variable_values=variables)
+        return result["ghAppTokenForJobRun"]

primitive/utils/files.py

@@ -1,15 +1,22 @@
 from typing import List, Tuple
 from pathlib import Path
 from loguru import logger
+import os
 
 
 def find_files_for_extension(source: Path, extensions: Tuple[str]) -> List[Path]:
     matching_files = []
     logger.debug(f"Looking for files at {source} with extensions {extensions}")
-    for dirpath, dirnames, filenames in source.walk():
+    # for those on python 3.12+
+    # for dirpath, dirnames, filenames in source.walk():
+    #     for filename in filenames:
+    #         if filename.endswith(extensions):
+    #             matching_files.append(dirpath.joinpath(filename))
+
+    for dirpath, dirnames, filenames in os.walk(source):
         for filename in filenames:
             if filename.endswith(extensions):
-                matching_files.append(dirpath.joinpath(filename))
+                matching_files.append(Path(dirpath).joinpath(filename))
     logger.debug(
         f"Found {len(matching_files)} following files that match: {matching_files}"
     )

{primitive-0.1.11.dist-info → primitive-0.1.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: primitive
-Version: 0.1.11
+Version: 0.1.12
 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

{primitive-0.1.11.dist-info → primitive-0.1.12.dist-info}/RECORD

@@ -1,34 +1,37 @@
-primitive/__about__.py,sha256=cEU7i5k8S0WhpV8qykCw-GDDdmS5X4Rl6vZ5cCkC7EE,129
+primitive/__about__.py,sha256=35zrg1jAPnEjmmKrCnH1TpiDYB4jwmc5kkbIoAHlG84,129
 primitive/__init__.py,sha256=bwKdgggKNVssJFVPfKSxqFMz4IxSr54WWbmiZqTMPNI,106
-primitive/cli.py,sha256=LXSpPtqQLxkOXsnsbs4UN3n7G_pHCz8AkBFCrk7Is4g,1796
-primitive/client.py,sha256=3RMRNY4ZiGT2OKnZtquWyrXjRSgDe5HlZvtFNKZErLQ,2083
-primitive/agent/actions.py,sha256=f6bv2va1YrDfrzOVK2kTnelRQ2oEvmrLAmsPrxMCh7E,2893
+primitive/cli.py,sha256=fWv1gnUs7U6yWRrv1fYjt2bNBclIvp6okKTxrl_Il24,1879
+primitive/client.py,sha256=dkbHg2Pbxl7NjIk2O6Shvz7TE7uDQezGQkNwmTzKdNs,2146
+primitive/agent/actions.py,sha256=8eQDvUy0THT-zVvSTVgrDAsthRdSIpmhCkDr0iTXqDE,3673
 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
 primitive/files/actions.py,sha256=f4JN3QFB2WXw-0JpnE-4-movnqtvXIpCrGd_9pdkeW4,2624
 primitive/files/commands.py,sha256=DDizo3xJnU3KLUBTMeeM72viVpnJinLwxs84tmqKhqo,810
+primitive/git/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+primitive/git/actions.py,sha256=fepcl5529w_hsaC6fBw9f-QHeyqNjGXz8HI5ebzbZMs,1386
+primitive/git/commands.py,sha256=64B2STTOn0dwVDmJHqEwekmIqKMfSyBBFwKg29Wt8Aw,1230
 primitive/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/graphql/sdk.py,sha256=BhCGmDtc4sNnH8CxbQSJyFwOZ-ZSqMtjsxMB3JRBhPw,1456
 primitive/hardware/actions.py,sha256=Ea3_2E3F_3WapV60g_mOIcpXhadoknwihR7slXyUWtk,18840
 primitive/hardware/commands.py,sha256=QE7LLeFdfOqlvz3JwdwJJRZAY3fHI1zB9kYmmDajpq0,1477
-primitive/lint/actions.py,sha256=FVigksFrADPDD3zHS0g0riyVrge7eEtwWk5PG2aZv9A,2352
+primitive/lint/actions.py,sha256=PDw0fkIkI5hHHjoHaAiXvQUocwHZkgB2mfI-LGMl6TI,2267
 primitive/lint/commands.py,sha256=3CZvkOEMpJspJWmaQzA5bpPKx0_VCijQIXA9l-eTnZE,487
 primitive/projects/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/projects/actions.py,sha256=h1rgt3tWUPye0gnFvIZzZz7ibTbxyBS4q9RryjkVLzY,3600
+primitive/projects/actions.py,sha256=4Fwq2KTBuhDy8SfEIqPhR_8rW2rAL76cGWIQS40a96A,4039
 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
 primitive/utils/config.py,sha256=DlFM5Nglo22WPtbpZSVtH7NX-PTMaKYlcrUE7GPRG4c,1058
-primitive/utils/files.py,sha256=Ug5UqrS9eZoTd08EFtgyyp8flsJTfsG9CwXBWmI-yFA,606
+primitive/utils/files.py,sha256=Yv__bQes3YIlzhOT9kVxtYhoA5CmUjPSvphl9PZ41k4,867
 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=-7UjQaBqSGHzEEyX8pNjeYFFP0P3lVnDV0OkgPz1qHU,1050
 primitive/utils/verible.py,sha256=QYczN1IvxODfj4jeq0nqjFuF0Oi0Zdx-Q32ySOJgcw8,2205
-primitive-0.1.11.dist-info/METADATA,sha256=-KZVIgcT4_othtd9SQqixiI7PhsJNcgUzeO7wQxRw7s,1818
-primitive-0.1.11.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-primitive-0.1.11.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
-primitive-0.1.11.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
-primitive-0.1.11.dist-info/RECORD,,
+primitive-0.1.12.dist-info/METADATA,sha256=Zpkvn5C7K77B5vWuLDBp1SATMc_sZM6iR4J15DVzN8w,1818
+primitive-0.1.12.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+primitive-0.1.12.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
+primitive-0.1.12.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
+primitive-0.1.12.dist-info/RECORD,,