dpml-prompt/src/constants.js

DPML-powered AI prompt framework - Revolutionary AI-First CLI system based on Deepractice Prompt Markup Language. Build sophisticated AI agents with structured prompts, memory systems, and execution frameworks.

/**
 * PromptX 系统常量配置
 * 统一管理命令格式、路径等配置信息
 */

const PromptXConfig = require('./lib/utils/promptxConfig')

// 缓存配置实例和命令前缀
let _config = null
let _cachedPrefix = null

/**
 * 获取配置实例
 */
function getConfig() {
  if (!_config) {
    _config = new PromptXConfig()
  }
  return _config
}

/**
 * 动态检测命令前缀
 * 优先级：环境变量 > 配置文件 > npm环境变量检测 > 默认值
 */
function detectCommandPrefix() {
  // 返回缓存的结果
  if (_cachedPrefix) {
    return _cachedPrefix
  }

  // 1. 环境变量优先（用于测试和自定义）
  if (process.env.DPML_COMMAND_PREFIX) {
    _cachedPrefix = process.env.DPML_COMMAND_PREFIX
    return _cachedPrefix
  }

  // 2. 尝试读取配置文件（同步方式，避免异步复杂性）
  try {
    const config = getConfig()
    const configPath = config.getPath('command-prefix')
    const fs = require('fs')
    if (fs.existsSync(configPath)) {
      _cachedPrefix = fs.readFileSync(configPath, 'utf8').trim()
      if (_cachedPrefix) {
        return _cachedPrefix
      }
    }
  } catch (error) {
    // 忽略读取错误，继续下一步检测
  }

  // 3. npm环境变量检测
  if (process.env.npm_execpath?.includes('npx') || 
      process.env.npm_config_user_agent?.includes('npx')) {
    _cachedPrefix = 'npx dpml-prompt@snapshot'
  } else {
    _cachedPrefix = 'npx dpml-prompt@snapshot' // 默认值保持安全
  }

  return _cachedPrefix
}

/**
 * 智能推测用户使用的命令前缀
 * 基于环境变量和执行路径的启发式判断
 */
function reconstructCommandPrefix() {
  // 最简单最直接的判断：如果有 npm_execpath 且包含 npx，就是 npx 调用
  if (process.env.npm_execpath && process.env.npm_execpath.includes('npx')) {
    return 'npx dpml-prompt@snapshot' // 默认 snapshot 版本
  }
  
  // 其他情况默认是全局安装
  return 'dpml-prompt'
}

/**
 * 保存命令前缀到配置文件
 * 在init命令中调用
 */
async function saveCommandPrefix() {
  try {
    const actualPrefix = reconstructCommandPrefix()
    const config = getConfig()
    await config.writeText('command-prefix', actualPrefix)
    
    // 更新缓存
    _cachedPrefix = actualPrefix
    
    return actualPrefix
  } catch (error) {
    console.warn('保存命令前缀失败:', error.message)
    return null
  }
}

// 动态生成命令常量（函数式）
function getCommands() {
  const prefix = detectCommandPrefix()
  return {
    INIT: `${prefix} init`,
    HELLO: `${prefix} hello`,
    ACTION: `${prefix} action`,
    LEARN: `${prefix} learn`,
    RECALL: `${prefix} recall`,
    REMEMBER: `${prefix} remember`,
    HELP: `${prefix} help`
  }
}

// 带参数的命令构建函数
function getBuildCommand() {
  const prefix = detectCommandPrefix()
  return {
    action: (roleId) => `${prefix} action ${roleId}`,
    learn: (resource) => `${prefix} learn ${resource}`,
    recall: (query = '') => `${prefix} recall${query ? ' ' + query : ''}`,
    remember: (content = '<content>') => `${prefix} remember${content !== '<content>' ? ' "' + content + '"' : ' <content>'}`
  }
}

// 为了向后兼容，保留原有的静态导出方式
// 但实际上是动态计算的
const COMMANDS = new Proxy({}, {
  get(target, prop) {
    return getCommands()[prop]
  }
})

const buildCommand = new Proxy({}, {
  get(target, prop) {
    return getBuildCommand()[prop]
  }
})

// 系统路径配置（静态）
const PATHS = {
  POUCH_DIR: '.promptx',
  MEMORY_DIR: '.promptx/memory',
  STATE_FILE: '.promptx/pouch.json',
  MEMORY_FILE: '.promptx/memory/declarative.md'
}

// 版本信息
const VERSION = '0.0.1'

// 系统状态
const STATES = {
  INITIALIZED: 'initialized',
  ROLE_DISCOVERY: 'role_discovery',
  ACTION_PLAN_GENERATED: 'action_plan_generated',
  LEARNED_ROLE: 'learned_role',
  MEMORY_SAVED: 'memory_saved',
  RECALL_WAITING: 'recall-waiting'
}

/**
 * 清除缓存（主要用于测试）
 */
function clearCache() {
  _cachedPrefix = null
  _config = null
}

// 导出
module.exports = {
  // 新的函数式API（推荐）
  getCommands,
  getBuildCommand,
  detectCommandPrefix,
  reconstructCommandPrefix,
  saveCommandPrefix,
  clearCache,
  
  // 向后兼容的静态API
  COMMANDS,
  buildCommand,
  
  // 其他静态常量
  PATHS,
  VERSION,
  STATES
}