@aweray/hsk-cli 0.2.2

package/bin/hsk-cli.js

@@ -0,0 +1,439 @@
+#!/usr/bin/env node
+
+const { program } = require('commander');
+const chalk = require('chalk');
+const tunnel = require('../lib/tunnel');
+const fileHosting = require('../lib/fileHosting');
+const download = require('../lib/download');
+const platform = require('../lib/platform');
+const format = require('../lib/format');
+const pidManager = require('../lib/pidManager');
+const version = require('../lib/version');
+const config = require('../lib/config');
+const build = require('../lib/build');
+const pack = require('../lib/pack');
+const resourceStore = require('../lib/resourceStore');
+const resourceChecker = require('../lib/resourceChecker');
+
+// === 全局选项 ===
+program
+  .name('hsk-cli')
+  .description('HSK CLI - 内网穿透 & 文件托管，支持 Windows/macOS/Linux 多平台')
+  .version(version)
+  .option('--format <format>', '输出格式: json | pretty | table | ndjson | csv (默认: pretty)', 'pretty')
+  .option('--dry-run', '预览模式，不执行实际操作');
+
+// 解析全局选项（在子命令之前）
+const globalOpts = program.opts();
+
+function getFormat(options) {
+  // --json 是 --format json 的快捷方式
+  if (options.json) return 'json';
+  return options.format || globalOpts.format || 'pretty';
+}
+
+function isDryRun(options) {
+  return options.dryRun || globalOpts.dryRun || false;
+}
+
+function handleOutput(data, formatType) {
+  if (formatType === 'pretty') {
+    console.log(format.outputFormat(data, 'pretty'));
+  } else {
+    console.log(format.outputFormat(data, formatType));
+  }
+}
+
+function handleError(err, formatType) {
+  const data = { success: false, error: err.message };
+  if (formatType === 'pretty') {
+    console.error(chalk.red('❌ 失败：') + err.message);
+  } else {
+    console.log(format.outputFormat(data, formatType));
+  }
+  process.exit(1);
+}
+
+// === tunnel 命令 ===
+const tunnelCmd = program
+  .command('tunnel')
+  .description('启动内网穿透隧道')
+  .requiredOption('--ip <ip>', '本地服务 IP 地址，如 127.0.0.1')
+  .requiredOption('--port <port>', '本地服务端口，如 9000')
+  .option('--arch <arch>', '强制指定架构 (win32, win64, macos-x64, macos-arm64, linux-x64, linux-arm64)')
+  .option('--force-download', '强制重新下载客户端二进制文件')
+  .option('--detach', '后台模式，启动后 CLI 立即退出，隧道在后台运行')
+  .option('--reuse', '复用已有隧道，不存在则创建')
+  .option('--json', '以 JSON 格式输出（等同于 --format json）')
+  .action(async (options) => {
+    const fmt = getFormat(options);
+    if (isDryRun(options)) {
+      const info = options.arch ? platform.fromString(options.arch) : platform.detect();
+      const binaryInfo = download.getBinaryInfo(info.platformKey);
+      const mode = options.detach ? '后台' : '前台';
+      console.log(format.dryRunInfo(
+        `${binaryInfo.filename} -ip ${options.ip} -port ${options.port}${options.detach ? ' (后台模式)' : ''}`,
+        {
+          目标IP: options.ip,
+          目标端口: options.port,
+          架构: `${info.platform} ${info.arch}`,
+          二进制文件: binaryInfo.filename,
+          版本: `v${binaryInfo.version}`,
+          运行模式: mode,
+          预期输出: '公网访问地址 (https://...)',
+        }
+      ));
+      return;
+    }
+    try {
+      if (options.forceDownload) {
+        const info = options.arch ? platform.fromString(options.arch) : platform.detect();
+        await download.ensureBinary(info.platformKey, true);
+      }
+      const result = await tunnel.start({
+        ip: options.ip,
+        port: options.port,
+        arch: options.arch,
+        detached: options.detach,
+        reuse: options.reuse
+      });
+
+      if (!options.detach && fmt === 'pretty') {
+        // 前台保活模式：打印成功后不退出，保持子进程运行
+        console.log(format.formatPretty(result));
+        console.log(chalk.gray('📡 隧道正在运行... (PID: ' + result.pid + ')'));
+        console.log(chalk.gray('    按 Ctrl+C 停止隧道\n'));
+        // 保持进程运行，不再 handleOutput（已经打印了）
+      } else {
+        handleOutput(result, fmt);
+      }
+    } catch (err) {
+      handleError(err, fmt);
+    }
+  });
+
+// === tunnel list 命令 ===
+program
+  .command('tunnel list')
+  .description('查看后台运行的内网穿透隧道')
+  .action(() => {
+    const records = pidManager.listPids();
+    const active = records.filter(r => r.isRunning);
+    const dead = records.filter(r => !r.isRunning);
+
+    if (active.length === 0 && dead.length === 0) {
+      console.log(chalk.gray('当前没有后台运行的隧道。'));
+      return;
+    }
+
+    if (active.length > 0) {
+      console.log(chalk.green('🟢 运行中的隧道：'));
+      active.forEach(r => {
+        console.log(`  PID: ${r.pid} | ${r.meta.ip}:${r.meta.port} | 启动: ${r.createdAt}`);
+        console.log(`  日志: ${r.logFile}`);
+      });
+    }
+
+    if (dead.length > 0) {
+      console.log(chalk.yellow('⚪ 已停止的隧道（残留记录）：'));
+      dead.forEach(r => {
+        console.log(`  PID: ${r.pid} | ${r.meta.ip}:${r.meta.port}`);
+      });
+      const cleaned = pidManager.cleanupDeadPids();
+      console.log(chalk.gray(`  已清理 ${cleaned} 条残留记录。`));
+    }
+  });
+
+// === tunnel stop 命令 ===
+program
+  .command('tunnel stop')
+  .description('停止内网穿透隧道')
+  .option('--pid <pid>', '指定要停止的 PID')
+  .option('--all', '停止所有后台隧道')
+  .action((options) => {
+    if (options.all) {
+      const count = pidManager.stopAll();
+      console.log(chalk.green(`✅ 已停止 ${count} 个后台隧道。`));
+      return;
+    }
+    if (options.pid) {
+      const pid = Number(options.pid);
+      if (isNaN(pid)) {
+        console.error(chalk.red('❌ 无效的 PID'));
+        process.exit(1);
+      }
+      const ok = pidManager.stopPid(pid);
+      if (ok) {
+        console.log(chalk.green(`✅ 已停止隧道 (PID: ${pid})`));
+      } else {
+        console.log(chalk.yellow(`⚠️  PID ${pid} 未找到或已停止`));
+      }
+      return;
+    }
+    console.error(chalk.red('❌ 请指定 --pid <PID> 或 --all'));
+    process.exit(1);
+  });
+
+// === host 命令 ===
+const hostCmd = program
+  .command('host <filePath>')
+  .description('上传文件到公网进行托管')
+  .option('--json', '以 JSON 格式输出（等同于 --format json）')
+  .option('--open', '上传完成后自动打开浏览器进行认领')
+  .option('--resource-id <id>', '资源 ID（传入则更新已有资源，否则创建新资源）')
+  .action(async (filePath, options) => {
+    const fmt = getFormat(options);
+    if (isDryRun(options)) {
+      const fs = require('fs');
+      const path = require('path');
+      const stats = fs.existsSync(filePath) ? fs.statSync(filePath) : null;
+      const uploadBase = fileHosting.getUploadBaseUrl();
+      console.log(format.dryRunInfo(
+        `POST ${fileHosting.TICKET_API}`,
+        {
+          文件路径: filePath,
+          文件大小: stats ? `${(stats.size / 1024 / 1024).toFixed(2)} MB` : '文件不存在',
+          资源ID: options.resourceId || '(新建)',
+          预期操作: '计算 SHA-256 → 获取 ticket → 上传/更新文件',
+          上传服务: options.resourceId ? `PUT ${uploadBase}/resource/update` : `POST ${uploadBase}/upload`,
+          预期输出: 'publicUrl + resourceId',
+        }
+      ));
+      return;
+    }
+    try {
+      const result = await fileHosting.host(filePath, { resourceId: options.resourceId });
+      handleOutput(result, fmt);
+      if (options.open && result.publicUrl) {
+        console.log(chalk.gray('🌐 正在打开浏览器...'));
+        await fileHosting.openClaimUrl(result.publicUrl);
+      }
+    } catch (err) {
+      handleError(err, fmt);
+    }
+  });
+
+// === download 命令 ===
+program
+  .command('download')
+  .description('预下载对应平台的客户端二进制文件')
+  .option('--arch <arch>', '指定架构，默认自动检测')
+  .action(async (options) => {
+    if (isDryRun(options)) {
+      const info = options.arch ? platform.fromString(options.arch) : platform.detect();
+      const binaryInfo = download.getBinaryInfo(info.platformKey);
+      console.log(format.dryRunInfo(
+        `下载 ${binaryInfo.filename}`,
+        { 架构: `${info.platform} ${info.arch}`, 版本: binaryInfo.version, 保存路径: `~/.hsk/bin/${binaryInfo.filename}` }
+      ));
+      return;
+    }
+    try {
+      const binPath = await download.ensureBinary(options.arch, true);
+      console.log(chalk.green('✅ 客户端已就绪：') + binPath);
+    } catch (err) {
+      console.error(chalk.red('❌ 下载失败：') + err.message);
+      process.exit(1);
+    }
+  });
+
+// === update 命令 ===
+program
+  .command('update')
+  .description('检查并更新客户端二进制文件到最新版本')
+  .option('--arch <arch>', '指定架构，默认自动检测')
+  .option('--force', '强制重新下载，即使本地已有')
+  .action(async (options) => {
+    const fs = require('fs');
+    const info = options.arch ? platform.fromString(options.arch) : platform.detect();
+    const binaryInfo = download.getBinaryInfo(info.platformKey);
+
+    if (isDryRun(options)) {
+      const binPath = download.getBinaryPath(binaryInfo.filename);
+      const hasLocal = fs.existsSync(binPath);
+      console.log(format.dryRunInfo(
+        hasLocal ? `重新下载 ${binaryInfo.filename}` : `下载 ${binaryInfo.filename}`,
+        {
+          架构: `${info.platform} ${info.arch}`,
+          版本: `v${binaryInfo.version}`,
+          本地状态: hasLocal ? '已存在' : '首次下载',
+          保存路径: `~/.hsk/bin/${binaryInfo.filename}`,
+          下载源: binaryInfo.downloadUrl
+        }
+      ));
+      return;
+    }
+
+    try {
+      const binPath = download.getBinaryPath(binaryInfo.filename);
+      const hasLocal = fs.existsSync(binPath);
+
+      if (hasLocal && !options.force) {
+        console.log(chalk.green(`✅ 本地已有 v${binaryInfo.version}: ${binaryInfo.filename}`));
+        console.log(chalk.gray(`   路径: ${binPath}`));
+        console.log(chalk.gray(`   使用 --force 强制重新下载`));
+        return;
+      }
+
+      if (options.force) {
+        console.log(chalk.cyan('🔄 强制重新下载...'));
+      } else {
+        console.log(chalk.cyan('⬇️ 首次下载...'));
+      }
+
+      const downloadedPath = await download.ensureBinary(options.arch, true);
+      console.log(chalk.green(`✅ 更新完成: ${downloadedPath}`));
+      console.log(chalk.gray(`   版本: v${binaryInfo.version}`));
+    } catch (err) {
+      console.error(chalk.red('❌ 更新失败：') + err.message);
+      process.exit(1);
+    }
+  });
+
+// === deploy 命令 ===
+program
+  .command('deploy')
+  .description('构建项目并上传构建产物到文件托管')
+  .option('--build-cmd <cmd>', '构建命令，如 "npm run build"')
+  .option('--build-dir <dir>', '构建输出目录，如 "dist"')
+  .option('--pack-output <file>', '打包输出文件名，如 "dist.zip"', 'dist.zip')
+  .option('--no-build', '跳过构建，直接打包上传')
+  .option('--no-clean', '上传后保留打包文件')
+  .option('--resource-id <id>', '资源 ID（更新已有资源）')
+  .option('--open', '上传完成后自动打开浏览器进行认领')
+  .option('--json', '以 JSON 格式输出（等同于 --format json）')
+  .action(async (options) => {
+    const fmt = getFormat(options);
+    const fs = require('fs');
+
+    // 读取配置（命令行 > 项目配置 > 全局配置 > 默认配置）
+    const cfg = config.getDeployConfig();
+    const buildCmd = options.buildCmd || cfg.buildCmd;
+    const buildDir = options.buildDir || cfg.buildDir;
+    const packOutput = options.packOutput || cfg.packOutput;
+    const cleanAfter = options.clean !== false && cfg.cleanAfterUpload;
+
+    if (isDryRun(options)) {
+      const info = options.arch ? platform.fromString(options.arch) : platform.detect();
+      console.log(format.dryRunInfo(
+        `deploy: ${buildCmd} → pack ${buildDir} → upload ${packOutput}`,
+        {
+          构建命令: buildCmd,
+          构建目录: buildDir,
+          打包文件: packOutput,
+          上传后清理: cleanAfter,
+          资源ID: options.resourceId || '(新建)'
+        }
+      ));
+      return;
+    }
+
+    try {
+      // 1. 构建
+      if (options.build !== false) {
+        console.log(chalk.cyan('🔨 开始构建...'));
+        console.log(chalk.gray(`   命令: ${buildCmd}`));
+        await build.execBuild(buildCmd);
+        console.log(chalk.green('✅ 构建完成'));
+      }
+
+      // 2. 检查构建目录
+      if (!fs.existsSync(buildDir)) {
+        throw new Error(`构建目录不存在: ${buildDir}`);
+      }
+
+      // 3. 打包
+      console.log(chalk.cyan(`📦 打包 ${buildDir}...`));
+      const packResult = await pack.packDir(buildDir, packOutput);
+      console.log(chalk.green(`✅ 打包完成: ${packResult.path}`));
+      console.log(chalk.gray(`   大小: ${(packResult.size / 1024 / 1024).toFixed(2)} MB`));
+
+      // 4. 上传
+      console.log(chalk.cyan('☁️  上传中...'));
+      const result = await fileHosting.host(packResult.path, { resourceId: options.resourceId });
+
+      // 5. 清理
+      if (cleanAfter) {
+        pack.cleanupPack(packResult.path);
+        console.log(chalk.gray('🧹 已清理打包文件'));
+      }
+
+      handleOutput(result, fmt);
+
+      if (options.open && result.publicUrl) {
+        console.log(chalk.gray('🌐 正在打开浏览器...'));
+        await fileHosting.openClaimUrl(result.publicUrl);
+      }
+    } catch (err) {
+      handleError(err, fmt);
+    }
+  });
+
+// === status 命令 ===
+program
+  .command('status')
+  .description('检查已发布资源的状态（仅内网穿透）')
+  .option('--json', '以 JSON 格式输出（等同于 --format json）')
+  .action(async (options) => {
+    const fmt = getFormat(options);
+    const all = resourceStore.getAllResources();
+    const tunnels = all.filter((r) => r.type === 'tunnel');
+
+    const results = [];
+    for (const r of tunnels) {
+      const check = await resourceChecker.checkResource(r);
+      results.push({ ...r, valid: check.valid, reason: check.reason, detail: check.detail });
+    }
+
+    if (fmt === 'json') {
+      console.log(JSON.stringify({ resources: results }, null, 2));
+    } else {
+      if (results.length === 0) {
+        console.log(chalk.gray('暂无隧道资源记录'));
+        return;
+      }
+      console.log(chalk.cyan('📋 隧道状态：\n'));
+      for (const r of results) {
+        const statusIcon = r.valid ? chalk.green('✅') : chalk.red('❌');
+        console.log(`${statusIcon} ${r.localIp}:${r.localPort}`);
+        console.log(`   URL: ${r.publicUrl || 'N/A'}`);
+        console.log(`   状态: ${r.status} | ${r.valid ? '有效' : '已失效'}`);
+        if (!r.valid) {
+          console.log(`   原因: ${r.detail || r.reason}`);
+        }
+        console.log();
+      }
+    }
+  });
+
+// === platform 命令 ===
+program
+  .command('platform')
+  .description('显示当前检测到的平台信息')
+  .action(() => {
+    const info = platform.detect();
+    console.log(JSON.stringify(info, null, 2));
+  });
+
+// === Shortcut 层命令（+ 前缀） ===
+// Commander 不直接支持 + 前缀，但我们可以解析原始参数
+// 如果用户输入 +tunnel 或 +host，转换为对应的子命令
+const rawArgs = process.argv.slice(2);
+if (rawArgs.length > 0 && rawArgs[0].startsWith('+')) {
+  const shortcut = rawArgs[0].slice(1); // 去掉 + 前缀
+  if (shortcut === 'tunnel') {
+    rawArgs[0] = 'tunnel';
+  } else if (shortcut === 'host') {
+    rawArgs[0] = 'host';
+  } else if (shortcut === 'deploy') {
+    rawArgs[0] = 'deploy';
+  } else {
+    console.error(chalk.red(`❌ 未知 Shortcut: +${shortcut}`));
+    console.log(chalk.gray('可用 Shortcuts: +tunnel, +host, +deploy'));
+    process.exit(1);
+  }
+  // 重新设置 process.argv 让 Commander 解析
+  process.argv = [process.argv[0], process.argv[1], ...rawArgs];
+}
+
+program.parse(process.argv);

