arkitekt-next 0.15.0__py3-none-any.whl → 0.17.0__py3-none-any.whl

arkitekt_next/bloks/socket.py

@@ -1,40 +0,0 @@
-from pydantic import BaseModel
-from typing import Dict, Any
-from blok import blok, InitContext, Option, ExecutionContext
-from blok.tree import YamlFile
-
-
-class AdminCredentials(BaseModel):
-    password: str
-    username: str
-    email: str
-
-
-@blok(
-    "live.arkitekt.docker_socket", description="Hosting the docker socket of the host"
-)
-class DockerSocketBlok:
-    def __init__(self) -> None:
-        self.docker_socket = "/var/run/docker.sock"
-        self.registered_configs = {}
-
-    def preflight(self, init: InitContext):
-        for key, value in init.kwargs.items():
-            setattr(self, key, value)
-
-    def register_socket(self, name: str) -> str:
-        self.registered_configs[name] = name
-        return self.docker_socket
-
-    def build(self, ex: ExecutionContext):
-        pass
-
-    def get_options(self):
-        config_path = Option(
-            subcommand="docker_socket",
-            help="Which docker_socket to use for configs",
-            default=self.docker_socket,
-            show_default=True,
-        )
-
-        return [config_path]

arkitekt_next/bloks/tailscale.py

@@ -1,90 +0,0 @@
-from pydantic import BaseModel
-from typing import Dict, Any
-from arkitekt_next.bloks.services.mount import MountService
-from blok import blok, InitContext, Option, ExecutionContext
-from arkitekt_next.bloks.services.gateway import GatewayService
-from blok.bloks.services.dns import DnsService
-from blok.tree import YamlFile
-
-
-@blok("live.arkitekt.tailscale", description="Tailscale is a zero config VPN")
-class TailscaleBlok:
-    def __init__(self, name: str = "arkitekt") -> None:
-        self.name = name
-        self.disable = False
-
-    def preflight(
-        self,
-        gateway: GatewayService,
-        mount_service: MountService,
-        dns: DnsService,
-        net_name: str,
-        auth_key: str,
-        host_name: str,
-        disable: bool = False,
-    ):
-        self.disable = disable
-        if self.disable:
-            return
-        assert auth_key, "You need to provide an auth_key"
-
-        self.caddy_service = gateway.get_internal_host()
-        self.caddy_port = gateway.get_http_port()
-        self.mount = mount_service.register_mount("tailscale_state", {})
-        self.tailnet_name = net_name
-        self.auth_key = auth_key
-        self.dns_service = dns.get_dns_result()
-        self.host_name = host_name
-
-    def build(self, ex: ExecutionContext):
-        if self.disable:
-            return
-
-        if not self.tailnet_name:
-            # Trying to automatically find the hostname
-            for i in self.dns_service.hostnames:
-                if i.endswith(".ts.net"):
-                    self.tailnet_name = i.split(".")[1]
-                    break
-
-        service = {
-            "image": "jhnnsrs/tailproxy:latest",
-            "volumes": [
-                f"{self.mount}:/var/lib/tailscale"  # Persist the tailscale state directory
-            ],
-            "environment": [
-                "TS_STATE_DIR=/var/lib/tailscale",
-                f"TS_AUTH_KEY={self.auth_key}",
-                f"TS_HOSTNAME={self.host_name}",
-                f"TS_TAILNET={self.tailnet_name}",
-                f"CADDY_TARGET={self.caddy_service}:{self.caddy_port}",
-            ],
-        }
-
-        ex.docker_compose.set_nested("services", "tailscale_proxy", service)
-
-    def get_options(self):
-        return [
-            Option(
-                subcommand="net_name",
-                help="The name of your tailnet",
-                type=str,
-            ),
-            Option(
-                subcommand="auth_key",
-                help="The auth_key of your tailnet",
-                type=str,
-            ),
-            Option(
-                subcommand="host_name",
-                help="The hostname of your tailnet",
-                default=self.name,
-                type=str,
-            ),
-            Option(
-                subcommand="disable",
-                help="Should we disable the creation of the tailscale service?",
-                default=self.disable,
-                type=str,
-            ),
-        ]