@contextstream/mcp-server 0.4.81 → 1.0.0

package/dist/hooks/pre-compact.js

@@ -1,317 +0,0 @@
-#!/usr/bin/env node
-
-// src/hooks/pre-compact.ts
-import * as fs from "node:fs";
-import * as path from "node:path";
-import { homedir } from "node:os";
-var ENABLED = process.env.CONTEXTSTREAM_PRECOMPACT_ENABLED !== "false";
-var AUTO_SAVE = process.env.CONTEXTSTREAM_PRECOMPACT_AUTO_SAVE !== "false";
-var API_URL = process.env.CONTEXTSTREAM_API_URL || "https://api.contextstream.io";
-var API_KEY = process.env.CONTEXTSTREAM_API_KEY || "";
-var WORKSPACE_ID = null;
-function loadConfigFromMcpJson(cwd) {
-  let searchDir = path.resolve(cwd);
-  for (let i = 0; i < 5; i++) {
-    if (!API_KEY) {
-      const mcpPath = path.join(searchDir, ".mcp.json");
-      if (fs.existsSync(mcpPath)) {
-        try {
-          const content = fs.readFileSync(mcpPath, "utf-8");
-          const config = JSON.parse(content);
-          const csEnv = config.mcpServers?.contextstream?.env;
-          if (csEnv?.CONTEXTSTREAM_API_KEY) {
-            API_KEY = csEnv.CONTEXTSTREAM_API_KEY;
-          }
-          if (csEnv?.CONTEXTSTREAM_API_URL) {
-            API_URL = csEnv.CONTEXTSTREAM_API_URL;
-          }
-        } catch {
-        }
-      }
-    }
-    if (!WORKSPACE_ID) {
-      const csConfigPath = path.join(searchDir, ".contextstream", "config.json");
-      if (fs.existsSync(csConfigPath)) {
-        try {
-          const content = fs.readFileSync(csConfigPath, "utf-8");
-          const csConfig = JSON.parse(content);
-          if (csConfig.workspace_id) {
-            WORKSPACE_ID = csConfig.workspace_id;
-          }
-        } catch {
-        }
-      }
-    }
-    const parentDir = path.dirname(searchDir);
-    if (parentDir === searchDir) break;
-    searchDir = parentDir;
-  }
-  if (!API_KEY) {
-    const homeMcpPath = path.join(homedir(), ".mcp.json");
-    if (fs.existsSync(homeMcpPath)) {
-      try {
-        const content = fs.readFileSync(homeMcpPath, "utf-8");
-        const config = JSON.parse(content);
-        const csEnv = config.mcpServers?.contextstream?.env;
-        if (csEnv?.CONTEXTSTREAM_API_KEY) {
-          API_KEY = csEnv.CONTEXTSTREAM_API_KEY;
-        }
-        if (csEnv?.CONTEXTSTREAM_API_URL) {
-          API_URL = csEnv.CONTEXTSTREAM_API_URL;
-        }
-      } catch {
-      }
-    }
-  }
-}
-function parseTranscript(transcriptPath) {
-  const activeFiles = /* @__PURE__ */ new Set();
-  const recentMessages = [];
-  const toolCalls = [];
-  const messages = [];
-  let startedAt = (/* @__PURE__ */ new Date()).toISOString();
-  let firstTimestamp = true;
-  try {
-    const content = fs.readFileSync(transcriptPath, "utf-8");
-    const lines = content.split("\n");
-    for (const line of lines) {
-      if (!line.trim()) continue;
-      try {
-        const entry = JSON.parse(line);
-        const msgType = entry.type || "";
-        const timestamp = entry.timestamp || (/* @__PURE__ */ new Date()).toISOString();
-        if (firstTimestamp && entry.timestamp) {
-          startedAt = entry.timestamp;
-          firstTimestamp = false;
-        }
-        if (msgType === "tool_use") {
-          const toolName = entry.name || "";
-          const toolInput = entry.input || {};
-          toolCalls.push({ name: toolName, input: toolInput });
-          if (["Read", "Write", "Edit", "NotebookEdit"].includes(toolName)) {
-            const filePath = toolInput.file_path || toolInput.notebook_path;
-            if (filePath) {
-              activeFiles.add(filePath);
-            }
-          } else if (toolName === "Glob") {
-            const pattern = toolInput.pattern;
-            if (pattern) {
-              activeFiles.add(`[glob:${pattern}]`);
-            }
-          }
-          messages.push({
-            role: "assistant",
-            content: `[Tool: ${toolName}]`,
-            timestamp,
-            tool_calls: { name: toolName, input: toolInput }
-          });
-        } else if (msgType === "tool_result") {
-          const resultContent = typeof entry.content === "string" ? entry.content.slice(0, 2e3) : JSON.stringify(entry.content || {}).slice(0, 2e3);
-          messages.push({
-            role: "tool",
-            content: resultContent,
-            timestamp,
-            tool_results: { name: entry.name }
-          });
-        } else if (msgType === "user" || entry.role === "user") {
-          const userContent = typeof entry.content === "string" ? entry.content : "";
-          if (userContent) {
-            messages.push({
-              role: "user",
-              content: userContent,
-              timestamp
-            });
-          }
-        } else if (msgType === "assistant" || entry.role === "assistant") {
-          const assistantContent = typeof entry.content === "string" ? entry.content : "";
-          if (assistantContent) {
-            messages.push({
-              role: "assistant",
-              content: assistantContent,
-              timestamp
-            });
-            if (assistantContent.length > 50) {
-              recentMessages.push(assistantContent.slice(0, 500));
-            }
-          }
-        }
-      } catch {
-        continue;
-      }
-    }
-  } catch {
-  }
-  return {
-    activeFiles: Array.from(activeFiles).slice(-20),
-    // Last 20 files
-    toolCallCount: toolCalls.length,
-    messageCount: messages.length,
-    lastTools: toolCalls.slice(-10).map((t) => t.name),
-    // Last 10 tool names
-    messages,
-    startedAt
-  };
-}
-async function saveFullTranscript(sessionId, transcriptData, trigger) {
-  if (!API_KEY) {
-    return { success: false, message: "No API key configured" };
-  }
-  if (transcriptData.messages.length === 0) {
-    return { success: false, message: "No messages to save" };
-  }
-  const payload = {
-    session_id: sessionId,
-    messages: transcriptData.messages,
-    started_at: transcriptData.startedAt,
-    source_type: "pre_compact",
-    title: `Pre-compaction save (${trigger})`,
-    metadata: {
-      trigger,
-      active_files: transcriptData.activeFiles,
-      tool_call_count: transcriptData.toolCallCount
-    },
-    tags: ["pre_compaction", trigger]
-  };
-  if (WORKSPACE_ID) {
-    payload.workspace_id = WORKSPACE_ID;
-  }
-  try {
-    const controller = new AbortController();
-    const timeoutId = setTimeout(() => controller.abort(), 1e4);
-    const response = await fetch(`${API_URL}/api/v1/transcripts`, {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json",
-        "X-API-Key": API_KEY
-      },
-      body: JSON.stringify(payload),
-      signal: controller.signal
-    });
-    clearTimeout(timeoutId);
-    if (response.ok) {
-      return { success: true, message: `Transcript saved (${transcriptData.messages.length} messages)` };
-    }
-    return { success: false, message: `API error: ${response.status}` };
-  } catch (error) {
-    return { success: false, message: String(error) };
-  }
-}
-async function saveSnapshot(sessionId, transcriptData, trigger) {
-  if (!API_KEY) {
-    return { success: false, message: "No API key configured" };
-  }
-  const snapshotContent = {
-    session_id: sessionId,
-    trigger,
-    captured_at: null,
-    // API will set timestamp
-    active_files: transcriptData.activeFiles,
-    tool_call_count: transcriptData.toolCallCount,
-    last_tools: transcriptData.lastTools,
-    auto_captured: true
-  };
-  const payload = {
-    event_type: "session_snapshot",
-    title: `Auto Pre-compaction Snapshot (${trigger})`,
-    content: JSON.stringify(snapshotContent),
-    importance: "high",
-    tags: ["session_snapshot", "pre_compaction", "auto_captured"],
-    source_type: "hook"
-  };
-  if (WORKSPACE_ID) {
-    payload.workspace_id = WORKSPACE_ID;
-  }
-  try {
-    const controller = new AbortController();
-    const timeoutId = setTimeout(() => controller.abort(), 5e3);
-    const response = await fetch(`${API_URL}/api/v1/memory/events`, {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json",
-        "X-API-Key": API_KEY
-      },
-      body: JSON.stringify(payload),
-      signal: controller.signal
-    });
-    clearTimeout(timeoutId);
-    if (response.ok) {
-      return { success: true, message: "Snapshot saved" };
-    }
-    return { success: false, message: `API error: ${response.status}` };
-  } catch (error) {
-    return { success: false, message: String(error) };
-  }
-}
-async function runPreCompactHook() {
-  if (!ENABLED) {
-    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 cwd = input.cwd || process.cwd();
-  loadConfigFromMcpJson(cwd);
-  const sessionId = input.session_id || "unknown";
-  const transcriptPath = input.transcript_path || "";
-  const trigger = input.trigger || "unknown";
-  const customInstructions = input.custom_instructions || "";
-  let transcriptData = {
-    activeFiles: [],
-    toolCallCount: 0,
-    messageCount: 0,
-    lastTools: [],
-    messages: [],
-    startedAt: (/* @__PURE__ */ new Date()).toISOString()
-  };
-  if (transcriptPath && fs.existsSync(transcriptPath)) {
-    transcriptData = parseTranscript(transcriptPath);
-  }
-  let autoSaveStatus = "";
-  if (AUTO_SAVE && API_KEY) {
-    const transcriptResult = await saveFullTranscript(sessionId, transcriptData, trigger);
-    if (transcriptResult.success) {
-      autoSaveStatus = `
-[ContextStream: ${transcriptResult.message}]`;
-    } else {
-      const { success, message } = await saveSnapshot(sessionId, transcriptData, trigger);
-      if (success) {
-        autoSaveStatus = `
-[ContextStream: Auto-saved snapshot with ${transcriptData.activeFiles.length} active files (transcript save failed: ${transcriptResult.message})]`;
-      } else {
-        autoSaveStatus = `
-[ContextStream: Auto-save failed - ${message}]`;
-      }
-    }
-  }
-  const filesList = transcriptData.activeFiles.slice(0, 5).join(", ") || "none detected";
-  const context = `[CONTEXT COMPACTION - ${trigger.toUpperCase()}]${autoSaveStatus}
-
-Active files detected: ${filesList}
-Tool calls in session: ${transcriptData.toolCallCount}
-
-After compaction, call session_init(is_post_compact=true) to restore context.${customInstructions ? `
-User instructions: ${customInstructions}` : ""}`;
-  console.log(
-    JSON.stringify({
-      additionalContext: context
-    })
-  );
-  process.exit(0);
-}
-var isDirectRun = process.argv[1]?.includes("pre-compact") || process.argv[2] === "pre-compact";
-if (isDirectRun) {
-  runPreCompactHook().catch(() => process.exit(0));
-}
-export {
-  runPreCompactHook
-};