@contextstream/mcp-server 0.4.81 → 1.0.0

package/dist/hooks/session-end.js

@@ -1,285 +0,0 @@
-#!/usr/bin/env node
-
-// src/hooks/session-end.ts
-import * as fs from "node:fs";
-import * as path from "node:path";
-import { homedir } from "node:os";
-var ENABLED = process.env.CONTEXTSTREAM_SESSION_END_ENABLED !== "false";
-var SAVE_TRANSCRIPT = process.env.CONTEXTSTREAM_SESSION_END_SAVE_TRANSCRIPT !== "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;
-var PROJECT_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 || !PROJECT_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) {
-            WORKSPACE_ID = csConfig.workspace_id;
-          }
-          if (csConfig.project_id && !PROJECT_ID) {
-            PROJECT_ID = csConfig.project_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 parseTranscriptStats(transcriptPath) {
-  const stats = {
-    messageCount: 0,
-    toolCallCount: 0,
-    duration: 0,
-    filesModified: [],
-    messages: [],
-    startedAt: (/* @__PURE__ */ new Date()).toISOString()
-  };
-  if (!transcriptPath || !fs.existsSync(transcriptPath)) {
-    return stats;
-  }
-  try {
-    const content = fs.readFileSync(transcriptPath, "utf-8");
-    const lines = content.split("\n");
-    let firstTimestamp = null;
-    let lastTimestamp = null;
-    const modifiedFiles = /* @__PURE__ */ new Set();
-    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 (entry.timestamp) {
-          const ts = new Date(entry.timestamp);
-          if (!firstTimestamp || ts < firstTimestamp) {
-            firstTimestamp = ts;
-            stats.startedAt = entry.timestamp;
-          }
-          if (!lastTimestamp || ts > lastTimestamp) {
-            lastTimestamp = ts;
-          }
-        }
-        if (msgType === "user" || entry.role === "user") {
-          stats.messageCount++;
-          const userContent = typeof entry.content === "string" ? entry.content : "";
-          if (userContent) {
-            stats.messages.push({
-              role: "user",
-              content: userContent,
-              timestamp
-            });
-          }
-        } else if (msgType === "assistant" || entry.role === "assistant") {
-          stats.messageCount++;
-          const assistantContent = typeof entry.content === "string" ? entry.content : "";
-          if (assistantContent) {
-            stats.messages.push({
-              role: "assistant",
-              content: assistantContent,
-              timestamp
-            });
-          }
-        } else if (msgType === "tool_use") {
-          stats.toolCallCount++;
-          const toolName = entry.name || "";
-          const toolInput = entry.input || {};
-          if (["Write", "Edit", "NotebookEdit"].includes(toolName)) {
-            const filePath = toolInput.file_path || toolInput.notebook_path;
-            if (filePath) {
-              modifiedFiles.add(filePath);
-            }
-          }
-          stats.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);
-          stats.messages.push({
-            role: "tool",
-            content: resultContent,
-            timestamp,
-            tool_results: { name: entry.name }
-          });
-        }
-      } catch {
-        continue;
-      }
-    }
-    if (firstTimestamp && lastTimestamp) {
-      stats.duration = Math.round((lastTimestamp.getTime() - firstTimestamp.getTime()) / 1e3);
-    }
-    stats.filesModified = Array.from(modifiedFiles);
-  } catch {
-  }
-  return stats;
-}
-async function saveFullTranscript(sessionId, stats, reason) {
-  if (!API_KEY) {
-    return { success: false, message: "No API key configured" };
-  }
-  if (stats.messages.length === 0) {
-    return { success: false, message: "No messages to save" };
-  }
-  const payload = {
-    session_id: sessionId,
-    messages: stats.messages,
-    started_at: stats.startedAt,
-    source_type: "session_end",
-    title: `Session transcript (${reason})`,
-    metadata: {
-      reason,
-      tool_call_count: stats.toolCallCount,
-      files_modified: stats.filesModified.slice(0, 20),
-      duration_seconds: stats.duration
-    },
-    tags: ["session_end", reason]
-  };
-  if (WORKSPACE_ID) {
-    payload.workspace_id = WORKSPACE_ID;
-  }
-  if (PROJECT_ID) {
-    payload.project_id = PROJECT_ID;
-  }
-  try {
-    const controller = new AbortController();
-    const timeoutId = setTimeout(() => controller.abort(), 15e3);
-    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 (${stats.messages.length} messages)` };
-    }
-    return { success: false, message: `API error: ${response.status}` };
-  } catch (error) {
-    return { success: false, message: String(error) };
-  }
-}
-async function finalizeSession(sessionId, stats, reason) {
-  if (!API_KEY) return;
-  if (SAVE_TRANSCRIPT && stats.messages.length > 0) {
-    await saveFullTranscript(sessionId, stats, reason);
-  }
-  const payload = {
-    event_type: "manual_note",
-    title: `Session Ended: ${reason}`,
-    content: JSON.stringify({
-      session_id: sessionId,
-      reason,
-      stats: {
-        messages: stats.messageCount,
-        tool_calls: stats.toolCallCount,
-        duration_seconds: stats.duration,
-        files_modified: stats.filesModified.length
-      },
-      files_modified: stats.filesModified.slice(0, 20),
-      ended_at: (/* @__PURE__ */ new Date()).toISOString()
-    }),
-    importance: "low",
-    tags: ["session", "end", reason],
-    source_type: "hook"
-  };
-  if (WORKSPACE_ID) {
-    payload.workspace_id = WORKSPACE_ID;
-  }
-  try {
-    const controller = new AbortController();
-    const timeoutId = setTimeout(() => controller.abort(), 5e3);
-    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);
-  } catch {
-  }
-}
-async function runSessionEndHook() {
-  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 reason = input.reason || "user_exit";
-  const stats = parseTranscriptStats(transcriptPath);
-  await finalizeSession(sessionId, stats, reason);
-  process.exit(0);
-}
-var isDirectRun = process.argv[1]?.includes("session-end") || process.argv[2] === "session-end";
-if (isDirectRun) {
-  runSessionEndHook().catch(() => process.exit(0));
-}
-export {
-  runSessionEndHook
-};