@weotro/dx 0.1.0

package/lib/cli/commands/stack.js

@@ -0,0 +1,427 @@
+import { exec, spawn } from 'node:child_process'
+import { promisify } from 'node:util'
+import readline from 'node:readline'
+import { isAbsolute, join } from 'node:path'
+import { existsSync, mkdirSync, readdirSync, rmSync, unlinkSync } from 'node:fs'
+import { homedir } from 'node:os'
+import { logger } from '../../logger.js'
+import { execManager } from '../../exec.js'
+
+const execPromise = promisify(exec)
+
+const DEFAULT_SERVICES = ['backend', 'front', 'admin']
+const DEFAULT_PRE_FLIGHT = {
+  pm2Reset: true,
+  killPorts: [3000, 3001, 3500],
+  forcePortCleanup: true,
+  cleanPaths: [
+    'apps/front/.next',
+    'dist/front',
+    'apps/front/.eslintcache',
+    'apps/admin-front/node_modules/.vite',
+    'dist/admin-front',
+    'apps/admin-front/.eslintcache',
+  ],
+  cleanTsBuildInfo: true,
+  cleanTsBuildInfoDirs: ['apps/front', 'apps/admin-front'],
+}
+
+class PM2StackManager {
+  constructor(options = {}) {
+    this.projectRoot = process.cwd()
+    this.pm2Bin = String(options.pm2Bin || 'pnpm pm2').trim()
+
+    const ecosystemConfig = options.ecosystemConfig || 'ecosystem.config.cjs'
+    this.configPath = this.resolvePath(ecosystemConfig)
+
+    this.services = Array.isArray(options.services) && options.services.length > 0
+      ? options.services.map(item => String(item).trim()).filter(Boolean)
+      : [...DEFAULT_SERVICES]
+
+    this.urls = options.urls && typeof options.urls === 'object' ? options.urls : {}
+
+    const incomingPreflight = options.preflight && typeof options.preflight === 'object'
+      ? options.preflight
+      : {}
+
+    this.preflight = {
+      pm2Reset: incomingPreflight.pm2Reset ?? DEFAULT_PRE_FLIGHT.pm2Reset,
+      killPorts: this.normalizePorts(incomingPreflight.killPorts, DEFAULT_PRE_FLIGHT.killPorts),
+      forcePortCleanup:
+        incomingPreflight.forcePortCleanup ?? DEFAULT_PRE_FLIGHT.forcePortCleanup,
+      cleanPaths: this.normalizeStringArray(
+        incomingPreflight.cleanPaths,
+        DEFAULT_PRE_FLIGHT.cleanPaths,
+      ),
+      cleanTsBuildInfo:
+        incomingPreflight.cleanTsBuildInfo ?? DEFAULT_PRE_FLIGHT.cleanTsBuildInfo,
+      cleanTsBuildInfoDirs: this.normalizeStringArray(
+        incomingPreflight.cleanTsBuildInfoDirs,
+        DEFAULT_PRE_FLIGHT.cleanTsBuildInfoDirs,
+      ),
+    }
+
+    this.isRunning = false
+  }
+
+  resolvePath(targetPath) {
+    if (isAbsolute(targetPath)) return targetPath
+    return join(this.projectRoot, targetPath)
+  }
+
+  normalizePorts(input, fallback) {
+    const source = Array.isArray(input) ? input : fallback
+    return source
+      .map(port => Number(port))
+      .filter(port => Number.isFinite(port) && port > 0)
+  }
+
+  normalizeStringArray(input, fallback) {
+    const source = Array.isArray(input) ? input : fallback
+    return source.map(item => String(item).trim()).filter(Boolean)
+  }
+
+  async start() {
+    if (!existsSync(this.configPath)) {
+      logger.error(`未找到 PM2 配置文件: ${this.configPath}`)
+      logger.info('请在 commands.json 的 start.stack.stack.ecosystemConfig 中配置正确路径')
+      throw new Error('PM2 配置文件不存在')
+    }
+
+    logger.step('启动 PM2 交互式服务栈')
+
+    await this.prepareBeforeStart()
+    await this.pm2Start()
+    this.isRunning = true
+    await this.showStatus()
+    this.startInteractive()
+  }
+
+  async prepareBeforeStart() {
+    logger.info('正在执行 stack 启动前检查...')
+
+    if (this.preflight.pm2Reset) {
+      await this.resetPm2State()
+    }
+
+    if (this.preflight.killPorts.length > 0) {
+      logger.info(`正在清理端口占用: ${this.preflight.killPorts.join(', ')}`)
+      await execManager.handlePortConflicts(
+        this.preflight.killPorts,
+        Boolean(this.preflight.forcePortCleanup),
+      )
+    }
+
+    this.cleanConfiguredPaths()
+
+    if (this.preflight.cleanTsBuildInfo) {
+      this.cleanTsBuildInfoFiles()
+    }
+  }
+
+  async resetPm2State() {
+    logger.info('正在重置 PM2 状态...')
+
+    try {
+      try {
+        await this.pm2Exec('delete all', { timeout: 5000 })
+      } catch {
+        // ignore
+      }
+
+      await this.pm2Exec('kill', { timeout: 5000 })
+      logger.success('PM2 守护进程已停止')
+    } catch {
+      logger.info('PM2 守护进程可能已停止')
+    }
+
+    const pm2Home = join(homedir(), '.pm2')
+    const stateFiles = ['dump.pm2', 'pm2.log', 'pm2.pid']
+
+    for (const file of stateFiles) {
+      const filePath = join(pm2Home, file)
+      try {
+        if (existsSync(filePath)) {
+          rmSync(filePath, { force: true })
+          logger.success(`已清理 PM2 文件: ${file}`)
+        }
+      } catch (error) {
+        logger.warn(`清理 PM2 文件失败 (${file}): ${error.message}`)
+      }
+    }
+
+    const logDir = join(this.projectRoot, 'logs', 'pm2')
+    if (!existsSync(logDir)) {
+      mkdirSync(logDir, { recursive: true })
+      logger.success('已创建 PM2 日志目录')
+    }
+  }
+
+  cleanConfiguredPaths() {
+    if (this.preflight.cleanPaths.length === 0) return
+
+    logger.info('正在清理缓存路径...')
+
+    for (const rawPath of this.preflight.cleanPaths) {
+      const targetPath = this.resolvePath(rawPath)
+      try {
+        if (existsSync(targetPath)) {
+          rmSync(targetPath, { recursive: true, force: true })
+          logger.success(`已清理: ${rawPath}`)
+        }
+      } catch (error) {
+        logger.warn(`清理失败 (${rawPath}): ${error.message}`)
+      }
+    }
+  }
+
+  cleanTsBuildInfoFiles() {
+    if (this.preflight.cleanTsBuildInfoDirs.length === 0) return
+
+    logger.info('正在清理 TypeScript 构建缓存...')
+
+    for (const rawDirPath of this.preflight.cleanTsBuildInfoDirs) {
+      const dirPath = this.resolvePath(rawDirPath)
+      if (!existsSync(dirPath)) continue
+
+      try {
+        const files = readdirSync(dirPath)
+        for (const file of files) {
+          if (!file.endsWith('.tsbuildinfo')) continue
+          const filePath = join(dirPath, file)
+          try {
+            unlinkSync(filePath)
+            logger.success(`已清理: ${join(rawDirPath, file)}`)
+          } catch (error) {
+            logger.warn(`清理失败 (${join(rawDirPath, file)}): ${error.message}`)
+          }
+        }
+      } catch (error) {
+        logger.warn(`读取目录失败 (${rawDirPath}): ${error.message}`)
+      }
+    }
+  }
+
+  async pm2Exec(args, options = {}) {
+    return execPromise(`${this.pm2Bin} ${args}`, options)
+  }
+
+  async pm2Start() {
+    logger.info('正在启动 PM2 服务...')
+    const { stderr } = await this.pm2Exec(`start "${this.configPath}"`, { timeout: 30000 })
+    if (stderr && !stderr.includes('[PM2]')) {
+      logger.warn(stderr)
+    }
+    logger.success('服务启动成功')
+    this.printServiceUrls()
+  }
+
+  printServiceUrls() {
+    const entries = Object.entries(this.urls)
+    if (entries.length === 0) return
+
+    console.log('')
+    logger.info('服务访问链接:')
+    for (const [service, url] of entries) {
+      console.log(`  ${service.padEnd(12)} → ${url}`)
+    }
+    console.log('')
+  }
+
+  async showStatus() {
+    try {
+      const { stdout } = await this.pm2Exec('list', { timeout: 5000 })
+      console.log(`\n${stdout}`)
+    } catch (error) {
+      logger.error(`获取状态失败: ${error.message}`)
+    }
+  }
+
+  ensureKnownService(service) {
+    if (this.services.includes(service)) return true
+    logger.error(`未知服务: ${service}`)
+    logger.info(`可用服务: ${this.services.join(', ')}`)
+    return false
+  }
+
+  async restart(service) {
+    if (!this.ensureKnownService(service)) return
+
+    logger.info(`正在重启 ${service}...`)
+    try {
+      await this.pm2Exec(`restart ${service}`)
+      logger.success(`${service} 重启成功`)
+      await this.showStatus()
+    } catch (error) {
+      logger.error(`重启失败: ${error.message}`)
+    }
+  }
+
+  async logs(service) {
+    if (!this.ensureKnownService(service)) return
+
+    logger.info(`查看 ${service} 日志（按 Ctrl+C 返回）...`)
+    console.log('')
+
+    const pm2Logs = spawn('bash', ['-lc', `${this.pm2Bin} logs ${service}`], {
+      stdio: 'inherit',
+    })
+
+    await new Promise(resolve => {
+      pm2Logs.on('exit', resolve)
+    })
+
+    console.log('')
+    this.showPrompt()
+  }
+
+  async stop(service) {
+    if (!this.ensureKnownService(service)) return
+
+    logger.info(`正在停止 ${service}...`)
+    try {
+      await this.pm2Exec(`stop ${service}`)
+      logger.success(`${service} 停止成功`)
+      await this.showStatus()
+    } catch (error) {
+      logger.error(`停止失败: ${error.message}`)
+    }
+  }
+
+  async monit() {
+    logger.info('打开 PM2 实时监控（按 q 退出）...')
+    console.log('')
+
+    const pm2Monit = spawn('bash', ['-lc', `${this.pm2Bin} monit`], {
+      stdio: 'inherit',
+    })
+
+    await new Promise(resolve => {
+      pm2Monit.on('exit', resolve)
+    })
+
+    console.log('')
+    this.showPrompt()
+  }
+
+  async stopAll() {
+    if (!this.isRunning) return
+
+    logger.step('正在停止所有服务...')
+    try {
+      await this.pm2Exec('stop all')
+      await this.pm2Exec('delete all')
+      logger.success('所有服务已停止')
+      this.isRunning = false
+    } catch (error) {
+      logger.error(`停止服务失败: ${error.message}`)
+    }
+  }
+
+  showHelp() {
+    console.log('\n可用命令:')
+    console.log('  r <service>   - 重启服务')
+    console.log('  l <service>   - 查看日志')
+    console.log('  s <service>   - 停止服务')
+    console.log('  list          - 显示服务状态')
+    console.log('  monit         - 打开实时监控')
+    console.log('  q / quit      - 停止所有服务并退出')
+    console.log('  help          - 显示此帮助信息')
+    console.log(`\n可用服务: ${this.services.join(', ')}\n`)
+  }
+
+  showPrompt() {
+    process.stdout.write('dx> ')
+  }
+
+  startInteractive() {
+    this.showHelp()
+    this.showPrompt()
+
+    const rl = readline.createInterface({
+      input: process.stdin,
+      output: process.stdout,
+      prompt: '',
+    })
+
+    rl.on('line', async line => {
+      const input = line.trim()
+      if (!input) {
+        this.showPrompt()
+        return
+      }
+
+      const [cmd, ...args] = input.split(/\s+/)
+
+      switch (cmd.toLowerCase()) {
+        case 'r':
+        case 'restart':
+          if (args[0]) await this.restart(args[0])
+          else logger.error('请指定服务名称，例如: r backend')
+          break
+
+        case 'l':
+        case 'logs':
+          if (args[0]) await this.logs(args[0])
+          else logger.error('请指定服务名称，例如: l backend')
+          return
+
+        case 's':
+        case 'stop':
+          if (args[0]) await this.stop(args[0])
+          else logger.error('请指定服务名称，例如: s backend')
+          break
+
+        case 'list':
+        case 'ls':
+          await this.showStatus()
+          break
+
+        case 'monit':
+        case 'monitor':
+          await this.monit()
+          return
+
+        case 'q':
+        case 'quit':
+        case 'exit':
+          await this.stopAll()
+          rl.close()
+          process.exit(0)
+          return
+
+        case 'help':
+        case 'h':
+        case '?':
+          this.showHelp()
+          break
+
+        default:
+          logger.warn(`未知命令: ${cmd}`)
+          logger.info('输入 help 查看可用命令')
+          break
+      }
+
+      this.showPrompt()
+    })
+
+    rl.on('close', async () => {
+      if (this.isRunning) {
+        console.log('\n')
+        await this.stopAll()
+      }
+      process.exit(0)
+    })
+
+    process.on('SIGINT', async () => {
+      console.log('\n')
+      await this.stopAll()
+      process.exit(0)
+    })
+  }
+}
+
+export async function runPm2Stack(options = {}) {
+  const manager = new PM2StackManager(options)
+  await manager.start()
+}

