@lumi-ai-lab/harness-data 0.0.19 → 0.0.21

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lumi-ai-lab/harness-data",
-  "version": "0.0.19",
+  "version": "0.0.21",
   "description": "Installer and updater for Harness Data",
   "type": "module",
   "bin": {

package/src/commands/install.js

@@ -57,6 +57,21 @@ function cleanupRuntimeBackups(backups) {
   for (const item of backups) fs.rmSync(item.backup, { recursive: true, force: true });
 }
 
+function mergeRuntimeConfig(runtimeDir, sourceDir) {
+  const targetDir = path.join(runtimeDir, "config");
+  fs.mkdirSync(targetDir, { recursive: true });
+  for (const file of fs.readdirSync(sourceDir)) {
+    const source = path.join(sourceDir, file);
+    const target = path.join(targetDir, file);
+    const stat = fs.statSync(source);
+    if (stat.isDirectory()) {
+      if (!fs.existsSync(target)) fs.cpSync(source, target, { recursive: true });
+      continue;
+    }
+    if (file.endsWith(".example") || !fs.existsSync(target)) fs.copyFileSync(source, target);
+  }
+}
+
 export async function installRuntimeBundle(runtimeDir, options = {}) {
   if (!options.force && fs.existsSync(path.join(runtimeDir, "agents")) &&
       fs.existsSync(path.join(runtimeDir, "config")) &&
@@ -102,7 +117,8 @@ export async function installRuntimeBundle(runtimeDir, options = {}) {
     fs.mkdirSync(path.join(stagedRoot, "config"), { recursive: true });
     for (const file of fs.readdirSync(configSource)) fs.copyFileSync(path.join(configSource, file), path.join(stagedRoot, "config", file));
 
-    for (const name of ["agents", "bootstrap", "config"]) replaceRuntimePath(runtimeDir, name, stagedRoot, backups);
+    for (const name of ["agents", "bootstrap"]) replaceRuntimePath(runtimeDir, name, stagedRoot, backups);
+    mergeRuntimeConfig(runtimeDir, path.join(stagedRoot, "config"));
     cleanupRuntimeBackups(backups);
   } catch (error) {
     restoreRuntimeBackups(backups);
@@ -328,7 +344,8 @@ export async function installCommand(options = {}) {
   blank();
 
   step(8, 8, "配置 Agent Hook");
-  const linkedAgents = linkAgents(runtimeDir, await chooseAgent(options));
+  const selectedAgent = await chooseAgent(options);
+  const linkedAgents = linkAgents(runtimeDir, selectedAgent);
   for (const [source, target] of linkedAgents) {
     if (fs.existsSync(path.join(runtimeDir, target))) ok(`${target} -> ${source}`);
   }
@@ -347,6 +364,7 @@ export async function installCommand(options = {}) {
     tools: manifest.installedTools || {},
     manifestSha256: manifestDigest(manifest),
     packageVersion: packageVersion(),
+    agent: selectedAgent,
     lastCheckAt: new Date().toISOString()
   });
   console.log(`安装完成：${runtimeDir}`);

package/src/commands/update.js

@@ -1,6 +1,6 @@
 import fs from "node:fs";
 import path from "node:path";
-import { confirm } from "../lib/prompt.js";
+import { chooseAgent, confirm } from "../lib/prompt.js";
 import { findWorkspaceDir, readUserState, writeState } from "../lib/paths.js";
 import { run } from "../lib/exec.js";
 import { installToolsFromManifest, manifestDigest, readManifest } from "../lib/manifest.js";
@@ -11,10 +11,15 @@ import { resolveLatestTool } from "../lib/tool-release.js";
 import { protocolFromUrl, runGitWithProtocol } from "../lib/git-auth.js";
 import { buildAndCheck, installRuntimeBundle, printDoctorSummary } from "./install.js";
 import { collectDoctor } from "./doctor.js";
+import { hasAnyAgentHook, linkAgents, writeLocalConfig } from "../lib/config.js";
 import { action, blank, header, ok, shortSha, skip, step, warn } from "../lib/log.js";
 
 export function isNonBlockingUpdateDoctorCheck(check) {
-  return check.name === "Agent hook" || check.name.startsWith("Agent hook .");
+  return check.name === "Agent hook" ||
+    check.name.startsWith("Agent hook .") ||
+    check.name === "CAS credentials file" ||
+    check.name === "CMR token" ||
+    check.name === "Indicators token";
 }
 
 async function npmLatest() {
@@ -47,7 +52,7 @@ async function maybeUpdateTool(runtimeDir, manifest, tool, options, state) {
     return null;
   }
   action(`发现更新：${tool.name} ${current.version || "unknown"} -> ${tag}`);
-  if (!(await confirm(`是否更新 ${tool.name}？`, { defaultNo: true }))) {
+  if (!(await confirm(`是否更新 ${tool.name}？`))) {
     skip(tool.name);
     options.skippedUpdates?.push(`${tool.name} ${tag}`);
     return null;
@@ -87,7 +92,7 @@ export async function updateWikis(runtimeDir, options, state) {
     return null;
   }
   action(`发现更新：harness-data-wikis ${shortSha(local)} -> ${shortSha(remote)}`);
-  if (!(await confirm("是否更新 harness-data-wikis？", { defaultNo: true }))) {
+  if (!(await confirm("是否更新 harness-data-wikis？"))) {
     skip("harness-data-wikis");
     options.skippedUpdates?.push(`harness-data-wikis ${shortSha(remote)}`);
     return null;
@@ -98,6 +103,20 @@ export async function updateWikis(runtimeDir, options, state) {
   return { commit };
 }
 
+export async function restoreAgentHooksIfMissing(runtimeDir, options = {}) {
+  if (hasAnyAgentHook(runtimeDir)) {
+    ok("Agent Hook 已配置");
+    return null;
+  }
+  action("未检测到 Agent Hook，重新配置");
+  const agent = await chooseAgent(options);
+  const linkedAgents = linkAgents(runtimeDir, agent);
+  for (const [source, target] of linkedAgents) {
+    if (fs.existsSync(path.join(runtimeDir, target))) ok(`${target} -> ${source}`);
+  }
+  return { agent, linkedAgents };
+}
+
 export async function checkUpdates(workspace, options = {}) {
   const state = readUserState();
   const latestInstaller = await npmLatest();
@@ -126,7 +145,7 @@ export async function updateCommand(options = {}) {
   const skipped = [];
   const trackingOptions = { ...options, skippedUpdates: skipped };
 
-  step(1, 6, "检查 installer");
+  step(1, 7, "检查 installer");
   const latestInstaller = await npmLatest();
   if (latestInstaller && latestInstaller !== packageVersion()) {
     warn(`installer 有新版本 ${packageVersion()} -> ${latestInstaller}`);
@@ -136,11 +155,11 @@ export async function updateCommand(options = {}) {
   }
   blank();
 
-  step(2, 6, "检查 runtime bundle");
+  step(2, 7, "检查 runtime bundle");
   const runtimeRelease = await latestRelease("lumi-ai-lab/harness-data", options);
   if (state.runtimeTag && state.runtimeTag !== runtimeRelease.tag_name) {
     action(`发现更新：runtime bundle ${state.runtimeTag} -> ${runtimeRelease.tag_name}`);
-    if (await confirm("是否更新 runtime bundle？", { defaultNo: true })) {
+    if (await confirm("是否更新 runtime bundle？")) {
       const bundle = await installRuntimeBundle(runtimeDir, { ...trackingOptions, force: true });
       runtimeTag = bundle.tag || runtimeRelease.tag_name;
       changed = true;
@@ -154,7 +173,7 @@ export async function updateCommand(options = {}) {
   }
   blank();
 
-  step(3, 6, "检查 CLI 工具");
+  step(3, 7, "检查 CLI 工具");
   for (const tool of manifest.tools || []) {
     if (state.installMode === "local-path" && tool.name !== "data-harness-cli") {
       skip(`${tool.name} 为本地路径模式，请手动检查`);
@@ -169,7 +188,7 @@ export async function updateCommand(options = {}) {
   }
   blank();
 
-  step(4, 6, "检查 Wikis 知识库");
+  step(4, 7, "检查 Wikis 知识库");
   const wikis = await updateWikis(runtimeDir, trackingOptions, state);
   if (wikis) {
     changed = true;
@@ -177,7 +196,11 @@ export async function updateCommand(options = {}) {
   }
   blank();
 
-  step(5, 6, "构建 Wikis 索引");
+  step(5, 7, "检查 Agent Hook");
+  const restoredAgent = await restoreAgentHooksIfMissing(runtimeDir, trackingOptions);
+  blank();
+
+  step(6, 7, "构建 Wikis 索引");
   if (changed) {
     await buildAndCheck(runtimeDir, trackingOptions);
   } else {
@@ -185,8 +208,10 @@ export async function updateCommand(options = {}) {
   }
   blank();
 
-  step(6, 6, "安装校验");
+  step(7, 7, "安装校验");
   if (changed) {
+    writeLocalConfig(runtimeDir, { overwrite: true });
+    ok("本地配置已刷新");
     const doctor = await collectDoctor(runtimeDir, trackingOptions);
     printDoctorSummary(doctor, { nonBlocking: isNonBlockingUpdateDoctorCheck });
     if (doctor.checks.some((check) => !check.ok && !isNonBlockingUpdateDoctorCheck(check))) throw new Error("doctor failed; update is incomplete");
@@ -195,6 +220,7 @@ export async function updateCommand(options = {}) {
       runtimeTag,
       tools: nextTools,
       manifestSha256: manifestDigest(manifest),
+      ...(restoredAgent ? { agent: restoredAgent.agent } : {}),
       lastCheckAt: new Date().toISOString()
     });
     blank();
@@ -209,6 +235,24 @@ export async function updateCommand(options = {}) {
       console.log("已跳过：");
       for (const item of skipped) console.log(`- ${item}`);
     }
+    if (restoredAgent) {
+      console.log("");
+      console.log("已恢复：");
+      console.log(`- Agent Hook ${restoredAgent.agent}`);
+    }
+  } else if (restoredAgent) {
+    ok("Agent Hook 已恢复");
+    writeState(runtimeDir, { ...state, agent: restoredAgent.agent, lastCheckAt: new Date().toISOString() });
+    blank();
+    console.log(`配置已恢复：${runtimeDir}`);
+    console.log("");
+    console.log("已恢复：");
+    console.log(`- Agent Hook ${restoredAgent.agent}`);
+    if (skipped.length) {
+      console.log("");
+      console.log("已跳过：");
+      for (const item of skipped) console.log(`- ${item}`);
+    }
   } else {
     skip("没有组件更新");
     writeState(runtimeDir, { ...state, lastCheckAt: new Date().toISOString() });

package/src/lib/config.js

@@ -21,6 +21,10 @@ export const agentLinks = {
 };
 export const agentChoiceText = agentChoices.join(", ");
 
+export function hasAnyAgentHook(workspace) {
+  return concreteAgentNames.some((name) => fs.existsSync(path.join(workspace, `.${name}`)));
+}
+
 export function writeLocalConfig(workspace, options = {}) {
   const configDir = path.join(workspace, "config");
   fs.mkdirSync(configDir, { recursive: true });