@contextstream/mcp-server 0.4.82 → 1.0.1

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

@@ -1,553 +0,0 @@
-#!/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,
-    index_wait_started_at: void 0,
-    index_wait_until: 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.index_wait_started_at = void 0;
-  target.entry.index_wait_until = void 0;
-  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 = 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();
-  for (const p of DISCOVERY_PATTERNS) {
-    if (patternLower.includes(p)) {
-      return true;
-    }
-  }
-  if (patternLower.startsWith("**/*.") || patternLower.startsWith("**/")) {
-    return true;
-  }
-  if (patternLower.includes("**") || patternLower.includes("*/")) {
-    return true;
-  }
-  return false;
-}
-function isDiscoveryGrep(filePath) {
-  if (!filePath || filePath === "." || filePath === "./" || filePath === "*" || filePath === "**") {
-    return true;
-  }
-  if (filePath.includes("*") || filePath.includes("**")) {
-    return true;
-  }
-  return false;
-}
-function isProjectIndexed(cwd) {
-  if (!fs2.existsSync(INDEX_STATUS_FILE)) {
-    return { isIndexed: false, isStale: false };
-  }
-  let data;
-  try {
-    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 = path2.resolve(cwd);
-  for (const [projectPath, info] of Object.entries(projects)) {
-    try {
-      const indexedPath = path2.resolve(projectPath);
-      if (cwdPath === indexedPath || cwdPath.startsWith(indexedPath + path2.sep)) {
-        const indexedAt = info.indexed_at;
-        if (indexedAt) {
-          try {
-            const indexedTime = new Date(indexedAt);
-            const now = /* @__PURE__ */ new Date();
-            const diffDays = (now.getTime() - indexedTime.getTime()) / (1e3 * 60 * 60 * 24);
-            if (diffDays > STALE_THRESHOLD_DAYS) {
-              return { isIndexed: true, isStale: true };
-            }
-          } catch {
-          }
-        }
-        return { isIndexed: true, isStale: false };
-      }
-    } catch {
-      continue;
-    }
-  }
-  return { isIndexed: false, isStale: false };
-}
-function extractCwd(input) {
-  if (input.cwd) return input.cwd;
-  if (input.workspace_roots?.length) return input.workspace_roots[0];
-  if (input.workspaceRoots?.length) return input.workspaceRoots[0];
-  return process.cwd();
-}
-function extractToolName(input) {
-  return input.tool_name || input.toolName || "";
-}
-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: {
-      hookEventName: "PreToolUse",
-      // Use additionalContext instead of deny - tool runs but Claude sees the message
-      additionalContext: `[CONTEXTSTREAM] ${message}`
-    }
-  };
-  fs2.appendFileSync(DEBUG_FILE, `[PreToolUse] REDIRECT (additionalContext): ${JSON.stringify(response)}
-`);
-  console.log(JSON.stringify(response));
-  process.exit(0);
-}
-function outputClineBlock(errorMessage, contextMod) {
-  const result = {
-    cancel: true,
-    errorMessage
-  };
-  if (contextMod) {
-    result.contextModification = contextMod;
-  }
-  console.log(JSON.stringify(result));
-  process.exit(0);
-}
-function outputClineAllow() {
-  console.log(JSON.stringify({ cancel: false }));
-  process.exit(0);
-}
-function outputCursorBlock(reason) {
-  console.log(
-    JSON.stringify({ permission: "deny", agent_message: reason, user_message: reason })
-  );
-  process.exit(0);
-}
-function outputCursorAllow() {
-  console.log(JSON.stringify({ permission: "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 (process.argv.includes("--editor=cursor")) {
-    return "cursor";
-  }
-  if (input.hookName !== void 0 || input.toolName !== void 0) {
-    return "cline";
-  }
-  if (input.hook_event_name !== void 0 || input.tool_name !== void 0) {
-    return "claude";
-  }
-  return "claude";
-}
-async function runPreToolUseHook() {
-  fs2.appendFileSync(DEBUG_FILE, `[PreToolUse] Hook invoked at ${(/* @__PURE__ */ new Date()).toISOString()}
-`);
-  console.error("[PreToolUse] Hook invoked at", (/* @__PURE__ */ new Date()).toISOString());
-  if (!ENABLED) {
-    fs2.appendFileSync(DEBUG_FILE, "[PreToolUse] Hook disabled, exiting\n");
-    console.error("[PreToolUse] Hook disabled, exiting");
-    process.exit(0);
-  }
-  let inputData = "";
-  for await (const chunk of process.stdin) {
-    inputData += chunk;
-  }
-  if (!inputData.trim()) {
-    process.exit(0);
-  }
-  let input;
-  try {
-    input = JSON.parse(inputData);
-  } catch {
-    process.exit(0);
-  }
-  const editorFormat = detectEditorFormat(input);
-  const cwd = extractCwd(input);
-  const tool = extractToolName(input);
-  const toolInput = extractToolInput(input);
-  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, isStale } = isProjectIndexed(cwd);
-  fs2.appendFileSync(DEBUG_FILE, `[PreToolUse] isIndexed=${isIndexed}, isStale=${isStale}
-`);
-  if (!isIndexed || isStale) {
-    allowTool(editorFormat, cwd, recordStateChange);
-  }
-  if (tool === "Glob") {
-    const pattern = toolInput?.pattern || "";
-    fs2.appendFileSync(DEBUG_FILE, `[PreToolUse] Glob pattern=${pattern}, isDiscovery=${isDiscoveryGlob(pattern)}
-`);
-    if (isDiscoveryGlob(pattern)) {
-      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.");
-      } else if (editorFormat === "cursor") {
-        outputCursorBlock(msg);
-      }
-      blockClaudeCode(msg);
-    }
-  } else if (tool === "Grep" || tool === "Search") {
-    const pattern = toolInput?.pattern || "";
-    const filePath = toolInput?.path || "";
-    if (pattern) {
-      if (filePath && !isDiscoveryGrep(filePath)) {
-        const msg = `STOP: Use Read("${filePath}") to view file content, or mcp__contextstream__search(mode="keyword", query="${pattern}") for codebase search.`;
-        if (editorFormat === "cline") {
-          outputClineBlock(msg, "[CONTEXTSTREAM] Use ContextStream search for code discovery.");
-        } else if (editorFormat === "cursor") {
-          outputCursorBlock(msg);
-        }
-        blockClaudeCode(msg);
-      } else {
-        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") {
-          outputCursorBlock(msg);
-        }
-        blockClaudeCode(msg);
-      }
-    }
-  } else if (tool === "Explore") {
-    const msg = 'Project index is current. Use mcp__contextstream__search(mode="auto", output_format="paths") instead of Explore for broad discovery.';
-    if (editorFormat === "cline") {
-      outputClineBlock(msg, "[CONTEXTSTREAM] Use ContextStream search for code discovery.");
-    } else if (editorFormat === "cursor") {
-      outputCursorBlock(msg);
-    }
-    blockClaudeCode(msg);
-  } else if (tool === "Task") {
-    const subagentTypeRaw = toolInput?.subagent_type || toolInput?.subagentType || "";
-    const subagentType = subagentTypeRaw.toLowerCase();
-    if (subagentType.includes("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") {
-        outputCursorBlock(msg);
-      }
-      blockClaudeCode(msg);
-    }
-    if (subagentType.includes("plan")) {
-      const msg = 'For planning, use mcp__contextstream__search(mode="auto", output_format="paths") for discovery, then save your plan 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") {
-        outputCursorBlock(msg);
-      }
-      blockClaudeCode(msg);
-    }
-  } else if (tool === "EnterPlanMode") {
-    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") {
-      outputCursorBlock(msg);
-    }
-    blockClaudeCode(msg);
-  }
-  if (tool === "list_files" || tool === "search_files") {
-    const pattern = toolInput?.path || toolInput?.regex || "";
-    if (isDiscoveryGlob(pattern) || isDiscoveryGrep(pattern)) {
-      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);
-    }
-  }
-  allowTool(editorFormat, cwd, recordStateChange);
-}
-var isDirectRun = process.argv[1]?.includes("pre-tool-use") || process.argv[2] === "pre-tool-use";
-if (isDirectRun) {
-  runPreToolUseHook().catch(() => process.exit(0));
-}
-export {
-  runPreToolUseHook
-};