package/lib/cli/commands/start.js

@@ -0,0 +1,58 @@
+import { logger } from '../../logger.js'
+
+export async function handleStart(cli, args) {
+  const service = args[0] || 'development'
+
+  const environment = cli.determineEnvironment()
+  const envKey = cli.normalizeEnvKey(environment)
+  const rawConfig = cli.commands.start[service]
+
+  if (!rawConfig) {
+    logger.error(`未找到启动配置: ${service}`)
+    process.exitCode = 1
+    return
+  }
+
+  let startConfig = rawConfig
+  const isRunnableConfig =
+    rawConfig &&
+    typeof rawConfig === 'object' &&
+    (rawConfig.command || rawConfig.internal || rawConfig.concurrent || rawConfig.sequential)
+
+  if (!isRunnableConfig && rawConfig && typeof rawConfig === 'object') {
+    startConfig = rawConfig[envKey] || null
+  }
+
+  if (!startConfig) {
+    logger.error(`启动目标 ${service} 未提供 ${cli.getEnvironmentFlagExample(envKey) || envKey} 环境配置。`)
+    process.exitCode = 1
+    return
+  }
+
+  logger.step(`启动 ${service} 服务 (${environment})`)
+
+  if (startConfig.concurrent && Array.isArray(startConfig.commands)) {
+    await cli.handleConcurrentCommands(startConfig.commands, 'start', envKey)
+    return
+  }
+
+  if (startConfig.sequential && Array.isArray(startConfig.commands)) {
+    await cli.handleSequentialCommands(startConfig.commands, envKey)
+    return
+  }
+
+  const ports = cli.collectStartPorts(service, startConfig, envKey)
+
+  if (envKey === 'development' && ports.length > 0) {
+    logger.info(`开发环境自动清理端口: ${ports.join(', ')}`)
+  }
+
+  const configToExecute = {
+    ...startConfig,
+    ...(ports.length > 0 ? { ports } : {}),
+    ...(envKey === 'development' ? { forcePortCleanup: true } : {}),
+  }
+
+  // 为执行阶段构造环境标志，确保 dotenv 选择正确层
+  await cli.executeCommand(configToExecute, cli.createExecutionFlags(environment))
+}

