@icyfenix-dmla/cli 2026.4.17-546

package/bin/dmla.js

@@ -0,0 +1,5 @@
+#!/usr/bin/env node
+/**
+ * DMLA CLI 入口
+ */
+import '../src/index.js'

package/package.json

@@ -0,0 +1,52 @@
+{
+  "name": "@icyfenix-dmla/cli",
+  "version": "2026.4.17-546",
+  "description": "DMLA 沙箱服务命令行工具",
+  "type": "module",
+  "main": "src/index.js",
+  "bin": {
+    "dmla": "./bin/dmla.js"
+  },
+  "scripts": {
+    "start": "node src/index.js",
+    "test": "node --experimental-vm-modules $(npm root)/jest/bin/jest.js"
+  },
+  "dependencies": {
+    "commander": "^12.1.0",
+    "chalk": "^5.3.0",
+    "enquirer": "^2.4.1",
+    "dockerode": "^4.0.2",
+    "express": "^4.21.2",
+    "cors": "^2.8.5"
+  },
+  "devDependencies": {
+    "jest": "^29.7.0"
+  },
+  "jest": {
+    "testEnvironment": "node",
+    "transform": {},
+    "moduleFileExtensions": [
+      "js",
+      "mjs"
+    ],
+    "testMatch": [
+      "**/tests/**/*.test.js"
+    ]
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "keywords": [
+    "dmla",
+    "sandbox",
+    "python",
+    "machine-learning",
+    "cli"
+  ],
+  "author": "icyfenix",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/icyfenix/dmla"
+  }
+}

package/src/commands/manage.js

