trickle-observe 0.2.95 → 0.2.96

package/dist/db-observer.d.ts

@@ -21,3 +21,13 @@ export declare function patchMysql2(mysqlModule: any, debug: boolean): void;
  * Patch better-sqlite3 to capture queries.
  */
 export declare function patchBetterSqlite3(dbConstructor: any, debug: boolean): void;
+/**
+ * Patch ioredis to capture Redis commands.
+ * Called from observe-register when ioredis is required.
+ */
+export declare function patchIoredis(ioredisModule: any, debug: boolean): void;
+/**
+ * Patch mongoose to capture MongoDB operations.
+ * Called from observe-register when mongoose is required.
+ */
+export declare function patchMongoose(mongooseModule: any, debug: boolean): void;

package/dist/db-observer.js

@@ -46,6 +46,8 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.patchPg = patchPg;
 exports.patchMysql2 = patchMysql2;
 exports.patchBetterSqlite3 = patchBetterSqlite3;
+exports.patchIoredis = patchIoredis;
+exports.patchMongoose = patchMongoose;
 const fs = __importStar(require("fs"));
 const path = __importStar(require("path"));
 let queriesFile = null;
@@ -290,3 +292,99 @@ function patchBetterSqlite3(dbConstructor, debug) {
     if (debug)
         console.log('[trickle/db] SQLite query tracing enabled');
 }
+/**
+ * Patch ioredis to capture Redis commands.
+ * Called from observe-register when ioredis is required.
+ */
+function patchIoredis(ioredisModule, debug) {
+    debugMode = debug;
+    const RedisClass = ioredisModule.default || ioredisModule;
+    const proto = RedisClass.prototype;
+    if (!proto || proto.sendCommand?.__trickle_patched)
+        return;
+    const origSendCommand = proto.sendCommand;
+    if (!origSendCommand)
+        return;
+    proto.sendCommand = function patchedSendCommand(command, ...rest) {
+        const cmdName = command?.name || 'UNKNOWN';
+        const cmdArgs = (command?.args || []).slice(0, 3).map((a) => typeof a === 'string' ? (a.length > 50 ? a.substring(0, 50) + '...' : a) : String(a).substring(0, 50));
+        const queryStr = `${cmdName.toUpperCase()} ${cmdArgs.join(' ')}`.trim();
+        const startTime = performance.now();
+        const result = origSendCommand.call(this, command, ...rest);
+        // ioredis returns a Promise
+        if (result && typeof result.then === 'function') {
+            result.then(() => {
+                writeQuery({
+                    kind: 'query', query: queryStr.substring(0, MAX_QUERY_LENGTH),
+                    durationMs: Math.round((performance.now() - startTime) * 100) / 100,
+                    rowCount: 1, timestamp: Date.now(),
+                });
+            }, (err) => {
+                writeQuery({
+                    kind: 'query', query: queryStr.substring(0, MAX_QUERY_LENGTH),
+                    durationMs: Math.round((performance.now() - startTime) * 100) / 100,
+                    rowCount: 0, error: err?.message?.substring(0, 200), timestamp: Date.now(),
+                });
+            });
+        }
+        return result;
+    };
+    proto.sendCommand.__trickle_patched = true;
+    if (debug)
+        console.log('[trickle/db] Redis (ioredis) query tracing enabled');
+}
+/**
+ * Patch mongoose to capture MongoDB operations.
+ * Called from observe-register when mongoose is required.
+ */
+function patchMongoose(mongooseModule, debug) {
+    debugMode = debug;
+    const Model = mongooseModule.Model;
+    if (!Model || Model.__trickle_patched)
+        return;
+    const methodsToWrap = [
+        'find', 'findOne', 'findById', 'findOneAndUpdate', 'findOneAndDelete',
+        'create', 'insertMany', 'updateOne', 'updateMany',
+        'deleteOne', 'deleteMany', 'countDocuments', 'aggregate',
+    ];
+    for (const method of methodsToWrap) {
+        const orig = Model[method];
+        if (!orig || orig.__trickle_patched)
+            continue;
+        Model[method] = function patchedMethod(...args) {
+            const collName = this.modelName || this.collection?.name || '?';
+            let filterStr = '';
+            if (args[0] && typeof args[0] === 'object') {
+                try {
+                    filterStr = ' ' + JSON.stringify(args[0]).substring(0, 200);
+                }
+                catch { }
+            }
+            const queryStr = `db.${collName}.${method}(${filterStr.trim()})`;
+            const startTime = performance.now();
+            const result = orig.apply(this, args);
+            // Mongoose methods return Query objects (thenables) or Promises
+            if (result && typeof result.then === 'function') {
+                result.then((res) => {
+                    const durationMs = Math.round((performance.now() - startTime) * 100) / 100;
+                    const rowCount = Array.isArray(res) ? res.length : (res ? 1 : 0);
+                    writeQuery({
+                        kind: 'query', query: queryStr.substring(0, MAX_QUERY_LENGTH),
+                        durationMs, rowCount, timestamp: Date.now(),
+                    });
+                }, (err) => {
+                    writeQuery({
+                        kind: 'query', query: queryStr.substring(0, MAX_QUERY_LENGTH),
+                        durationMs: Math.round((performance.now() - startTime) * 100) / 100,
+                        rowCount: 0, error: err?.message?.substring(0, 200), timestamp: Date.now(),
+                    });
+                });
+            }
+            return result;
+        };
+        Model[method].__trickle_patched = true;
+    }
+    Model.__trickle_patched = true;
+    if (debug)
+        console.log('[trickle/db] MongoDB (mongoose) query tracing enabled');
+}

