deploy-k3s.py

# SPDX-FileCopyrightText: 2026 Kristofer Älvring <kristofer@mainly.ai>
# SPDX-FileCopyrightText: 2026 Leah Lundqvist <leah@mainly.ai>
#
# SPDX-License-Identifier: GPL-2.0-only

import json
import os
import subprocess
import time

reg = os.environ.get("REGISTRY", "registry.production.mainly.ai")
should_build = os.environ.get("ONLY_BUILD", "ws,vault,webservices,webstatic").split(",")


def try_os_system(cmd, errmsg):
    print(f"$ {cmd}")
    res = os.system(cmd)
    if res != 0 and "IGNORE_DOCKER_BUILD_ERRORS" not in os.environ:
        print(f"!! {errmsg} !!")
        exit()


def load_config():
    config_keys = [
        "VITE_DEPLOYMENT_BASE",
        "VITE_COPILOT_URL",
        "VITE_STRIPE_PK",
        "VITE_TURNSTILE_SITEKEY",
        "VITE_NODE_ENV",
        "VITE_ASKAI2_URL",
        "VITE_ASKAI2_DIFF_FIX_URL",
        "VITE_TEMPLATE_REPO_ID",
    ]
    config = {}
    if os.path.exists("deploy_config.json"):
        with open("deploy_config.json") as f:
            config = json.load(f)

    changed = False
    for key in config_keys:
        if key not in config:
            config[key] = input(f"{key}=")
            changed = True

    if changed:
        with open("deploy_config.json", "w") as f:
            json.dump(config, f)

    return config


def build_images():
    config = load_config()
    envstr = " ".join([f"{k}={v}" for k, v in config.items()])

    if "webstatic" in should_build:
        try_os_system(
            f'{envstr} NODE_OPTIONS="--max-old-space-size=4096" npm run build',
            "Failed to build webstatic",
        )

    if "webservices" in should_build:
        try_os_system(
            "rm -rf apps/webservice/dist/", "Failed to remove webservice dist directory"
        )
        try_os_system(
            "mkdir -p apps/webservice/dist/",
            "Failed to create webservice dist directory",
        )
        try_os_system(
            "cp -r ../mainlyai/dist/*.whl apps/webservice/dist/",
            "Failed to copy webservice wheel",
        )

    if "ws" in should_build:
        try_os_system(
            "cd apps/ws-gateway && cargo build --release", "Failed to build ws-gateway"
        )
    if "vault" in should_build:
        try_os_system("cd apps/vault && cargo build --release", "Failed to build vault")

    # Generate unique build hash for service worker cache invalidation
    build_hash = hex(int(time.time() * 1000))[2:]  # Millisecond timestamp as hex

    if "ws" in should_build:
        try_os_system(
            f"docker build --platform linux/amd64 -t {reg}/ws-gateway -f apps/ws-gateway/Dockerfile apps/ws-gateway",
            "Failed to build ws-gateway image",
        )
    if "vault" in should_build:
        try_os_system(
            f"docker build --platform linux/amd64 -t {reg}/vault -f apps/vault/Dockerfile apps/vault",
            "Failed to build vault image",
        )
    if "webservices" in should_build:
        try_os_system(
            f"docker build --platform linux/amd64 -t {reg}/webservices -f apps/webservice/Dockerfile apps/webservice",
            "Failed to build webservice image",
        )
    if "webstatic" in should_build:
        print(f"Building webstatic with build hash: {build_hash}")
        try_os_system(
            f"docker build --platform linux/amd64 --build-arg BUILD_HASH={build_hash} -t {reg}/webstatic -f apps/Dockerfile apps",
            "Failed to build webstatic image",
        )
    if "ws" in should_build:
        try_os_system(
            f"docker push {reg}/ws-gateway ", "Failed to push ws-gateway  image"
        )
    if "vault" in should_build:
        try_os_system(f"docker push {reg}/vault ", "Failed to push vault image")
    if "webservices" in should_build:
        try_os_system(
            f"docker push {reg}/webservices", "Failed to push webservice image"
        )
    if "webstatic" in should_build:
        try_os_system(f"docker push {reg}/webstatic", "Failed to push webstatic image")


def deploy():
    print(
        "! kubeconfig is currently uncopied, ssh into the host to restart the deployments"
    )


#   try_os_system('kubectl apply -f webservices.yml', 'Failed to create namespace')
#   try_os_system('kubectl rollout restart deployment ws-gateway', 'Failed to restart ws-gateway deployment')
#   try_os_system('kubectl rollout restart deployment vault', 'Failed to restart vault deployment')
#   try_os_system('kubectl rollout restart deployment webservices', 'Failed to restart webservices deployment')
#   try_os_system('kubectl rollout restart deployment webstatic', 'Failed to restart webstatic deployment')

if __name__ == "__main__":
    build_images()

    print("Do you want to deploy? (y/n) ")
    res = input()
    if res != "y":
        print("Aborting")
        exit()

    deploy()