@contextstream/mcp-server 0.4.62 → 0.4.64

package/dist/hooks/teammate-idle.js

@@ -0,0 +1,197 @@
+#!/usr/bin/env node
+
+// src/hooks/common.ts
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { homedir } from "node:os";
+var DEFAULT_API_URL = "https://api.contextstream.io";
+function readHookInput() {
+  try {
+    return JSON.parse(fs.readFileSync(0, "utf8"));
+  } catch {
+    return {};
+  }
+}
+function writeHookOutput(output) {
+  const payload = output && (output.additionalContext || output.blocked || output.reason) ? {
+    hookSpecificOutput: output.additionalContext ? {
+      hookEventName: output.hookEventName,
+      additionalContext: output.additionalContext
+    } : void 0,
+    additionalContext: output.additionalContext,
+    blocked: output.blocked,
+    reason: output.reason
+  } : {};
+  console.log(JSON.stringify(payload));
+}
+function extractCwd(input) {
+  const cwd = typeof input.cwd === "string" && input.cwd.trim() ? input.cwd.trim() : process.cwd();
+  return cwd;
+}
+function loadHookConfig(cwd) {
+  let apiUrl = process.env.CONTEXTSTREAM_API_URL || DEFAULT_API_URL;
+  let apiKey = process.env.CONTEXTSTREAM_API_KEY || "";
+  let jwt = process.env.CONTEXTSTREAM_JWT || "";
+  let workspaceId = process.env.CONTEXTSTREAM_WORKSPACE_ID || null;
+  let projectId = process.env.CONTEXTSTREAM_PROJECT_ID || null;
+  let searchDir = path.resolve(cwd);
+  for (let i = 0; i < 6; i++) {
+    if (!apiKey && !jwt) {
+      const mcpPath = path.join(searchDir, ".mcp.json");
+      if (fs.existsSync(mcpPath)) {
+        try {
+          const config = JSON.parse(fs.readFileSync(mcpPath, "utf8"));
+          const env = config.mcpServers?.contextstream?.env;
+          if (env?.CONTEXTSTREAM_API_KEY) apiKey = env.CONTEXTSTREAM_API_KEY;
+          if (env?.CONTEXTSTREAM_JWT) jwt = env.CONTEXTSTREAM_JWT;
+          if (env?.CONTEXTSTREAM_API_URL) apiUrl = env.CONTEXTSTREAM_API_URL;
+          if (env?.CONTEXTSTREAM_WORKSPACE_ID && !workspaceId) workspaceId = env.CONTEXTSTREAM_WORKSPACE_ID;
+          if (env?.CONTEXTSTREAM_PROJECT_ID && !projectId) projectId = env.CONTEXTSTREAM_PROJECT_ID;
+        } catch {
+        }
+      }
+    }
+    if (!workspaceId || !projectId) {
+      const localConfigPath = path.join(searchDir, ".contextstream", "config.json");
+      if (fs.existsSync(localConfigPath)) {
+        try {
+          const localConfig = JSON.parse(fs.readFileSync(localConfigPath, "utf8"));
+          if (localConfig.workspace_id && !workspaceId) workspaceId = localConfig.workspace_id;
+          if (localConfig.project_id && !projectId) projectId = localConfig.project_id;
+        } catch {
+        }
+      }
+    }
+    const parentDir = path.dirname(searchDir);
+    if (parentDir === searchDir) break;
+    searchDir = parentDir;
+  }
+  if (!apiKey && !jwt) {
+    const homeMcpPath = path.join(homedir(), ".mcp.json");
+    if (fs.existsSync(homeMcpPath)) {
+      try {
+        const config = JSON.parse(fs.readFileSync(homeMcpPath, "utf8"));
+        const env = config.mcpServers?.contextstream?.env;
+        if (env?.CONTEXTSTREAM_API_KEY) apiKey = env.CONTEXTSTREAM_API_KEY;
+        if (env?.CONTEXTSTREAM_JWT) jwt = env.CONTEXTSTREAM_JWT;
+        if (env?.CONTEXTSTREAM_API_URL) apiUrl = env.CONTEXTSTREAM_API_URL;
+      } catch {
+      }
+    }
+  }
+  return { apiUrl, apiKey, jwt, workspaceId, projectId };
+}
+function isConfigured(config) {
+  return Boolean(config.apiKey || config.jwt);
+}
+function authHeaders(config) {
+  if (config.apiKey) {
+    return { "X-API-Key": config.apiKey };
+  }
+  if (config.jwt) {
+    return { Authorization: `Bearer ${config.jwt}` };
+  }
+  return {};
+}
+async function apiRequest(config, apiPath, init = {}) {
+  const response = await fetch(`${config.apiUrl}${apiPath}`, {
+    method: init.method || "GET",
+    headers: {
+      "Content-Type": "application/json",
+      ...authHeaders(config)
+    },
+    body: init.body !== void 0 ? JSON.stringify(init.body) : void 0
+  });
+  if (!response.ok) {
+    throw new Error(`${response.status} ${response.statusText}`);
+  }
+  const json = await response.json();
+  if (json && typeof json === "object" && "data" in json) {
+    return json.data;
+  }
+  return json;
+}
+async function postMemoryEvent(config, title, content, tags, eventType = "operation") {
+  if (!isConfigured(config) || !config.workspaceId) return;
+  await apiRequest(config, "/memory/events", {
+    method: "POST",
+    body: {
+      workspace_id: config.workspaceId,
+      project_id: config.projectId || void 0,
+      event_type: eventType,
+      title,
+      content: typeof content === "string" ? content : JSON.stringify(content),
+      metadata: {
+        tags,
+        source: "mcp_hook",
+        captured_at: (/* @__PURE__ */ new Date()).toISOString()
+      }
+    }
+  });
+}
+async function listPendingTasks(config, limit = 5) {
+  if (!isConfigured(config) || !config.workspaceId) return [];
+  try {
+    const params = new URLSearchParams();
+    params.set("workspace_id", config.workspaceId);
+    params.set("status", "pending");
+    params.set("limit", String(limit));
+    if (config.projectId) params.set("project_id", config.projectId);
+    const result = await apiRequest(config, `/tasks?${params.toString()}`);
+    if (Array.isArray(result)) return result;
+    if (Array.isArray(result?.items)) return result.items;
+    if (Array.isArray(result?.tasks)) return result.tasks;
+    return [];
+  } catch {
+    return [];
+  }
+}
+
+// src/hooks/teammate-idle.ts
+function firstString(input, keys) {
+  for (const key of keys) {
+    const value = input[key];
+    if (typeof value === "string" && value.trim()) return value.trim();
+  }
+  return void 0;
+}
+async function runTeammateIdleHook() {
+  const input = readHookInput();
+  const cwd = extractCwd(input);
+  const config = loadHookConfig(cwd);
+  const teammateName = firstString(input, ["teammate_name", "teammateName", "agent_name"]) || "teammate";
+  const teamName = firstString(input, ["team_name", "teamName"]) || "team";
+  if (!isConfigured(config)) {
+    writeHookOutput();
+    return;
+  }
+  const pendingTasks = await listPendingTasks(config, 5);
+  await postMemoryEvent(
+    config,
+    "Teammate idle",
+    {
+      teammate_name: teammateName,
+      team_name: teamName,
+      pending_tasks: pendingTasks.slice(0, 5),
+      timestamp: (/* @__PURE__ */ new Date()).toISOString()
+    },
+    ["hook", "teammate", "idle"]
+  ).catch(() => {
+  });
+  const shouldRedirect = process.env.CONTEXTSTREAM_TEAMMATE_IDLE_REDIRECT !== "false";
+  if (shouldRedirect && pendingTasks.length > 0) {
+    const firstTask = pendingTasks[0];
+    const taskTitle = typeof firstTask.title === "string" && firstTask.title || typeof firstTask.subject === "string" && firstTask.subject || typeof firstTask.name === "string" && firstTask.name || "pending task";
+    const message = `Pending ContextStream task available: ${taskTitle}. Continue and complete this task before idling.`;
+    writeHookOutput({ additionalContext: message, blocked: true, reason: message });
+    return;
+  }
+  writeHookOutput();
+}
+var isDirectRun = process.argv[1]?.includes("teammate-idle") || process.argv[2] === "teammate-idle";
+if (isDirectRun) {
+  runTeammateIdleHook().catch(() => process.exit(0));
+}
+export {
+  runTeammateIdleHook
+};