package/dist/observe-register.js

@@ -1148,6 +1148,24 @@ if (enabled) {
             }
             catch { /* not critical */ }
         }
+        // Redis (ioredis)
+        if (request === 'ioredis' && !expressPatched.has('ioredis')) {
+            expressPatched.add('ioredis');
+            try {
+                const { patchIoredis } = require(path_1.default.join(__dirname, 'db-observer.js'));
+                patchIoredis(exports, debug);
+            }
+            catch { /* not critical */ }
+        }
+        // MongoDB (mongoose)
+        if (request === 'mongoose' && !expressPatched.has('mongoose')) {
+            expressPatched.add('mongoose');
+            try {
+                const { patchMongoose } = require(path_1.default.join(__dirname, 'db-observer.js'));
+                patchMongoose(exports, debug);
+            }
+            catch { /* not critical */ }
+        }
         // Resolve to absolute path for dedup — do this FIRST since bundlers like
         // tsx/esbuild may use path aliases (e.g., @config/env) that don't start
         // with './' or '/'. We need the resolved path to decide if it's user code.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "trickle-observe",
-  "version": "0.2.95",
+  "version": "0.2.96",
   "description": "Runtime type observability for JavaScript applications",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/db-observer.ts

@@ -281,3 +281,113 @@ export function patchBetterSqlite3(dbConstructor: any, debug: boolean): void {
 
   if (debug) console.log('[trickle/db] SQLite query tracing enabled');
 }
