@yuchenrx/pmt-cli 0.1.2

package/src/command-config.ts

@@ -0,0 +1,244 @@
+import os from 'node:os'
+import path from 'node:path'
+import {
+  createDefaultConfig,
+  ensurePromptsRoot,
+  getGlobalConfigPath,
+  getProjectConfigPath,
+  loadMergedConfig,
+  readConfigFile,
+  resolvePromptsRoot,
+  writeConfigFile,
+  type ConfigScope,
+} from './config'
+import { formatCommand } from './cli-name'
+import { ask } from './prompt'
+import { fileExists, isFile, isSubpath, isUrl, listFiles, resolvePath } from './helpers'
+
+
+const log = (text: string) => console.log(text)
+const fail = (text: string) => {
+  console.error(text)
+  process.exitCode = 1
+}
+
+const requireConfig = async (scope: ConfigScope) => {
+  const config = await readConfigFile(scope)
+  if (!config) {
+    const configPath = getConfigPathByScope(scope)
+    fail(`未找到配置文件: ${configPath}`)
+    return null
+  }
+  return config
+}
+
+const getConfigPathByScope = (scope: ConfigScope) => {
+  if (scope === 'global') return getGlobalConfigPath()
+  return getProjectConfigPath()
+}
+
+const getBaseDir = (scope: ConfigScope) => {
+  if (scope === 'global') return os.homedir()
+  return process.cwd()
+}
+
+const resolvePromptsPath = (base: string | null, source: string) => {
+  if (!base) return null
+  return resolvePath(source, base)
+}
+
+const resolveExisting = (
+  primary: string,
+  primaryExists: boolean,
+  fallback: string | null,
+  fallbackExists: boolean,
+) => {
+  if (primaryExists) return primary
+  if (fallbackExists && fallback) return fallback
+  return null
+}
+
+const listPromptSuggestions = async (root: string | null) => {
+  if (!root) return []
+  const files = await listFiles(root)
+  return files.sort()
+}
+
+const ensurePromptRepo = async (promptsRoot: string) => {
+  const gitDir = resolvePath('.git', promptsRoot)
+  const exists = await fileExists(gitDir)
+  if (exists) return
+
+  const { spawnSync } = await import('node:child_process')
+  const result = spawnSync('git', ['init'], { cwd: promptsRoot, encoding: 'utf8' })
+  if (result.status !== 0) {
+    const message = result.stderr || result.stdout || 'git init 失败'
+    log(`初始化 git 失败: ${message.trim()}`)
+  }
+}
+
+export const initConfig = async (scope: ConfigScope) => {
+  const configPath = getConfigPathByScope(scope)
+  const exists = await fileExists(configPath)
+  if (exists) {
+    log(`配置已存在: ${configPath}`)
+    return
+  }
+
+  const config = createDefaultConfig()
+  if (scope === 'global') {
+    const rootDefault = '.promptslot/prompts'
+    const promptRoot = await ask('全局 prompts 目录', rootDefault)
+    config.prompts_root = promptRoot
+  } else {
+    const dirDefault = 'prompts'
+    const promptDir = await ask('项目 prompts 目录', dirDefault)
+    config.prompts_dir = promptDir
+  }
+
+  const linkDefault = await ask('默认 link 文件名', config.link_default ?? 'AGENTS.md')
+  config.link_default = linkDefault
+
+
+  if (scope === 'project') {
+    const defaultSpec = resolvePath('规范.md')
+    const hasSpec = await fileExists(defaultSpec)
+    if (hasSpec) {
+      config.sources.spec = '规范.md'
+      config.current = 'spec'
+    }
+  }
+
+  await writeConfigFile(scope, config)
+  const promptsRoot = await ensurePromptsRoot(config, getBaseDir(scope))
+
+  if (scope === 'global' && promptsRoot) {
+    await ensurePromptRepo(promptsRoot)
+  }
+
+  log(`已生成配置: ${configPath}`)
+}
+
+export const setSource = async (name: string, source: string, scope: ConfigScope) => {
+  if (!name || !source) {
+    fail(`用法: ${formatCommand('set <name> <file>')}`)
+    return
+  }
+
+  const loaded = await requireConfig(scope)
+  if (!loaded) return
+
+  if (isUrl(source)) {
+    loaded.config.sources[name] = source
+    loaded.config.current = name
+    await writeConfigFile(scope, loaded.config)
+    log(`已设置 ${name} -> ${source}`)
+    return
+  }
+
+  const baseDir = getBaseDir(scope)
+  const promptsRoot = resolvePromptsRoot(loaded.config, loaded.baseDir)
+  const baseResolved = resolvePath(source, baseDir)
+  const promptsResolved = resolvePromptsPath(promptsRoot, source)
+  const baseExists = await fileExists(baseResolved)
+  let promptsExists = false
+  if (promptsResolved) {
+    promptsExists = await fileExists(promptsResolved)
+  }
+  const resolved = resolveExisting(
+    baseResolved,
+    baseExists,
+    promptsResolved,
+    promptsExists,
+  )
+
+  if (!resolved) {
+    const suggestions = await listPromptSuggestions(promptsRoot)
+    if (suggestions.length) {
+      log(`可选文件: ${suggestions.join(', ')}`)
+    }
+    fail(`文件不存在: ${baseResolved}`)
+    return
+  }
+
+  const isFilePath = await isFile(resolved)
+  if (!isFilePath) {
+    fail(`目标不是文件: ${resolved}`)
+    return
+  }
+  let relativeToPrompts: string | null = null
+  if (promptsRoot) {
+    relativeToPrompts = path.relative(promptsRoot, resolved)
+  }
+  const relativeToBase = path.relative(baseDir, resolved)
+  let stored = relativeToBase
+  if (promptsRoot && relativeToPrompts && isSubpath(promptsRoot, resolved)) {
+    stored = relativeToPrompts
+  }
+
+  loaded.config.sources[name] = stored
+  loaded.config.current = name
+  await writeConfigFile(scope, loaded.config)
+  log(`已设置 ${name} -> ${stored}`)
+}
+
+export const useSource = async (name: string, scope: ConfigScope) => {
+  if (!name) {
+    fail(`用法: ${formatCommand('use <name>')}`)
+    return
+  }
+
+  if (scope === 'global') {
+    const loaded = await requireConfig(scope)
+    if (!loaded) return
+
+    if (!loaded.config.sources[name]) {
+      fail(`未找到命名值: ${name}`)
+      return
+    }
+
+    loaded.config.current = name
+    await writeConfigFile(scope, loaded.config)
+    log(`已切换到 ${name}`)
+    return
+  }
+
+  const merged = await loadMergedConfig()
+  if (!merged.hasConfig) {
+    fail(`未找到配置文件，请先运行: ${formatCommand('init')}`)
+    return
+  }
+
+  if (!merged.resolvedSources[name]) {
+    fail(`未找到命名值: ${name}`)
+    return
+  }
+
+  const project = await requireConfig(scope)
+  if (!project) return
+  project.config.current = name
+  await writeConfigFile(scope, project.config)
+  log(`已切换到 ${name}`)
+}
+
+export const listSources = async () => {
+  const merged = await loadMergedConfig()
+  if (!merged.hasConfig) {
+    fail(`未找到配置文件，请先运行: ${formatCommand('init')}`)
+    return
+  }
+
+  const entries = Object.entries(merged.resolvedSources)
+  if (!entries.length) {
+    log('未配置任何 source')
+    return
+  }
+
+  entries.forEach(([name, source]) => {
+    let tag = ''
+    if (merged.config.current === name) {
+      tag = ' (current)'
+    }
+    log(`${name}: ${source}${tag}`)
+  })
+}

