arkitekt-next 0.7.8__py3-none-any.whl

arkitekt_next/cli/commands/run/prod.py

@@ -0,0 +1,57 @@
+import rich_click as click
+from arkitekt_next.cli.options import *
+import asyncio
+from arkitekt_next.cli.ui import construct_run_panel
+from importlib import import_module
+from .utils import import_builder
+from arkitekt_next.constants import DEFAULT_ARKITEKT_URL
+
+async def run_app(app):
+    async with app:
+        await app.rekuest.run()
+
+
+@click.command("prod")
+@click.option(
+    "--url",
+    help="The fakts url for connection",
+    default=DEFAULT_ARKITEKT_URL,
+    envvar="FAKTS_URL",
+)
+@with_builder
+@with_token
+@with_instance_id
+@with_headless
+@with_log_level
+@with_skip_cache
+@click.pass_context
+def prod(ctx, entrypoint=None, builder=None, **builder_kwargs):
+    """Runs the app in production mode
+
+    \n
+    You can specify the builder to use with the --builder flag. By default, the easy builder is used, which is designed to be easy to use and to get started with.
+
+    """
+
+    manifest = get_manifest(ctx)
+    console = get_console(ctx)
+    entrypoint = entrypoint or manifest.entrypoint
+
+    builder = import_builder(builder)
+
+    with console.status("Loading entrypoint module..."):
+        try:
+            import_module(entrypoint)
+        except ModuleNotFoundError as e:
+            console.print(f"Could not find entrypoint module {entrypoint}")
+            raise e
+
+    app = builder(
+        **manifest.to_builder_dict(),
+        **builder_kwargs,
+    )
+
+    panel = construct_run_panel(app)
+    console.print(panel)
+
+    asyncio.run(run_app(app))

arkitekt_next/cli/commands/run/utils.py

@@ -0,0 +1,10 @@
+from importlib import import_module
+from typing import Callable
+from arkitekt_next.apps.types import App
+
+
+def import_builder(builder: str) -> Callable[..., App]:
+    module_path, function_name = builder.rsplit(".", 1)
+    module = import_module(module_path)
+    function = getattr(module, function_name)
+    return function

arkitekt_next/cli/commands/server/down.py

@@ -0,0 +1,56 @@
+import rich_click as click
+from click import Context
+from dokker.projects.dokker import DokkerProject
+from dokker.loggers.print import PrintLogger
+from dokker.deployment import Deployment
+from typing import Optional
+import os
+from .utils import compile_options
+
+
+@click.command()
+@click.option(
+    "--name",
+    help="The name of the deployment",
+    default=None,
+    required=False,
+    type=click.Choice(compile_options()),
+)
+@click.pass_context
+def down(
+    ctx: Context,
+    name: Optional[str] = None,
+) -> None:
+    """
+    Down a deployment
+
+    Removing a deployment will stop all containers and call docker compose down
+    on the project. This will remove all containers and networks created by the
+    deployment. The deployment will still be available in the .dokker folder
+    However (depending on your settings) data that was stored in volumes managed
+    by the deployment will be removed.
+
+    If you want to simple Stop the deployment, use the stop command instead.
+
+    """
+    if not name:
+        options = compile_options()
+        if not options:
+            raise click.ClickException(
+                "No deployments found. Please run arkitekt_next server init first"
+            )
+
+        name = options[0]
+
+    print(f"Running {name}")
+
+    project = DokkerProject(
+        name=name,
+    )
+
+    deployment = Deployment(project=project, logger=PrintLogger())
+    with deployment:
+        deployment.down()
+        print("Shutting down...")
+
+    print("Done")

arkitekt_next/cli/commands/server/init.py

