@contextstream/mcp-server 0.4.60 → 0.4.62

package/dist/hooks/post-write.js

@@ -74,7 +74,7 @@ var INDEXABLE_EXTENSIONS = /* @__PURE__ */ new Set([
   ".prisma",
   ".proto"
 ]);
-var MAX_FILE_SIZE = 1024 * 1024;
+var MAX_FILE_SIZE = 5 * 1024 * 1024;
 function extractFilePath(input) {
   if (input.tool_input) {
     const filePath = input.tool_input.file_path || input.tool_input.notebook_path || input.tool_input.path;
@@ -274,6 +274,13 @@ async function indexFile(filePath, projectId, apiUrl, apiKey, projectRoot) {
   if (!response.ok) {
     throw new Error(`API error: ${response.status} ${response.statusText}`);
   }
+  try {
+    const body = await response.json();
+    if (body?.data?.status === "cooldown" || body?.data?.status === "daily_limit_exceeded") {
+      return;
+    }
+  } catch {
+  }
 }
 function findProjectRoot(filePath) {
   let currentDir = path.dirname(path.resolve(filePath));

package/dist/hooks/pre-tool-use.js

@@ -1,13 +1,159 @@
 #!/usr/bin/env node
 
 // src/hooks/pre-tool-use.ts
+import * as fs2 from "node:fs";
+import * as path2 from "node:path";
+import { homedir as homedir2 } from "node:os";
+
+// src/hooks/prompt-state.ts
 import * as fs from "node:fs";
 import * as path from "node:path";
 import { homedir } from "node:os";
+var STATE_PATH = path.join(homedir(), ".contextstream", "prompt-state.json");
+function defaultState() {
+  return { workspaces: {} };
+}
+function nowIso() {
+  return (/* @__PURE__ */ new Date()).toISOString();
+}
+function ensureStateDir() {
+  try {
+    fs.mkdirSync(path.dirname(STATE_PATH), { recursive: true });
+  } catch {
+  }
+}
+function normalizePath(input) {
+  try {
+    return path.resolve(input);
+  } catch {
+    return input;
+  }
+}
+function workspacePathsMatch(a, b) {
+  const left = normalizePath(a);
+  const right = normalizePath(b);
+  return left === right || left.startsWith(`${right}${path.sep}`) || right.startsWith(`${left}${path.sep}`);
+}
+function readState() {
+  try {
+    const content = fs.readFileSync(STATE_PATH, "utf8");
+    const parsed = JSON.parse(content);
+    if (!parsed || typeof parsed !== "object" || !parsed.workspaces) {
+      return defaultState();
+    }
+    return parsed;
+  } catch {
+    return defaultState();
+  }
+}
+function writeState(state) {
+  try {
+    ensureStateDir();
+    fs.writeFileSync(STATE_PATH, JSON.stringify(state, null, 2), "utf8");
+  } catch {
+  }
+}
+function getOrCreateEntry(state, cwd) {
+  if (!cwd.trim()) return null;
+  const exact = state.workspaces[cwd];
+  if (exact) return { key: cwd, entry: exact };
+  for (const [trackedCwd, trackedEntry] of Object.entries(state.workspaces)) {
+    if (workspacePathsMatch(trackedCwd, cwd)) {
+      return { key: trackedCwd, entry: trackedEntry };
+    }
+  }
+  const created = {
+    require_context: false,
+    require_init: false,
+    last_context_at: void 0,
+    last_state_change_at: void 0,
+    updated_at: nowIso()
+  };
+  state.workspaces[cwd] = created;
+  return { key: cwd, entry: created };
+}
+function cleanupStale(maxAgeSeconds) {
+  const state = readState();
+  const now = Date.now();
+  let changed = false;
+  for (const [cwd, entry] of Object.entries(state.workspaces)) {
+    const updated = new Date(entry.updated_at);
+    if (Number.isNaN(updated.getTime())) continue;
+    const ageSeconds = (now - updated.getTime()) / 1e3;
+    if (ageSeconds > maxAgeSeconds) {
+      delete state.workspaces[cwd];
+      changed = true;
+    }
+  }
+  if (changed) {
+    writeState(state);
+  }
+}
+function clearContextRequired(cwd) {
+  if (!cwd.trim()) return;
+  const state = readState();
+  const target = getOrCreateEntry(state, cwd);
+  if (!target) return;
+  target.entry.require_context = false;
+  target.entry.last_context_at = nowIso();
+  target.entry.updated_at = nowIso();
+  writeState(state);
+}
+function isContextRequired(cwd) {
+  if (!cwd.trim()) return false;
+  const state = readState();
+  const target = getOrCreateEntry(state, cwd);
+  return Boolean(target?.entry.require_context);
+}
+function clearInitRequired(cwd) {
+  if (!cwd.trim()) return;
+  const state = readState();
+  const target = getOrCreateEntry(state, cwd);
+  if (!target) return;
+  target.entry.require_init = false;
+  target.entry.updated_at = nowIso();
+  writeState(state);
+}
+function isInitRequired(cwd) {
+  if (!cwd.trim()) return false;
+  const state = readState();
+  const target = getOrCreateEntry(state, cwd);
+  return Boolean(target?.entry.require_init);
+}
+function markStateChanged(cwd) {
+  if (!cwd.trim()) return;
+  const state = readState();
+  const target = getOrCreateEntry(state, cwd);
+  if (!target) return;
+  target.entry.last_state_change_at = nowIso();
+  target.entry.updated_at = nowIso();
+  writeState(state);
+}
+function isContextFreshAndClean(cwd, maxAgeSeconds) {
+  if (!cwd.trim()) return false;
+  const state = readState();
+  const target = getOrCreateEntry(state, cwd);
+  const entry = target?.entry;
+  if (!entry?.last_context_at) return false;
+  const contextAt = new Date(entry.last_context_at);
+  if (Number.isNaN(contextAt.getTime())) return false;
+  const ageSeconds = (Date.now() - contextAt.getTime()) / 1e3;
+  if (ageSeconds < 0 || ageSeconds > maxAgeSeconds) return false;
+  if (entry.last_state_change_at) {
+    const changedAt = new Date(entry.last_state_change_at);
+    if (!Number.isNaN(changedAt.getTime()) && changedAt.getTime() > contextAt.getTime()) {
+      return false;
+    }
+  }
+  return true;
+}
+
+// src/hooks/pre-tool-use.ts
 var ENABLED = process.env.CONTEXTSTREAM_HOOK_ENABLED !== "false";
-var INDEX_STATUS_FILE = path.join(homedir(), ".contextstream", "indexed-projects.json");
+var INDEX_STATUS_FILE = path2.join(homedir2(), ".contextstream", "indexed-projects.json");
 var DEBUG_FILE = "/tmp/pretooluse-hook-debug.log";
 var STALE_THRESHOLD_DAYS = 7;
+var CONTEXT_FRESHNESS_SECONDS = 120;
 var DISCOVERY_PATTERNS = ["**/*", "**/", "src/**", "lib/**", "app/**", "components/**"];
 function isDiscoveryGlob(pattern) {
   const patternLower = pattern.toLowerCase();
@@ -34,22 +180,22 @@ function isDiscoveryGrep(filePath) {
   return false;
 }
 function isProjectIndexed(cwd) {
-  if (!fs.existsSync(INDEX_STATUS_FILE)) {
+  if (!fs2.existsSync(INDEX_STATUS_FILE)) {
     return { isIndexed: false, isStale: false };
   }
   let data;
   try {
-    const content = fs.readFileSync(INDEX_STATUS_FILE, "utf-8");
+    const content = fs2.readFileSync(INDEX_STATUS_FILE, "utf-8");
     data = JSON.parse(content);
   } catch {
     return { isIndexed: false, isStale: false };
   }
   const projects = data.projects || {};
-  const cwdPath = path.resolve(cwd);
+  const cwdPath = path2.resolve(cwd);
   for (const [projectPath, info] of Object.entries(projects)) {
     try {
-      const indexedPath = path.resolve(projectPath);
-      if (cwdPath === indexedPath || cwdPath.startsWith(indexedPath + path.sep)) {
+      const indexedPath = path2.resolve(projectPath);
+      if (cwdPath === indexedPath || cwdPath.startsWith(indexedPath + path2.sep)) {
         const indexedAt = info.indexed_at;
         if (indexedAt) {
           try {
@@ -82,6 +228,97 @@ function extractToolName(input) {
 function extractToolInput(input) {
   return input.tool_input || input.parameters || input.toolParameters || {};
 }
+function normalizeContextstreamToolName(toolName) {
+  const trimmed = toolName.trim();
+  if (!trimmed) return null;
+  const lower = trimmed.toLowerCase();
+  const prefixed = "mcp__contextstream__";
+  if (lower.startsWith(prefixed)) {
+    return lower.slice(prefixed.length);
+  }
+  if (lower.startsWith("contextstream__")) {
+    return lower.slice("contextstream__".length);
+  }
+  if (lower === "init" || lower === "context") {
+    return lower;
+  }
+  return null;
+}
+function actionFromToolInput(toolInput) {
+  const maybeAction = toolInput?.action;
+  return typeof maybeAction === "string" ? maybeAction.trim().toLowerCase() : "";
+}
+function isContextstreamReadOnlyOperation(toolName, toolInput) {
+  const action = actionFromToolInput(toolInput);
+  switch (toolName) {
+    case "workspace":
+      return action === "list" || action === "get";
+    case "memory":
+      return action === "list_docs" || action === "list_events" || action === "list_todos" || action === "list_tasks" || action === "list_transcripts" || action === "list_nodes" || action === "decisions" || action === "get_doc" || action === "get_event" || action === "get_task" || action === "get_todo" || action === "get_transcript";
+    case "session":
+      return action === "get_lessons" || action === "get_plan" || action === "list_plans" || action === "recall";
+    case "help":
+      return action === "version" || action === "tools" || action === "auth";
+    case "project":
+      return action === "list" || action === "get" || action === "index_status";
+    case "reminder":
+      return action === "list" || action === "active";
+    case "context":
+    case "init":
+      return true;
+    default:
+      return false;
+  }
+}
+function isLikelyStateChangingTool(toolLower, toolInput, isContextstreamCall, normalizedContextstreamTool) {
+  if (isContextstreamCall && normalizedContextstreamTool) {
+    return !isContextstreamReadOnlyOperation(normalizedContextstreamTool, toolInput);
+  }
+  if ([
+    "read",
+    "read_file",
+    "grep",
+    "glob",
+    "search",
+    "grep_search",
+    "code_search",
+    "semanticsearch",
+    "codebase_search",
+    "list_files",
+    "search_files",
+    "search_files_content",
+    "find_files",
+    "find_by_name",
+    "ls",
+    "cat",
+    "view"
+  ].includes(toolLower)) {
+    return false;
+  }
+  const writeMarkers = [
+    "write",
+    "edit",
+    "create",
+    "delete",
+    "remove",
+    "rename",
+    "move",
+    "patch",
+    "apply",
+    "insert",
+    "append",
+    "replace",
+    "update",
+    "commit",
+    "push",
+    "install",
+    "exec",
+    "run",
+    "bash",
+    "shell"
+  ];
+  return writeMarkers.some((marker) => toolLower.includes(marker));
+}
 function blockClaudeCode(message) {
   const response = {
     hookSpecificOutput: {
@@ -90,7 +327,7 @@ function blockClaudeCode(message) {
       additionalContext: `[CONTEXTSTREAM] ${message}`
     }
   };
-  fs.appendFileSync(DEBUG_FILE, `[PreToolUse] REDIRECT (additionalContext): ${JSON.stringify(response)}
+  fs2.appendFileSync(DEBUG_FILE, `[PreToolUse] REDIRECT (additionalContext): ${JSON.stringify(response)}
 `);
   console.log(JSON.stringify(response));
   process.exit(0);
@@ -118,6 +355,25 @@ function outputCursorAllow() {
   console.log(JSON.stringify({ decision: "allow" }));
   process.exit(0);
 }
+function blockWithMessage(editorFormat, message) {
+  if (editorFormat === "cline") {
+    outputClineBlock(message, "[CONTEXTSTREAM] Follow ContextStream startup requirements.");
+  } else if (editorFormat === "cursor") {
+    outputCursorBlock(message);
+  }
+  blockClaudeCode(message);
+}
+function allowTool(editorFormat, cwd, recordStateChange) {
+  if (recordStateChange) {
+    markStateChanged(cwd);
+  }
+  if (editorFormat === "cline") {
+    outputClineAllow();
+  } else if (editorFormat === "cursor") {
+    outputCursorAllow();
+  }
+  process.exit(0);
+}
 function detectEditorFormat(input) {
   if (input.hookName !== void 0 || input.toolName !== void 0) {
     return "cline";
@@ -128,11 +384,11 @@ function detectEditorFormat(input) {
   return "claude";
 }
 async function runPreToolUseHook() {
-  fs.appendFileSync(DEBUG_FILE, `[PreToolUse] Hook invoked at ${(/* @__PURE__ */ new Date()).toISOString()}
+  fs2.appendFileSync(DEBUG_FILE, `[PreToolUse] Hook invoked at ${(/* @__PURE__ */ new Date()).toISOString()}
 `);
   console.error("[PreToolUse] Hook invoked at", (/* @__PURE__ */ new Date()).toISOString());
   if (!ENABLED) {
-    fs.appendFileSync(DEBUG_FILE, "[PreToolUse] Hook disabled, exiting\n");
+    fs2.appendFileSync(DEBUG_FILE, "[PreToolUse] Hook disabled, exiting\n");
     console.error("[PreToolUse] Hook disabled, exiting");
     process.exit(0);
   }
@@ -153,28 +409,52 @@ async function runPreToolUseHook() {
   const cwd = extractCwd(input);
   const tool = extractToolName(input);
   const toolInput = extractToolInput(input);
-  fs.appendFileSync(DEBUG_FILE, `[PreToolUse] tool=${tool}, cwd=${cwd}, editorFormat=${editorFormat}
+  const toolLower = tool.toLowerCase();
+  const normalizedContextstreamTool = normalizeContextstreamToolName(tool);
+  const isContextstreamCall = normalizedContextstreamTool !== null;
+  const recordStateChange = isLikelyStateChangingTool(
+    toolLower,
+    toolInput,
+    isContextstreamCall,
+    normalizedContextstreamTool
+  );
+  fs2.appendFileSync(DEBUG_FILE, `[PreToolUse] tool=${tool}, cwd=${cwd}, editorFormat=${editorFormat}
 `);
+  cleanupStale(180);
+  if (isInitRequired(cwd)) {
+    if (isContextstreamCall && normalizedContextstreamTool === "init") {
+      clearInitRequired(cwd);
+    } else {
+      const required = "mcp__contextstream__init(...)";
+      const msg = `First call required for this session: ${required}. Run it before any other MCP tool. Then call mcp__contextstream__context(user_message="...", save_exchange=true, session_id="<session-id>").`;
+      blockWithMessage(editorFormat, msg);
+    }
+  }
+  if (isContextRequired(cwd)) {
+    if (isContextstreamCall && normalizedContextstreamTool === "context") {
+      clearContextRequired(cwd);
+    } else if (isContextstreamCall && normalizedContextstreamTool === "init") {
+    } else if (isContextstreamCall && normalizedContextstreamTool && isContextstreamReadOnlyOperation(normalizedContextstreamTool, toolInput) && isContextFreshAndClean(cwd, CONTEXT_FRESHNESS_SECONDS)) {
+    } else {
+      const msg = 'First call required for this prompt: mcp__contextstream__context(user_message="...", save_exchange=true, session_id="<session-id>"). Run it before any other MCP tool.';
+      blockWithMessage(editorFormat, msg);
+    }
+  }
   const { isIndexed } = isProjectIndexed(cwd);
-  fs.appendFileSync(DEBUG_FILE, `[PreToolUse] isIndexed=${isIndexed}
+  fs2.appendFileSync(DEBUG_FILE, `[PreToolUse] isIndexed=${isIndexed}
 `);
   if (!isIndexed) {
-    fs.appendFileSync(DEBUG_FILE, `[PreToolUse] Project not indexed, allowing
+    fs2.appendFileSync(DEBUG_FILE, `[PreToolUse] Project not indexed, allowing
 `);
-    if (editorFormat === "cline") {
-      outputClineAllow();
-    } else if (editorFormat === "cursor") {
-      outputCursorAllow();
-    }
-    process.exit(0);
+    allowTool(editorFormat, cwd, recordStateChange);
   }
   if (tool === "Glob") {
     const pattern = toolInput?.pattern || "";
-    fs.appendFileSync(DEBUG_FILE, `[PreToolUse] Glob pattern=${pattern}, isDiscovery=${isDiscoveryGlob(pattern)}
+    fs2.appendFileSync(DEBUG_FILE, `[PreToolUse] Glob pattern=${pattern}, isDiscovery=${isDiscoveryGlob(pattern)}
 `);
     if (isDiscoveryGlob(pattern)) {
-      const msg = `STOP: Use mcp__contextstream__search(mode="auto", query="${pattern}") instead of Glob.`;
-      fs.appendFileSync(DEBUG_FILE, `[PreToolUse] Intercepting discovery glob: ${msg}
+      const msg = `This project index is current. Use mcp__contextstream__search(mode="auto", query="${pattern}") instead of Glob for faster, richer code results.`;
+      fs2.appendFileSync(DEBUG_FILE, `[PreToolUse] Intercepting discovery glob: ${msg}
 `);
       if (editorFormat === "cline") {
         outputClineBlock(msg, "[CONTEXTSTREAM] Use ContextStream search for code discovery.");
@@ -196,7 +476,7 @@ async function runPreToolUseHook() {
         }
         blockClaudeCode(msg);
       } else {
-        const msg = `STOP: Use mcp__contextstream__search(mode="auto", query="${pattern}") instead of ${tool}.`;
+        const msg = `This project index is current. Use mcp__contextstream__search(mode="auto", query="${pattern}") instead of ${tool} for faster, richer code results.`;
         if (editorFormat === "cline") {
           outputClineBlock(msg, "[CONTEXTSTREAM] Use ContextStream search for code discovery.");
         } else if (editorFormat === "cursor") {
@@ -208,7 +488,7 @@ async function runPreToolUseHook() {
   } else if (tool === "Task") {
     const subagentType = toolInput?.subagent_type?.toLowerCase() || "";
     if (subagentType === "explore") {
-      const msg = 'STOP: Use mcp__contextstream__search(mode="auto") instead of Task(Explore).';
+      const msg = 'Project index is current. Use mcp__contextstream__search(mode="auto") instead of Task(Explore) for broad discovery.';
       if (editorFormat === "cline") {
         outputClineBlock(msg, "[CONTEXTSTREAM] Use ContextStream search for code discovery.");
       } else if (editorFormat === "cursor") {
@@ -217,7 +497,7 @@ async function runPreToolUseHook() {
       blockClaudeCode(msg);
     }
     if (subagentType === "plan") {
-      const msg = 'STOP: Use mcp__contextstream__session(action="capture_plan") for planning. ContextStream plans persist across sessions.';
+      const msg = 'After your plan is ready, save it with mcp__contextstream__session(action="capture_plan"). Then create tasks with mcp__contextstream__memory(action="create_task", title="...", plan_id="...").';
       if (editorFormat === "cline") {
         outputClineBlock(msg, "[CONTEXTSTREAM] Use ContextStream plans for persistence.");
       } else if (editorFormat === "cursor") {
@@ -226,7 +506,7 @@ async function runPreToolUseHook() {
       blockClaudeCode(msg);
     }
   } else if (tool === "EnterPlanMode") {
-    const msg = 'STOP: Use mcp__contextstream__session(action="capture_plan", title="...", steps=[...]) instead of EnterPlanMode. ContextStream plans persist across sessions and are searchable.';
+    const msg = 'After finalizing your plan, save it to ContextStream (not a local markdown file): mcp__contextstream__session(action="capture_plan", title="...", steps=[...]). Then create tasks with mcp__contextstream__memory(action="create_task", title="...", plan_id="...").';
     if (editorFormat === "cline") {
       outputClineBlock(msg, "[CONTEXTSTREAM] Use ContextStream plans for persistence.");
     } else if (editorFormat === "cursor") {
@@ -237,20 +517,16 @@ async function runPreToolUseHook() {
   if (tool === "list_files" || tool === "search_files") {
     const pattern = toolInput?.path || toolInput?.regex || "";
     if (isDiscoveryGlob(pattern) || isDiscoveryGrep(pattern)) {
-      const msg = `Use mcp__contextstream__search(mode="auto", query="${pattern}") instead of ${tool}. ContextStream search is indexed and faster.`;
+      const msg = `Project index is current. Use mcp__contextstream__search(mode="auto", query="${pattern}") instead of ${tool} for faster, richer code results.`;
       if (editorFormat === "cline") {
         outputClineBlock(msg, "[CONTEXTSTREAM] Use ContextStream search for code discovery.");
       } else if (editorFormat === "cursor") {
         outputCursorBlock(msg);
       }
+      blockClaudeCode(msg);
     }
   }
-  if (editorFormat === "cline") {
-    outputClineAllow();
-  } else if (editorFormat === "cursor") {
-    outputCursorAllow();
-  }
-  process.exit(0);
+  allowTool(editorFormat, cwd, recordStateChange);
 }
 var isDirectRun = process.argv[1]?.includes("pre-tool-use") || process.argv[2] === "pre-tool-use";
 if (isDirectRun) {