ronds_ai 0.1.10 → 0.1.11

package/README.md

@@ -49,9 +49,29 @@ npx ronds_ai@latest check record
 - 当前 Node 版本
 - Node 是否满足 `>=16`
 - 当前目录下读取到的 `git user.email`
-- 当前 `worker_id` 配置来源和取值
+- 当前 `MCP_TRACKER_WORKER_ID` 的来源和取值
 - `~/.profile` 路径
 
+输出 JSON 后，命令还会继续在终端里显示当前 `MCP_TRACKER_WORKER_ID`，并询问是否要修改。
+如果选择修改：
+
+- Windows：写入用户级持久环境变量，新的终端可直接读取
+- Linux：写入 `~/.profile`，新的 shell 可读取
+
+如果不想进入交互流程，可加上：
+
+```bash
+ronds_ai check record --no-prompt
+```
+
+或：
+
+```bash
+npx ronds_ai@latest check record --no-prompt
+```
+
+交互式修改后，命令会再次显示更新后的 `MCP_TRACKER_WORKER_ID`。
+
 输出是一个 JSON，例如：
 
 ```json

package/bin/ronds_ai.js

@@ -1,9 +1,61 @@
 #!/usr/bin/env node
 
 const { runCodeRecord, saveCliError } = require('../lib/code_record');
-const { runCheckRecord } = require('../lib/check_record');
+const {
+  ENVIRONMENT_VARIABLE_NAME,
+  getRefreshNotice,
+  getPersistentWorkerHint,
+  isPersistentWorkerUpdateSupported,
+  runCheckRecord,
+  savePersistentWorkerId,
+} = require('../lib/check_record');
 const { runDoctor } = require('../lib/doctor');
 const { deployHooks } = require('../lib/hooks_deploy');
+const { promptForText, promptYesNo } = require('../lib/skills_prompt');
+
+function writeWorkerSummary(result) {
+  process.stderr.write(`${ENVIRONMENT_VARIABLE_NAME}: ${result.workerId.value || '（空）'}\n`);
+  process.stderr.write(`来源: ${result.workerId.source || '未设置'}\n`);
+}
+
+async function maybeUpdateWorkerId() {
+  const initialResult = runCheckRecord(process.cwd());
+  writeWorkerSummary(initialResult);
+  process.stderr.write(`${getPersistentWorkerHint()}\n`);
+
+  if (!isPersistentWorkerUpdateSupported()) {
+    return;
+  }
+
+  const shouldModify = await promptYesNo(`是否要修改 ${ENVIRONMENT_VARIABLE_NAME}？`);
+  if (!shouldModify) {
+    return;
+  }
+
+  const nextValue = await promptForText(`请输入新的 ${ENVIRONMENT_VARIABLE_NAME}：`, initialResult.workerId.value);
+  if (!nextValue) {
+    process.stderr.write(`输入为空，已跳过更新 ${ENVIRONMENT_VARIABLE_NAME}。\n`);
+    return;
+  }
+
+  const saveResult = savePersistentWorkerId(nextValue);
+  process.stderr.write(`${ENVIRONMENT_VARIABLE_NAME} 已保存到 ${saveResult.location}。\n`);
+
+  const refreshNotice = getRefreshNotice();
+  if (refreshNotice) {
+    process.stderr.write(`${refreshNotice}\n`);
+  }
+
+  writeWorkerSummary(runCheckRecord(process.cwd()));
+}
+
+function shouldPromptWorkerUpdate(args) {
+  return !args.includes('--no-prompt');
+}
+
+function stripCheckFlags(args) {
+  return args.filter((value) => value !== '--no-prompt');
+}
 
 const SUPPORTED_SOURCES = new Set(['claude', 'cursor', 'codex']);
 
@@ -47,7 +99,8 @@ async function runHooksCommand(args) {
 }
 
 async function runCheckCommand(args) {
-  const [target] = args;
+  const filteredArgs = stripCheckFlags(args);
+  const [target] = filteredArgs;
 
   if (target !== 'record') {
     throw new Error('Unsupported check command');
@@ -56,6 +109,10 @@ async function runCheckCommand(args) {
   const result = runCheckRecord(process.cwd());
   process.stdout.write(`${JSON.stringify(result, null, 2)}\n`);
 
+  if (shouldPromptWorkerUpdate(args)) {
+    await maybeUpdateWorkerId();
+  }
+
   if (!result.ok) {
     process.exitCode = 1;
   }

package/lib/check_record.js

@@ -1,106 +1,240 @@
-const fs = require('fs');
-const os = require('os');
-const path = require('path');
-const { execFileSync } = require('child_process');
-
-const ENVIRONMENT_VARIABLE_NAME = process.env.ENVIRONMENT_VARIABLE_NAME || 'MCP_TRACKER_WORKER_ID';
-const MINIMUM_NODE_MAJOR = 16;
-
-function readGitUserEmail(baseDir) {
-  try {
-    return execFileSync('git', ['config', 'user.email'], {
-      cwd: baseDir,
-      encoding: 'utf8',
-      stdio: ['ignore', 'pipe', 'pipe'],
-    })
-      .trim()
-      .replace(/^["'“”]+/, '')
-      .replace(/["'“”]+$/, '');
-  } catch {
-    return '';
-  }
-}
-
-function getProfileFilePath() {
-  return path.join(os.homedir(), '.profile');
-}
-
-function getCurrentWorkerId() {
-  const envValue = String(process.env[ENVIRONMENT_VARIABLE_NAME] || '').trim();
-  if (envValue) {
-    return {
-      name: ENVIRONMENT_VARIABLE_NAME,
-      value: envValue,
-      source: 'env',
-    };
-  }
-
-  const profileFile = getProfileFilePath();
-  if (!fs.existsSync(profileFile)) {
-    return {
-      name: ENVIRONMENT_VARIABLE_NAME,
-      value: '',
-      source: '',
-    };
-  }
-
-  const lines = fs.readFileSync(profileFile, 'utf8').split('\n');
-  const prefix = `export ${ENVIRONMENT_VARIABLE_NAME}=`;
-
-  for (let index = lines.length - 1; index >= 0; index -= 1) {
-    const line = lines[index].trim();
-    if (!line.startsWith(prefix)) {
-      continue;
-    }
-
-    const value = line
-      .slice(prefix.length)
-      .trim()
-      .replace(/^"/, '')
-      .replace(/"$/, '');
-
-    return {
-      name: ENVIRONMENT_VARIABLE_NAME,
-      value,
-      source: 'profile',
-    };
-  }
-
-  return {
-    name: ENVIRONMENT_VARIABLE_NAME,
-    value: '',
-    source: '',
-  };
-}
-
-function parseNodeMajor(version) {
-  const normalized = String(version || '').trim();
-  const match = normalized.match(/^v?(\d+)/);
-  return match ? Number(match[1]) : 0;
-}
-
-function runCheckRecord(targetDir = process.cwd()) {
-  const baseDir = path.resolve(targetDir);
-  const nodeVersion = process.version;
-  const nodeMajor = parseNodeMajor(nodeVersion);
-  const gitUserEmail = readGitUserEmail(baseDir);
-  const worker = getCurrentWorkerId();
-  const nodeSatisfied = nodeMajor >= MINIMUM_NODE_MAJOR;
-
-  return {
-    ok: nodeSatisfied,
-    targetDir: baseDir,
-    node: {
-      version: nodeVersion,
-      requirement: `>=${MINIMUM_NODE_MAJOR}`,
-      satisfied: nodeSatisfied,
-    },
-    gitUserEmail,
-    workerId: worker,
-    profileFile: getProfileFilePath(),
-  };
-}
-
-module.exports = {
-  runCheckRecord,
-};
+const fs = require('fs');
+const os = require('os');
+const path = require('path');
+const { execFileSync } = require('child_process');
+
+const ENVIRONMENT_VARIABLE_NAME = process.env.ENVIRONMENT_VARIABLE_NAME || 'MCP_TRACKER_WORKER_ID';
+const MINIMUM_NODE_MAJOR = 16;
+const SUPPORTED_PERSISTENT_PLATFORMS = new Set(['win32', 'linux']);
+
+function readGitUserEmail(baseDir) {
+  try {
+    return execFileSync('git', ['config', 'user.email'], {
+      cwd: baseDir,
+      encoding: 'utf8',
+      stdio: ['ignore', 'pipe', 'pipe'],
+    })
+      .trim()
+      .replace(/^["'“”]+/, '')
+      .replace(/["'“”]+$/, '');
+  } catch {
+    return '';
+  }
+}
+
+function getProfileFilePath() {
+  return path.join(os.homedir(), '.profile');
+}
+
+function stripShellQuotes(value) {
+  return String(value || '')
+    .trim()
+    .replace(/^['"]/, '')
+    .replace(/['"]$/, '');
+}
+
+function readWorkerIdFromProfile() {
+  const profileFile = getProfileFilePath();
+  if (!fs.existsSync(profileFile)) {
+    return '';
+  }
+
+  const lines = fs.readFileSync(profileFile, 'utf8').split('\n');
+  const prefix = `export ${ENVIRONMENT_VARIABLE_NAME}=`;
+
+  for (let index = lines.length - 1; index >= 0; index -= 1) {
+    const line = lines[index].trim();
+    if (!line.startsWith(prefix)) {
+      continue;
+    }
+
+    return stripShellQuotes(line.slice(prefix.length));
+  }
+
+  return '';
+}
+
+function readWorkerIdFromWindowsUserEnv() {
+  try {
+    return execFileSync('powershell', ['-NoProfile', '-Command', `[Environment]::GetEnvironmentVariable('${ENVIRONMENT_VARIABLE_NAME}', 'User')`], {
+      encoding: 'utf8',
+      stdio: ['ignore', 'pipe', 'pipe'],
+    }).trim();
+  } catch {
+    return '';
+  }
+}
+
+function getCurrentWorkerId() {
+  const envValue = String(process.env[ENVIRONMENT_VARIABLE_NAME] || '').trim();
+  if (envValue) {
+    return {
+      name: ENVIRONMENT_VARIABLE_NAME,
+      value: envValue,
+      source: 'env',
+    };
+  }
+
+  if (process.platform === 'win32') {
+    const userEnvValue = readWorkerIdFromWindowsUserEnv();
+    if (userEnvValue) {
+      return {
+        name: ENVIRONMENT_VARIABLE_NAME,
+        value: userEnvValue,
+        source: 'user_env',
+      };
+    }
+  }
+
+  const profileValue = readWorkerIdFromProfile();
+  if (profileValue) {
+    return {
+      name: ENVIRONMENT_VARIABLE_NAME,
+      value: profileValue,
+      source: 'profile',
+    };
+  }
+
+  return {
+    name: ENVIRONMENT_VARIABLE_NAME,
+    value: '',
+    source: '',
+  };
+}
+
+function escapeProfileValue(value) {
+  return String(value).replace(/\\/g, '\\\\').replace(/"/g, '\\"');
+}
+
+function upsertProfileWorkerId(value) {
+  const profileFile = getProfileFilePath();
+  const nextLine = `export ${ENVIRONMENT_VARIABLE_NAME}="${escapeProfileValue(value)}"`;
+  const currentContent = fs.existsSync(profileFile)
+    ? fs.readFileSync(profileFile, 'utf8')
+    : '';
+  const lines = currentContent ? currentContent.split('\n') : [];
+  const prefix = `export ${ENVIRONMENT_VARIABLE_NAME}=`;
+  let updated = false;
+
+  for (let index = lines.length - 1; index >= 0; index -= 1) {
+    if (!lines[index].trim().startsWith(prefix)) {
+      continue;
+    }
+
+    lines[index] = nextLine;
+    updated = true;
+    break;
+  }
+
+  const nextLines = updated ? lines : lines.concat(nextLine);
+  const nextContent = `${nextLines.join('\n').replace(/\n*$/, '\n')}`;
+  if (nextContent !== currentContent) {
+    fs.writeFileSync(profileFile, nextContent, 'utf8');
+  }
+
+  return {
+    location: profileFile,
+    source: 'profile',
+  };
+}
+
+function setWindowsUserWorkerId(value) {
+  execFileSync('setx', [ENVIRONMENT_VARIABLE_NAME, String(value)], {
+    stdio: ['ignore', 'pipe', 'pipe'],
+  });
+
+  return {
+    location: 'Windows user environment',
+    source: 'user_env',
+  };
+}
+
+function setPersistentWorkerId(value) {
+  const normalizedValue = String(value || '').trim();
+  if (!normalizedValue) {
+    throw new Error(`${ENVIRONMENT_VARIABLE_NAME} cannot be empty`);
+  }
+
+  if (!SUPPORTED_PERSISTENT_PLATFORMS.has(process.platform)) {
+    throw new Error(`Persistent ${ENVIRONMENT_VARIABLE_NAME} updates are not supported on ${process.platform}`);
+  }
+
+  if (process.platform === 'win32') {
+    return setWindowsUserWorkerId(normalizedValue);
+  }
+
+  return upsertProfileWorkerId(normalizedValue);
+}
+
+function savePersistentWorkerId(value) {
+  const normalizedValue = String(value || '').trim();
+  const result = setPersistentWorkerId(normalizedValue);
+  process.env[ENVIRONMENT_VARIABLE_NAME] = normalizedValue;
+  return result;
+}
+
+function isPersistentWorkerUpdateSupported() {
+  return SUPPORTED_PERSISTENT_PLATFORMS.has(process.platform);
+}
+
+function getPersistentWorkerHint() {
+  if (process.platform === 'win32') {
+    return '会写入 Windows 用户级环境变量，新开的终端可以读取到。';
+  }
+
+  if (process.platform === 'linux') {
+    return `会写入 ${getProfileFilePath()}，新开的 shell 可以读取到。`;
+  }
+
+  return `当前平台暂不支持持久化修改 ${ENVIRONMENT_VARIABLE_NAME}。`;
+}
+
+function getRefreshNotice() {
+  if (process.platform === 'win32') {
+    return '请打开新的终端，让 shell 继承更新后的用户环境变量。';
+  }
+
+  if (process.platform === 'linux') {
+    return `请打开新的 shell，或执行 source ${getProfileFilePath()} 以在当前 shell 中加载新变量。`;
+  }
+
+  return '';
+}
+
+function parseNodeMajor(version) {
+  const normalized = String(version || '').trim();
+  const match = normalized.match(/^v?(\d+)/);
+  return match ? Number(match[1]) : 0;
+}
+
+function runCheckRecord(targetDir = process.cwd()) {
+  const baseDir = path.resolve(targetDir);
+  const nodeVersion = process.version;
+  const nodeMajor = parseNodeMajor(nodeVersion);
+  const gitUserEmail = readGitUserEmail(baseDir);
+  const worker = getCurrentWorkerId();
+  const nodeSatisfied = nodeMajor >= MINIMUM_NODE_MAJOR;
+
+  return {
+    ok: nodeSatisfied,
+    targetDir: baseDir,
+    node: {
+      version: nodeVersion,
+      requirement: `>=${MINIMUM_NODE_MAJOR}`,
+      satisfied: nodeSatisfied,
+    },
+    gitUserEmail,
+    workerId: worker,
+    profileFile: getProfileFilePath(),
+  };
+}
+
+module.exports = {
+  ENVIRONMENT_VARIABLE_NAME,
+  getCurrentWorkerId,
+  getPersistentWorkerHint,
+  getRefreshNotice,
+  isPersistentWorkerUpdateSupported,
+  runCheckRecord,
+  savePersistentWorkerId,
+};

package/lib/skills_prompt.js

@@ -81,11 +81,50 @@ async function promptForScope() {
   }
 }
 
+async function promptYesNo(message) {
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+  });
+
+  try {
+    process.stderr.write(`${message} [y/N]\n`);
+    const answer = String(await new Promise((resolve) => {
+      rl.question('> ', resolve);
+    })).trim().toLowerCase();
+    return answer === 'y' || answer === 'yes';
+  } finally {
+    rl.close();
+  }
+}
+
+async function promptForText(message, defaultValue = '') {
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+  });
+
+  try {
+    process.stderr.write(`${message}\n`);
+    if (defaultValue) {
+      process.stderr.write(`Current value: ${defaultValue}\n`);
+    }
+    const answer = await new Promise((resolve) => {
+      rl.question('> ', resolve);
+    });
+    return String(answer).trim();
+  } finally {
+    rl.close();
+  }
+}
+
 module.exports = {
   SCOPE_CHOICES,
   TOOL_CHOICES,
   normalizeChoiceInput,
   normalizeCsvInput,
   promptForScope,
+  promptForText,
   promptForTools,
+  promptYesNo,
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ronds_ai",
-  "version": "0.1.10",
+  "version": "0.1.11",
   "description": "CLI for reporting AI code edit events.",
   "bin": {
     "ronds_ai": "bin/ronds_ai.js"