package/src/command-link.ts

@@ -0,0 +1,126 @@
+import { promises as fs } from 'node:fs'
+import path from 'node:path'
+import { formatCommand } from './cli-name'
+import type { ConfigScope } from './config'
+import { ensureTargetExtension, fileExists, getUrlBasename, isUrl, listFiles, parseContentDisposition } from './helpers'
+import { createAutoLink, resolvePromptFile } from './link-manager'
+import { loadPromptRoot } from './prompt-root'
+import { upsertLink } from './links-store'
+import { recordLink } from './link-index'
+import { ensureRenameDaemon } from './rename-daemon'
+
+const log = (text: string) => console.log(text)
+const fail = (text: string) => {
+  console.error(text)
+  process.exitCode = 1
+}
+
+const resolveTargetPath = (targetInput: string | undefined, fallback: string) => {
+  let target = fallback
+  if (targetInput) {
+    const trimmed = targetInput.trim()
+    if (trimmed) {
+      target = trimmed
+    }
+  }
+  return path.resolve(process.cwd(), target)
+}
+
+const listPromptNames = async (root: string) => {
+  const files = await listFiles(root)
+  return files.sort()
+}
+
+const resolveRoot = async (scope?: ConfigScope) => {
+  const info = await loadPromptRoot(scope)
+  if (info) return info.root
+  return null
+}
+
+const resolveRemoteFilename = async (url: string) => {
+  const response = await fetch(url, { method: 'HEAD' }).catch(() => null)
+  if (!response) return null
+  if (!response.ok) return null
+
+  const header = response.headers.get('content-disposition')
+  return parseContentDisposition(header)
+}
+
+const resolveDefaultTargetName = async (sourcePath: string) => {
+  if (isUrl(sourcePath)) {
+    const remoteName = await resolveRemoteFilename(sourcePath)
+    if (remoteName) return remoteName
+    return getUrlBasename(sourcePath)
+  }
+
+  return path.basename(sourcePath)
+}
+
+const resolveLinkSource = async (root: string, name: string) => {
+  if (isUrl(name)) return name
+
+  const source = await resolvePromptFile(root, name)
+  if (source) return source
+
+  const suggestions = await listPromptNames(root)
+  if (suggestions.length) {
+    log(`可选文件: ${suggestions.join(', ')}`)
+  }
+  return null
+}
+
+export const linkFile = async (name: string, target?: string, scope?: ConfigScope) => {
+  if (!name) {
+    fail(`用法: ${formatCommand('link <name> [file] [scope]')}`)
+    return
+  }
+
+  const root = await resolveRoot(scope)
+  if (!root && !isUrl(name)) {
+    fail('未配置 prompts_root，请先运行 init 配置全局目录')
+    return
+  }
+
+  const sourcePath = await resolveLinkSource(root ?? '', name)
+  if (!sourcePath) {
+    fail(`未找到 prompt 文件: ${name}`)
+    return
+  }
+
+  let baseTarget = ''
+  if (target) {
+    baseTarget = resolveTargetPath(target, target)
+  } else {
+    const defaultName = await resolveDefaultTargetName(sourcePath)
+    if (!defaultName) {
+      fail('无法确定目标文件名，请显式传入目标路径')
+      return
+    }
+    baseTarget = resolveTargetPath(defaultName, defaultName)
+  }
+
+  const targetPath = ensureTargetExtension(baseTarget, sourcePath)
+  const exists = await fileExists(targetPath)
+  if (exists) {
+    fail(`目标文件已存在: ${targetPath}`)
+    return
+  }
+
+  await fs.mkdir(path.dirname(targetPath), { recursive: true })
+  const { mode } = await createAutoLink(sourcePath, targetPath)
+  await upsertLink(targetPath, { source: sourcePath, mode })
+  await recordLink(sourcePath, targetPath, process.cwd())
+
+  if (mode === 'hardlink') {
+    log(`已创建硬链接: ${targetPath}`)
+    return
+  }
+
+  if (mode === 'symlink') {
+    log(`已创建软链接: ${targetPath}`)
+    return
+  }
+
+  log(`已创建 link 文件: ${targetPath}`)
+  void ensureRenameDaemon()
+}

