shieldcortex 2.0.0

package/hooks/clawdbot/cortex-memory/handler.js

@@ -0,0 +1,279 @@
+/**
+ * Cortex Memory Hook — Persistent brain-like memory for Clawdbot/Moltbot
+ *
+ * Integrates ShieldCortex MCP server via mcporter to provide:
+ * - Auto-extraction of important session content on /new
+ * - Context injection on agent bootstrap
+ * - Keyword-triggered memory saves
+ */
+import { execFile } from "node:child_process";
+import fs from "node:fs/promises";
+
+// ==================== CORTEX MCP HELPER ====================
+
+/**
+ * Call a ShieldCortex MCP tool via mcporter
+ * @param {string} tool - Tool name (e.g., "remember", "recall", "get_context")
+ * @param {Record<string, string>} args - Tool arguments as key:value pairs
+ * @returns {Promise<string|null>} Raw stdout or null on error
+ */
+function callCortex(tool, args = {}) {
+  return new Promise((resolve) => {
+    const cmdArgs = [
+      "mcporter", "call", "--stdio",
+      "npx -y shieldcortex",
+      tool,
+    ];
+    for (const [key, value] of Object.entries(args)) {
+      // Escape single quotes by doubling them (FTS5-safe)
+      const safe = String(value).replace(/'/g, "''");
+      cmdArgs.push(`${key}:${safe}`);
+    }
+
+    execFile("npx", cmdArgs, {
+      timeout: 15000,
+      maxBuffer: 1024 * 256,
+    }, (err, stdout) => {
+      if (err) {
+        console.error(`[cortex-memory] mcporter error (${tool}):`, err.message);
+        resolve(null);
+        return;
+      }
+      resolve(stdout?.trim() || null);
+    });
+  });
+}
+
+// ==================== CONTENT EXTRACTION ====================
+
+const PATTERNS = {
+  architecture: [
+    /\b(?:architecture|designed|structured|pattern|approach)\b.*?(?:uses?|is|with)\b/i,
+    /\b(?:created|implemented|refactored|built|set up)\b/i,
+    /\b(?:decided?\s+to|going\s+with|chose|opted?\s+for|using)\b/i,
+  ],
+  error: [
+    /\b(?:fixed|resolved|solved)\s+(?:by|with|using)\b/i,
+    /\b(?:the\s+)?(?:solution|fix|root\s*cause|bug)\s+(?:was|is)\b/i,
+  ],
+  learning: [
+    /\b(?:learned|discovered|turns?\s+out|figured\s+out|realized)\b/i,
+    /\b(?:TIL|today\s+I\s+learned)\b/i,
+  ],
+  preference: [
+    /\b(?:always|never|prefer|don't\s+like|should\s+always)\b/i,
+  ],
+  note: [
+    /\b(?:important|remember|key\s+point|crucial|note)\s*:/i,
+  ],
+};
+
+/**
+ * Extract high-salience content from session messages
+ * @param {string[]} messages - Array of "role: content" strings
+ * @returns {Array<{title: string, content: string, category: string}>}
+ */
+function extractMemories(messages) {
+  const extracted = [];
+  const seen = new Set();
+
+  for (const msg of messages) {
+    if (!msg.startsWith("assistant:")) continue;
+    const text = msg.slice("assistant:".length).trim();
+    if (text.length < 20) continue;
+
+    for (const [category, patterns] of Object.entries(PATTERNS)) {
+      for (const pattern of patterns) {
+        if (pattern.test(text)) {
+          const title = text.slice(0, 80).replace(/["\n]/g, " ").trim();
+          if (seen.has(title)) break;
+          seen.add(title);
+
+          extracted.push({
+            title,
+            content: text.slice(0, 500),
+            category,
+          });
+          break;
+        }
+      }
+      if (extracted.length >= 5) break;
+    }
+    if (extracted.length >= 5) break;
+  }
+
+  return extracted;
+}
+
+// ==================== SESSION FILE READER ====================
+
+/**
+ * Read recent messages from a session JSONL file
+ * @param {string} sessionFilePath
+ * @returns {Promise<string[]>} Array of "role: content" strings
+ */
+async function getRecentMessages(sessionFilePath) {
+  try {
+    const content = await fs.readFile(sessionFilePath, "utf-8");
+    const lines = content.trim().split("\n");
+    const recentLines = lines.slice(-30);
+
+    const messages = [];
+    for (const line of recentLines) {
+      try {
+        const entry = JSON.parse(line);
+        if (entry.type === "message" && entry.message) {
+          const msg = entry.message;
+          if ((msg.role === "user" || msg.role === "assistant") && msg.content) {
+            const text = Array.isArray(msg.content)
+              ? msg.content.find((c) => c.type === "text")?.text
+              : msg.content;
+            if (text && !text.startsWith("/")) {
+              messages.push(`${msg.role}: ${text}`);
+            }
+          }
+        }
+      } catch {
+        // Skip invalid lines
+      }
+    }
+    return messages;
+  } catch {
+    return [];
+  }
+}
+
+// ==================== EVENT HANDLERS ====================
+
+/**
+ * Handle command:new — extract memories from ending session
+ */
+async function onSessionEnd(event) {
+  const context = event.context || {};
+  const sessionEntry = context.previousSessionEntry || context.sessionEntry || {};
+  const sessionFile = sessionEntry.sessionFile;
+
+  if (!sessionFile) {
+    console.log("[cortex-memory] No session file found, skipping extraction");
+    return;
+  }
+
+  const messages = await getRecentMessages(sessionFile);
+  if (messages.length === 0) {
+    console.log("[cortex-memory] No messages to extract");
+    return;
+  }
+
+  const memories = extractMemories(messages);
+  if (memories.length === 0) {
+    console.log("[cortex-memory] No high-salience content found");
+    return;
+  }
+
+  let saved = 0;
+  for (const mem of memories) {
+    const result = await callCortex("remember", {
+      title: mem.title,
+      content: mem.content,
+      category: mem.category,
+      project: "clawdbot",
+      scope: "global",
+      importance: "high",
+      tags: "auto-extracted,clawdbot-hook",
+    });
+    if (result) saved++;
+  }
+
+  console.log(`[cortex-memory] Saved ${saved}/${memories.length} memories from session`);
+}
+
+/**
+ * Handle agent:bootstrap — inject past context into agent
+ */
+async function onBootstrap(event) {
+  const context = event.context || {};
+  if (!Array.isArray(context.bootstrapFiles)) return;
+
+  const result = await callCortex("get_context", {
+    query: "clawdbot session context",
+    format: "summary",
+  });
+
+  if (!result || result.length < 20) {
+    console.log("[cortex-memory] No context to inject");
+    return;
+  }
+
+  context.bootstrapFiles.push({
+    name: "CORTEX_MEMORY.md",
+    content: `# Past Session Context (from ShieldCortex)\n\n${result}`,
+  });
+
+  console.log("[cortex-memory] Injected past context into bootstrap");
+}
+
+/**
+ * Handle command events — check for keyword triggers
+ */
+async function onKeywordTrigger(event) {
+  if (event.action === "new" || event.action === "stop") return;
+
+  const context = event.context || {};
+  const sessionEntry = context.sessionEntry || {};
+  const lastMessage = context.lastUserMessage || sessionEntry.lastUserMessage;
+  if (!lastMessage || typeof lastMessage !== "string") return;
+
+  const lower = lastMessage.toLowerCase();
+  const triggers = ["remember this", "don't forget", "dont forget"];
+  const triggered = triggers.some((t) => lower.includes(t));
+  if (!triggered) return;
+
+  let content = lastMessage;
+  for (const t of triggers) {
+    const idx = lower.indexOf(t);
+    if (idx !== -1) {
+      content = lastMessage.slice(idx + t.length).replace(/^[:\s]+/, "").trim();
+      break;
+    }
+  }
+
+  if (content.length < 5) return;
+
+  const title = content.slice(0, 80).replace(/["\n]/g, " ").trim();
+
+  const result = await callCortex("remember", {
+    title,
+    content: content.slice(0, 500),
+    category: "note",
+    project: "clawdbot",
+    scope: "global",
+    importance: "critical",
+    tags: "keyword-trigger,clawdbot-hook",
+  });
+
+  if (result) {
+    event.messages.push(`Saved to Cortex memory: "${title}"`);
+    console.log(`[cortex-memory] Keyword trigger saved: ${title}`);
+  }
+}
+
+// ==================== MAIN HANDLER ====================
+
+const cortexMemoryHandler = async (event) => {
+  try {
+    if (event.type === "command" && event.action === "new") {
+      await onSessionEnd(event);
+    } else if (event.type === "agent" && event.action === "bootstrap") {
+      await onBootstrap(event);
+    } else if (event.type === "command") {
+      await onKeywordTrigger(event);
+    }
+  } catch (err) {
+    console.error(
+      "[cortex-memory] Error:",
+      err instanceof Error ? err.message : String(err)
+    );
+  }
+};
+
+export default cortexMemoryHandler;

package/package.json

@@ -0,0 +1,73 @@
+{
+  "name": "shieldcortex",
+  "version": "2.0.0",
+  "description": "Security layer for AI agent memory — firewall, trust scoring, sensitivity classification, and audit for any memory backend",
+  "main": "dist/index.js",
+  "type": "module",
+  "scripts": {
+    "build": "tsc",
+    "build:all": "tsc && cd dashboard && npm run build",
+    "start": "node dist/index.js",
+    "start:api": "node dist/index.js --mode api",
+    "start:dashboard": "node dist/index.js --dashboard",
+    "dev": "tsx src/index.ts",
+    "dev:api": "tsx src/index.ts --mode api",
+    "dev:dashboard": "tsx src/index.ts --dashboard",
+    "watch": "tsc --watch",
+    "test": "node --experimental-vm-modules node_modules/jest/bin/jest.js",
+    "test:watch": "node --experimental-vm-modules node_modules/jest/bin/jest.js --watch",
+    "test:coverage": "node --experimental-vm-modules node_modules/jest/bin/jest.js --coverage",
+    "audit:security": "npm audit --audit-level=moderate",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "claude",
+    "mcp",
+    "memory",
+    "ai",
+    "context"
+  ],
+  "author": "Michael Kyriacou",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/mkdelta221/shieldcortex.git"
+  },
+  "homepage": "https://github.com/mkdelta221/shieldcortex#readme",
+  "bugs": {
+    "url": "https://github.com/mkdelta221/shieldcortex/issues"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "dependencies": {
+    "@huggingface/transformers": "^3.7.2",
+    "@modelcontextprotocol/sdk": "^1.0.0",
+    "better-sqlite3": "^11.0.0",
+    "cors": "^2.8.5",
+    "express": "^4.21.0",
+    "ws": "^8.18.0",
+    "zod": "^3.23.0"
+  },
+  "devDependencies": {
+    "@types/better-sqlite3": "^7.6.11",
+    "@types/cors": "^2.8.17",
+    "@types/express": "^5.0.0",
+    "@types/jest": "^30.0.0",
+    "@types/node": "^22.0.0",
+    "@types/ws": "^8.5.13",
+    "jest": "^30.2.0",
+    "ts-jest": "^29.4.6",
+    "tsx": "^4.19.0",
+    "typescript": "^5.6.0"
+  },
+  "bin": {
+    "shieldcortex": "dist/index.js"
+  },
+  "files": [
+    "dist",
+    "dashboard",
+    "hooks",
+    "scripts"
+  ]
+}