@frostbridge/imdl 0.1.5 → 0.1.7

package/dist/index.js

@@ -726,6 +726,23 @@ function setPaused(paused) {
     }
   }
 }
+var SESSION_FILE = join3(IMDL_DIR, ".last-session");
+function getLastSessionId() {
+  try {
+    if (existsSync4(SESSION_FILE)) {
+      return readFileSync(SESSION_FILE, "utf-8").trim() || "unknown";
+    }
+  } catch {
+  }
+  return "unknown";
+}
+function saveLastSessionId(sessionId) {
+  try {
+    ensureImdlDir();
+    writeFileSync(SESSION_FILE, sessionId, { mode: 384 });
+  } catch {
+  }
+}
 
 // src/mcp/reporter.ts
 var THROTTLE_FILE = join4(getImdlDir(), "last-mcp-report");
@@ -2784,11 +2801,12 @@ async function sendBatch(apiUrl, sessionId, events, developerId) {
   if (config.authToken) {
     headers["X-IMDL-Key"] = config.authToken;
   }
+  const agentType = events.find((e) => e.agentType)?.agentType;
   try {
     const res = await fetch(`${apiUrl}/api/sessions/${sessionId}/events`, {
       method: "POST",
       headers,
-      body: JSON.stringify({ events, developerId }),
+      body: JSON.stringify({ events, developerId, ...agentType && { agentType } }),
       signal: AbortSignal.timeout(FLUSH_TIMEOUT)
     });
     return res.ok;
@@ -3195,7 +3213,9 @@ var CHAT_DATA_KEYS = [
   "workbench.panel.aichat.view.aichat.chatdata",
   "workbench.panel.aichat.view.aichat.chatdata.v2",
   "composer.composerData",
-  "composer.allComposers"
+  "composer.allComposers",
+  "aiService.prompts",
+  "aiService.generations"
 ];
 var CursorAdapter = class _CursorAdapter {
   agentType = "cursor";
@@ -3232,6 +3252,11 @@ var CursorAdapter = class _CursorAdapter {
    */
   findVscdbFiles(baseDir) {
     const files = [];
+    const directDb = join16(baseDir, "state.vscdb");
+    if (existsSync17(directDb)) {
+      files.push(directDb);
+      return files;
+    }
     try {
       const entries = readdirSync4(baseDir);
       for (const entry of entries) {
@@ -3262,9 +3287,23 @@ var CursorAdapter = class _CursorAdapter {
     try {
       db = new Database(dbPath, { readonly: true, fileMustExist: true });
       const rows = this.queryAIChatRows(db);
+      let hasGenerations = false;
       for (const row of rows) {
         const parsed = this.parseJsonValue(row.value);
         if (!parsed) continue;
+        if (row.key === "aiService.generations") {
+          const genEvents = this.extractFromGenerations(parsed, dbPath, since);
+          events.push(...genEvents);
+          if (genEvents.length > 0) hasGenerations = true;
+          continue;
+        }
+        if (row.key === "aiService.prompts") {
+          if (!hasGenerations) {
+            const promptEvents = this.extractFromPrompts(parsed, dbPath, since);
+            events.push(...promptEvents);
+          }
+          continue;
+        }
         const conversations = this.extractConversations(parsed);
         for (const conv of conversations) {
           const convEvents = this.conversationToEvents(conv, dbPath, since);
@@ -3280,6 +3319,57 @@ var CursorAdapter = class _CursorAdapter {
     }
     return events;
   }
+  /**
+   * Extract events from aiService.generations format:
+   * [{unixMs, generationUUID, type, textDescription}, ...]
+   */
+  extractFromGenerations(data, dbPath, since) {
+    if (!Array.isArray(data)) return [];
+    const events = [];
+    const sessionId = this.sessionIdFromPath(dbPath);
+    for (const gen of data) {
+      if (!gen || typeof gen !== "object") continue;
+      const { unixMs, textDescription, type } = gen;
+      if (!textDescription || typeof textDescription !== "string") continue;
+      if (!unixMs || typeof unixMs !== "number") continue;
+      if (since && unixMs <= since.getTime()) continue;
+      const genType = type;
+      if (genType === "apply") continue;
+      if (this.isNoisePrompt(textDescription)) continue;
+      events.push({
+        sessionId,
+        type: "user_prompt",
+        prompt: textDescription.slice(0, 1e4),
+        timestamp: new Date(unixMs).toISOString(),
+        agentType: "cursor"
+      });
+    }
+    return events;
+  }
+  /**
+   * Extract events from aiService.prompts format:
+   * [{text, commandType}, ...]
+   * No timestamps — only use if generations is empty.
+   */
+  extractFromPrompts(data, dbPath, _since) {
+    if (!Array.isArray(data) || data.length === 0) return [];
+    const events = [];
+    const sessionId = this.sessionIdFromPath(dbPath);
+    for (const prompt of data) {
+      if (!prompt || typeof prompt !== "object") continue;
+      const { text } = prompt;
+      if (!text || typeof text !== "string") continue;
+      if (this.isNoisePrompt(text)) continue;
+      events.push({
+        sessionId,
+        type: "user_prompt",
+        prompt: text.slice(0, 1e4),
+        timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+        agentType: "cursor"
+      });
+    }
+    return events;
+  }
   /**
    * Query the ItemTable for rows with AI-related keys.
    */
@@ -3608,6 +3698,42 @@ async function collectPrompts() {
       }
     }
   }
+  const cursorWorkspaces = CursorAdapter.getAllWorkspaces();
+  if (cursorWorkspaces.length > 0) {
+    const cursorAdapter = new CursorAdapter();
+    for (const ws of cursorWorkspaces) {
+      try {
+        const events = await cursorAdapter.extractEvents({
+          kind: "directory",
+          path: ws,
+          since: new Date(state.lastRun)
+        });
+        for (const event of events) {
+          let violation;
+          if (event.prompt) {
+            const scan = scanPromptForSecrets(event.prompt);
+            if (scan.hasSecrets) {
+              violation = createPromptViolation(scan.findings);
+            }
+          }
+          const buffered = {
+            sessionId: event.sessionId,
+            type: event.type,
+            toolName: event.toolName,
+            toolInput: event.toolInput ? redactObject(event.toolInput) : void 0,
+            prompt: event.prompt ? redactObject(event.prompt) : void 0,
+            timestamp: event.timestamp,
+            cwd: event.cwd,
+            agentType: "cursor",
+            violation
+          };
+          appendEvent(buffered);
+          collected++;
+        }
+      } catch {
+      }
+    }
+  }
   if (collected > 0) {
     try {
       await tryFlush();
@@ -5690,8 +5816,9 @@ async function handleHook(type) {
       behaviorRiskScore = computeRiskScore(signals);
     }
   }
+  const sessionId = event.session_id || getLastSessionId();
   const buffered = {
-    sessionId: event.session_id,
+    sessionId,
     type: hookType,
     toolName: event.tool_name,
     toolInput: redactObject(event.tool_input),
@@ -5703,6 +5830,9 @@ async function handleHook(type) {
     behaviorSignals,
     behaviorRiskScore
   };
+  if (event.session_id) {
+    saveLastSessionId(event.session_id);
+  }
   appendEvent(buffered);
   try {
     await tryFlush();

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "restricted"
   },
-  "version": "0.1.5",
+  "version": "0.1.7",
   "description": "IMDL — Intelligent Mediation & Detection Layer. AI agent security CLI.",
   "files": [
     "dist"