WIP create command

Author: matthew-bernardo

cli/src/stacksync_cli/commands/create.py

@@ -1,16 +1,69 @@
-from ..utils import with_auth
+import os
+import re
+import shutil
+import tempfile
+import urllib.request
+import zipfile
+
 import click
 
+from ..utils import with_auth
+
 
 @click.command()
 @click.argument("app-name", type=str, required=True)
-@click.option("--crawl", type=str, required=True, help="A URL which should be crawled by AI to generate the app.")
-@click.option("--prompt", type=str, required=True, help="A prompt to guide the AI in generating the app.")
+@click.option("--crawl", type=str, required=False, help="A URL which should be crawled by AI to generate the app.")
+@click.option("--prompt", type=str, required=False, help="A prompt to guide the AI in generating the app.")
 @with_auth
-def create(api_key: str, app_name: str, crawl: str, prompt: str) -> None:
+def create(app_name: str, crawl: str, prompt: str, api_key: str) -> None:
     """
-    The dev command create a temporary private app in a Stacksync workspace,
-    runs a Stacksync application on localhost,
-    and uses the local bridge to route traffic to the application.
+    Create a new app folder in the current directory from the connector template.
     """
-    print("Oh boy", api_key)
+    target_dir = os.path.join(os.getcwd(), _to_kebab_case(app_name))
+    if os.path.exists(target_dir):
+        raise click.ClickException(
+            f"Destination already exists: {os.path.basename(target_dir)!r}"
+        )
+
+    temp_folder = _get_templates_folder()
+    template_dir = _get_connector_template_dir(temp_folder)
+
+    try:
+        shutil.copytree(template_dir, target_dir)
+    finally:
+        shutil.rmtree(temp_folder, ignore_errors=True)
+
+    click.secho(f"Created app in {target_dir}", fg="green")
+
+
+def _to_kebab_case(value: str) -> str:
+    normalized = re.sub(r"([a-z0-9])([A-Z])", r"\1-\2", value.strip())
+    normalized = re.sub(r"[^A-Za-z0-9]+", "-", normalized)
+    normalized = normalized.strip("-").lower()
+    if not normalized:
+        raise click.ClickException("App name must contain at least one letter or number.")
+    return normalized
+
+
+def _get_templates_folder() -> str:
+    """
+    This function downloads a zip from Github and extracts it to a temporary folder.
+    """
+    temp_folder = tempfile.mkdtemp()
+    zip_url = "https://github.com/stacksyncdata/cdk/archive/refs/heads/prod.zip"
+    zip_path = os.path.join(temp_folder, "cdk.zip")
+    with urllib.request.urlopen(zip_url) as response, open(zip_path, "wb") as out_file:
+        shutil.copyfileobj(response, out_file)
+    with zipfile.ZipFile(zip_path, "r") as zip_ref:
+        zip_ref.extractall(temp_folder)
+    return temp_folder
+
+
+def _get_connector_template_dir(temp_folder: str) -> str:
+    extracted_root = os.path.join(temp_folder, "cdk-prod")
+    template_dir = os.path.join(extracted_root, "templates", "connector")
+    if not os.path.isdir(template_dir):
+        raise click.ClickException(
+            "Failed to locate the connector template in the downloaded archive."
+        )
+    return template_dir

cli/src/stacksync_cli/commands/env.py

@@ -0,0 +1,15 @@
+import click
+from ..utils import DEFAULTS, get_env_file, update_env_file
+
+def complete_env_key(ctx: click.Context, param: click.Parameter, incomplete: str):
+    return [key for key in DEFAULTS if key.startswith(incomplete)]
+
+
+@click.command(help="Get or set environment variables. Available keys: " + ", ".join(DEFAULTS.keys()))
+@click.argument("key", type=str, required=True, shell_complete=complete_env_key)
+@click.argument("value", type=str, required=False)
+def env(key: str, value: str) -> None:
+    if (not value):
+        click.echo(get_env_file()[key])
+    else:
+        update_env_file({key: value})

cli/src/stacksync_cli/main.py

@@ -3,6 +3,7 @@
 from .commands.dev import dev as dev_cmd
 from .commands.create import create as create_cmd
 from .commands.hello import hello as hello_cmd
+from .commands.env import env as env_cmd
 
 
 @click.group(context_settings={"help_option_names": ["-h", "--help"]})
@@ -14,3 +15,4 @@ def cli() -> None:
 cli.add_command(dev_cmd)
 cli.add_command(create_cmd)
 cli.add_command(hello_cmd)
+cli.add_command(env_cmd)

cli/src/stacksync_cli/utils.py

@@ -21,9 +21,11 @@ def print_logo() -> None:
     )
 
 
+# TODO: Move this somewhere nicer. They're not very sensitive so this will do the trick for now.
 DEFAULTS = {
-    "local_bridge_url": "ws://localhost:8787",
-    "login_url": "http://localhost:8080/cdk/login",
+    "local_bridge_url": "https://local-bridge-prod-besg-1011549189584.europe-west1.run.app/",
+    "login_url": "https://app.stacksync.com/cdk/login",
+    "templates_source_branch": "prod",
 }