package/dist/hooks/user-prompt-submit.js

@@ -283,6 +283,7 @@ var API_KEY = process.env.CONTEXTSTREAM_API_KEY || "";
 var WORKSPACE_ID = null;
 var PROJECT_ID = null;
 var REMINDER = `[CONTEXTSTREAM] On the first message in every session call mcp__contextstream__init(...), then call mcp__contextstream__context(user_message="...", save_exchange=true, session_id="<session-id>") FIRST before any other tool. On subsequent messages, default to context first. Narrow bypass is allowed only for immediate read-only ContextStream calls when prior context is fresh and no state-changing tool has run. Response contains dynamic rules, lessons, preferences.
+SEARCH-FIRST: Use mcp__contextstream__search(mode="auto") before Glob/Grep/Read/Explore/Task/EnterPlanMode. In planning, never use EnterPlanMode or Task(Explore) for file-by-file discovery.
 COMMON MEMORY CALLS: list docs via memory(action="list_docs"), list lessons via session(action="get_lessons"), list plans via session(action="list_plans"), list tasks/todos via memory(action="list_tasks"|"list_todos").
 [END]`;
 var FULL_REMINDER = `[CONTEXTSTREAM RULES - MANDATORY]
@@ -296,7 +297,7 @@ var FULL_REMINDER = `[CONTEXTSTREAM RULES - MANDATORY]
 
 2. FOR CODE SEARCH: Check index status, then search appropriately
    \u26A0\uFE0F BEFORE searching: mcp__contextstream__project(action="index_status")