@@ -0,0 +1,334 @@
+/**
+ * 管理命令（安装、更新、诊断）
+ */
+import chalk from 'chalk'
+import Docker from 'dockerode'
+import { spawn, execSync } from 'child_process'
+import http from 'http'
+import path from 'path'
+import { fileURLToPath } from 'url'
+import fs from 'fs'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = path.dirname(__filename)
+
+const docker = new Docker()
+
+// 配置
+const CONFIG = {
+  imageCpu: 'dmla-sandbox:cpu',
+  imageGpu: 'dmla-sandbox:gpu',
+  dockerhubRegistry: 'icyfenix',
+  tcrRegistry: 'ccr.ccs.tencentyun.com/icyfenix',
+  imageName: 'dmla-sandbox',
+  defaultPort: 3001
+}
+
+/**
+ * 获取镜像仓库地址
+ */
+function getRegistryUrl(registry) {
+  if (registry === 'tcr') {
+    return `${CONFIG.tcrRegistry}/${CONFIG.imageName}`
+  }
+  return `${CONFIG.dockerhubRegistry}/${CONFIG.imageName}`
+}
+
+/**
+ * 安装镜像
+ */
+export async function installImages(types, registry = 'dockerhub') {
+  const registryUrl = getRegistryUrl(registry)
+
+  console.log(chalk.gray(`   从 ${registry === 'tcr' ? '腾讯云 TCR' : 'Docker Hub'} 拉取镜像`))
+
+  for (const type of types) {
+    console.log()
+    console.log(chalk.bold(`📥 拉取 ${type.toUpperCase()} 版本镜像...`))
+
+    const remoteImage = `${registryUrl}:${type}`
+    const localImage = type === 'gpu' ? CONFIG.imageGpu : CONFIG.imageCpu
+
+    try {
+      // 拉取镜像
+      await pullImageWithProgress(remoteImage)
+
+      // Tag 为本地名称
+      console.log(chalk.gray(`   重命名为 ${localImage}...`))
+      const image = docker.getImage(remoteImage)
+      await image.tag({ repo: CONFIG.imageName, tag: type })
+
+      console.log(chalk.green(`✅ ${type.toUpperCase()} 镜像安装完成`))
+    } catch (error) {
+      console.log(chalk.red(`❌ ${type.toUpperCase()} 镜像安装失败: ${error.message}`))
+    }
+  }
+
+  console.log()
+  console.log(chalk.green('🎉 镜像安装完成'))
+  console.log(chalk.yellow('💡 提示: 运行 dmla start 启动服务'))
+}
+
+/**
+ * 带进度显示的镜像拉取
+ */
+async function pullImageWithProgress(imageName) {
+  return new Promise((resolve, reject) => {
+    docker.pull(imageName, (err, stream) => {
+      if (err) {
+        reject(err)
+        return
+      }
+
+      // 解析进度
+      docker.modem.followProgress(stream, (err, output) => {
+        if (err) {
+          reject(err)
+        } else {
+          resolve(output)
+        }
+      }, (event) => {
+        // 显示进度
+        if (event.status) {
+          let progress = event.status
+          if (event.progress) {
+            progress += ` ${event.progress}`
+          }
+          if (event.id) {
+            console.log(chalk.gray(`   [${event.id}] ${progress}`))
+          } else {
+            console.log(chalk.gray(`   ${progress}`))
+          }
+        }
+      })
+    })
+  })
+}
+
+/**
+ * 更新所有组件
+ */
+export async function updateAll(registry = 'dockerhub') {
+  console.log()
+
+  // 更新 npm 包
+  console.log(chalk.bold('📦 更新 npm 包'))
+  try {
+    console.log(chalk.gray('   执行 npm update -g @icyfenix-dmla/cli...'))
+    execSync('npm update -g @icyfenix-dmla/cli', { stdio: 'inherit' })
+    console.log(chalk.green('✅ npm 包已更新'))
+  } catch (error) {
+    console.log(chalk.yellow('⚠️ npm 包更新失败或已是最新版本'))
+  }
+
+  console.log()
+
+  // 检查并更新镜像
+  console.log(chalk.bold('🖼️  检查 Docker 镜像更新'))
+  const registryUrl = getRegistryUrl(registry)
+
+  for (const type of ['cpu', 'gpu']) {
+    const remoteImage = `${registryUrl}:${type}`
+    const localImage = type === 'gpu' ? CONFIG.imageGpu : CONFIG.imageCpu
+
+    console.log(chalk.gray(`   检查 ${type.toUpperCase()} 版本...`))
+
+    try {
+      // 检查本地镜像是否存在
+      let localImageInfo = null
+      try {
+        localImageInfo = await docker.getImage(localImage).inspect()
+      } catch {
+        // 本地镜像不存在，需要拉取
+      }
+
+      // 拉取最新镜像
+      console.log(chalk.gray(`   拉取最新 ${type.toUpperCase()} 镜像...`))
+      await pullImageWithProgress(remoteImage)
+
+      // 获取拉取的镜像信息
+      const remoteImageInfo = await docker.getImage(remoteImage).inspect()
+
+      // 比较镜像 ID
+      if (localImageInfo && localImageInfo.Id === remoteImageInfo.Id) {
+        console.log(chalk.green(`✅ ${type.toUpperCase()} 镜像已是最新版本`))
+      } else {
+        // Tag 为本地名称
+        console.log(chalk.gray(`   重命名为 ${localImage}...`))
+        const image = docker.getImage(remoteImage)
+        await image.tag({ repo: CONFIG.imageName, tag: type })
+
+        console.log(chalk.green(`✅ ${type.toUpperCase()} 镜像已更新`))
+      }
+    } catch (error) {
+      console.log(chalk.yellow(`⚠️ ${type.toUpperCase()} 镜像更新失败: ${error.message}`))
+    }
+  }
+
+  console.log()
+  console.log(chalk.green('🎉 更新完成'))
+}
+
+/**
+ * 环境诊断
+ */
+export async function runDoctor() {
+  console.log()
+  const issues = []
+
+  // ───────────────────────────────────────────────────────────
+  // Docker 检查
+  // ───────────────────────────────────────────────────────────
+  console.log(chalk.bold('🐳 Docker 环境'))
+
+  try {
+    const dockerInfo = await docker.info()
+    console.log(chalk.green('   ✅ Docker 已安装'))
+    console.log(chalk.gray(`   版本: ${dockerInfo.ServerVersion || '未知'}`))
+
+    // 检查版本是否满足要求
+    const minVersion = '20.10'
+    if (dockerInfo.ServerVersion && dockerInfo.ServerVersion < minVersion) {
+      issues.push(`Docker 版本过低，建议升级到 ${minVersion} 或更高`)
+    }
+  } catch (error) {
+    console.log(chalk.red('   ❌ Docker 未安装或未运行'))
+    issues.push('请安装 Docker 并确保服务正在运行')
+  }
+
+  console.log()
+
+  // ───────────────────────────────────────────────────────────
+  // 镜像检查
+  // ───────────────────────────────────────────────────────────
+  console.log(chalk.bold('🖼️  Docker 镜像'))
+
+  const cpuImage = CONFIG.imageCpu
+  const gpuImage = CONFIG.imageGpu
+
+  let cpuExists = false
+  let gpuExists = false
+
+  try {
+    const cpuInfo = await docker.getImage(cpuImage).inspect()
+    cpuExists = true
+    console.log(chalk.green(`   ✅ CPU 镜像已安装`))
+    console.log(chalk.gray(`   大小: ${Math.round(cpuInfo.Size / 1024 / 1024)} MB`))
+  } catch {
+    console.log(chalk.red(`   ❌ CPU 镜像未安装`))
+    issues.push('运行 dmla install --cpu 安装 CPU 镜像')
+  }
+
+  try {
+    const gpuInfo = await docker.getImage(gpuImage).inspect()
+    gpuExists = true
+    console.log(chalk.green(`   ✅ GPU 镜像已安装`))
+    console.log(chalk.gray(`   大小: ${Math.round(gpuInfo.Size / 1024 / 1024)} MB`))
+  } catch {
+    console.log(chalk.yellow(`   ⚠️ GPU 镜像未安装`))
+    console.log(chalk.gray('   (可选，仅在需要 GPU 时安装)'))
+  }
+
+  console.log()
+
+  // ───────────────────────────────────────────────────────────
+  // GPU 检查
+  // ───────────────────────────────────────────────────────────
+  console.log(chalk.bold('🎮 GPU 驱动'))
+
+  try {
+    const output = execSync('nvidia-smi -L', { timeout: 5000, encoding: 'utf8' })
+    if (output.includes('GPU')) {
+      console.log(chalk.green('   ✅ NVIDIA GPU 可用'))
+      const lines = output.split('\n').filter(l => l.trim())
+      lines.forEach(line => console.log(chalk.gray(`   ${line.trim()}`)))
+
+      // 检查 GPU 镜像
+      if (!gpuExists) {
+        console.log(chalk.yellow('   💡 检测到 GPU，建议安装 GPU 镜像'))
+        issues.push('运行 dmla install --gpu 安装 GPU 镜像')
+      }
+    } else {
+      console.log(chalk.gray('   GPU 不可用'))
+    }
+  } catch {
+    console.log(chalk.gray('   GPU 不可用'))
+    console.log(chalk.gray('   (如果需要 GPU，请安装 NVIDIA 驱动)'))
+  }
+
+  console.log()
+
+  // ───────────────────────────────────────────────────────────
+  // 端口检查
+  // ───────────────────────────────────────────────────────────
+  console.log(chalk.bold('🔌 端口可用性'))
+
+  const port = CONFIG.defaultPort
+  const portAvailable = await checkPortAvailable(port)
+
+  if (portAvailable) {
+    console.log(chalk.green(`   ✅ 端口 ${port} 可用`))
+  } else {
+    console.log(chalk.red(`   ❌ 端口 ${port} 已被占用`))
+    issues.push(`端口 ${port} 已被占用，使用 --port 指定其他端口`)
+  }
+
+  console.log()
+
+  // ───────────────────────────────────────────────────────────
+  // 网络连通性
+  // ───────────────────────────────────────────────────────────
+  console.log(chalk.bold('🌐 网络连通性'))
+
+  // 测试 Docker Hub
+  console.log(chalk.gray('   测试 Docker Hub 连接...'))
+  try {
+    execSync('docker pull icyfenix/dmla-sandbox:cpu --quiet', { timeout: 10000 })
+    console.log(chalk.green('   ✅ Docker Hub 连接正常'))
+  } catch {
+    console.log(chalk.yellow('   ⚠️ Docker Hub 连接超时或受限'))
+  }
+
+  // 测试 TCR
+  console.log(chalk.gray('   测试腾讯云 TCR 连接...'))
+  try {
+    execSync('docker pull ccr.ccs.tencentyun.com/icyfenix/dmla-sandbox:cpu --quiet', { timeout: 10000 })
+    console.log(chalk.green('   ✅ TCR 连接正常'))
+  } catch {
+    console.log(chalk.yellow('   ⚠️ TCR 连接超时或受限'))
+  }
+
+  console.log()
+
+  // ───────────────────────────────────────────────────────────
+  // 问题汇总
+  // ───────────────────────────────────────────────────────────
+  if (issues.length > 0) {
+    console.log(chalk.bold.red('❌ 发现以下问题：'))
+    console.log()
+    issues.forEach((issue, i) => {
+      console.log(chalk.red(`   ${i + 1}. ${issue}`))
+    })
+    console.log()
+    console.log(chalk.yellow('💡 请根据上述提示解决问题后再次运行 dmla doctor'))
+  } else {
+    console.log(chalk.bold.green('✅ 所有检查通过，环境正常'))
+    console.log()
+    console.log(chalk.gray('💡 运行 dmla start 启动服务'))
+  }
+}
+
+/**
+ * 检查端口是否可用
+ */
+async function checkPortAvailable(port) {
+  return new Promise((resolve) => {
+    const server = http.createServer()
+    server.once('error', () => resolve(false))
+    server.once('listening', () => {
+      server.close()
+      resolve(true)
+    })
+    server.listen(port)
+  })
+}