package/src/command-open.ts

@@ -0,0 +1,47 @@
+import path from 'node:path'
+import { readConfigFile, type ConfigScope } from './config'
+import { formatCommand } from './cli-name'
+
+const fail = (text: string) => {
+  console.error(text)
+  process.exitCode = 1
+}
+
+const openFolder = async (folder: string) => {
+  const { spawn } = await import('node:child_process')
+  let command = 'xdg-open'
+  const args = [folder]
+
+  if (process.platform === 'win32') {
+    command = 'explorer'
+  } else if (process.platform === 'darwin') {
+    command = 'open'
+  }
+
+  const proc = spawn(command, args, {
+    detached: true,
+    stdio: 'ignore',
+  })
+  proc.unref()
+}
+
+const resolveConfigPath = async (scope?: ConfigScope) => {
+  if (scope === 'project') return readConfigFile('project')
+  if (scope === 'global') return readConfigFile('global')
+
+  const project = await readConfigFile('project')
+  if (project) return project
+
+  return readConfigFile('global')
+}
+
+export const openConfigFolder = async (scope?: ConfigScope) => {
+  const config = await resolveConfigPath(scope)
+  if (!config) {
+    fail(`未找到配置文件，请先运行: ${formatCommand('init')}`)
+    return
+  }
+
+  const folder = path.dirname(config.configPath)
+  await openFolder(folder)
+}