package/lib/cli/commands/worktree.js

@@ -0,0 +1,145 @@
+import { logger } from '../../logger.js'
+import { confirmManager } from '../../confirm.js'
+
+export async function handleWorktree(cli, args) {
+  const worktreeManager = await cli.getWorktreeManager()
+  logger.warn('注意：该封装与原生 git worktree 行为不同，勿混用')
+  const action = args[0]
+  const issueNumber = args[1]
+  // 解析可选的基础分支（位置参数或 --base/-b 标志）
+  let baseBranch = null
+  // 位置参数作为第3个无标志参数传入
+  if (args[2] && !String(args[2]).startsWith('-')) {
+    baseBranch = args[2]
+  }
+  // 支持 --base/-b 标志（从原始参数中解析，包含所有标志）
+  const allArgs = cli.args
+  const baseIdx = allArgs.indexOf('--base')
+  const shortBaseIdx = allArgs.indexOf('-b')
+  if (!baseBranch) {
+    if (baseIdx !== -1 && baseIdx + 1 < allArgs.length) baseBranch = allArgs[baseIdx + 1]
+    else if (shortBaseIdx !== -1 && shortBaseIdx + 1 < allArgs.length) baseBranch = allArgs[shortBaseIdx + 1]
+  }
+
+  if (!action) {
+    logger.error('请指定 worktree 操作: make, del, list, clean')
+    logger.info('用法:')
+    logger.info('  dx worktree make <issue_number> [base]  - 创建新的 worktree（可选基础分支）')
+    logger.info('  dx worktree del <issue_number> [issue_number2] ...  - 删除指定 worktree（支持批量）')
+    logger.info('  dx worktree del --all            - 删除所有 issue 相关 worktree')
+    logger.info('  dx worktree list                 - 列出所有 worktree')
+    logger.info('  dx worktree clean                - 清理无效的 worktree')
+    logger.info('')
+    logger.info('选项:')
+    logger.info('  --base <branch>, -b <branch>     - 指定基础分支（make 命令专用）')
+    logger.info('  --all                            - 删除所有 worktree（del 命令专用）')
+    logger.info('  -Y, --yes                        - 跳过所有确认提示（非交互式）')
+    logger.info('')
+    logger.info('示例:')
+    logger.info('  dx worktree make 88              - 从 main 分支创建 worktree')
+    logger.info('  dx worktree make 88 dev          - 从 dev 分支创建 worktree')
+    logger.info('  dx worktree make 88 --base dev   - 使用标志指定基础分支')
+    logger.info('  dx worktree del 88               - 删除单个 worktree')
+    logger.info('  dx worktree del 88 89 90         - 批量删除多个 worktree')
+    logger.info('  dx worktree del --all            - 删除所有 worktree（需确认）')
+    logger.info('  dx worktree del --all -Y         - 删除所有（跳过确认）')
+    logger.warn('注意：该封装与原生 git worktree 行为不同，勿混用')
+    process.exitCode = 1
+    return
+  }
+
+  switch (action) {
+    case 'make':
+      if (!issueNumber) {
+        logger.error('请指定 issue 编号')
+        logger.info('用法: dx worktree make <issue_number> [base]  或  dx worktree make <issue_number> --base <branch>')
+        logger.info('示例: dx worktree make 88 dev  或  dx worktree make 88 --base dev')
+        process.exitCode = 1
+        return
+      }
+      await worktreeManager.make(issueNumber, {
+        force: Boolean(cli.flags.Y),
+        baseBranch,
+      })
+      break
+
+    case 'del':
+      // 互斥校验：--all 不能与 issue 编号同时使用
+      // args[0] 是 action，args[1] 开始才是 issue 编号
+      if (cli.flags.all && args.length > 1) {
+        logger.error('--all 标志不能与 issue 编号同时使用')
+        logger.info('用法: dx worktree del --all  或  dx worktree del <issue_number> ...')
+        process.exitCode = 1
+        return
+      }
+
+      // 批量删除所有 worktree
+      if (cli.flags.all) {
+        const allIssues = await worktreeManager.getAllIssueWorktrees()
+
+        if (allIssues.length === 0) {
+          logger.info('没有找到 issue 相关的 worktree')
+          return
+        }
+
+        logger.info(`\n找到 ${allIssues.length} 个 issue worktree:`)
+        allIssues.forEach(issue => {
+          logger.info(`  - issue-${issue}`)
+        })
+
+        // 安全确认（除非 -Y）
+        if (!cli.flags.Y) {
+          const confirmed = await confirmManager.confirm(
+            `\n确定要删除所有 ${allIssues.length} 个 worktree 吗？(这将永久删除工作目录)`,
+            false,
+            false,
+          )
+          if (!confirmed) {
+            logger.info('操作已取消')
+            return
+          }
+        }
+
+        await worktreeManager.del(allIssues, { force: Boolean(cli.flags.Y) })
+        return
+      }
+
+      // 删除指定 issue 编号（原逻辑保持不变）
+      if (!issueNumber) {
+        logger.error('请指定一个或多个 issue 编号，或使用 --all 删除所有')
+        logger.info('用法: dx worktree del <issue_number> [issue_number2] ...')
+        logger.info('      dx worktree del --all  # 删除所有 issue 相关 worktree')
+        logger.info('示例: dx worktree del 123 456 789  # 批量删除指定 worktree')
+        logger.info('      dx worktree del --all -Y      # 删除所有（跳过确认）')
+        logger.info('选项: -Y, --yes  # 跳过所有确认提示')
+        process.exitCode = 1
+        return
+      }
+
+      // 收集所有 issue 编号（从第二个参数开始的所有非标志参数）
+      const issueNumbers = [issueNumber]
+      for (let i = 2; i < args.length; i++) {
+        const arg = args[i]
+        if (arg && !arg.startsWith('-')) {
+          issueNumbers.push(arg)
+        }
+      }
+
+      await worktreeManager.del(issueNumbers, { force: Boolean(cli.flags.Y) })
+      break
+
+    case 'list':
+      await worktreeManager.list()
+      break
+
+    case 'clean':
+      await worktreeManager.clean()
+      break
+
+    default:
+      logger.error(`未知的 worktree 操作: ${action}`)
+      logger.info('可用操作: make, del, list, clean')
+      logger.info('使用 dx worktree --help 查看详细用法')
+      logger.warn('注意：该封装与原生 git worktree 行为不同，勿混用')
+  }
+}