package/src/commands/server.js

@@ -0,0 +1,293 @@
+/**
+ * 服务管理命令
+ */
+import chalk from 'chalk'
+import Docker from 'dockerode'
+import { spawn } from 'child_process'
+import http from 'http'
+import path from 'path'
+import { fileURLToPath } from 'url'
+import fs from 'fs'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = path.dirname(__filename)
+
+const docker = new Docker()
+
+// 配置
+const CONFIG = {
+  imageCpu: 'dmla-sandbox:cpu',
+  imageGpu: 'dmla-sandbox:gpu',
+  defaultPort: 3001
+}
+
+/**
+ * 检查端口是否可用
+ */
+async function checkPortAvailable(port) {
+  return new Promise((resolve) => {
+    const server = http.createServer()
+    server.once('error', () => resolve(false))
+    server.once('listening', () => {
+      server.close()
+      resolve(true)
+    })
+    server.listen(port)
+  })
+}
+
+/**
+ * 检查镜像是否存在
+ */
+async function checkImageExists(type) {
+  const image = type === 'gpu' ? CONFIG.imageGpu : CONFIG.imageCpu
+  try {
+    await docker.getImage(image).inspect()
+    return true
+  } catch {
+    return false
+  }
+}
+
+/**
+ * 检查 GPU 是否可用
+ */
+async function checkGPUAvailable() {
+  try {
+    // 尝试运行 nvidia-smi 命令
+    const result = await new Promise((resolve, reject) => {
+      const proc = spawn('nvidia-smi', ['-L'], { timeout: 5000 })
+      let output = ''
+      proc.stdout.on('data', (data) => output += data.toString())
+      proc.stderr.on('data', (data) => output += data.toString())
+      proc.on('close', (code) => {
+        if (code === 0) resolve(output)
+        else reject(new Error('nvidia-smi failed'))
+      })
+      proc.on('error', reject)
+    })
+    return result.includes('GPU')
+  } catch {
+    return false
+  }
+}
+
+/**
+ * 检查服务是否运行
+ */
+async function checkServiceRunning(port) {
+  return new Promise((resolve) => {
+    const req = http.request({
+      hostname: 'localhost',
+      port: port,
+      path: '/api/health',
+      method: 'GET',
+      timeout: 2000
+    }, (res) => {
+      resolve(res.statusCode === 200)
+    })
+    req.on('error', () => resolve(false))
+    req.on('timeout', () => {
+      req.destroy()
+      resolve(false)
+    })
+    req.end()
+  })
+}
+
+/**
+ * 查找运行中的服务容器
+ */
+async function findServiceContainer() {
+  try {
+    const containers = await docker.listContainers({ all: true })
+    // 查找 dmla 服务容器
+    for (const container of containers) {
+      if (container.Names.some(name => name.includes('dmla-server'))) {
+        return container
+      }
+    }
+    return null
+  } catch {
+    return null
+  }
+}
+
+/**
+ * 启动服务
+ */
+export async function startServer(port, useGpu = false) {
+  // 检查端口
+  const portAvailable = await checkPortAvailable(port)
+  if (!portAvailable) {
+    console.log(chalk.red(`❌ 端口 ${port} 已被占用`))
+    console.log(chalk.yellow('💡 提示: 使用 --port 选项指定其他端口'))
+    return
+  }
+
+  // 检查镜像
+  const imageType = useGpu ? 'gpu' : 'cpu'
+  const imageExists = await checkImageExists(imageType)
+  if (!imageExists) {
+    console.log(chalk.red(`❌ 镜像 ${useGpu ? CONFIG.imageGpu : CONFIG.imageCpu} 不存在`))
+    console.log(chalk.yellow('💡 提示: 运行 dmla install 安装镜像'))
+    return
+  }
+
+  // 检查服务是否已运行
+  const alreadyRunning = await checkServiceRunning(port)
+  if (alreadyRunning) {
+    console.log(chalk.green(`✅ 服务已在端口 ${port} 运行`))
+    return
+  }
+
+  // 启动服务
+  console.log(chalk.gray('   正在启动...'))
+
+  try {
+    // 使用 spawn 启动 server 进程
+    const serverPath = path.resolve(__dirname, '../../../local-server/src/index.js')
+
+    // 如果 server 文件不存在，说明是独立安装模式，需要启动内置服务
+    const standaloneServerPath = path.resolve(__dirname, '../server/index.js')
+
+    const actualServerPath = fs.existsSync(serverPath) ? serverPath :
+                             fs.existsSync(standaloneServerPath) ? standaloneServerPath : null
+
+    if (!actualServerPath) {
+      console.log(chalk.red('❌ 找不到服务入口文件'))
+      console.log(chalk.yellow('💡 提示: 确保正确安装了 @icyfenix-dmla/cli'))
+      return
+    }
+
+    const env = {
+      ...process.env,
+      PORT: port.toString(),
+      USE_GPU: useGpu ? 'true' : 'false'
+    }
+
+    const serverProcess = spawn('node', [actualServerPath], {
+      env,
+      stdio: 'inherit',
+      detached: true
+    })
+
+    serverProcess.unref()
+
+    // 等待服务启动
+    console.log(chalk.gray('   等待服务就绪...'))
+    let attempts = 0
+    const maxAttempts = 30
+
+    while (attempts < maxAttempts) {
+      const running = await checkServiceRunning(port)
+      if (running) {
+        console.log(chalk.green(`✅ 服务已启动: http://localhost:${port}`))
+        console.log(chalk.gray(`   健康检查: http://localhost:${port}/api/health`))
+        return
+      }
+      await new Promise(resolve => setTimeout(resolve, 500))
+      attempts++
+    }
+
+    console.log(chalk.yellow('⚠️ 服务启动超时，请检查日志'))
+  } catch (error) {
+    console.log(chalk.red(`❌ 启动失败: ${error.message}`))
+  }
+}
+
+/**
+ * 停止服务
+ */
+export async function stopServer() {
+  // 查找运行中的容器
+  const container = await findServiceContainer()
+
+  if (container) {
+    try {
+      const containerObj = docker.getContainer(container.Id)
+      await containerObj.stop()
+      await containerObj.remove()
+      console.log(chalk.green('✅ 服务已停止'))
+    } catch (error) {
+      console.log(chalk.red(`❌ 停止失败: ${error.message}`))
+    }
+  } else {
+    // 尝试通过端口查找进程
+    console.log(chalk.yellow('⚠️ 未找到运行中的服务容器'))
+    console.log(chalk.gray('   提示: 服务可能以非容器模式运行'))
+  }
+}
+
+/**
+ * 获取状态
+ */
+export async function getStatus() {
+  console.log()
+
+  // 检查 npm 包版本
+  console.log(chalk.bold('📦 npm 包版本'))
+  try {
+    const pkgPath = path.resolve(__dirname, '../package.json')
+    const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf8'))
+    console.log(chalk.gray(`   @icyfenix-dmla/cli: ${pkg.version}`))
+  } catch {
+    console.log(chalk.gray('   版本信息不可用'))
+  }
+
+  console.log()
+
+  // 检查镜像
+  console.log(chalk.bold('🖼️  Docker 镜像'))
+  const cpuExists = await checkImageExists('cpu')
+  const gpuExists = await checkImageExists('gpu')
+  console.log(chalk.gray(`   CPU: ${cpuExists ? chalk.green('已安装') : chalk.red('未安装')}`))
+  console.log(chalk.gray(`   GPU: ${gpuExists ? chalk.green('已安装') : chalk.red('未安装')}`))
+
+  console.log()
+
+  // 检查 GPU
+  console.log(chalk.bold('🎮 GPU 状态'))
+  const gpuAvailable = await checkGPUAvailable()
+  if (gpuAvailable) {
+    console.log(chalk.green('   GPU 可用'))
+    try {
+      const proc = spawn('nvidia-smi', ['-L'])
+      proc.stdout.on('data', (data) => {
+        const lines = data.toString().split('\n').filter(l => l.trim())
+        lines.forEach(line => console.log(chalk.gray(`   ${line}`)))
+      })
+    } catch {}
+  } else {
+    console.log(chalk.gray('   GPU 不可用'))
+  }
+
+  console.log()
+
+  // 检查服务
+  console.log(chalk.bold('🚀 服务状态'))
+  const running = await checkServiceRunning(CONFIG.defaultPort)
+  if (running) {
+    console.log(chalk.green(`   服务运行中 (端口 ${CONFIG.defaultPort})`))
+    try {
+      // 获取详细状态
+      const healthUrl = `http://localhost:${CONFIG.defaultPort}/api/sandbox/health`
+      http.get(healthUrl, (res) => {
+        let data = ''
+        res.on('data', (chunk) => data += chunk)
+        res.on('end', () => {
+          try {
+            const health = JSON.parse(data)
+            if (health.images) {
+              console.log(chalk.gray(`   CPU 镜像: ${health.images.cpu ? '就绪' : '未就绪'}`))
+              console.log(chalk.gray(`   GPU 镜像: ${health.images.gpu ? '就绪' : '未就绪'}`))
+            }
+          } catch {}
+        })
+      })
+    } catch {}
+  } else {
+    console.log(chalk.gray('   服务未运行'))
+    console.log(chalk.yellow('   提示: 运行 dmla start 启动服务'))
+  }
+}