package/index.js

@@ -0,0 +1,11 @@
+const tunnel = require('./lib/tunnel');
+const fileHosting = require('./lib/fileHosting');
+const download = require('./lib/download');
+const platform = require('./lib/platform');
+
+module.exports = {
+  tunnel,
+  fileHosting,
+  download,
+  platform
+};

package/lib/build.js

@@ -0,0 +1,27 @@
+const { spawn } = require('child_process');
+
+/**
+ * 执行构建命令
+ * @param {string} buildCmd - 构建命令，如 "npm run build"
+ * @param {string} cwd - 工作目录
+ */
+function execBuild(buildCmd, cwd = process.cwd()) {
+  return new Promise((resolve, reject) => {
+    const [cmd, ...args] = buildCmd.split(' ');
+    const proc = spawn(cmd, args, { cwd, stdio: 'inherit', shell: true });
+
+    proc.on('close', (code) => {
+      if (code === 0) {
+        resolve();
+      } else {
+        reject(new Error(`Build 失败，退出码: ${code}`));
+      }
+    });
+
+    proc.on('error', (err) => {
+      reject(new Error(`Build 执行失败: ${err.message}`));
+    });
+  });
+}
+
+module.exports = { execBuild };

package/lib/config.js

@@ -0,0 +1,63 @@
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+
+const DEFAULT_CONFIG = {
+  deploy: {
+    buildCmd: 'npm run build',
+    buildDir: 'dist',
+    packFormat: 'zip',
+    packOutput: 'dist.zip',
+    cleanAfterUpload: true
+  }
+};
+
+/**
+ * 从当前目录向上查找 package.json
+ */
+function findProjectConfig() {
+  let cwd = process.cwd();
+  while (cwd !== path.dirname(cwd)) {
+    const pkgPath = path.join(cwd, 'package.json');
+    if (fs.existsSync(pkgPath)) {
+      try {
+        const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf8'));
+        return pkg.hsk || null;
+      } catch (e) {
+        return null;
+      }
+    }
+    cwd = path.dirname(cwd);
+  }
+  return null;
+}
+
+/**
+ * 读取全局配置 ~/.hsk/config.json
+ */
+function findGlobalConfig() {
+  const globalPath = path.join(os.homedir(), '.hsk', 'config.json');
+  if (fs.existsSync(globalPath)) {
+    try {
+      return JSON.parse(fs.readFileSync(globalPath, 'utf8'));
+    } catch (e) {
+      return null;
+    }
+  }
+  return null;
+}
+
+/**
+ * 获取 deploy 配置，优先级：命令行 > 项目配置 > 全局配置 > 默认配置
+ */
+function getDeployConfig() {
+  const project = findProjectConfig();
+  const global = findGlobalConfig();
+  return {
+    ...DEFAULT_CONFIG.deploy,
+    ...(global?.deploy || {}),
+    ...(project?.deploy || {})
+  };
+}
+
+module.exports = { getDeployConfig, DEFAULT_CONFIG, findProjectConfig, findGlobalConfig };

