@dropout-ai/runtime 0.3.3 → 0.3.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dropout-ai/runtime",
-  "version": "0.3.3",
+  "version": "0.3.4",
   "description": "Invisible Node.js runtime for capturing AI interactions.",
   "main": "src/index.js",
   "scripts": {

package/src/index.js

@@ -1,8 +1,7 @@
 /**
  * @dropout-ai/runtime
  * Universal AI Interaction Capture for Node.js & Next.js
- * Role: Passive observer.
- * Behavior: Fire-and-forget raw JSON to Supabase.
+ * Patches: fetch, http.request, https.request
  * Capability: Captures Genkit, OpenAI, LangChain, Axios, and standard fetch.
  */
 
@@ -19,355 +18,300 @@ const KNOWN_AI_DOMAINS = [
   'aiplatform.googleapis.com', 'api.groq.com', 'api.mistral.ai', 'api.cohere.ai'
 ];
 
-// Global State
-const GLOBAL_OBJ = typeof window !== 'undefined' ? window : global;
-let activeConfig = {
-  projectId: null,
-  apiKey: null,
-  captureEndpoint: SUPABASE_FUNCTION_URL,
-  maxOutputBytes: 32768,
-  privacyMode: (typeof process !== 'undefined' && process.env.DROPOUT_PRIVACY_MODE) || 'full',
-  debug: false // Added debug flag
-};
-
-let isPatched = false;
-let turnIndex = 0;
-let lastTurnConfirmed = true;
-let lastPromptHash = null;
-let lastResponseHash = null;
-let instance = null;
-
-// --- UTILS ---
-
-function log(msg, ...args) {
-  if (activeConfig.debug) console.log(`[Dropout] ${msg}`, ...args);
-}
+class Dropout {
+  constructor(config = {}) {
+    // 1. Validation
+    if (!config.apiKey || !config.projectId) {
+      console.warn("[Dropout] ⚠️ Initialization Skipped: Missing apiKey or projectId.");
+      return;
+    }
 
-function generateSessionId() {
-  try {
-    return crypto.randomUUID();
-  } catch (e) {
-    return 'sess_' + Math.random().toString(36).substring(2, 12) + Date.now().toString(36);
-  }
-}
+    // 2. Config Setup
+    this.config = config;
+    this.projectId = config.projectId;
+    this.apiKey = config.apiKey;
+    this.debug = config.debug || false;
+    this.privacy = config.privacy || 'full';
+    this.captureEndpoint = SUPABASE_FUNCTION_URL;
+
+    // 3. Singleton Guard
+    if (global.__dropout_initialized__) {
+      if (this.debug) console.log("[Dropout] ℹ️ Already initialized. Skipping patch.");
+      return;
+    }
+    global.__dropout_initialized__ = true;
 
-function hash(text) {
-  if (!text) return null;
-  try {
-    return crypto.createHash('sha256').update(text.toLowerCase().trim()).digest('hex');
-  } catch (e) { return 'hash_err'; }
-}
+    // 4. Initialize Identity
+    if (!global.__dropout_session_id__) {
+      global.__dropout_session_id__ = this.generateSessionId();
+    }
 
-function normalize(url, body) {
-  let provider = 'custom';
-  let model = 'unknown';
-
-  if (url) {
-    const u = url.toLowerCase();
-    if (u.includes('openai')) provider = 'openai';
-    else if (u.includes('anthropic')) provider = 'anthropic';
-    else if (u.includes('google') || u.includes('generative') || u.includes('aiplatform')) provider = 'google';
-    else if (u.includes('groq')) provider = 'groq';
-    else if (u.includes('mistral')) provider = 'mistral';
-    else if (u.includes('cohere')) provider = 'cohere';
-    else if (u.includes('localhost') || u.includes('127.0.0.1')) provider = 'local';
-  }
+    // 5. Start the Wiretap
+    if (this.debug) console.log(`[Dropout] 🟢 Initialized for Project: ${this.projectId}`);
 
-  if (body) {
-    try {
-      const parsed = typeof body === 'string' ? JSON.parse(body) : body;
-      if (parsed.model) model = parsed.model;
-    } catch (e) { }
+    // Bind methods to 'this' to avoid scope loss
+    this.log = this.log.bind(this);
+    this.isAiRequest = this.isAiRequest.bind(this);
+    this.patchNetwork();
   }
-  return { provider, model };
-}
-
-function isAiRequest(url, bodyString) {
-  if (!url) return false;
-  // Guard: Infinite Loop Prevention
-  if (url.includes(activeConfig.captureEndpoint)) return false;
 
-  // 1. Check Known Domains
-  if (KNOWN_AI_DOMAINS.some(d => url.includes(d))) return true;
-
-  // 2. Heuristic Check (for custom endpoints)
-  if (bodyString) {
-    return (bodyString.includes('"model"') || bodyString.includes('"messages"')) &&
-      (bodyString.includes('"user"') || bodyString.includes('"prompt"'));
+  // --- UTILS ---
+  log(msg, ...args) {
+    if (this.debug) console.log(`[Dropout] ${msg}`, ...args);
   }
-  return false;
-}
-
-// --- EMITTER (Authenticated) ---
-
-function emit(payload) {
-  // 1. Guard: Do not emit if not initialized
-  if (!activeConfig.apiKey || !activeConfig.projectId) return;
-
-  // 2. Construct Final Payload
-  const finalPayload = {
-    ...payload,
-    project_id: activeConfig.projectId,
 
-    // ✅ Ensure these columns are always populated for the DB
-    content_blob: payload.content, // Map content to blob
-    received_at: new Date().toISOString()
-  };
-
-  log(`🚀 Sending Capture (${payload.turn_role})`);
-
-  // 3. Fire and Forget using HTTPS (Bypassing our own patches)
-  const req = https.request(activeConfig.captureEndpoint, {
-    method: 'POST',
-    headers: {
-      'Content-Type': 'application/json',
-      'x-dropout-key': activeConfig.apiKey
+  generateSessionId() {
+    try {
+      return crypto.randomUUID();
+    } catch (e) {
+      return 'sess_' + Math.random().toString(36).substring(2, 12) + Date.now().toString(36);
     }
-  });
-
-  req.on('error', (e) => log("❌ Upload Failed", e.message));
-  req.write(JSON.stringify(finalPayload));
-  req.end();
-}
-
-// --- MONKEY PATCH ---
-
-function applyPatch() {
-  if (isPatched) return;
-
-  // --- A. PATCH FETCH (Next.js / OpenAI) ---
-  if (GLOBAL_OBJ.fetch) {
-    GLOBAL_OBJ.__dropout_original_fetch__ = GLOBAL_OBJ.fetch;
-    GLOBAL_OBJ.fetch = async function (input, init) {
-      const url = typeof input === 'string' ? input : (input && input.url);
-
-      // Check body string safely
-      let bodyStr = "";
-      if (init && init.body) {
-        try { bodyStr = typeof init.body === 'string' ? init.body : JSON.stringify(init.body); } catch (e) { }
-      }
-
-      // Guard
-      if (!isAiRequest(url, bodyStr)) {
-        return GLOBAL_OBJ.__dropout_original_fetch__(input, init);
-      }
-
-      log(`⚡ [FETCH] Intercepting: ${url}`);
-
-      const start = Date.now();
+  }
 
-      // Calculate Turn
-      let activeTurn;
-      if (lastTurnConfirmed) {
-        activeTurn = turnIndex++;
-        lastTurnConfirmed = false;
-      } else {
-        activeTurn = turnIndex > 0 ? turnIndex - 1 : 0;
-      }
+  hash(text) {
+    if (!text) return null;
+    try {
+      return crypto.createHash('sha256').update(text.toLowerCase().trim()).digest('hex');
+    } catch (e) { return 'hash_err'; }
+  }
 
-      const { provider, model } = normalize(url, bodyStr);
-      const pHash = hash(bodyStr);
-
-      // Emit Request (User)
-      emit({
-        session_id: GLOBAL_OBJ.__dropout_session_id__,
-        turn_index: activeTurn,
-        direction: 'user_to_ai',
-        turn_role: 'user',
-        provider,
-        model,
-        content: activeConfig.privacyMode === 'full' ? bodyStr : null,
-        content_hash: pHash,
-        metadata_flags: {
-          retry_like: pHash === lastPromptHash ? 1 : 0,
-          method: 'FETCH',
-          url: url
-        }
-      });
-      lastPromptHash = pHash;
+  normalize(url, body) {
+    let provider = 'custom';
+    let model = 'unknown';
+
+    if (url) {
+      const u = url.toLowerCase();
+      if (u.includes('openai')) provider = 'openai';
+      else if (u.includes('anthropic')) provider = 'anthropic';
+      else if (u.includes('google') || u.includes('gemini') || u.includes('aiplatform')) provider = 'google';
+      else if (u.includes('groq')) provider = 'groq';
+      else if (u.includes('mistral')) provider = 'mistral';
+      else if (u.includes('cohere')) provider = 'cohere';
+      else if (u.includes('localhost') || u.includes('127.0.0.1')) provider = 'local';
+    }
 
-      // Actual Call
-      let response;
+    if (body) {
       try {
-        response = await GLOBAL_OBJ.__dropout_original_fetch__(input, init);
-      } catch (err) { throw err; }
+        const parsed = typeof body === 'string' ? JSON.parse(body) : body;
+        if (parsed.model) model = parsed.model;
+      } catch (e) { }
+    }
+    return { provider, model };
+  }
 
-      const latency = Date.now() - start;
+  isAiRequest(url, bodyString) {
+    if (!url) return false;
+    // Guard: Infinite Loop Prevention
+    if (url.includes(this.captureEndpoint)) return false;
 
-      // Emit Response (Assistant)
-      try {
-        const cloned = response.clone();
-        const oText = await cloned.text();
-        const oHash = hash(oText);
-
-        emit({
-          session_id: GLOBAL_OBJ.__dropout_session_id__,
-          turn_index: activeTurn,
-          direction: 'ai_to_user',
-          turn_role: 'assistant',
-          latency_ms: latency,
-          provider,
-          model,
-          content: activeConfig.privacyMode === 'full' ? oText : null,
-          content_hash: oHash,
-          metadata_flags: {
-            non_adaptive_response: oHash === lastResponseHash ? 1 : 0,
-            turn_boundary_confirmed: 1,
-            status: response.status
-          }
-        });
-        lastResponseHash = oHash;
-      } catch (e) { log("⚠️ Failed to read response body"); }
+    // 1. Check Known Domains
+    if (KNOWN_AI_DOMAINS.some(d => url.includes(d))) return true;
 
-      lastTurnConfirmed = true;
-      return response;
-    };
-    log("✅ Patch Applied: global.fetch");
+    // 2. Heuristic Check (for custom endpoints)
+    if (bodyString) {
+      return (bodyString.includes('"model"') || bodyString.includes('"messages"')) &&
+        (bodyString.includes('"user"') || bodyString.includes('"prompt"'));
+    }
+    return false;
   }
 
-  // --- B. PATCH HTTP/HTTPS (Axios / Genkit / Node SDKs) ---
-  const patchNodeRequest = (module, moduleName) => {
-    const originalRequest = module.request;
-    module.request = function (...args) {
-      // Resolve URL
-      let url;
-      let options;
-      if (typeof args[0] === 'string') {
-        url = args[0];
-        options = args[1] || {};
-      } else {
-        options = args[0] || {};
-        const protocol = options.protocol || 'https:';
-        const host = options.hostname || options.host || 'localhost';
-        const path = options.path || '/';
-        url = `${protocol}//${host}${path}`;
-      }
-
-      // We can't check body here yet, so we rely on URL
-      if (!isAiRequest(url, null)) {
-        return originalRequest.apply(this, args);
-      }
+  // --- EMITTER ---
+  emit(payload) {
+    // 1. Guard
+    if (!this.apiKey || !this.projectId) return;
+
+    // 2. Construct Payload
+    const finalPayload = {
+      ...payload,
+      project_id: this.projectId,
+      content_blob: payload.content, // Map for new schema
+      content: payload.content,      // Map for old schema (fallback)
+      received_at: new Date().toISOString()
+    };
 
-      log(`⚡ [${moduleName.toUpperCase()}] Intercepting: ${url}`);
+    this.log(`🚀 Sending Capture (${payload.turn_role})`);
 
-      const start = Date.now();
-      const clientRequest = originalRequest.apply(this, args);
+    // 3. Fire and Forget using HTTPS (Bypassing our own patches via pure request)
+    // We use a clean request to avoid triggering our own hooks if we used global.fetch
+    const req = https.request(this.captureEndpoint, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'x-dropout-key': this.apiKey
+      }
+    });
 
-      // Capture Request Body
-      const reqChunks = [];
-      const originalWrite = clientRequest.write;
-      const originalEnd = clientRequest.end;
+    req.on('error', (e) => this.log("❌ Upload Failed", e.message));
+    req.write(JSON.stringify(finalPayload));
+    req.end();
+  }
 
-      clientRequest.write = function (...writeArgs) {
-        if (writeArgs[0]) reqChunks.push(Buffer.from(writeArgs[0]));
-        return originalWrite.apply(this, writeArgs);
-      };
+  // --- CORE PATCHING ---
+  patchNetwork() {
+    const _this = this; // Capture instance
 
-      clientRequest.end = function (...endArgs) {
-        if (endArgs[0]) reqChunks.push(Buffer.from(endArgs[0]));
+    // --- A. PATCH FETCH ---
+    if (global.fetch) {
+      const originalFetch = global.fetch;
+      global.fetch = async function (input, init) {
+        const url = typeof input === 'string' ? input : input?.url;
 
-        // Request Complete - Emit User Turn
-        const reqBody = Buffer.concat(reqChunks).toString('utf8');
-        const { provider, model } = normalize(url, reqBody);
-        const pHash = hash(reqBody);
+        // Safe Body Check
+        let bodyStr = "";
+        if (init && init.body) {
+          try { bodyStr = typeof init.body === 'string' ? init.body : JSON.stringify(init.body); } catch (e) { }
+        }
 
-        let activeTurn;
-        if (lastTurnConfirmed) {
-          activeTurn = turnIndex++;
-          lastTurnConfirmed = false;
-        } else {
-          activeTurn = turnIndex > 0 ? turnIndex - 1 : 0;
+        // Guard
+        if (!_this.isAiRequest(url, bodyStr)) {
+          return originalFetch.apply(this, arguments);
         }
 
-        // Save state on request object for response handler
-        clientRequest._dropout_turn = activeTurn;
-        clientRequest._dropout_provider = provider;
-        clientRequest._dropout_model = model;
+        _this.log(`⚡ [FETCH] Intercepting: ${url}`);
+        const start = Date.now();
 
-        emit({
-          session_id: GLOBAL_OBJ.__dropout_session_id__,
-          turn_index: activeTurn,
-          direction: 'user_to_ai',
+        // Emit Request
+        const { provider, model } = _this.normalize(url, bodyStr);
+        _this.emit({
+          session_id: global.__dropout_session_id__,
+          turn_index: 0,
           turn_role: 'user',
           provider,
           model,
-          content: activeConfig.privacyMode === 'full' ? reqBody : null,
-          content_hash: pHash,
-          metadata_flags: { method: moduleName.toUpperCase(), url: url }
+          content: _this.privacy === 'full' ? bodyStr : null,
+          metadata_flags: { method: 'FETCH', url: url }
         });
-        lastPromptHash = pHash;
 
-        return originalEnd.apply(this, endArgs);
-      };
-
-      // Capture Response
-      clientRequest.on('response', (res) => {
-        const resChunks = [];
-        res.on('data', (chunk) => resChunks.push(chunk));
-        res.on('end', () => {
-          const resBody = Buffer.concat(resChunks).toString('utf8');
-          const latency = Date.now() - start;
-          const oHash = hash(resBody);
-
-          emit({
-            session_id: GLOBAL_OBJ.__dropout_session_id__,
-            turn_index: clientRequest._dropout_turn || 0,
-            direction: 'ai_to_user',
+        // Execute
+        let response;
+        try {
+          response = await originalFetch.apply(this, arguments);
+        } catch (err) { throw err; }
+
+        const latency = Date.now() - start;
+
+        // Emit Response
+        try {
+          const cloned = response.clone();
+          const oText = await cloned.text();
+          _this.emit({
+            session_id: global.__dropout_session_id__,
+            turn_index: 0,
             turn_role: 'assistant',
             latency_ms: latency,
-            provider: clientRequest._dropout_provider,
-            model: clientRequest._dropout_model,
-            content: activeConfig.privacyMode === 'full' ? resBody : null,
-            content_hash: oHash,
-            metadata_flags: { status: res.statusCode }
+            provider,
+            model,
+            content: _this.privacy === 'full' ? oText : null,
+            metadata_flags: { status: response.status }
           });
+        } catch (e) { _this.log("⚠️ Failed to read response body"); }
 
-          lastResponseHash = oHash;
-          lastTurnConfirmed = true;
-        });
-      });
+        return response;
+      };
+      this.log("✅ Patch Applied: global.fetch");
+    }
 
-      return clientRequest;
-    };
-  };
+    // --- B. PATCH NODE HTTP/HTTPS ---
+    const patchNodeRequest = (module, moduleName) => {
+      const originalRequest = module.request;
+      module.request = function (...args) {
+        let url;
+        // Argument Resolution (url, options, callback) or (options, callback)
+        if (typeof args[0] === 'string') {
+          url = args[0];
+        } else {
+          const opts = args[0] || {};
+          const protocol = opts.protocol || (moduleName === 'https' ? 'https:' : 'http:');
+          const host = opts.hostname || opts.host || 'localhost';
+          const path = opts.path || '/';
+          url = `${protocol}//${host}${path}`;
+        }
 
-  patchNodeRequest(https, 'https');
-  patchNodeRequest(http, 'http');
-  log("✅ Patch Applied: http/https");
+        if (!_this.isAiRequest(url, null)) {
+          return originalRequest.apply(this, args);
+        }
 
-  isPatched = true;
-}
+        _this.log(`⚡ [${moduleName.toUpperCase()}] Intercepting: ${url}`);
+        const start = Date.now();
 
-// --- MAIN CLASS EXPORT ---
+        const clientRequest = originalRequest.apply(this, args);
 
-class Dropout {
-  constructor(config = {}) {
-    if (instance) return instance;
+        // Capture Buffers
+        const reqChunks = [];
+        const originalWrite = clientRequest.write;
+        const originalEnd = clientRequest.end;
 
-    if (!config.apiKey || !config.projectId) {
-      console.warn("[Dropout] Missing apiKey or projectId. Tracking disabled.");
-      return;
-    }
+        // SAFE WRITE PATCH
+        clientRequest.write = function (...writeArgs) {
+          const chunk = writeArgs[0];
+          // Only buffer if it's data (String or Buffer), ignore objects/callbacks
+          if (chunk && (typeof chunk === 'string' || Buffer.isBuffer(chunk))) {
+            reqChunks.push(Buffer.from(chunk));
+          }
+          return originalWrite.apply(this, writeArgs);
+        };
+
+        // SAFE END PATCH
+        clientRequest.end = function (...endArgs) {
+          const chunk = endArgs[0];
+          // Only buffer if it's data
+          if (chunk && (typeof chunk === 'string' || Buffer.isBuffer(chunk))) {
+            reqChunks.push(Buffer.from(chunk));
+          }
 
-    // Update Global Configuration
-    activeConfig.apiKey = config.apiKey;
-    activeConfig.projectId = config.projectId;
-    if (config.privacyMode) activeConfig.privacyMode = config.privacyMode;
-    activeConfig.debug = config.debug || false;
+          // Emit Request
+          const reqBody = Buffer.concat(reqChunks).toString('utf8');
+          const { provider, model } = _this.normalize(url, reqBody);
+
+          // Attach state to request for response handler
+          clientRequest._dropout_meta = { provider, model };
+
+          _this.emit({
+            session_id: global.__dropout_session_id__,
+            turn_index: 0,
+            turn_role: 'user',
+            provider,
+            model,
+            content: _this.privacy === 'full' ? reqBody : null,
+            metadata_flags: { method: moduleName.toUpperCase(), url: url }
+          });
 
-    // Initialize Identity
-    if (!GLOBAL_OBJ.__dropout_session_id__) {
-      GLOBAL_OBJ.__dropout_session_id__ = generateSessionId();
-    }
+          return originalEnd.apply(this, endArgs);
+        };
+
+        // Capture Response
+        clientRequest.on('response', (res) => {
+          const resChunks = [];
+          res.on('data', (chunk) => resChunks.push(chunk));
+          res.on('end', () => {
+            const resBody = Buffer.concat(resChunks).toString('utf8');
+            const latency = Date.now() - start;
+            const meta = clientRequest._dropout_meta || {};
+
+            _this.emit({
+              session_id: global.__dropout_session_id__,
+              turn_index: 0,
+              turn_role: 'assistant',
+              latency_ms: latency,
+              provider: meta.provider,
+              model: meta.model,
+              content: _this.privacy === 'full' ? resBody : null,
+              metadata_flags: { status: res.statusCode }
+            });
+          });
+        });
+
+        return clientRequest;
+      };
+    };
 
-    // Apply the patch
-    if (activeConfig.debug) console.log(`[Dropout] 🟢 Initialized for Project: ${config.projectId}`);
-    applyPatch();
-    instance = this;
+    patchNodeRequest(https, 'https');
+    patchNodeRequest(http, 'http');
+    this.log("✅ Patch Applied: http/https");
   }
 }
 
-// Auto-Start
+// Auto-Start (Node Preload)
 if (process.env.DROPOUT_PROJECT_ID && process.env.DROPOUT_API_KEY) {
   new Dropout({
     projectId: process.env.DROPOUT_PROJECT_ID,