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

primitive/cli.py

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

primitive/client.py

@@ -9,6 +9,7 @@ from .hardware.actions import Hardware
 from .lint.actions import Lint
 from .agent.actions import Agent
 from .git.actions import Git
+from .daemons.actions import Daemons
 
 from loguru import logger
 
@@ -61,6 +62,7 @@ class Primitive:
         self.lint: Lint = Lint(self)
         self.agent: Agent = Agent(self)
         self.git: Git = Git(self)
+        self.daemons: Daemons = Daemons(self)
 
     def get_host_config(self):
         self.full_config = read_config_file()

primitive/daemons/actions.py

@@ -0,0 +1,75 @@
+import platform
+import typing
+
+if typing.TYPE_CHECKING:
+    from ..client import Primitive
+
+from .launch_agents import (
+    full_launch_agent_install,
+    full_launch_agent_uninstall,
+    start_launch_agent,
+    stop_launch_agent,
+    view_launch_agent_logs,
+)
+
+from .launch_service import (
+    full_service_install,
+    full_service_uninstall,
+    start_service,
+    stop_service,
+    view_service_logs,
+)
+
+
+class Daemons:
+    def __init__(self, primitive) -> None:
+        self.primitive: Primitive = primitive
+        self.os_family = platform.system()
+
+    def install(self):
+        result = True
+        if self.os_family == "Darwin":
+            full_launch_agent_install()
+        elif self.os_family == "Linux":
+            full_service_install()
+        elif self.os_family == "Windows":
+            print("Not Implemented")
+        return result
+
+    def uninstall(self):
+        result = True
+        if self.os_family == "Darwin":
+            full_launch_agent_uninstall()
+        elif self.os_family == "Linux":
+            full_service_uninstall()
+        elif self.os_family == "Windows":
+            print("Not Implemented")
+        return result
+
+    def stop(self) -> bool:
+        result = True
+        if self.os_family == "Darwin":
+            result = stop_launch_agent()
+        elif self.os_family == "Linux":
+            stop_service()
+        elif self.os_family == "Windows":
+            print("Not Implemented")
+        return result
+
+    def start(self) -> bool:
+        result = True
+        if self.os_family == "Darwin":
+            result = start_launch_agent()
+        elif self.os_family == "Linux":
+            start_service()
+        elif self.os_family == "Windows":
+            print("Not Implemented")
+        return result
+
+    def logs(self):
+        if self.os_family == "Darwin":
+            view_launch_agent_logs()
+        elif self.os_family == "Linux":
+            view_service_logs()
+        elif self.os_family == "Windows":
+            print("Not Implemented")

primitive/daemons/commands.py