package/lib/download.js

@@ -0,0 +1,87 @@
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const fetch = require('node-fetch');
+const chalk = require('chalk');
+const platform = require('./platform');
+
+const BIN_DIR = path.join(os.homedir(), '.hsk', 'bin');
+const VERSIONS_JSON = require('../versions.json');
+
+function ensureBinDir() {
+  if (!fs.existsSync(BIN_DIR)) {
+    fs.mkdirSync(BIN_DIR, { recursive: true });
+  }
+  return BIN_DIR;
+}
+
+function getBinaryPath(filename) {
+  return path.join(ensureBinDir(), filename);
+}
+
+function isBinaryExists(filename) {
+  return fs.existsSync(getBinaryPath(filename));
+}
+
+// 根据平台信息获取对应的版本配置
+function getBinaryInfo(platformKey) {
+  const binaryConfig = VERSIONS_JSON.binaries[platformKey];
+  if (!binaryConfig) {
+    throw new Error(`该平台暂无可用二进制: ${platformKey}。可用平台: ${Object.keys(VERSIONS_JSON.binaries).join(', ')}`);
+  }
+  return {
+    version: VERSIONS_JSON.version,
+    filename: binaryConfig.filename,
+    downloadUrl: `${VERSIONS_JSON.downloadBaseUrl}/${binaryConfig.filename}`
+  };
+}
+
+async function downloadBinary(binaryInfo, showProgress = true) {
+  const { filename, downloadUrl, version } = binaryInfo;
+  const binPath = getBinaryPath(filename);
+
+  if (showProgress) {
+    console.log(chalk.gray(`⬇️  下载 ${filename} (v${version})...`));
+    console.log(chalk.gray(`   来源: ${downloadUrl}`));
+  }
+
+  const response = await fetch(downloadUrl);
+  if (!response.ok) {
+    throw new Error(`下载失败: HTTP ${response.status} - ${response.statusText}\nURL: ${downloadUrl}`);
+  }
+
+  const buffer = await response.buffer();
+  fs.writeFileSync(binPath, buffer);
+
+  if (process.platform !== 'win32') {
+    fs.chmodSync(binPath, 0o755);
+  }
+
+  if (showProgress) {
+    console.log(chalk.green(`✅ 下载完成: ${binPath}`));
+  }
+  return binPath;
+}
+
+// 获取当前平台需要的二进制，自动下载/更新
+async function ensureBinary(archOverride, forceDownload = false) {
+  const info = archOverride ? platform.fromString(archOverride) : platform.detect();
+  const binaryInfo = getBinaryInfo(info.platformKey);
+  const binPath = getBinaryPath(binaryInfo.filename);
+
+  // 检查是否已有正确版本（通过文件名中的版本号判断）
+  if (!forceDownload && isBinaryExists(binaryInfo.filename)) {
+    console.log(chalk.gray(`📦 使用本地缓存: ${binaryInfo.filename}`));
+    return binPath;
+  }
+
+  return downloadBinary(binaryInfo, true);
+}
+
+// 获取当前平台版本信息
+function getCurrentVersionInfo() {
+  const info = platform.detect();
+  return getBinaryInfo(info.platformKey);
+}
+
+module.exports = { ensureBinary, downloadBinary, getBinaryPath, getBinaryInfo, getCurrentVersionInfo, BIN_DIR };