@sanqianx/claude-ai-workbench 0.13.2 → 0.13.4

package/src/backend/lib/prompt-registry.js

@@ -1,166 +1,166 @@
-// Prompt Registry — reads/writes claude-prompts.json and renders templated prompts.
-//
-// File location: <KB_ROOT>/claude-prompts.json
-// Schema: { schema: 'claude-prompts/v1', prompts: { <key>: { description, model, permissionMode,
-//            allowedTools, systemPrompt, userPrompt } } }
-//
-// Variable interpolation: {{KEY}} placeholders in systemPrompt/userPrompt are replaced
-// with values from the `vars` argument. Unknown placeholders are left untouched (so
-// incomplete prompts don't crash — they show the raw token for diagnosis).
-
-const fs = require('fs');
-const path = require('path');
-
-const { getDataDir } = require('./data-dir');
-const { getWorkbenchConfig } = require('./runtime-config');
-const SCHEMA = 'claude-prompts/v1';
-
-const PLACEHOLDER_RE = /\{\{\s*([A-Z_][A-Z0-9_]*)\s*\}\}/g;
-
-function normalizePromptsConfig(cfg) {
-  const normalized = cfg && typeof cfg === 'object' ? cfg : {};
-  if (!normalized.prompts || typeof normalized.prompts !== 'object') normalized.prompts = {};
-  if (!normalized.schema) normalized.schema = SCHEMA;
-  return normalized;
-}
-
-function promptsPath() {
-  const cfg = getWorkbenchConfig();
-  return path.join(getDataDir(), cfg.promptsFileName || 'claude-prompts.json');
-}
-
-function bundledPromptsPath() {
-  const cfg = getWorkbenchConfig();
-  return cfg.bundledPromptsPath || path.resolve(__dirname, '..', '..', 'assets', 'claude-prompts.json');
-}
-
-function readBundledPrompts() {
-  const file = bundledPromptsPath();
-  if (!fs.existsSync(file)) return { schema: SCHEMA, prompts: {} };
-  try {
-    return normalizePromptsConfig(JSON.parse(fs.readFileSync(file, 'utf-8')));
-  } catch {
-    return { schema: SCHEMA, prompts: {} };
-  }
-}
-
-function persistPrompts(cfg) {
-  const file = promptsPath();
-  fs.mkdirSync(path.dirname(file), { recursive: true });
-  fs.writeFileSync(file, JSON.stringify(cfg, null, 2) + '\n', 'utf-8');
-}
-
-function withBundledDefaults(cfg) {
-  const current = normalizePromptsConfig(cfg);
-  const bundled = readBundledPrompts();
-  let changed = false;
-  const prompts = { ...current.prompts };
-  for (const [key, prompt] of Object.entries(bundled.prompts || {})) {
-    if (!Object.prototype.hasOwnProperty.call(prompts, key)) {
-      prompts[key] = prompt;
-      changed = true;
-    }
-  }
-  return {
-    cfg: {
-      schema: current.schema || bundled.schema || SCHEMA,
-      prompts,
-    },
-    changed,
-  };
-}
-
-function readPrompts() {
-  const file = promptsPath();
-  if (!fs.existsSync(file)) {
-    const seeded = withBundledDefaults({ schema: SCHEMA, prompts: {} }).cfg;
-    if (Object.keys(seeded.prompts).length) {
-      persistPrompts(seeded);
-    }
-    return seeded;
-  }
-  try {
-    const result = withBundledDefaults(JSON.parse(fs.readFileSync(file, 'utf-8')));
-    if (result.changed) persistPrompts(result.cfg);
-    return result.cfg;
-  } catch (e) {
-    throw new Error(`failed to parse ${file}: ${e.message}`);
-  }
-}
-
-function writePrompts(cfg) {
-  if (!cfg || typeof cfg !== 'object') throw new Error('prompts config must be an object');
-  if (cfg.prompts && typeof cfg.prompts !== 'object') throw new Error('prompts field must be an object');
-  const normalized = {
-    schema: cfg.schema || SCHEMA,
-    prompts: cfg.prompts || {},
-  };
-  for (const [key, val] of Object.entries(normalized.prompts)) {
-    if (!key || typeof key !== 'string') throw new Error(`invalid prompt key: ${key}`);
-    if (!val || typeof val !== 'object') throw new Error(`prompt "${key}" must be an object`);
-    if (typeof val.userPrompt !== 'string' || !val.userPrompt) {
-      throw new Error(`prompt "${key}" must have a non-empty userPrompt`);
-    }
-  }
-  persistPrompts(normalized);
-  return normalized;
-}
-
-function listPromptKeys() {
-  return Object.keys(readPrompts().prompts);
-}
-
-function getPrompt(key) {
-  const cfg = readPrompts();
-  const p = cfg.prompts[key];
-  if (!p) return null;
-  return p;
-}
-
-function interpolate(text, vars) {
-  if (typeof text !== 'string') return text;
-  if (!vars || typeof vars !== 'object') return text;
-  return text.replace(PLACEHOLDER_RE, (_, name) => {
-    if (Object.prototype.hasOwnProperty.call(vars, name)) {
-      const v = vars[name];
-      return v == null ? '' : String(v);
-    }
-    return `{{${name}}}`;
-  });
-}
-
-// Render the prompt for a given key, returning all fields needed to spawn claude.
-// Returns null if key doesn't exist.
-function renderPrompt(key, vars) {
-  const p = getPrompt(key);
-  if (!p) return null;
-  const safeVars = vars || {};
-  return {
-    key,
-    description: p.description || '',
-    model: p.model || 'sonnet',
-    permissionMode: p.permissionMode || 'bypassPermissions',
-    allowedTools: Array.isArray(p.allowedTools) ? p.allowedTools.slice() : ['Read', 'Grep', 'Glob'],
-    systemPrompt: interpolate(p.systemPrompt || '', safeVars),
-    userPrompt: interpolate(p.userPrompt, safeVars),
-  };
-}
-
-const exported = {
-  promptsPath,
-  bundledPromptsPath,
-  SCHEMA,
-  readPrompts,
-  writePrompts,
-  listPromptKeys,
-  getPrompt,
-  renderPrompt,
-  interpolate,
-};
-
-Object.defineProperties(exported, {
-  PROMPTS_PATH: { enumerable: true, get: promptsPath },
-  BUNDLED_PROMPTS_PATH: { enumerable: true, get: bundledPromptsPath },
-});
-
-module.exports = exported;
+// Prompt Registry — reads/writes claude-prompts.json and renders templated prompts.
+//
+// File location: <KB_ROOT>/claude-prompts.json
+// Schema: { schema: 'claude-prompts/v1', prompts: { <key>: { description, model, permissionMode,
+//            allowedTools, systemPrompt, userPrompt } } }
+//
+// Variable interpolation: {{KEY}} placeholders in systemPrompt/userPrompt are replaced
+// with values from the `vars` argument. Unknown placeholders are left untouched (so
+// incomplete prompts don't crash — they show the raw token for diagnosis).
+
+const fs = require('fs');
+const path = require('path');
+
+const { getDataDir } = require('./data-dir');
+const { getWorkbenchConfig } = require('./runtime-config');
+const SCHEMA = 'claude-prompts/v1';
+
+const PLACEHOLDER_RE = /\{\{\s*([A-Z_][A-Z0-9_]*)\s*\}\}/g;
+
+function normalizePromptsConfig(cfg) {
+  const normalized = cfg && typeof cfg === 'object' ? cfg : {};
+  if (!normalized.prompts || typeof normalized.prompts !== 'object') normalized.prompts = {};
+  if (!normalized.schema) normalized.schema = SCHEMA;
+  return normalized;
+}
+
+function promptsPath() {
+  const cfg = getWorkbenchConfig();
+  return path.join(getDataDir(), cfg.promptsFileName || 'claude-prompts.json');
+}
+
+function bundledPromptsPath() {
+  const cfg = getWorkbenchConfig();
+  return cfg.bundledPromptsPath || path.resolve(__dirname, '..', '..', 'assets', 'claude-prompts.json');
+}
+
+function readBundledPrompts() {
+  const file = bundledPromptsPath();
+  if (!fs.existsSync(file)) return { schema: SCHEMA, prompts: {} };
+  try {
+    return normalizePromptsConfig(JSON.parse(fs.readFileSync(file, 'utf-8')));
+  } catch {
+    return { schema: SCHEMA, prompts: {} };
+  }
+}
+
+function persistPrompts(cfg) {
+  const file = promptsPath();
+  fs.mkdirSync(path.dirname(file), { recursive: true });
+  fs.writeFileSync(file, JSON.stringify(cfg, null, 2) + '\n', 'utf-8');
+}
+
+function withBundledDefaults(cfg) {
+  const current = normalizePromptsConfig(cfg);
+  const bundled = readBundledPrompts();
+  let changed = false;
+  const prompts = { ...current.prompts };
+  for (const [key, prompt] of Object.entries(bundled.prompts || {})) {
+    if (!Object.prototype.hasOwnProperty.call(prompts, key)) {
+      prompts[key] = prompt;
+      changed = true;
+    }
+  }
+  return {
+    cfg: {
+      schema: current.schema || bundled.schema || SCHEMA,
+      prompts,
+    },
+    changed,
+  };
+}
+
+function readPrompts() {
+  const file = promptsPath();
+  if (!fs.existsSync(file)) {
+    const seeded = withBundledDefaults({ schema: SCHEMA, prompts: {} }).cfg;
+    if (Object.keys(seeded.prompts).length) {
+      persistPrompts(seeded);
+    }
+    return seeded;
+  }
+  try {
+    const result = withBundledDefaults(JSON.parse(fs.readFileSync(file, 'utf-8')));
+    if (result.changed) persistPrompts(result.cfg);
+    return result.cfg;
+  } catch (e) {
+    throw new Error(`failed to parse ${file}: ${e.message}`);
+  }
+}
+
+function writePrompts(cfg) {
+  if (!cfg || typeof cfg !== 'object') throw new Error('prompts config must be an object');
+  if (cfg.prompts && typeof cfg.prompts !== 'object') throw new Error('prompts field must be an object');
+  const normalized = {
+    schema: cfg.schema || SCHEMA,
+    prompts: cfg.prompts || {},
+  };
+  for (const [key, val] of Object.entries(normalized.prompts)) {
+    if (!key || typeof key !== 'string') throw new Error(`invalid prompt key: ${key}`);
+    if (!val || typeof val !== 'object') throw new Error(`prompt "${key}" must be an object`);
+    if (typeof val.userPrompt !== 'string' || !val.userPrompt) {
+      throw new Error(`prompt "${key}" must have a non-empty userPrompt`);
+    }
+  }
+  persistPrompts(normalized);
+  return normalized;
+}
+
+function listPromptKeys() {
+  return Object.keys(readPrompts().prompts);
+}
+
+function getPrompt(key) {
+  const cfg = readPrompts();
+  const p = cfg.prompts[key];
+  if (!p) return null;
+  return p;
+}
+
+function interpolate(text, vars) {
+  if (typeof text !== 'string') return text;
+  if (!vars || typeof vars !== 'object') return text;
+  return text.replace(PLACEHOLDER_RE, (_, name) => {
+    if (Object.prototype.hasOwnProperty.call(vars, name)) {
+      const v = vars[name];
+      return v == null ? '' : String(v);
+    }
+    return `{{${name}}}`;
+  });
+}
+
+// Render the prompt for a given key, returning all fields needed to spawn claude.
+// Returns null if key doesn't exist.
+function renderPrompt(key, vars) {
+  const p = getPrompt(key);
+  if (!p) return null;
+  const safeVars = vars || {};
+  return {
+    key,
+    description: p.description || '',
+    model: p.model || 'sonnet',
+    permissionMode: p.permissionMode || 'bypassPermissions',
+    allowedTools: Array.isArray(p.allowedTools) ? p.allowedTools.slice() : ['Read', 'Grep', 'Glob'],
+    systemPrompt: interpolate(p.systemPrompt || '', safeVars),
+    userPrompt: interpolate(p.userPrompt, safeVars),
+  };
+}
+
+const exported = {
+  promptsPath,
+  bundledPromptsPath,
+  SCHEMA,
+  readPrompts,
+  writePrompts,
+  listPromptKeys,
+  getPrompt,
+  renderPrompt,
+  interpolate,
+};
+
+Object.defineProperties(exported, {
+  PROMPTS_PATH: { enumerable: true, get: promptsPath },
+  BUNDLED_PROMPTS_PATH: { enumerable: true, get: bundledPromptsPath },
+});
+
+module.exports = exported;