package/src/index.js

@@ -0,0 +1,104 @@
+/**
+ * DMLA CLI 入口
+ * 沙箱服务命令行管理工具
+ */
+import { program } from 'commander'
+import chalk from 'chalk'
+import { startServer, stopServer, getStatus } from './commands/server.js'
+import { installImages, updateAll, runDoctor } from './commands/manage.js'
+
+const VERSION = '0.0.0' // 将在发布时由 workflow 更新
+
+program
+  .name('dmla')
+  .description('DMLA 沙箱服务命令行管理工具')
+  .version(VERSION)
+
+// ─────────────────────────────────────────────────────────────
+// start 命令
+// ─────────────────────────────────────────────────────────────
+program
+  .command('start')
+  .description('启动沙箱服务')
+  .option('-p, --port <number>', '服务端口', '3001')
+  .option('--gpu', '使用 GPU 镜像')
+  .action(async (options) => {
+    const port = parseInt(options.port, 10)
+    const useGpu = options.gpu
+    console.log(chalk.blue('🚀 启动 DMLA 沙箱服务...'))
+    console.log(chalk.gray(`   端口: ${port}`))
+    console.log(chalk.gray(`   镜像: ${useGpu ? 'GPU' : 'CPU'}`))
+    await startServer(port, useGpu)
+  })
+
+// ─────────────────────────────────────────────────────────────
+// stop 命令
+// ─────────────────────────────────────────────────────────────
+program
+  .command('stop')
+  .description('停止运行中的沙箱服务')
+  .action(async () => {
+    console.log(chalk.blue('🛑 停止 DMLA 沙箱服务...'))
+    await stopServer()
+  })
+
+// ─────────────────────────────────────────────────────────────
+// status 命令
+// ─────────────────────────────────────────────────────────────
+program
+  .command('status')
+  .description('查看服务状态')
+  .action(async () => {
+    console.log(chalk.blue('📊 DMLA 沙箱服务状态'))
+    await getStatus()
+  })
+
+// ─────────────────────────────────────────────────────────────
+// install 命令
+// ─────────────────────────────────────────────────────────────
+program
+  .command('install')
+  .description('安装 Docker 镜像')
+  .option('--cpu', '仅安装 CPU 版本')
+  .option('--gpu', '仅安装 GPU 版本')
+  .option('--all', '安装所有镜像（默认）')
+  .option('-r, --registry <type>', '镜像仓库 (dockerhub/tcr)', 'dockerhub')
+  .action(async (options) => {
+    const registry = options.registry
+    let types = []
+
+    if (options.cpu) types.push('cpu')
+    if (options.gpu) types.push('gpu')
+    if (types.length === 0 || options.all) types = ['cpu', 'gpu']
+
+    console.log(chalk.blue('📦 安装 DMLA Docker 镜像...'))
+    console.log(chalk.gray(`   仓库: ${registry}`))
+    console.log(chalk.gray(`   类型: ${types.join(', ')}`))
+
+    await installImages(types, registry)
+  })
+
+// ─────────────────────────────────────────────────────────────
+// update 命令
+// ─────────────────────────────────────────────────────────────
+program
+  .command('update')
+  .description('更新 npm 包和 Docker 镜像')
+  .option('-r, --registry <type>', '镜像仓库 (dockerhub/tcr)', 'dockerhub')
+  .action(async (options) => {
+    console.log(chalk.blue('🔄 更新 DMLA...'))
+    await updateAll(options.registry)
+  })
+
+// ─────────────────────────────────────────────────────────────
+// doctor 命令
+// ─────────────────────────────────────────────────────────────
+program
+  .command('doctor')
+  .description('诊断安装环境')
+  .action(async () => {
+    console.log(chalk.blue('🔍 DMLA 环境诊断'))
+    await runDoctor()
+  })
+
+program.parse()

package/tests/cli.test.js

@@ -0,0 +1,34 @@
+/**
+ * CLI 命令单元测试
+ */
+import { describe, it, expect, beforeAll } from '@jest/globals'
+import { fileURLToPath } from 'url'
+import path from 'path'
+import fs from 'fs'
+
+// ESM 中获取 __dirname
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = path.dirname(__filename)
+
+// 基础测试：确保模块可导入
+describe('CLI Module', () => {
+  beforeAll(() => {
+    // 设置测试环境
+  })
+
+  it('should have correct package.json', async () => {
+    const pkgPath = path.resolve(__dirname, '../package.json')
+    const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf8'))
+
+    expect(pkg.name).toBe('@icyfenix-dmla/cli')
+    expect(pkg.bin).toBeDefined()
+    expect(pkg.bin.dmla).toBe('./bin/dmla.js')
+  })
+
+  it('should have commander dependency', async () => {
+    const pkgPath = path.resolve(__dirname, '../package.json')
+    const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf8'))
+
+    expect(pkg.dependencies.commander).toBeDefined()
+  })
+})