@icyfenix-dmla/cli 2026.5.2-1955 → 2026.5.2-2022

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@icyfenix-dmla/cli",
-  "version": "2026.5.2-1955",
+  "version": "2026.5.2-2022",
   "description": "DMLA 沙箱服务命令行工具",
   "type": "module",
   "main": "src/index.js",

package/src/commands/data.js

@@ -208,9 +208,8 @@ async function showMainMenu(dataPath) {
     { name: '2', message: '下载数据集' },
     { name: '3', message: '查看数据集列表' },
     { name: '4', message: '清空数据内容' },
-    { name: '5', message: '显示实际路径' },
-    { name: '6', message: '删除数据卷' },
-    { name: '7', message: '退出' }
+    { name: '5', message: '删除数据卷' },
+    { name: '6', message: '退出' }
   ]
 
   const { action } = await prompt({
@@ -277,35 +276,6 @@ async function mountPath() {
   console.log(chalk.yellow('提示: 需要重启沙箱服务才能生效 (dmla stop && dmla start)'))
 }
 
-/**
- * 显示实际路径
- */
-function showPath() {
-  const dataPath = getDataVolumePath()
-  const stats = getDirectoryStats(dataPath)
-
-  console.log()
-  console.log(chalk.bold('数据卷信息'))
-  console.log()
-  console.log(chalk.gray(`挂载路径: ${dataPath}`))
-  console.log(chalk.gray(`存在状态: ${fs.existsSync(dataPath) ? '已创建' : '未创建'}`))
-  console.log(chalk.gray(`数据集数量: ${stats.datasets}`))
-  console.log(chalk.gray(`模型数量: ${stats.models}`))
-  console.log()
-
-  // 显示子目录
-  if (fs.existsSync(dataPath)) {
-    console.log(chalk.bold('目录结构:'))
-    const subDirs = ['datasets', 'models', 'outputs', 'cache']
-    for (const dir of subDirs) {
-      const fullPath = path.join(dataPath, dir)
-      const exists = fs.existsSync(fullPath)
-      console.log(chalk.gray(`  ${dir}/ ${exists ? '(已存在)' : '(不存在)'}`))
-    }
-  }
-  console.log()
-}
-
 /**
  * 清空数据内容
  */
@@ -419,54 +389,87 @@ async function downloadDatasets() {
     return
   }
 
-  // 显示可用数据集
-  console.log('可用数据集:')
-  console.log()
+  // 检查 Git 环境
+  try {
+    execSync('git --version', { stdio: 'pipe' })
+  } catch {
+    console.log(chalk.red('❌ Git 未安装'))
+    console.log(chalk.yellow('下载数据集需要 Git，请先安装: https://git-scm.com/downloads'))
+    return
+  }
 
-  for (let i = 0; i < DATASETS.length; i++) {
-    const dataset = DATASETS[i]
+  // 构建选项列表
+  const choices = DATASETS.map((dataset, index) => {
     const downloaded = isDatasetDownloaded(dataPath, dataset.id)
-    const status = downloaded ? chalk.green('[x]') : chalk.gray('[ ]')
-    const downloadedTag = downloaded ? chalk.gray(' (已下载)') : chalk.gray(` (${dataset.size})`)
-    console.log(`  ${i + 1}. ${status} ${dataset.name}${downloadedTag}`)
-  }
+    const isMnist = dataset.id === 'mnist'
 
-  console.log()
-  console.log(chalk.gray('提示: 输入序号选择数据集，q 返回'))
-  console.log()
+    let message = `${dataset.name} (${dataset.size})`
+    if (downloaded) {
+      message += ' [已下载]'
+    } else if (isMnist) {
+      message += ' [训练时自动下载]'
+    }
 
-  const { selection } = await prompt({
-    type: 'input',
-    name: 'selection',
-    message: '选择要下载的数据集序号'
+    return {
+      name: index.toString(),
+      message,
+      disabled: downloaded || isMnist
+    }
   })
 
-  if (selection.toLowerCase() === 'q') {
-    return
-  }
+  // 操作提示
+  console.log(chalk.gray('操作: 上下键移动，空格勾选/取消，回车确认，ESC 返回'))
+  console.log()
 
-  const index = parseInt(selection) - 1
-  if (index < 0 || index >= DATASETS.length) {
-    console.log(chalk.yellow('无效的选择'))
-    return
-  }
+  try {
+    const { selected } = await prompt({
+      type: 'multiselect',
+      name: 'selected',
+      message: '选择要下载的数据集',
+      choices,
+      hint: '空格选择，回车确认下载',
+      warn: '已下载或自动下载类型'
+    })
 
-  const dataset = DATASETS[index]
+    if (!selected || selected.length === 0) {
+      console.log(chalk.yellow('未选择任何数据集'))
+      return
+    }
 
-  // 检查是否已下载
-  if (isDatasetDownloaded(dataPath, dataset.id)) {
-    console.log(chalk.yellow(`${dataset.name} 已下载`))
-    return
-  }
+    // 下载选中的数据集
+    for (const indexStr of selected) {
+      const index = parseInt(indexStr)
+      const dataset = DATASETS[index]
 
-  // MNIST 使用特殊处理
-  if (dataset.id === 'mnist') {
-    console.log(chalk.yellow('MNIST 数据集将在训练时通过 torchvision 自动下载'))
-    return
+      console.log()
+      console.log(chalk.cyan(`────────────────────────────────────`))
+
+    // MNIST 特殊处理
+    if (dataset.id === 'mnist') {
+      console.log(chalk.yellow('MNIST 数据集将在训练时通过 torchvision 自动下载'))
+      continue
+    }
+
+    // 检查是否已下载
+    if (isDatasetDownloaded(dataPath, dataset.id)) {
+      console.log(chalk.yellow(`${dataset.name} 已下载，跳过`))
+      continue
+    }
+
+    await downloadDataset(dataPath, dataset)
   }
 
-  // 开始下载
-  await downloadDataset(dataPath, dataset)
+  console.log()
+  console.log(chalk.cyan(`────────────────────────────────────`))
+  console.log(chalk.green('所有选中的数据集已处理完成'))
+  } catch (error) {
+    // 用户按 ESC 或 Ctrl+C 取消
+    if (error.message && error.message.includes('cancel')) {
+      console.log(chalk.gray('返回上一级'))
+      return
+    }
+    throw error
+  }
 }
 
 /**
@@ -653,12 +656,9 @@ export async function runDataTUI() {
         await clearData()
         break
       case 5:
-        showPath()
-        break
-      case 6:
         await removeData()
         break
-      case 7:
+      case 6:
         console.log()
         console.log(chalk.gray('已退出数据管理'))
         console.log()