-   \u2705 IF indexed & fresh: Use mcp__contextstream__search(mode="auto", query="...")
+   \u2705 IF indexed & fresh: Use mcp__contextstream__search(mode="auto", query="...") BEFORE Glob/Grep/Read/Explore/Task/EnterPlanMode
    \u2705 IF NOT indexed OR stale: Use local tools (Glob/Grep/Read) directly
    \u2705 IF search returns 0 results: Fallback to local tools (Glob/Grep/Read)
 
@@ -309,7 +310,8 @@ var FULL_REMINDER = `[CONTEXTSTREAM RULES - MANDATORY]
 4. FOR PLANS & TASKS: Use ContextStream, not file-based plans
    \u2705 Plans: mcp__contextstream__session(action="capture_plan", ...)
    \u2705 Tasks: mcp__contextstream__memory(action="create_task", ...)
-   \u274C DO NOT use EnterPlanMode or write plans to markdown files
+   \u274C DO NOT use EnterPlanMode or Task(subagent_type="Explore") for file-by-file discovery
+   \u2705 For planning discovery: mcp__contextstream__search(mode="auto", query="...", output_format="paths")
 
 5. CHECK THESE from context() response:
    - Lessons: Past mistakes to avoid (shown as warnings)
@@ -664,7 +666,7 @@ function buildEnhancedReminder(ctx, isNewSession2, versionNotice) {
 2. Wait for indexing: if \`init\` returns \`indexing_status: "started"\`, files are being indexed
 3. Generate a unique session_id (e.g., "session-" + timestamp or UUID) - use this for ALL context() calls
 4. Call \`context(user_message="...", save_exchange=true, session_id="<your-session-id>")\` for task-specific context
-5. Use \`search(mode="auto")\` for code discovery (not Glob/Grep/Read)
+5. Use \`search(mode="auto")\` for code discovery (not Glob/Grep/Read/Explore/Task/EnterPlanMode)
 
 `);
   }
@@ -827,7 +829,7 @@ async function runUserPromptSubmitHook() {
     const newSession = isNewSession(input, editorFormat);
     const [ctx, versionNotice] = await Promise.all([fetchSessionContext(), versionNoticePromise, saveExchangePromise]);
     if (editorFormat === "cursor") {
-      let cursorReminder = ctx?.lessons?.length ? `[CONTEXTSTREAM] \u26A0\uFE0F ${ctx.lessons.length} lessons from past mistakes. Call context(save_exchange=true, session_id="...") FIRST. Use search(mode="auto") before Glob/Grep. After file edits: project(action="index").` : `[CONTEXTSTREAM] Call context(save_exchange=true, session_id="...") FIRST. Use search(mode="auto") before Glob/Grep/Read. After file edits: project(action="index").`;
+      let cursorReminder = ctx?.lessons?.length ? `[CONTEXTSTREAM] \u26A0\uFE0F ${ctx.lessons.length} lessons from past mistakes. Call context(save_exchange=true, session_id="...") FIRST. Use search(mode="auto") before Glob/Grep/Read/Explore/Task/EnterPlanMode. After file edits: project(action="index").` : `[CONTEXTSTREAM] Call context(save_exchange=true, session_id="...") FIRST. Use search(mode="auto") before Glob/Grep/Read/Explore/Task/EnterPlanMode. After file edits: project(action="index").`;
       if (versionNotice?.behind) {
         cursorReminder += ` [UPDATE v${versionNotice.current}\u2192${versionNotice.latest}]`;
       }