+
+/**
+ * Patch ioredis to capture Redis commands.
+ * Called from observe-register when ioredis is required.
+ */
+export function patchIoredis(ioredisModule: any, debug: boolean): void {
+  debugMode = debug;
+
+  const RedisClass = ioredisModule.default || ioredisModule;
+  const proto = RedisClass.prototype;
+  if (!proto || (proto.sendCommand as any)?.__trickle_patched) return;
+
+  const origSendCommand = proto.sendCommand;
+  if (!origSendCommand) return;
+
+  proto.sendCommand = function patchedSendCommand(command: any, ...rest: any[]): any {
+    const cmdName = command?.name || 'UNKNOWN';
+    const cmdArgs = (command?.args || []).slice(0, 3).map((a: any) =>
+      typeof a === 'string' ? (a.length > 50 ? a.substring(0, 50) + '...' : a) : String(a).substring(0, 50)
+    );
+    const queryStr = `${cmdName.toUpperCase()} ${cmdArgs.join(' ')}`.trim();
+
+    const startTime = performance.now();
+    const result = origSendCommand.call(this, command, ...rest);
+
+    // ioredis returns a Promise
+    if (result && typeof result.then === 'function') {
+      result.then(
+        () => {
+          writeQuery({
+            kind: 'query', query: queryStr.substring(0, MAX_QUERY_LENGTH),
+            durationMs: Math.round((performance.now() - startTime) * 100) / 100,
+            rowCount: 1, timestamp: Date.now(),
+          });
+        },
+        (err: any) => {
+          writeQuery({
+            kind: 'query', query: queryStr.substring(0, MAX_QUERY_LENGTH),
+            durationMs: Math.round((performance.now() - startTime) * 100) / 100,
+            rowCount: 0, error: err?.message?.substring(0, 200), timestamp: Date.now(),
+          });
+        }
+      );
+    }
+    return result;
+  };
+  (proto.sendCommand as any).__trickle_patched = true;
+
+  if (debug) console.log('[trickle/db] Redis (ioredis) query tracing enabled');
+}
+
+/**
+ * Patch mongoose to capture MongoDB operations.
+ * Called from observe-register when mongoose is required.
+ */
+export function patchMongoose(mongooseModule: any, debug: boolean): void {
+  debugMode = debug;
+
+  const Model = mongooseModule.Model;
+  if (!Model || (Model as any).__trickle_patched) return;
+
+  const methodsToWrap = [
+    'find', 'findOne', 'findById', 'findOneAndUpdate', 'findOneAndDelete',
+    'create', 'insertMany', 'updateOne', 'updateMany',
+    'deleteOne', 'deleteMany', 'countDocuments', 'aggregate',
+  ];
+
+  for (const method of methodsToWrap) {
+    const orig = Model[method];
+    if (!orig || (orig as any).__trickle_patched) continue;
+
+    Model[method] = function patchedMethod(this: any, ...args: any[]): any {
+      const collName = this.modelName || this.collection?.name || '?';
+      let filterStr = '';
+      if (args[0] && typeof args[0] === 'object') {
+        try { filterStr = ' ' + JSON.stringify(args[0]).substring(0, 200); } catch {}
+      }
+      const queryStr = `db.${collName}.${method}(${filterStr.trim()})`;
+
+      const startTime = performance.now();
+      const result = orig.apply(this, args);
+
+      // Mongoose methods return Query objects (thenables) or Promises
+      if (result && typeof result.then === 'function') {
+        result.then(
+          (res: any) => {
+            const durationMs = Math.round((performance.now() - startTime) * 100) / 100;
+            const rowCount = Array.isArray(res) ? res.length : (res ? 1 : 0);
+            writeQuery({
+              kind: 'query', query: queryStr.substring(0, MAX_QUERY_LENGTH),
+              durationMs, rowCount, timestamp: Date.now(),
+            });
+          },
+          (err: any) => {
+            writeQuery({
+              kind: 'query', query: queryStr.substring(0, MAX_QUERY_LENGTH),
+              durationMs: Math.round((performance.now() - startTime) * 100) / 100,
+              rowCount: 0, error: err?.message?.substring(0, 200), timestamp: Date.now(),
+            });
+          }
+        );
+      }
+      return result;
+    };
+    (Model[method] as any).__trickle_patched = true;
+  }
+  (Model as any).__trickle_patched = true;
+
+  if (debug) console.log('[trickle/db] MongoDB (mongoose) query tracing enabled');
+}

package/src/observe-register.ts

@@ -1136,6 +1136,24 @@ if (enabled) {
       } catch { /* not critical */ }
     }
 
+    // Redis (ioredis)
+    if (request === 'ioredis' && !expressPatched.has('ioredis')) {
+      expressPatched.add('ioredis');
+      try {
+        const { patchIoredis } = require(path.join(__dirname, 'db-observer.js'));
+        patchIoredis(exports, debug);
+      } catch { /* not critical */ }
+    }
+
+    // MongoDB (mongoose)
+    if (request === 'mongoose' && !expressPatched.has('mongoose')) {
+      expressPatched.add('mongoose');
+      try {
+        const { patchMongoose } = require(path.join(__dirname, 'db-observer.js'));
+        patchMongoose(exports, debug);
+      } catch { /* not critical */ }
+    }
+
     // Resolve to absolute path for dedup — do this FIRST since bundlers like
     // tsx/esbuild may use path aliases (e.g., @config/env) that don't start
     // with './' or '/'. We need the resolved path to decide if it's user code.