@@ -0,0 +1,65 @@
+import click
+
+from ..utils.printer import print_result
+import typing
+
+if typing.TYPE_CHECKING:
+    from ..client import Primitive
+
+
+@click.group()
+@click.pass_context
+def cli(context):
+    """Daemon"""
+    pass
+
+
+@cli.command("install")
+@click.pass_context
+def install_daemon_command(context):
+    """Install the full primitive daemon"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    result = primitive.daemons.install()
+    print_result(message=result, context=context)
+
+
+@cli.command("uninstall")
+@click.pass_context
+def uninstall_daemon_command(context):
+    """Uninstall the full primitive Daemon"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    result = primitive.daemons.uninstall()
+    print_result(message=result, context=context)
+
+
+@cli.command("stop")
+@click.pass_context
+def stop_daemon_command(context):
+    """Stop primitive Daemon"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    result = primitive.daemons.stop()
+    message = "stopping primitive daemon"
+    if context.obj["JSON"]:
+        message = result
+    print_result(message=message, context=context)
+
+
+@cli.command("start")
+@click.pass_context
+def start_daemon_command(context):
+    """Start primitive Daemon"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    result = primitive.daemons.start()
+    message = "starting primitive daemon"
+    if context.obj["JSON"]:
+        message = result
+    print_result(message=message, context=context)
+
+
+@cli.command("logs")
+@click.pass_context
+def log_daemon_command(context):
+    """Logs from primitive Daemon"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    result = primitive.daemons.logs()
+    print_result(message=result, context=context)

primitive/daemons/launch_agents.py

@@ -0,0 +1,144 @@
+import os
+from pathlib import Path
+import subprocess
+
+HOME_DIRECTORY = Path.home()
+CURRENT_USER = str(HOME_DIRECTORY.expanduser()).lstrip("/Users/")
+
+PRIMITIVE_BINARY_PATH = Path(HOME_DIRECTORY / ".pyenv" / "shims" / "primitive")
+PRIMITIVE_LAUNCH_AGENT_FILEPATH = Path(
+    HOME_DIRECTORY / "Library" / "LaunchAgents" / "tech.primitive.agent.plist"
+)
+PRIMITIVE_LAUNCH_AGENT_LOGS = Path(
+    HOME_DIRECTORY / "Library" / "Logs" / "tech.primitive.agent.log"
+)
+PRIMITIVE_LAUNCH_AGENT_LABEL = "tech.primitive.agent"
+PRIMITIVE_LAUNCH_AGENT_WORKING_DIR = Path(
+    HOME_DIRECTORY / "Logs" / "tech.primitive.agent.log"
+)
+
+
+def stop_launch_agent():
+    try:
+        stop_existing_process = f"launchctl stop {PRIMITIVE_LAUNCH_AGENT_LABEL}"
+        subprocess.check_output(stop_existing_process.split(" "))
+        return True
+    except subprocess.CalledProcessError as exception:
+        print("stop_launch_agent: ", exception)
+        return False
+
+
+def start_launch_agent():
+    try:
+        start_new_agent = f"launchctl start {PRIMITIVE_LAUNCH_AGENT_LABEL}"
+        subprocess.check_output(start_new_agent.split(" "))
+        return True
+    except subprocess.CalledProcessError as exception:
+        print("start_launch_agent: ", exception)
+        return False
+
+
+def unload_launch_agent():
+    try:
+        remove_existing_agent = f"launchctl unload -w {PRIMITIVE_LAUNCH_AGENT_FILEPATH}"
+        subprocess.check_output(remove_existing_agent.split(" "))
+        return True
+    except subprocess.CalledProcessError as exception:
+        print("remove_launch_agent: ", exception)
+        return False
+
+
+def load_launch_agent():
+    try:
+        load_new_plist = f"launchctl load -w {PRIMITIVE_LAUNCH_AGENT_FILEPATH}"
+        subprocess.check_output(load_new_plist.split(" "))
+        return True
+    except subprocess.CalledProcessError as exception:
+        print("load_launch_agent: ", exception)
+        return False
+
+
+def create_stdout_file():
+    if not PRIMITIVE_LAUNCH_AGENT_LOGS.exists():
+        PRIMITIVE_LAUNCH_AGENT_LOGS.parent.mkdir(parents=True, exist_ok=True)
+        PRIMITIVE_LAUNCH_AGENT_LOGS.touch()
+
+
+def delete_stdout_file():
+    if PRIMITIVE_LAUNCH_AGENT_LOGS.exists():
+        PRIMITIVE_LAUNCH_AGENT_LOGS.unlink()
+
+
+def populate_fresh_launch_agent():
+    PRIMITIVE_LAUNCH_AGENT_LOGS.parent.mkdir(parents=True, exist_ok=True)
+    PRIMITIVE_LAUNCH_AGENT_LOGS.touch()
+
+    if PRIMITIVE_LAUNCH_AGENT_FILEPATH.exists():
+        PRIMITIVE_LAUNCH_AGENT_FILEPATH.unlink()
+    PRIMITIVE_LAUNCH_AGENT_FILEPATH.parent.mkdir(parents=True, exist_ok=True)
+    PRIMITIVE_LAUNCH_AGENT_FILEPATH.touch()
+
+    PRIMITIVE_LAUNCH_AGENT_FILEPATH.write_text(
+        f"""<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+    <key>KeepAlive</key>
+    <true/>
+    <key>Label</key>
+    <string>{PRIMITIVE_LAUNCH_AGENT_LABEL}</string>
+    <key>LimitLoadToSessionType</key>
+	<array>
+		<string>Aqua</string>
+		<string>Background</string>
+		<string>LoginWindow</string>
+		<string>StandardIO</string>
+	</array>
+    <key>ProgramArguments</key>
+    <array>
+        <string>{PRIMITIVE_BINARY_PATH}</string>
+        <string>agent</string>
+    </array>
+    <key>RunAtLoad</key>
+    <true/>
+    <key>StandardErrorPath</key>
+    <string>{PRIMITIVE_LAUNCH_AGENT_LOGS}</string>
+    <key>StandardOutPath</key>
+    <string>{PRIMITIVE_LAUNCH_AGENT_LOGS}</string>
+</dict>
+</plist>"""  # noqa: E501
+    )
+    PRIMITIVE_LAUNCH_AGENT_FILEPATH.chmod(0o644)
+    verify_launch_agent()
+
+
+def verify_launch_agent():
+    plutil_check = f"plutil -lint {PRIMITIVE_LAUNCH_AGENT_FILEPATH}"
+    try:
+        subprocess.check_output(plutil_check.split(" "))
+        return True
+    except subprocess.CalledProcessError as exception:
+        print("verify_launch_agent: ", exception)
+        return False
+
+
+def view_launch_agent_logs():
+    follow_logs = f"tail -f -n +1 {PRIMITIVE_LAUNCH_AGENT_LOGS}"
+    os.system(follow_logs)
+
+
+def full_launch_agent_install():
+    stop_launch_agent()
+    unload_launch_agent()
+    populate_fresh_launch_agent()
+    create_stdout_file()
+    load_launch_agent()
+    start_launch_agent()
+
+
+def full_launch_agent_uninstall():
+    stop_launch_agent()
+    unload_launch_agent()
+    if PRIMITIVE_LAUNCH_AGENT_FILEPATH.exists():
+        PRIMITIVE_LAUNCH_AGENT_FILEPATH.unlink()
+    delete_stdout_file()

primitive/daemons/launch_service.py

@@ -0,0 +1,170 @@
+import os
+import configparser
+import subprocess
+from pathlib import Path
+
+HOME_DIRECTORY = Path.home()
+
+PRIMITIVE_BINARY_PATH = Path(HOME_DIRECTORY / ".pyenv" / "shims" / "primitive")
+
+PRIMITIVE_AGENT_SERVICE = "tech.primitive.agent.service"
+PRIMITIVE_AGENT_LOGS = "tech.primitive.agent.log"
+PRIMITIVE_AGENT_SERVICE_FILEPATH = Path(
+    HOME_DIRECTORY / ".config" / "systemd" / "user" / PRIMITIVE_AGENT_SERVICE
+)
+PRIMITIVE_AGENT_LOGS_FILEPATH = Path(
+    HOME_DIRECTORY / ".cache" / "primitive" / PRIMITIVE_AGENT_LOGS
+)
+
+
+def stop_service():
+    try:
+        if is_service_active():
+            stop_existing_service = f"systemctl --user stop {PRIMITIVE_AGENT_SERVICE}"
+            subprocess.check_output(stop_existing_service.split(" "))
+        return True
+    except subprocess.CalledProcessError as exception:
+        print("stop_service: ", exception)
+        return False
+
+
+def is_service_active():
+    try:
+        is_service_active = (
+            f"systemctl --user show {PRIMITIVE_AGENT_SERVICE} -p ActiveState --value"
+        )
+        output = subprocess.check_output(is_service_active.split(" ")).decode().strip()
+        return output == "active"
+    except subprocess.CalledProcessError as exception:
+        print("is_service_active: ", exception)
+        return False
+
+
+def disable_service():
+    try:
+        if is_service_enabled():
+            disable_existing_service = (
+                f"systemctl --user disable {PRIMITIVE_AGENT_SERVICE}"
+            )
+            subprocess.check_output(disable_existing_service.split(" "))
+        return True
+    except subprocess.CalledProcessError as exception:
+        print("disable_service: ", exception)
+        return False
+
+
+def is_service_enabled():
+    try:
+        is_service_active = (
+            f"systemctl --user show {PRIMITIVE_AGENT_SERVICE} -p UnitFileState --value"  # noqa
+        )
+        output = subprocess.check_output(is_service_active.split(" ")).decode().strip()
+        return output == "enabled"
+    except subprocess.CalledProcessError as exception:
+        print("is_service_enabled: ", exception)
+        return False
+
+
+def populate_service_file():
+    PRIMITIVE_AGENT_LOGS_FILEPATH.parent.mkdir(parents=True, exist_ok=True)
+    PRIMITIVE_AGENT_LOGS_FILEPATH.touch()
+
+    if PRIMITIVE_AGENT_SERVICE_FILEPATH.exists():
+        PRIMITIVE_AGENT_SERVICE_FILEPATH.unlink()
+    PRIMITIVE_AGENT_SERVICE_FILEPATH.parent.mkdir(parents=True, exist_ok=True)
+    PRIMITIVE_AGENT_SERVICE_FILEPATH.touch()
+
+    config = configparser.ConfigParser()
+    config.optionxform = str  # type: ignore
+
+    config["Unit"] = {
+        "Description": "Primitive Agent",
+        "After": "network.target",
+    }
+
+    config["Service"] = {
+        "ExecStart": f"{PRIMITIVE_BINARY_PATH} agent",
+        "Restart": "always",
+        "StandardError": f"append:{PRIMITIVE_AGENT_LOGS_FILEPATH}",
+        "StandardOutput": f"append:{PRIMITIVE_AGENT_LOGS_FILEPATH}",
+    }
+
+    config["Install"] = {
+        "WantedBy": "default.target",
+    }
+
+    try:
+        with open(PRIMITIVE_AGENT_SERVICE_FILEPATH, "w") as service_file:
+            config.write(service_file)
+    except IOError as exception:
+        print(f"populate_service_file: {exception}")
+
+    PRIMITIVE_AGENT_SERVICE_FILEPATH.chmod(0o644)
+    verify_service_file()
+
+
+def verify_service_file():
+    systemctl_check = (
+        f"systemctl --user show {PRIMITIVE_AGENT_SERVICE} -p CanStart --value"
+    )
+    try:
+        output = subprocess.check_output(systemctl_check.split(" ")).decode().strip()
+        if output == "no":
+            raise Exception(f"{systemctl_check} yielded {output}")
+        return True
+    except subprocess.CalledProcessError as exception:
+        print("verify_service_file: ", exception)
+        return False
+
+
+def create_stdout_file():
+    if not PRIMITIVE_AGENT_LOGS_FILEPATH.exists():
+        PRIMITIVE_AGENT_LOGS_FILEPATH.parent.mkdir(parents=True, exist_ok=True)
+        PRIMITIVE_AGENT_LOGS_FILEPATH.touch()
+
+
+def delete_stdout_file():
+    if PRIMITIVE_AGENT_LOGS_FILEPATH.exists():
+        PRIMITIVE_AGENT_LOGS_FILEPATH.unlink()
+
+
+def enable_service():
+    try:
+        enable_service = f"systemctl --user enable {PRIMITIVE_AGENT_SERVICE}"
+        subprocess.check_output(enable_service.split(" "))
+        return True
+    except subprocess.CalledProcessError as exception:
+        print("enable_service: ", exception)
+        return False
+
+
+def start_service():
+    try:
+        start_new_service = f"systemctl --user start {PRIMITIVE_AGENT_SERVICE}"
+        subprocess.check_output(start_new_service.split(" "))
+        return True
+    except subprocess.CalledProcessError as exception:
+        print("start_service: ", exception)
+        return False
+
+
+def view_service_logs():
+    follow_logs = f"tail -f -n +1 {PRIMITIVE_AGENT_LOGS_FILEPATH}"
+    os.system(follow_logs)
+
+
+def full_service_install():
+    stop_service()
+    disable_service()
+    populate_service_file()
+    create_stdout_file()
+    enable_service()
+    start_service()
+
+
+def full_service_uninstall():
+    stop_service()
+    disable_service()
+    if PRIMITIVE_AGENT_SERVICE_FILEPATH.exists():
+        PRIMITIVE_AGENT_SERVICE_FILEPATH.unlink()
+    delete_stdout_file()

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: primitive
-Version: 0.1.12
+Version: 0.1.13
 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.12.dist-info → primitive-0.1.13.dist-info}/RECORD

@@ -1,12 +1,16 @@
-primitive/__about__.py,sha256=35zrg1jAPnEjmmKrCnH1TpiDYB4jwmc5kkbIoAHlG84,129
+primitive/__about__.py,sha256=xMBcV0dgZBAcVvP4QqmQUK6YrJ-xPRUcJbx7NA4hQWw,129
 primitive/__init__.py,sha256=bwKdgggKNVssJFVPfKSxqFMz4IxSr54WWbmiZqTMPNI,106
-primitive/cli.py,sha256=fWv1gnUs7U6yWRrv1fYjt2bNBclIvp6okKTxrl_Il24,1879
-primitive/client.py,sha256=dkbHg2Pbxl7NjIk2O6Shvz7TE7uDQezGQkNwmTzKdNs,2146
+primitive/cli.py,sha256=2IEAdHj93nmWaDauyua3hw0kFsqhW2zPPVB7W05d7-Q,1978
+primitive/client.py,sha256=A2Ir9KRnm7VptG6iUba4zihx12SzQZLnsF19neyJmdY,2229
 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/daemons/actions.py,sha256=Nt3yNtbBhen0jK4sRsH_N7AP3UBuyL48VaUhtC7wYq8,2015
+primitive/daemons/commands.py,sha256=-Muh-6ib4uAVtPn_67AcMrDwuCwYlCnRQozCi2Xurmk,1726
+primitive/daemons/launch_agents.py,sha256=xR_JDL93mt5-fSbj-v3qvYFvLqfn486fPNbNxZOb1wU,4412
+primitive/daemons/launch_service.py,sha256=3ktr0n0rSNA-IV6hQ8hbG50L2Y3V8-IeW79C8CrZi8U,5203
 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
@@ -30,8 +34,8 @@ primitive/utils/memory_size.py,sha256=4xfha21kW82nFvOTtDFx9Jk2ZQoEhkfXii-PGNTpIU
 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.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,,
+primitive-0.1.13.dist-info/METADATA,sha256=fUowLloXzAcRrLMhUloZAmRpiOLiVMXH8PZspjLb0YE,1818
+primitive-0.1.13.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+primitive-0.1.13.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
+primitive-0.1.13.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
+primitive-0.1.13.dist-info/RECORD,,