import tab from "@bomb.sh/tab/citty";
import { defineCommand } from "citty";
import { db } from "./db";
import { down } from "./down";
import { exec } from "./exec";
import { exportCommand } from "./export";
import { logs } from "./logs";
import { ps } from "./ps";
import { restart } from "./restart";
import { rollback } from "./rollback";
import { s3 } from "./s3";
import { start } from "./start";
import { stop } from "./stop";
import { up } from "./up";

export const main = defineCommand({
  meta: {
    name: "kuber",
    version: "1.3.0",
    description: "Docker Compose -> K8s translation layer",
  },
  args: {
    config: {
      type: "string",
      description: "Path to a kuber configuration file",
    },
  },
  subCommands: {
    db,
    down,
    export: exportCommand,
    exec,
    logs,
    ps,
    restart,
    rollback,
    s3,
    start,
    stop,
    up,
  },
});

let completion: ReturnType<typeof tab> | undefined;

export function initializeCompletion() {
  completion ??= tab(main);
  return completion;
}