package/src/backend/lib/runtime-config.js

@@ -1,121 +1,121 @@
-const path = require('path');
-const os = require('os');
-
-const PRESETS = {
-  workbench: {
-    appName: 'claude-ai-workbench',
-    defaultDataDirName: '.claude-ai-workbench',
-    dataDirEnvVars: ['CLAUDE_AI_WORKBENCH_DATA_DIR', 'KB_DATA_DIR', 'DEV_TASK_RADAR_CLAUDE_DATA_DIR'],
-    sessionDirName: 'claude-workbench',
-    sessionSchema: 'claude-workbench-session/v1',
-    legacySessionDirNames: ['claude-terminal'],
-    legacySessionSchemas: ['devtask-radar-claude-session/v1'],
-    suppressHooksEnv: '',
-    fakeClaudeEnvVar: 'KB_AUTOMATION_FAKE_CLAUDE',
-    staleActiveEnvVar: 'KB_STALE_ACTIVE_MS',
-  },
-  projectKnowledge: {
-    appName: 'project-knowledge',
-    defaultDataDirName: '.project-knowledge',
-    dataDirEnvVars: ['KB_DATA_DIR'],
-    sessionDirName: 'claude-workbench',
-    sessionSchema: 'claude-workbench-session/v1',
-    legacySessionDirNames: [],
-    legacySessionSchemas: [],
-    suppressHooksEnv: '',
-    fakeClaudeEnvVar: 'KB_AUTOMATION_FAKE_CLAUDE',
-    staleActiveEnvVar: 'KB_STALE_ACTIVE_MS',
-  },
-  devTaskRadar: {
-    appName: 'devtask-radar',
-    defaultDataDirName: '.devtask-radar',
-    dataDirEnvVars: ['DEV_TASK_RADAR_CLAUDE_DATA_DIR', 'DEV_TASK_RADAR_ANALYSIS_DATA_DIR', 'KB_DATA_DIR'],
-    sessionDirName: 'claude-terminal',
-    sessionSchema: 'devtask-radar-claude-session/v1',
-    legacySessionDirNames: ['claude-workbench'],
-    legacySessionSchemas: ['claude-workbench-session/v1'],
-    suppressHooksEnv: 'DEV_TASK_RADAR_SUPPRESS_HOOKS',
-    fakeClaudeEnvVar: 'KB_AUTOMATION_FAKE_CLAUDE',
-    staleActiveEnvVar: 'KB_STALE_ACTIVE_MS',
-  },
-};
-
-const DEFAULTS = {
-  ...PRESETS.workbench,
-  aiRootDirName: '_ai',
-  projectsFileName: 'projects.json',
-  promptsFileName: 'claude-prompts.json',
-  bundledPromptsPath: path.resolve(__dirname, '..', '..', 'assets', 'claude-prompts.json'),
-  standardSubdirs: ['drafts', 'runs', 'context-packs', 'backups'],
-  suppressHooksValue: '1',
-};
-
-let runtimeConfig = { ...DEFAULTS };
-
-function normalizeArray(value) {
-  if (!value) return [];
-  return Array.isArray(value) ? value.filter(Boolean).map(String) : [String(value)];
-}
-
-function mergeConfig(base, override = {}) {
-  const next = { ...base, ...(override || {}) };
-  next.dataDirEnvVars = normalizeArray(next.dataDirEnvVars);
-  next.legacySessionDirNames = normalizeArray(next.legacySessionDirNames);
-  next.legacySessionSchemas = normalizeArray(next.legacySessionSchemas);
-  next.standardSubdirs = normalizeArray(next.standardSubdirs);
-  return next;
-}
-
-function configureWorkbench(options = {}) {
-  runtimeConfig = mergeConfig(runtimeConfig, options);
-  return getWorkbenchConfig();
-}
-
-function configurePreset(name, options = {}) {
-  const preset = PRESETS[name];
-  if (!preset) throw new Error(`unknown Claude AI workbench preset: ${name}`);
-  runtimeConfig = mergeConfig({ ...DEFAULTS, ...preset }, options);
-  return getWorkbenchConfig();
-}
-
-function getWorkbenchConfig() {
-  return mergeConfig({}, runtimeConfig);
-}
-
-function firstEnv(names) {
-  for (const name of normalizeArray(names)) {
-    const value = process.env[name];
-    if (value) return value;
-  }
-  return '';
-}
-
-function resolveDataDir() {
-  const cfg = getWorkbenchConfig();
-  const explicit = cfg.dataDir || firstEnv(cfg.dataDirEnvVars);
-  return explicit
-    ? path.resolve(explicit)
-    : path.join(os.homedir(), cfg.defaultDataDirName || '.claude-ai-workbench');
-}
-
-function sessionDirNames() {
-  const cfg = getWorkbenchConfig();
-  return [cfg.sessionDirName || 'claude-workbench', ...normalizeArray(cfg.legacySessionDirNames)]
-    .filter((value, index, arr) => value && arr.indexOf(value) === index);
-}
-
-function sessionSchemas() {
-  const cfg = getWorkbenchConfig();
-  return [cfg.sessionSchema || 'claude-workbench-session/v1', ...normalizeArray(cfg.legacySessionSchemas)]
-    .filter((value, index, arr) => value && arr.indexOf(value) === index);
-}
-
-module.exports = {
-  PRESETS,
-  configureWorkbench,
-  configurePreset,
-  getWorkbenchConfig,
-  resolveDataDir,
-  sessionDirNames,
-  sessionSchemas,
-};
+const path = require('path');
+const os = require('os');
+
+const PRESETS = {
+  workbench: {
+    appName: 'claude-ai-workbench',
+    defaultDataDirName: '.claude-ai-workbench',
+    dataDirEnvVars: ['CLAUDE_AI_WORKBENCH_DATA_DIR', 'KB_DATA_DIR', 'DEV_TASK_RADAR_CLAUDE_DATA_DIR'],
+    sessionDirName: 'claude-workbench',
+    sessionSchema: 'claude-workbench-session/v1',
+    legacySessionDirNames: ['claude-terminal'],
+    legacySessionSchemas: ['devtask-radar-claude-session/v1'],
+    suppressHooksEnv: '',
+    fakeClaudeEnvVar: 'KB_AUTOMATION_FAKE_CLAUDE',
+    staleActiveEnvVar: 'KB_STALE_ACTIVE_MS',
+  },
+  projectKnowledge: {
+    appName: 'project-knowledge',
+    defaultDataDirName: '.project-knowledge',
+    dataDirEnvVars: ['KB_DATA_DIR'],
+    sessionDirName: 'claude-workbench',
+    sessionSchema: 'claude-workbench-session/v1',
+    legacySessionDirNames: [],
+    legacySessionSchemas: [],
+    suppressHooksEnv: '',
+    fakeClaudeEnvVar: 'KB_AUTOMATION_FAKE_CLAUDE',
+    staleActiveEnvVar: 'KB_STALE_ACTIVE_MS',
+  },
+  devTaskRadar: {
+    appName: 'devtask-radar',
+    defaultDataDirName: '.devtask-radar',
+    dataDirEnvVars: ['DEV_TASK_RADAR_CLAUDE_DATA_DIR', 'DEV_TASK_RADAR_ANALYSIS_DATA_DIR', 'KB_DATA_DIR'],
+    sessionDirName: 'claude-terminal',
+    sessionSchema: 'devtask-radar-claude-session/v1',
+    legacySessionDirNames: ['claude-workbench'],
+    legacySessionSchemas: ['claude-workbench-session/v1'],
+    suppressHooksEnv: 'DEV_TASK_RADAR_SUPPRESS_HOOKS',
+    fakeClaudeEnvVar: 'KB_AUTOMATION_FAKE_CLAUDE',
+    staleActiveEnvVar: 'KB_STALE_ACTIVE_MS',
+  },
+};
+
+const DEFAULTS = {
+  ...PRESETS.workbench,
+  aiRootDirName: '_ai',
+  projectsFileName: 'projects.json',
+  promptsFileName: 'claude-prompts.json',
+  bundledPromptsPath: path.resolve(__dirname, '..', '..', 'assets', 'claude-prompts.json'),
+  standardSubdirs: ['drafts', 'runs', 'context-packs', 'backups'],
+  suppressHooksValue: '1',
+};
+
+let runtimeConfig = { ...DEFAULTS };
+
+function normalizeArray(value) {
+  if (!value) return [];
+  return Array.isArray(value) ? value.filter(Boolean).map(String) : [String(value)];
+}
+
+function mergeConfig(base, override = {}) {
+  const next = { ...base, ...(override || {}) };
+  next.dataDirEnvVars = normalizeArray(next.dataDirEnvVars);
+  next.legacySessionDirNames = normalizeArray(next.legacySessionDirNames);
+  next.legacySessionSchemas = normalizeArray(next.legacySessionSchemas);
+  next.standardSubdirs = normalizeArray(next.standardSubdirs);
+  return next;
+}
+
+function configureWorkbench(options = {}) {
+  runtimeConfig = mergeConfig(runtimeConfig, options);
+  return getWorkbenchConfig();
+}
+
+function configurePreset(name, options = {}) {
+  const preset = PRESETS[name];
+  if (!preset) throw new Error(`unknown Claude AI workbench preset: ${name}`);
+  runtimeConfig = mergeConfig({ ...DEFAULTS, ...preset }, options);
+  return getWorkbenchConfig();
+}
+
+function getWorkbenchConfig() {
+  return mergeConfig({}, runtimeConfig);
+}
+
+function firstEnv(names) {
+  for (const name of normalizeArray(names)) {
+    const value = process.env[name];
+    if (value) return value;
+  }
+  return '';
+}
+
+function resolveDataDir() {
+  const cfg = getWorkbenchConfig();
+  const explicit = cfg.dataDir || firstEnv(cfg.dataDirEnvVars);
+  return explicit
+    ? path.resolve(explicit)
+    : path.join(os.homedir(), cfg.defaultDataDirName || '.claude-ai-workbench');
+}
+
+function sessionDirNames() {
+  const cfg = getWorkbenchConfig();
+  return [cfg.sessionDirName || 'claude-workbench', ...normalizeArray(cfg.legacySessionDirNames)]
+    .filter((value, index, arr) => value && arr.indexOf(value) === index);
+}
+
+function sessionSchemas() {
+  const cfg = getWorkbenchConfig();
+  return [cfg.sessionSchema || 'claude-workbench-session/v1', ...normalizeArray(cfg.legacySessionSchemas)]
+    .filter((value, index, arr) => value && arr.indexOf(value) === index);
+}
+
+module.exports = {
+  PRESETS,
+  configureWorkbench,
+  configurePreset,
+  getWorkbenchConfig,
+  resolveDataDir,
+  sessionDirNames,
+  sessionSchemas,
+};