@@ -0,0 +1,74 @@
+import rich_click as click
+from click import Context
+from dokker.projects.contrib.konstruktor import KonstruktorProject, RemoteRepo
+import asyncio
+from typing import Optional
+import os
+from arkitekt_next.cli.vars import get_console
+
+DEFAULT_REPO_URL = "https://arkitekt_next.live/repo.json"
+
+
+@click.command()
+@click.option(
+    "--repo-url",
+    help="The repo to use",
+    default=DEFAULT_REPO_URL,
+    prompt="What repo do you want to use?",
+)
+@click.option(
+    "--channel",
+    help="The channel to use",
+    default="beta",
+    prompt="What channel do you want to use?",
+)
+@click.option(
+    "--name",
+    help="The name of the deployment",
+    default=None,
+    required=False,
+)
+@click.option(
+    "--overwrite",
+    help="Should we overwrite the old deployment?",
+    default=False,
+    is_flag=True,
+)
+@click.option(
+    "--setup",
+    "-s",
+    "setup",
+    help="Additional setup variables",
+    type=(str, str),
+    multiple=True,
+)
+@click.pass_context
+def init(
+    ctx: Context,
+    channel: str,
+    repo_url: str,
+    overwrite: bool = False,
+    setup: Optional[list] = None,
+    name: Optional[str] = None,
+) -> None:
+    """
+    Initializes a new server
+
+    """
+
+    extra_content = dict(setup or [])
+
+    project = KonstruktorProject(
+        channel=channel,
+        repo=RemoteRepo(url=repo_url),
+        reinit_if_exists=overwrite,
+        name=name,
+        extra_context=extra_content,
+    )
+
+    try:
+        asyncio.run(project.ainititialize())
+        get_console(ctx).print("Done :)")
+    except Exception as e:
+        get_console(ctx).print_exception()
+        raise click.ClickException("Failed to initialize project") from e

arkitekt_next/cli/commands/server/inspect.py

@@ -0,0 +1,59 @@
+import rich_click as click
+from click import Context
+from dokker.projects.dokker import DokkerProject
+from dokker.loggers.print import PrintLogger
+from dokker.deployment import Deployment
+from typing import Optional
+import os
+from .utils import compile_options
+from rich.table import Table
+from rich.live import Live
+from arkitekt_next.cli.vars import get_console
+import json
+
+DEFAULT_REPO_URL = (
+    "https://raw.githubusercontent.com/jhnnsrs/konstruktor/master/repo/channels.json"
+)
+
+
+@click.command()
+@click.option(
+    "--name",
+    help="The name of the deployment",
+    default=None,
+    required=False,
+    type=click.Choice(compile_options()),
+)
+@click.pass_context
+def inspect(
+    ctx: Context,
+    name: Optional[str] = None,
+) -> None:
+    """
+    Inspect a deployment
+
+    Inspect helps you find information about a deployment. It will show you
+    information about the containers, networks and volumes created by the
+    deployment. If you have not run arkitekt_next server init before, this will
+    fail.
+
+    """
+    if not name:
+        options = compile_options()
+        if not options:
+            raise click.ClickException(
+                "No deployments found. Please run arkitekt_next server init first"
+            )
+
+        name = options[0]
+
+    project = DokkerProject(
+        name=name,
+    )
+
+    console = get_console(ctx)
+
+    deployment = Deployment(project=project)
+
+    with deployment:
+        print(json.dumps(deployment.inspect().dict(), indent=4))

arkitekt_next/cli/commands/server/main.py

@@ -0,0 +1,33 @@
+import rich_click as click
+from .init import init
+from .up import up
+from .down import down
+from .inspect import inspect
+from .remove import remove
+from .open import open
+from .stop import stop
+from click import Context
+
+
+@click.group()
+@click.pass_context
+def server(ctx: Context) -> None:
+    """Manages your arkitekt_next deployments
+
+    You can directly deploy your app to a server with the arkitekt_next server command. This will
+    create a local docker-compose.yml file and a .arkitekt_next folder in your project. You can then
+    use the arkitekt_next server build command to build your app for port and the arkitekt_next server run
+    command to run your app on the server.
+
+    """
+
+    pass
+
+
+server.add_command(init, "init")
+server.add_command(up, "up")
+server.add_command(down, "down")
+server.add_command(remove, "remove")
+server.add_command(inspect, "inspect")
+server.add_command(open, "open")
+server.add_command(stop, "stop")

