Auto commit: 2026-03-24 13:04

src/routes/docker.ts

@@ -0,0 +1,90 @@
+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)