feat: 实现动态命令前缀检测机制 - 新增 PromptXConfig 工具类统一管理 .promptx 目录配置文件 - 重构 constants.js 为函数式，支持动态命令前缀检测 - init 命令自动保存用户实际使用的命令前缀 - 优先级：环境变量 > 配置文件 > npm环境检测 > 默认值 - 解决 AI 提示命令与用户实际使用不一致的问题 - 完整的 E2E 测试覆盖所有使用场景 核心价值：用户怎么调用 init，AI 就提示相同的命令前缀，确保 AI-First CLI 的用户体验一致性
This commit is contained in:
sean
200
src/tests/commands/command-prefix.e2e.test.js
Normal file
200
src/tests/commands/command-prefix.e2e.test.js
Normal file
@ -0,0 +1,200 @@
|
||||
const path = require('path')
|
||||
const fs = require('fs-extra')
|
||||
const { execSync } = require('child_process')
|
||||
const tmp = require('tmp')
|
||||
|
||||
const PromptXConfig = require('../../lib/utils/promptxConfig')
|
||||
const PouchCLI = require('../../lib/core/pouch/PouchCLI')
|
||||
|
||||
describe('命令前缀动态检测 E2E', () => {
|
||||
let tempDir
|
||||
let originalCwd
|
||||
let config
|
||||
|
||||
beforeEach(async () => {
|
||||
// 创建临时目录
|
||||
tempDir = tmp.dirSync({ unsafeCleanup: true }).name
|
||||
originalCwd = process.cwd()
|
||||
process.chdir(tempDir)
|
||||
|
||||
config = new PromptXConfig(tempDir)
|
||||
|
||||
// 静默console输出
|
||||
jest.spyOn(console, 'log').mockImplementation(() => {})
|
||||
jest.spyOn(console, 'warn').mockImplementation(() => {})
|
||||
|
||||
// 清除constants.js的缓存，避免测试间污染
|
||||
delete require.cache[require.resolve('../../constants.js')]
|
||||
const { clearCache } = require('../../constants.js')
|
||||
clearCache()
|
||||
})
|
||||
|
||||
afterEach(async () => {
|
||||
process.chdir(originalCwd)
|
||||
try {
|
||||
await fs.remove(tempDir)
|
||||
} catch (error) {
|
||||
// 忽略清理失败
|
||||
}
|
||||
|
||||
// 恢复console输出
|
||||
jest.restoreAllMocks()
|
||||
})
|
||||
|
||||
describe('init命令保存命令前缀', () => {
|
||||
test('npx方式调用时应保存npx前缀', async () => {
|
||||
// 模拟npx调用init命令
|
||||
process.argv = ['node', 'npx_cache/dpml-prompt@snapshot/dist/bin/promptx.js', 'init']
|
||||
process.env.npm_execpath = '/usr/local/lib/node_modules/npm/bin/npx-cli.js'
|
||||
|
||||
// 导入并执行init命令
|
||||
const cli = new PouchCLI()
|
||||
|
||||
await cli.execute('init', [])
|
||||
|
||||
// 验证保存的命令前缀
|
||||
const savedPrefix = await config.readText('command-prefix')
|
||||
expect(savedPrefix).toBe('npx dpml-prompt@snapshot')
|
||||
})
|
||||
|
||||
test('全局安装调用时应保存直接前缀', async () => {
|
||||
// 模拟全局安装调用
|
||||
process.argv = ['node', '/usr/local/bin/dpml-prompt', 'init']
|
||||
delete process.env.npm_execpath
|
||||
|
||||
const cli = new PouchCLI()
|
||||
|
||||
await cli.execute('init', [])
|
||||
|
||||
const savedPrefix = await config.readText('command-prefix')
|
||||
expect(savedPrefix).toBe('dpml-prompt')
|
||||
})
|
||||
|
||||
test('指定版本号时应正确保存', async () => {
|
||||
process.argv = ['node', 'npx_cache/dpml-prompt@latest/dist/bin/promptx.js', 'init']
|
||||
process.env.npm_execpath = '/usr/local/lib/node_modules/npm/bin/npx-cli.js'
|
||||
|
||||
const cli = new PouchCLI()
|
||||
|
||||
await cli.execute('init', [])
|
||||
|
||||
const savedPrefix = await config.readText('command-prefix')
|
||||
expect(savedPrefix).toBe('npx dpml-prompt@snapshot') // 简化逻辑只会返回默认snapshot版本
|
||||
})
|
||||
})
|
||||
|
||||
describe('constants.js动态读取', () => {
|
||||
test('存在配置文件时应使用保存的前缀', async () => {
|
||||
// 预先保存配置
|
||||
await config.writeText('command-prefix', 'npx dpml-prompt@0.0.2')
|
||||
|
||||
// 重新require constants.js以触发动态读取
|
||||
delete require.cache[require.resolve('../../constants.js')]
|
||||
const constants = require('../../constants.js')
|
||||
|
||||
const commands = constants.getCommands()
|
||||
expect(commands.INIT).toBe('npx dpml-prompt@0.0.2 init')
|
||||
expect(commands.HELLO).toBe('npx dpml-prompt@0.0.2 hello')
|
||||
})
|
||||
|
||||
test('不存在配置文件时应使用默认前缀', async () => {
|
||||
// 确保配置文件不存在
|
||||
await config.remove('command-prefix')
|
||||
|
||||
delete require.cache[require.resolve('../../constants.js')]
|
||||
const constants = require('../../constants.js')
|
||||
|
||||
const commands = constants.getCommands()
|
||||
expect(commands.INIT).toBe('npx dpml-prompt@snapshot init')
|
||||
})
|
||||
|
||||
test('环境变量应能覆盖配置文件', async () => {
|
||||
// 保存配置文件
|
||||
await config.writeText('command-prefix', 'npx dpml-prompt@snapshot')
|
||||
|
||||
// 设置环境变量
|
||||
process.env.DPML_COMMAND_PREFIX = 'my-custom-prefix'
|
||||
|
||||
delete require.cache[require.resolve('../../constants.js')]
|
||||
const constants = require('../../constants.js')
|
||||
|
||||
const commands = constants.getCommands()
|
||||
expect(commands.INIT).toBe('my-custom-prefix init')
|
||||
|
||||
// 清理环境变量
|
||||
delete process.env.DPML_COMMAND_PREFIX
|
||||
})
|
||||
})
|
||||
|
||||
describe('各种命令格式解析', () => {
|
||||
const testCases = [
|
||||
{
|
||||
name: 'npx最新版本',
|
||||
argv: ['node', '/tmp/.npm/_npx/1234/lib/node_modules/dpml-prompt/src/bin/promptx.js', 'init'],
|
||||
hasNpxEnv: true,
|
||||
expected: 'npx dpml-prompt@snapshot'
|
||||
},
|
||||
{
|
||||
name: 'npx指定版本',
|
||||
argv: ['node', '/tmp/.npm/_npx/1234/lib/node_modules/dpml-prompt@0.1.0/src/bin/promptx.js', 'init'],
|
||||
hasNpxEnv: true,
|
||||
expected: 'npx dpml-prompt@snapshot'
|
||||
},
|
||||
{
|
||||
name: 'npx snapshot版本',
|
||||
argv: ['node', '/tmp/.npm/_npx/1234/lib/node_modules/dpml-prompt@snapshot/src/bin/promptx.js', 'init'],
|
||||
hasNpxEnv: true,
|
||||
expected: 'npx dpml-prompt@snapshot'
|
||||
},
|
||||
{
|
||||
name: '全局安装',
|
||||
argv: ['node', '/usr/local/bin/dpml-prompt', 'init'],
|
||||
hasNpxEnv: false,
|
||||
expected: 'dpml-prompt'
|
||||
},
|
||||
{
|
||||
name: '开发模式',
|
||||
argv: ['node', '/Users/dev/PromptX/src/bin/promptx.js', 'init'],
|
||||
hasNpxEnv: false,
|
||||
expected: 'dpml-prompt'
|
||||
}
|
||||
]
|
||||
|
||||
testCases.forEach(({ name, argv, hasNpxEnv, expected }) => {
|
||||
test(`${name}: ${argv[1]} → ${expected}`, async () => {
|
||||
process.argv = argv
|
||||
|
||||
// 根据测试配置设置环境变量
|
||||
if (hasNpxEnv) {
|
||||
process.env.npm_execpath = '/usr/local/lib/node_modules/npm/bin/npx-cli.js'
|
||||
} else {
|
||||
delete process.env.npm_execpath
|
||||
}
|
||||
|
||||
const cli = new PouchCLI()
|
||||
|
||||
await cli.execute('init', [])
|
||||
|
||||
const savedPrefix = await config.readText('command-prefix')
|
||||
expect(savedPrefix).toBe(expected)
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
describe('缓存性能', () => {
|
||||
test('第二次调用应使用缓存，不重新检测', async () => {
|
||||
// 第一次调用保存配置
|
||||
await config.writeText('command-prefix', 'npx dpml-prompt@snapshot')
|
||||
|
||||
// 模拟多次调用constants
|
||||
delete require.cache[require.resolve('../../constants.js')]
|
||||
const constants1 = require('../../constants.js')
|
||||
|
||||
delete require.cache[require.resolve('../../constants.js')]
|
||||
const constants2 = require('../../constants.js')
|
||||
|
||||
// 两次调用应该返回相同结果
|
||||
expect(constants1.getCommands().INIT).toBe(constants2.getCommands().INIT)
|
||||
})
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user