package/src/command-search.ts

@@ -0,0 +1,52 @@
+import { linkFile } from './command-link'
+import type { ConfigScope } from './config'
+import { formatCommand } from './cli-name'
+import { listFiles } from './helpers'
+import { loadPromptRoot } from './prompt-root'
+
+const log = (text: string) => console.log(text)
+const fail = (text: string) => {
+  console.error(text)
+  process.exitCode = 1
+}
+
+const pickFromList = async (names: string[]) => {
+  if (!process.stdin.isTTY) return null
+  if (!names.length) return null
+
+  const module = await import('enquirer')
+  const AutoComplete = (module as unknown as { AutoComplete: new (...args: unknown[]) => { run: () => Promise<string> } }).AutoComplete
+
+  const prompt = new AutoComplete({
+    name: 'prompt',
+    message: '选择 prompt 文件',
+    choices: names,
+    limit: 10,
+  })
+
+  const selected = await prompt.run().catch(() => null)
+  if (typeof selected === 'string') return selected
+  return null
+}
+
+export const searchAndLink = async (target?: string, scope?: ConfigScope) => {
+  const info = await loadPromptRoot(scope)
+  if (!info) {
+    fail(`未配置 prompts_root，请先运行: ${formatCommand('init')}`)
+    return
+  }
+
+  const names = await listFiles(info.root)
+  if (!names.length) {
+    log('未找到任何 prompt 文件')
+    return
+  }
+
+  const selected = await pickFromList(names.sort())
+  if (!selected) {
+    log('未选择任何文件')
+    return
+  }
+
+  await linkFile(selected, target, scope)
+}

package/src/command-status.ts

@@ -0,0 +1,27 @@
+import path from 'node:path'
+import { readIndex } from './link-index'
+import { readLinks } from './links-store'
+import { fileExists } from './helpers'
+
+const log = (text: string) => console.log(text)
+
+export const statusLinks = async () => {
+  const links = await readLinks()
+  const entries = Object.entries(links)
+  if (!entries.length) {
+    log('未发现 link 记录')
+    return
+  }
+
+  for (const [target, entry] of entries) {
+    const targetPath = path.resolve(process.cwd(), target)
+    const exists = await fileExists(targetPath)
+    const flag = exists ? 'ok' : 'missing'
+    log(`${target} -> ${entry.source} [${entry.mode}] ${flag}`)
+  }
+
+  const index = await readIndex()
+  if (index.roots.length) {
+    log(`roots: ${index.roots.join(', ')}`)
+  }
+}

package/src/commands.ts