arkitekt_next/cli/commands/server/open.py

@@ -0,0 +1,66 @@
+import rich_click as click
+from click import Context
+from dokker.projects.dokker import DokkerProject
+from dokker.loggers.print import PrintLogger
+from dokker.deployment import Deployment
+from typing import Optional, List
+import os
+from .utils import compile_options
+from rich.table import Table
+from rich.live import Live
+from arkitekt_next.cli.vars import get_console
+import webbrowser
+
+DEFAULT_REPO_URL = (
+    "https://raw.githubusercontent.com/jhnnsrs/konstruktor/master/repo/channels.json"
+)
+
+
+@click.command()
+@click.option(
+    "--name",
+    help="The name of the deployment",
+    default=None,
+    required=False,
+    type=click.Choice(compile_options()),
+)
+@click.argument("services", nargs=-1, required=False)
+@click.pass_context
+def open(
+    ctx: Context,
+    name: Optional[str] = None,
+    services: List[str] = None,
+) -> None:
+    """
+    Opens a service in the browser
+
+    This will open the service in the browser. If you have not run arkitekt_next server init
+    before, this will fail.
+
+
+    """
+    if not name:
+        options = compile_options()
+        if not options:
+            raise click.ClickException(
+                "No deployments found. Please run arkitekt_next server init first"
+            )
+
+        name = options[0]
+
+    project = DokkerProject(
+        name=name,
+    )
+
+    console = get_console(ctx)
+
+    deployment = Deployment(project=project)
+
+    if not services:
+        services = ["orkestrator"]
+
+    with deployment:
+        for service in services:
+            link = deployment.inspect().find_service(service).get_label("arkitekt_next.link")
+            console.print(f"Opening {service} at [link={link}]{link}[/link]")
+            webbrowser.open(link)

arkitekt_next/cli/commands/server/remove.py

@@ -0,0 +1,60 @@
+import rich_click as click
+from click import Context
+from dokker.projects.dokker import DokkerProject
+from dokker.loggers.print import PrintLogger
+from dokker.deployment import Deployment
+from typing import Optional
+import os
+from .utils import compile_options
+
+
+@click.command()
+@click.option(
+    "--name",
+    help="The name of the deployment",
+    default=None,
+    required=False,
+    type=click.Choice(compile_options()),
+)
+@click.pass_context
+def remove(
+    ctx: Context,
+    name: Optional[str] = None,
+) -> None:
+    """
+    Remove a deployment
+
+    Removing a deployment will stop all containers and call docker compose down
+    on the project. This will remove all containers and networks created by the
+    deployment. As opposed to the down command, the deployment will also be
+    removed from the .dokker folder, and all data stored in volumes managed by
+    the deployment will be removed.
+
+    This step is irreversible, so use with caution.
+
+    """
+    if not name:
+        options = compile_options()
+        if not options:
+            raise click.ClickException(
+                "No deployments found. Please run arkitekt_next server init first"
+            )
+
+        name = options[0]
+
+    print(f"Running {name}")
+
+    project = DokkerProject(
+        name=name,
+    )
+
+    deployment = Deployment(project=project, logger=PrintLogger())
+
+    with deployment:
+        print("Shutting down...")
+        deployment.down()
+
+        print("Removing...")
+        deployment.remove()
+
+    print("Done")

arkitekt_next/cli/commands/server/stop.py

