@angelitosystems/nest-devtools 1.0.8 → 1.0.9

package/dist/index.cjs

@@ -76,25 +76,6 @@ function recordSpan(requestId, layer, label, options) {
 }
 
 // src/instrumentation/http.ts
-var REQUEST_HEADERS = [
-  "content-type",
-  "content-length",
-  "accept",
-  "origin",
-  "referer",
-  "user-agent",
-  "x-forwarded-for",
-  "x-request-id",
-  "x-forwarded-proto"
-];
-var RESPONSE_HEADERS = [
-  "content-type",
-  "content-length",
-  "location",
-  "set-cookie",
-  "x-request-id",
-  "x-response-time"
-];
 var WRAPPED = /* @__PURE__ */ Symbol("nest-devtools.wrapped");
 var HttpInstrumentation = class {
   constructor(ctx) {
@@ -192,6 +173,7 @@ var HttpInstrumentation = class {
     };
     spans.push(middlewareSpan);
     const capturedBody = this.ctx.config.capture.requests && this.ctx.config.maxPayloadBytes > 0 ? captureRequestPreview(req, this.redactor) : void 0;
+    const responseCapture = createResponseCapture(res, this.ctx.config.maxPayloadBytes, this.redactor);
     const onFinished = () => {
       const finishedAt = Date.now();
       middlewareSpan.duration = Math.max(0, finishedAt - middlewareSpan.startedAt);
@@ -205,7 +187,8 @@ var HttpInstrumentation = class {
       };
       spans.push(responseSpan);
       const responseHeaders = this.collectResponseHeaders(res);
-      const responsePreview = this.ctx.config.capture.requests && this.ctx.config.maxPayloadBytes > 0 ? captureResponsePreview(res, this.redactor) : void 0;
+      const responsePreview = responseCapture.read();
+      const requestBody = this.ctx.config.capture.requests && this.ctx.config.maxPayloadBytes > 0 ? captureRequestPreview(req, this.redactor) : void 0;
       const errored = res.statusCode >= 500;
       const completed = {
         requestId,
@@ -217,10 +200,11 @@ var HttpInstrumentation = class {
         duration: finishedAt - startedAt,
         startedAt,
         timeline: spans,
+        requestHeaders: started.headers,
         query: parseQuery(rawUrl),
         headers: responseHeaders,
         responsePreview,
-        requestBody: capturedBody,
+        requestBody: requestBody ?? capturedBody,
         errored
       };
       emit("request.completed", completed);
@@ -245,8 +229,7 @@ var HttpInstrumentation = class {
   }
   collectRequestHeaders(req) {
     const out = {};
-    for (const key of REQUEST_HEADERS) {
-      const value = req.headers[key];
+    for (const [key, value] of Object.entries(req.headers)) {
       if (value === void 0) continue;
       out[key] = this.redactor.isSensitive(key) ? "[REDACTED]" : asString(value).slice(0, 200);
     }
@@ -254,12 +237,9 @@ var HttpInstrumentation = class {
   }
   collectResponseHeaders(res) {
     const out = {};
-    const headers = res._headers;
-    if (!headers || typeof headers !== "object") return out;
+    const headers = typeof res.getHeaders === "function" ? res.getHeaders() : {};
     for (const rawKey of Object.keys(headers)) {
       const key = rawKey.toLowerCase();
-      const ok = RESPONSE_HEADERS.some((h) => h.toLowerCase() === key);
-      if (!ok) continue;
       const raw = headers[rawKey] ?? "";
       const value = String(raw);
       if (value === "") continue;
@@ -336,16 +316,38 @@ function captureRequestPreview(req, redactor) {
     return void 0;
   }
 }
-function captureResponsePreview(res, redactor) {
-  try {
-    const resLike = res;
-    const body = resLike.body;
-    if (body === void 0 || body === null) return void 0;
-    if (typeof body === "string") return redactor.redactString(body);
-    return redactor.serialize(body);
-  } catch {
-    return void 0;
-  }
+function createResponseCapture(res, maxBytes, redactor) {
+  if (maxBytes <= 0) return { read: () => void 0 };
+  const chunks = [];
+  let size = 0;
+  const originalWrite = res.write.bind(res);
+  const originalEnd = res.end.bind(res);
+  const capture2 = (chunk) => {
+    if (chunk === void 0 || chunk === null || size >= maxBytes) return;
+    const buffer = Buffer.isBuffer(chunk) ? chunk : Buffer.from(String(chunk));
+    const remaining = maxBytes - size;
+    chunks.push(buffer.subarray(0, remaining));
+    size += Math.min(buffer.length, remaining);
+  };
+  res.write = (chunk, ...args) => {
+    capture2(chunk);
+    return originalWrite(chunk, ...args);
+  };
+  res.end = (chunk, ...args) => {
+    capture2(chunk);
+    return originalEnd(chunk, ...args);
+  };
+  res.once("finish", () => {
+    res.write = originalWrite;
+    res.end = originalEnd;
+  });
+  return {
+    read: () => {
+      if (chunks.length === 0) return void 0;
+      const raw = Buffer.concat(chunks).toString("utf8");
+      return redactor.redactString(raw);
+    }
+  };
 }
 
 // src/instrumentation/console.ts
@@ -662,6 +664,13 @@ var AppExplorer = class {
   extractMembers(node) {
     const controllers = [];
     const providers = [];
+    const controllersMap = node.controllers;
+    if (controllersMap instanceof Map) {
+      for (const [id, wrapper] of controllersMap) {
+        const member = this.memberFromWrapper(id, wrapper, "controller");
+        if (member) controllers.push(member);
+      }
+    }
     const providersMap = node.providers;
     if (providersMap instanceof Map) {
       for (const [id, wrapper] of providersMap) {
@@ -676,10 +685,10 @@ var AppExplorer = class {
     }
     return { controllers, providers };
   }
-  memberFromWrapper(id, wrapper) {
+  memberFromWrapper(id, wrapper, forcedType) {
     const w = wrapper;
     const name = w?.metatype?.name ?? (typeof id === "string" ? id.replace(/^[A-Z_0-9]+:/, "") : "unknown");
-    const type = classify(name, w?.instance);
+    const type = forcedType ?? classify(name, w?.instance);
     const routes = type === "controller" ? extractControllerRoutes(w?.metatype) : void 0;
     return { name, type, routes };
   }
@@ -836,6 +845,7 @@ var WebsocketInstrumentation = class {
       }
       if (method === "handleMessage" || method === "handleEvent") {
         const payload = this.extractPayload(args);
+        this.emitMessage("received", instance, args, payload, requestId, started);
         recordSpan(requestId, "websocket", `ws.receive:${this.estimateEvent(args)}`, {
           detail: `gateway=${instance.constructor?.name} payload=${this.redactor.serialize(payload)}`
         });
@@ -844,6 +854,7 @@ var WebsocketInstrumentation = class {
         const result = await Promise.resolve(original.call(instance, ...args));
         if (method === "handleMessage" || method === "handleEvent") {
           const payload = this.extractPayload(args);
+          this.emitMessage("sent", instance, args, payload, requestId, started);
           recordSpan(requestId, "websocket", `ws.send:${this.estimateEvent(args)}`, {
             detail: `gateway=${instance.constructor?.name} payload=${this.redactor.serialize(payload)}`,
             status: "ok"
@@ -861,6 +872,21 @@ var WebsocketInstrumentation = class {
       }
     };
   }
+  emitMessage(direction, instance, args, payload, requestId, startedAt) {
+    const preview = this.redactor.redact(payload);
+    const serialized = this.redactor.serialize(payload);
+    emit("websocket.message", {
+      projectId: this.projectId,
+      gateway: instance.constructor?.name ?? this.estimateGatewayName(instance),
+      event: this.estimateEvent(args),
+      direction,
+      payloadSize: Buffer.byteLength(serialized, "utf8"),
+      payloadPreview: preview,
+      requestId,
+      duration: Date.now() - startedAt,
+      timestamp: Date.now()
+    });
+  }
   extractPayload(args) {
     for (const arg of args) {
       if (arg && typeof arg === "object" && !Buffer.isBuffer(arg) && !ArrayBuffer.isView(arg)) {
@@ -938,6 +964,23 @@ var DatabaseInstrumentation = class {
       if (!app) return false;
       const dataSource = this.resolveTypeOrmDataSource(app);
       if (!dataSource) return false;
+      if (typeof dataSource.query === "function") {
+        const originalQuery = dataSource.query.bind(dataSource);
+        dataSource.__nestDevToolsOriginalQuery = originalQuery;
+        dataSource.query = async (query, parameters) => {
+          const started = Date.now();
+          this.startQuery(String(query), parameters ?? []);
+          try {
+            const result = await originalQuery(query, parameters);
+            this.endQuery(String(query), parameters ?? [], Date.now() - started);
+            return result;
+          } catch (error) {
+            this.endQuery(String(query), parameters ?? [], Date.now() - started, error);
+            throw error;
+          }
+        };
+        return true;
+      }
       const listener = {
         beforeQuery: (query, parameters) => {
           this.startQuery(query, parameters);
@@ -990,6 +1033,22 @@ var DatabaseInstrumentation = class {
       if (!app) return false;
       const prisma = this.resolvePrismaClient(app);
       if (!prisma) return false;
+      if (typeof prisma.$use === "function") {
+        prisma.$use(async (params, next) => {
+          const label = `${params.model ?? "prisma"}.${params.action ?? "query"}`;
+          const started = Date.now();
+          this.startQuery(label, [params.args]);
+          try {
+            const result = await next(params);
+            this.endQuery(label, [params.args], Date.now() - started);
+            return result;
+          } catch (error) {
+            this.endQuery(label, [params.args], Date.now() - started, error);
+            throw error;
+          }
+        });
+        return true;
+      }
       const original = prisma.$queryRaw ?? prisma.$executeRaw ?? null;
       if (!original) return false;
       const wrapped = (query, parameters) => {
@@ -1133,8 +1192,8 @@ var DatabaseInstrumentation = class {
   }
   resolveTypeOrmDataSource(app) {
     try {
-      const dataSource = app.get(require("@nestjs/typeorm").TypeOrmModule)?.options?.DataSource ?? null;
-      if (dataSource && typeof dataSource.query === "function") return dataSource;
+      const dataSource = this.findProvider(app, (value) => typeof value?.query === "function" && typeof value?.manager === "object");
+      if (dataSource) return dataSource;
       return null;
     } catch {
       return null;
@@ -1142,7 +1201,7 @@ var DatabaseInstrumentation = class {
   }
   resolvePrismaClient(app) {
     try {
-      const prisma = app.get("PrismaService") ?? app.get("PrismaClient") ?? app.get("prisma") ?? null;
+      const prisma = this.safeGet(app, "PrismaService") ?? this.safeGet(app, "PrismaClient") ?? this.safeGet(app, "prisma") ?? this.findProvider(app, (value) => typeof value?.$queryRaw === "function");
       if (prisma && typeof prisma.$queryRaw === "function") return prisma;
       return null;
     } catch {
@@ -1151,13 +1210,34 @@ var DatabaseInstrumentation = class {
   }
   resolveMongoose(app) {
     try {
-      const mongoose = app.get("MongooseService") ?? app.get("Mongoose") ?? app.get("mongoose") ?? null;
+      const mongoose = this.safeGet(app, "MongooseService") ?? this.safeGet(app, "Mongoose") ?? this.safeGet(app, "mongoose") ?? this.findProvider(app, (value) => typeof value?.plugin === "function");
       if (mongoose && typeof mongoose.plugin === "function") return mongoose;
       return null;
     } catch {
       return null;
     }
   }
+  safeGet(app, token) {
+    try {
+      return app?.get?.(token, { strict: false }) ?? null;
+    } catch {
+      return null;
+    }
+  }
+  findProvider(app, predicate) {
+    try {
+      const modules = app?.container?.getModules?.() ?? /* @__PURE__ */ new Map();
+      for (const module2 of modules.values()) {
+        for (const wrapper of module2.providers?.values?.() ?? []) {
+          const instance = wrapper?.instance;
+          if (instance && predicate(instance)) return instance;
+        }
+      }
+      return null;
+    } catch {
+      return null;
+    }
+  }
 };
 
 // src/instrumentation/queue.ts
@@ -1330,7 +1410,7 @@ function detectNestJsVersion() {
 }
 
 // src/version.ts
-var SDK_VERSION = "0.1.0";
+var SDK_VERSION = "1.0.9";
 
 // src/banner.ts
 var RESET = "\x1B[0m";