@@ -0,0 +1,165 @@
+import { Command } from 'commander'
+import { applyCurrent, insertPlaceholder } from './command-apply'
+import { initConfig, listSources, setSource, useSource } from './command-config'
+import { linkFile } from './command-link'
+import { searchAndLink } from './command-search'
+import { openConfigFolder } from './command-open'
+import { statusLinks } from './command-status'
+import { runRenameDaemon } from './rename-daemon'
+import { resolveScope } from './scope'
+import type { ConfigScope } from './config'
+
+
+export const registerCommands = (program: Command) => {
+  program
+    .command('init')
+    .description('初始化全局配置')
+    .action(async () => {
+      await initConfig('global')
+    })
+
+  program
+    .command('set')
+    .description('设置并切换值')
+    .argument('<name>', '命名 key，如 spec')
+    .argument('<file>', 'prompt 文件路径')
+    .argument('[scope]', 'project | global，默认 global')
+    .action(async (name, file, scopeInput) => {
+      const { scope, error } = resolveScope(scopeInput)
+      if (error) {
+        console.error(error)
+        process.exitCode = 1
+        return
+      }
+      await setSource(name, file, scope)
+    })
+
+  program
+    .command('use')
+    .description('切换到已保存的值')
+    .argument('<name>', '命名 key，如 spec')
+    .argument('[scope]', 'project | global，默认 global')
+    .action(async (name, scopeInput) => {
+      const { scope, error } = resolveScope(scopeInput)
+      if (error) {
+        console.error(error)
+        process.exitCode = 1
+        return
+      }
+      await useSource(name, scope)
+    })
+
+
+  program
+    .command('insert')
+    .description('插入占位符')
+    .argument('<file>', '目标文件路径')
+    .action(async (file) => {
+      await insertPlaceholder(file)
+    })
+
+  program
+    .command('apply')
+    .description('应用当前值')
+    .argument('[files...]', '指定要替换的文件列表')
+    .action(async (files: string[] = []) => {
+      await applyCurrent(files)
+    })
+
+  program
+    .command('link')
+    .description('创建 link 文件')
+    .argument('[name]', 'prompt 文件名（可省略进入交互）')
+    .argument('[file]', '目标文件路径')
+    .argument('[scope]', 'project | global，默认 global')
+    .option('-i', '交互搜索')
+    .action(async (name, file, scopeInput, options) => {
+      const interactive = options.i || !name || name === '-'
+      let targetFile = file
+      let scopeArg = scopeInput
+
+      if (interactive) {
+        if (name && name !== '-') {
+          targetFile = name
+          scopeArg = file
+        }
+        let scope: ConfigScope | undefined
+        if (scopeArg) {
+          const result = resolveScope(scopeArg)
+          if (result.error) {
+            console.error(result.error)
+            process.exitCode = 1
+            return
+          }
+          scope = result.scope
+        }
+        await searchAndLink(targetFile, scope)
+        return
+      }
+
+      let scope: ConfigScope | undefined
+      if (scopeInput) {
+        const result = resolveScope(scopeInput)
+        if (result.error) {
+          console.error(result.error)
+          process.exitCode = 1
+          return
+        }
+        scope = result.scope
+      }
+      await linkFile(name, file, scope)
+    })
+
+  program
+    .command('search')
+    .description('交互搜索并创建 link')
+    .argument('[file]', '目标文件路径')
+    .argument('[scope]', 'project | global，默认 global')
+    .action(async (file, scopeInput) => {
+      let scope: ConfigScope | undefined
+      if (scopeInput) {
+        const result = resolveScope(scopeInput)
+        if (result.error) {
+          console.error(result.error)
+          process.exitCode = 1
+          return
+        }
+        scope = result.scope
+      }
+      await searchAndLink(file, scope)
+    })
+
+  program
+    .command('open-config')
+    .description('打开配置所在文件夹')
+    .argument('[scope]', 'project | global，默认 global')
+    .action(async (scopeInput) => {
+      let scope: ConfigScope | undefined
+      if (scopeInput) {
+        const result = resolveScope(scopeInput)
+        if (result.error) {
+          console.error(result.error)
+          process.exitCode = 1
+          return
+        }
+        scope = result.scope
+      }
+      await openConfigFolder(scope)
+    })
+    .alias('open')
+
+  program
+    .command('status')
+    .description('查看 link 状态')
+    .action(async () => {
+      await statusLinks()
+    })
+
+  program
+    .command('rename-daemon', { hidden: true })
+    .description('内部重命名监听')
+    .action(async () => {
+      await runRenameDaemon()
+    })
+
+}