@@ -0,0 +1,56 @@
+import rich_click as click
+from click import Context
+from dokker.projects.dokker import DokkerProject
+from dokker.loggers.print import PrintLogger
+from dokker.deployment import Deployment
+from typing import Optional
+import os
+from .utils import compile_options
+
+
+@click.command()
+@click.option(
+    "--name",
+    help="The name of the deployment",
+    default=None,
+    required=False,
+    type=click.Choice(compile_options()),
+)
+@click.pass_context
+def stop(
+    ctx: Context,
+    name: Optional[str] = None,
+) -> None:
+    """
+    Stop a deployment
+
+    Stopping a deployment will stop all containers and call docker compose stop
+    on the project. This will stop all containers and networks created by the
+    deployment. The deployment will still be available in the .dokker folder
+
+    This should not be confused with the down command, which will remove all
+    containers and remove all networks created by the deployment.
+
+    """
+    if not name:
+        options = compile_options()
+        if not options:
+            raise click.ClickException(
+                "No deployments found. Please run arkitekt_next server init first"
+            )
+
+        name = options[0]
+
+    print(f"Stopping {name}")
+
+    project = DokkerProject(
+        name=name,
+    )
+
+    deployment = Deployment(project=project, logger=PrintLogger())
+
+    with deployment:
+        print("Shutting down...")
+        deployment.stop()
+
+    print("Done")

arkitekt_next/cli/commands/server/up.py

@@ -0,0 +1,70 @@
+import rich_click as click
+from click import Context
+from dokker.projects.dokker import DokkerProject
+from dokker.loggers.print import PrintLogger
+from dokker.deployment import Deployment
+from typing import Optional
+import os
+from .utils import compile_options
+from rich.table import Table
+from rich.live import Live
+from arkitekt_next.cli.vars import get_console
+
+
+DEFAULT_REPO_URL = (
+    "https://raw.githubusercontent.com/jhnnsrs/konstruktor/master/repo/channels.json"
+)
+
+
+@click.command()
+@click.option(
+    "--name",
+    help="The name of the deployment",
+    default=None,
+    required=False,
+    type=click.Choice(compile_options()),
+)
+@click.pass_context
+def up(
+    ctx: Context,
+    name: Optional[str] = None,
+) -> None:
+    """
+    Ups a deployment
+
+    Up starts your deployment. This will create all containers and networks
+    needed for your deployment. If you have not run arkitekt_next server init
+    before, this will fail.
+
+
+    """
+    if not name:
+        options = compile_options()
+        if not options:
+            raise click.ClickException(
+                "No deployments found. Please run arkitekt_next server init first"
+            )
+
+        name = options[0]
+
+    print(f"Running {name}")
+
+    project = DokkerProject(
+        name=name,
+    )
+
+    console = get_console(ctx)
+
+    logger = PrintLogger(print_function=lambda x: console.log(x))
+
+    deployment = Deployment(project=project, logger=logger)
+
+    with deployment:
+        port = deployment.inspect().find_service("lok").get_label("arkitekt_next.link")
+
+        status = console.status(
+            f"[bold green]Running on the Lok Server Port {port} Press Ctrl+c once to cancel..."
+        )
+
+        with status:
+            x = deployment.up(detach=False)

arkitekt_next/cli/commands/server/utils.py

@@ -0,0 +1,33 @@
+import os
+from typing import List
+
+
+def compile_options() -> List[str]:
+    """Compiles the available options for the server up and down commands:
+
+    All deployments in the .arkitekt_next/deployments folder
+    """
+    deployments = os.path.join(os.getcwd(), ".dokker")
+
+    available_deployments = []
+
+    if not os.path.exists(deployments):
+        return available_deployments
+
+    for deployment in os.listdir(deployments):
+        if os.path.isdir(os.path.join(deployments, deployment)):
+            available_deployments.append(deployment)
+
+    return available_deployments
+
+
+def build_server_path(name: str) -> str:
+    """Builds the path to the server
+
+    Args:
+        name (str): The name of the server
+
+    Returns:
+        str: The path to the server
+    """
+    return os.path.join(os.getcwd(), ".arkitekt_next", "servers", name)