@hiveai/mcp 0.9.6 → 0.9.8

package/dist/index.js

@@ -1912,6 +1912,7 @@ var CodeMapInputSchema = {
     "Approximate token budget for the response. When the matching set exceeds it, files are ranked by export density (exports per LOC) and the highest-signal ones are kept first. Omit to disable budgeting (legacy behavior)."
   )
 };
+var CodeMapInputZod = z18.object(CodeMapInputSchema);
 async function codeMapTool(input, ctx) {
   const map = await loadCodeMap2(ctx.paths);
   if (!map) {
@@ -3450,8 +3451,9 @@ When done, respond with: "Imported N memories: [list of IDs]" or "Nothing action
 }
 
 // src/server.ts
+import { loadConfigSync } from "@hiveai/core";
 var SERVER_NAME = "haive";
-var SERVER_VERSION = "0.9.6";
+var SERVER_VERSION = "0.9.8";
 function jsonResult(data) {
   return {
     content: [
@@ -3462,15 +3464,70 @@ function jsonResult(data) {
     ]
   };
 }
+var ENFORCEMENT_PROFILE_TOOLS = /* @__PURE__ */ new Set([
+  "get_briefing",
+  "mem_save",
+  "mem_tried",
+  "mem_search",
+  "mem_get",
+  "mem_update",
+  "mem_verify",
+  "mem_relevant_to",
+  "code_map",
+  "pre_commit_check"
+]);
+var BRIEFING_TOOLS = /* @__PURE__ */ new Set(["get_briefing", "mem_relevant_to"]);
+var MUTATING_TOOLS = /* @__PURE__ */ new Set([
+  "mem_save",
+  "mem_tried",
+  "mem_observe",
+  "mem_session_end",
+  "bootstrap_project_save",
+  "mem_update",
+  "mem_approve",
+  "mem_reject",
+  "mem_delete",
+  "runtime_journal_append",
+  "pattern_detect"
+]);
 function createHaiveServer(options = {}) {
   const context = createContext(options);
+  const config = loadConfigSync(context.paths);
+  const toolProfile = options.env?.HAIVE_TOOL_PROFILE ?? config.enforcement?.toolProfile ?? "enforcement";
+  const requireBriefingFirst = options.env?.HAIVE_REQUIRE_BRIEFING_FIRST === "0" ? false : config.enforcement?.requireBriefingFirst ?? true;
+  let briefingLoaded = false;
   const tracker = new SessionTracker(context);
   void tracker.init();
   const server = new McpServer(
     { name: SERVER_NAME, version: SERVER_VERSION },
     { capabilities: { tools: {}, prompts: {} } }
   );
-  server.tool(
+  const shouldRegisterTool = (name) => toolProfile === "full" || ENFORCEMENT_PROFILE_TOOLS.has(name);
+  const registerTool = (name, description, schema, handler) => {
+    if (!shouldRegisterTool(name)) return;
+    const tool = server.tool.bind(server);
+    tool(
+      name,
+      description,
+      schema,
+      async (input) => {
+        if (BRIEFING_TOOLS.has(name)) {
+          briefingLoaded = true;
+          return await handler(input);
+        }
+        if (requireBriefingFirst && MUTATING_TOOLS.has(name) && !briefingLoaded) {
+          return jsonResult({
+            error: "haive_briefing_required",
+            message: "This hAIve project requires get_briefing or mem_relevant_to before state-changing hAIve tools. Call get_briefing({ task: '...' }) first.",
+            tool: name
+          });
+        }
+        return await handler(input);
+      }
+    );
+  };
+  const shouldRegisterPrompt = (name) => toolProfile === "full" || name === "bootstrap_project" || name === "post_task";
+  registerTool(
     "mem_save",
     [
       "Save a piece of knowledge as a persistent memory that survives across AI sessions.",
@@ -3502,7 +3559,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memSave(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_suggest_topic",
     [
       "Propose a stable `topic` key (topic-upsert) from type + short title.",
@@ -3519,7 +3576,7 @@ function createHaiveServer(options = {}) {
     MemSuggestTopicInputSchema,
     async (input) => jsonResult(await memSuggestTopic(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_tried",
     [
       "Record a FAILED approach so future agents don't repeat the same mistake.",
@@ -3546,7 +3603,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memTried(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_observe",
     [
       "Capture a code-level discovery made WHILE READING existing code.",
@@ -3578,7 +3635,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memObserve(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_session_end",
     [
       "Save an end-of-session recap so the NEXT session starts with fresh context.",
@@ -3607,7 +3664,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memSessionEnd(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "get_briefing",
     [
       "\u2B50 DEFAULT-FIRST for coding agents on any repo where `haive init` ran: call this BEFORE",
@@ -3658,7 +3715,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await getBriefing(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_search",
     [
       "Search memories by keyword or semantic similarity.",
@@ -3690,7 +3747,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memSearch(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_timeline",
     [
       "Chronological view of related memories: by shared frontmatter.topic OR expanded from a seed id",
@@ -3706,7 +3763,7 @@ function createHaiveServer(options = {}) {
     MemTimelineInputSchema,
     async (input) => jsonResult(await memTimeline(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_for_files",
     [
       "Surface memories relevant to the files you are currently editing.",
@@ -3730,7 +3787,7 @@ function createHaiveServer(options = {}) {
     MemForFilesInputSchema,
     async (input) => jsonResult(await memForFiles(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_get",
     [
       "Fetch a single memory by its full id with all details.",
@@ -3746,7 +3803,7 @@ function createHaiveServer(options = {}) {
     MemGetInputSchema,
     async (input) => jsonResult(await memGet(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_list",
     [
       "List memories with optional filters. Use for browsing, not for task onboarding.",
@@ -3766,7 +3823,7 @@ function createHaiveServer(options = {}) {
     MemListInputSchema,
     async (input) => jsonResult(await memList(input, context))
   );
-  server.tool(
+  registerTool(
     "get_project_context",
     [
       "Read .ai/project-context.md (and optionally a module context) directly.",
@@ -3784,7 +3841,7 @@ function createHaiveServer(options = {}) {
     GetProjectContextInputSchema,
     async (input) => jsonResult(await getProjectContext(input, context))
   );
-  server.tool(
+  registerTool(
     "bootstrap_project_save",
     [
       "Persist the project context document (.ai/project-context.md) or a module",
@@ -3802,7 +3859,7 @@ function createHaiveServer(options = {}) {
     BootstrapProjectSaveInputSchema,
     async (input) => jsonResult(await bootstrapProjectSave(input, context))
   );
-  server.tool(
+  registerTool(
     "code_map",
     [
       "Look up where symbols (classes, functions, interfaces) are defined in the codebase.",
@@ -3823,7 +3880,7 @@ function createHaiveServer(options = {}) {
     CodeMapInputSchema,
     async (input) => jsonResult(await codeMapTool(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_resolve_project",
     [
       "Diagnostics: resolve which project root hAIve is using (never throws).",
@@ -3839,7 +3896,7 @@ function createHaiveServer(options = {}) {
     MemResolveProjectInputSchema,
     async (input) => jsonResult(await memResolveProject(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_update",
     [
       "Update the body, tags, or anchor of an existing memory in-place.",
@@ -3862,7 +3919,7 @@ function createHaiveServer(options = {}) {
     MemUpdateInputSchema,
     async (input) => jsonResult(await memUpdate(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_verify",
     [
       "Check whether memory anchor paths and symbols still exist in the current code.",
@@ -3881,7 +3938,7 @@ function createHaiveServer(options = {}) {
     MemVerifyInputSchema,
     async (input) => jsonResult(await memVerify(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_approve",
     [
       "Mark a memory as validated (trusted, approved by a human or the team).",
@@ -3897,7 +3954,7 @@ function createHaiveServer(options = {}) {
     MemApproveInputSchema,
     async (input) => jsonResult(await memApprove(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_reject",
     [
       "Mark a memory as rejected and record a reason.",
@@ -3915,7 +3972,7 @@ function createHaiveServer(options = {}) {
     MemRejectInputSchema,
     async (input) => jsonResult(await memReject(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_pending",
     [
       "List memories in 'proposed' status awaiting review, sorted by read count.",
@@ -3931,7 +3988,7 @@ function createHaiveServer(options = {}) {
     MemPendingInputSchema,
     async (input) => jsonResult(await memPending(input, context))
   );
-  server.tool(
+  registerTool(
     "mem_delete",
     [
       "Permanently delete a memory by id.",
@@ -3948,7 +4005,7 @@ function createHaiveServer(options = {}) {
     MemDeleteInputSchema,
     async (input) => jsonResult(await memDelete(input, context))
   );
-  server.tool(
+  registerTool(
     "get_recap",
     [
       "Return ONLY the most recent session_recap. Cheaper than get_briefing when",
@@ -3966,7 +4023,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await getRecap(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_relevant_to",
     [
       "One-shot ranked memories for a task \u2014 use instead of get_briefing when",
@@ -3992,7 +4049,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memRelevantTo(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "code_search",
     [
       "Semantic search over the codebase \u2014 finds exported symbols (functions, classes,",
@@ -4015,7 +4072,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await codeSearch(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "why_this_file",
     [
       "One-shot file-context lookup: combines recent git history, memories anchored",
@@ -4035,7 +4092,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await whyThisFile(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "anti_patterns_check",
     [
       "Scan a diff (or set of paths) against documented attempt/gotcha memories.",
@@ -4058,7 +4115,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await antiPatternsCheck(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_distill",
     [
       "Cluster recurring observations / failed attempts so a human can collapse",
@@ -4083,7 +4140,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memDistill(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "why_this_decision",
     [
       "Trace the genealogy of a memory (especially decision/architecture):",
@@ -4105,7 +4162,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await whyThisDecision(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_conflicts_with",
     [
       "Detect memories that potentially CONTRADICT a given memory.",
@@ -4131,7 +4188,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memConflicts(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_conflict_candidates",
     [
       "Bulk scan for conflict CANDIDATES (not proof):",
@@ -4152,7 +4209,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await memConflictCandidates(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "runtime_journal_append",
     [
       "Append one line to `.ai/.runtime/session-journal.ndjson` \u2014 machine-local session continuity.",
@@ -4166,7 +4223,7 @@ function createHaiveServer(options = {}) {
     RuntimeJournalAppendInputSchema,
     async (input) => jsonResult(await runtimeJournalAppend(input, context))
   );
-  server.tool(
+  registerTool(
     "runtime_journal_tail",
     [
       "Read the last N entries from the runtime session journal (parsed JSON lines).",
@@ -4178,7 +4235,7 @@ function createHaiveServer(options = {}) {
     RuntimeJournalTailInputSchema,
     async (input) => jsonResult(await runtimeJournalTail(input, context))
   );
-  server.tool(
+  registerTool(
     "pre_commit_check",
     [
       "One-shot 'should I block this commit?' check. Combines three signals:",
@@ -4203,7 +4260,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await preCommitCheck(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "pattern_detect",
     [
       "Heuristic memory detector \u2014 finds knowledge worth saving WITHOUT calling an LLM.",
@@ -4234,7 +4291,7 @@ function createHaiveServer(options = {}) {
       return jsonResult(await patternDetect(input, context));
     }
   );
-  server.tool(
+  registerTool(
     "mem_diff",
     [
       "Compare two memories side-by-side to decide if they should be merged.",
@@ -4251,40 +4308,46 @@ function createHaiveServer(options = {}) {
     MemDiffInputSchema,
     async (input) => jsonResult(await memDiff(input, context))
   );
-  server.prompt(
-    "bootstrap_project",
-    [
-      "Analyze the project codebase and write .ai/project-context.md \u2014 run once after haive init.",
-      "The AI explores the directory structure, reads key files (package.json, README, config),",
-      "identifies the tech stack, architectural patterns, key modules, and conventions,",
-      "then persists everything via bootstrap_project_save.",
-      "For multi-component projects, run with module param to create .ai/modules/<name>/context.md."
-    ].join(" "),
-    BootstrapProjectArgsSchema,
-    (args) => bootstrapProjectPrompt(args, context)
-  );
-  server.prompt(
-    "post_task",
-    [
-      "\u2B50 Post-task reflection \u2014 run at the end of every session to capture what you learned:",
-      "failed approaches (mem_tried), new conventions/decisions/gotchas (mem_save),",
-      "code discoveries (mem_observe), and an end-of-session recap (mem_session_end).",
-      "In autopilot mode a minimal recap saves automatically; calling this produces a richer one."
-    ].join(" "),
-    PostTaskArgsSchema,
-    (args) => postTaskPrompt(args, context)
-  );
-  server.prompt(
-    "import_docs",
-    [
-      "Import knowledge from a document (README, ADR, wiki, API spec) as hAIve memories.",
-      "Pass the full document content; the AI extracts up to 10 actionable memories",
-      "(conventions, decisions, gotchas, architecture) and saves them via mem_save.",
-      "Good candidates: ADRs, onboarding docs, runbooks, team wikis."
-    ].join(" "),
-    ImportDocsArgsSchema,
-    (args) => importDocsPrompt(args, context)
-  );
+  if (shouldRegisterPrompt("bootstrap_project")) {
+    server.prompt(
+      "bootstrap_project",
+      [
+        "Analyze the project codebase and write .ai/project-context.md \u2014 run once after haive init.",
+        "The AI explores the directory structure, reads key files (package.json, README, config),",
+        "identifies the tech stack, architectural patterns, key modules, and conventions,",
+        "then persists everything via bootstrap_project_save.",
+        "For multi-component projects, run with module param to create .ai/modules/<name>/context.md."
+      ].join(" "),
+      BootstrapProjectArgsSchema,
+      (args) => bootstrapProjectPrompt(args, context)
+    );
+  }
+  if (shouldRegisterPrompt("post_task")) {
+    server.prompt(
+      "post_task",
+      [
+        "\u2B50 Post-task reflection \u2014 run at the end of every session to capture what you learned:",
+        "failed approaches (mem_tried), new conventions/decisions/gotchas (mem_save),",
+        "code discoveries (mem_observe), and an end-of-session recap (mem_session_end).",
+        "In autopilot mode a minimal recap saves automatically; calling this produces a richer one."
+      ].join(" "),
+      PostTaskArgsSchema,
+      (args) => postTaskPrompt(args, context)
+    );
+  }
+  if (shouldRegisterPrompt("import_docs")) {
+    server.prompt(
+      "import_docs",
+      [
+        "Import knowledge from a document (README, ADR, wiki, API spec) as hAIve memories.",
+        "Pass the full document content; the AI extracts up to 10 actionable memories",
+        "(conventions, decisions, gotchas, architecture) and saves them via mem_save.",
+        "Good candidates: ADRs, onboarding docs, runbooks, team wikis."
+      ].join(" "),
+      ImportDocsArgsSchema,
+      (args) => importDocsPrompt(args, context)
+    );
+  }
   return { server, context, tracker };
 }
 function parseMcpCliArgs(argv) {
@@ -4309,7 +4372,7 @@ function printHaiveMcpVersion() {
   console.log(SERVER_VERSION);
 }
 async function runHaiveMcpStdio(options) {
-  const { server, context } = createHaiveServer({ root: options.root });
+  const { server, context } = createHaiveServer({ root: options.root, env: process.env });
   console.error(
     `[haive-mcp] starting server v${SERVER_VERSION} (project root: ${context.paths.root})`
   );