import { app } from '../app.ts';
import { z } from 'zod';
import { chalk } from '@/module/chalk.ts';
import { spawn } from 'node:child_process';
import { useKey } from '@kevisual/use-config';

app.route({
  path: 'docker',
  key: 'login',
  description: '登录 Docker 镜像仓库',
  metadata: {
    args: {
      registry: z.string().optional().describe('Docker 镜像仓库地址'),
    }
  }
}).define(async (ctx) => {
  const registry = ctx.args.registry || 'default';
  let DOCKER_USERNAME = useKey('DOCKER_USERNAME') as string;
  let DOCKER_PASSWORD = useKey('DOCKER_PASSWORD') as string;
  let DOCKER_REGISTRY = useKey('DOCKER_REGISTRY') as string;

  if (registry !== 'default') {
    DOCKER_USERNAME = 'cnb';
    DOCKER_PASSWORD = useKey('CNB_TOKEN') as string;
    DOCKER_REGISTRY = 'docker.cnb.cool';
  }
  if (!DOCKER_USERNAME || !DOCKER_PASSWORD) {
    console.log(chalk.red('请先配置 DOCKER_USERNAME 和 DOCKER_PASSWORD'));
    return;
  }
  const loginProcess = spawn('docker', [
    'login',
    '--username',
    DOCKER_USERNAME,
    DOCKER_REGISTRY,
    '--password-stdin'
  ], {
    stdio: ['pipe', 'inherit', 'inherit']
  });

  loginProcess.stdin.write(DOCKER_PASSWORD + '\n');
  loginProcess.stdin.end();

  loginProcess.on('close', (code) => {
    if (code === 0) {
      console.log(chalk.green('登录成功'));
    } else {
      console.log(chalk.red(`登录失败，退出码：${code}`));
    }
  });
}).addTo(app)

app.route({
  path: 'helm',
  key: 'login',
  description: '登录 Helm 镜像仓库',
  metadata: {
    args: {}
  }
}).define(async () => {
  let DOCKER_USERNAME = 'cnb';
  let DOCKER_PASSWORD = useKey('CNB_TOKEN') as string;

  if (!DOCKER_PASSWORD) {
    console.log(chalk.red('请先配置 CNB_TOKEN'));
    return;
  }

  const helmLoginProcess = spawn('helm', [
    'registry',
    'login',
    '--username',
    DOCKER_USERNAME,
    '--password-stdin',
    'helm.cnb.cool'
  ], {
    stdio: ['pipe', 'inherit', 'inherit']
  });

  helmLoginProcess.stdin.write(DOCKER_PASSWORD + '\n');
  helmLoginProcess.stdin.end();

  helmLoginProcess.on('close', (code) => {
    if (code === 0) {
      console.log(chalk.green('Helm 登录成功'));
    } else {
      console.log(chalk.red(`Helm 登录失败，退出码：${code}`));
    }
  });
}).addTo(app)