@neat.is/core 0.5.1 → 0.5.3

package/dist/index.cjs

@@ -4673,14 +4673,21 @@ function engineFromImage(image) {
 // src/extract/databases/dotenv.ts
 var CONNECTION_KEYS = /* @__PURE__ */ new Set([
   "DATABASE_URL",
+  "DATABASE_URI",
   "DB_URL",
+  "DB_URI",
   "POSTGRES_URL",
+  "POSTGRES_URI",
   "POSTGRESQL_URL",
+  "POSTGRESQL_URI",
   "MYSQL_URL",
+  "MYSQL_URI",
+  "MONGODB_URL",
   "MONGODB_URI",
   "MONGO_URL",
   "MONGO_URI",
-  "REDIS_URL"
+  "REDIS_URL",
+  "REDIS_URI"
 ]);
 function parseDotenvLine(line) {
   const trimmed = line.trim();
@@ -7766,11 +7773,42 @@ async function readPackageJson(scanPath) {
   const raw = await import_node_fs26.promises.readFile(pkgPath, "utf8");
   return JSON.parse(raw);
 }
+var HOOK_WALK_SKIP_DIRS = /* @__PURE__ */ new Set([
+  "node_modules",
+  "dist",
+  "build",
+  "out",
+  "coverage",
+  "neat-out"
+]);
 async function findHookFiles(scanPath) {
-  const entries = await import_node_fs26.promises.readdir(scanPath);
-  return entries.filter(
-    (e) => (e.startsWith("instrumentation") || e.startsWith("otel-init")) && /\.(ts|js)$/.test(e)
-  ).sort();
+  const found = [];
+  const walk3 = async (dir) => {
+    const entries = await import_node_fs26.promises.readdir(dir, { withFileTypes: true }).catch(() => []);
+    for (const entry of entries) {
+      if (entry.isDirectory()) {
+        if (entry.name.startsWith(".") || HOOK_WALK_SKIP_DIRS.has(entry.name)) continue;
+        await walk3(import_node_path43.default.join(dir, entry.name));
+      } else if (entry.isFile()) {
+        if ((entry.name.startsWith("instrumentation") || entry.name.startsWith("otel-init")) && /\.(ts|js|cjs|mjs)$/.test(entry.name)) {
+          const rel = import_node_path43.default.relative(scanPath, import_node_path43.default.join(dir, entry.name));
+          found.push(rel.split(import_node_path43.default.sep).join("/"));
+        }
+      }
+    }
+  };
+  await walk3(scanPath);
+  return found.sort();
+}
+async function pickPrimaryHookFile(scanPath, hookFiles, snippet2) {
+  let fallback = null;
+  for (const file of hookFiles) {
+    const content = await import_node_fs26.promises.readFile(import_node_path43.default.join(scanPath, file), "utf8");
+    const patched = splicedContent(content, snippet2);
+    if (patched !== null) return { file, content, patched };
+    if (fallback === null) fallback = { file, content };
+  }
+  return { file: fallback.file, content: fallback.content, patched: null };
 }
 function extendLogPath() {
   return process.env.NEAT_EXTEND_LOG ?? import_node_path43.default.join(import_node_os2.default.homedir(), ".neat", "extend-log.ndjson");
@@ -7863,7 +7901,13 @@ async function applyExtension(ctx, args, options) {
       return { library: args.library, filesTouched: [], depsAdded: [], installOutput: "", alreadyApplied: true };
     }
   }
-  const primaryFile = hookFiles[0];
+  const primary = await pickPrimaryHookFile(ctx.scanPath, hookFiles, args.registration_snippet);
+  if (primary.patched === null) {
+    throw new Error(
+      `Could not find instrumentation insertion point in ${hookFiles.join(", ")}. Expected __INSTRUMENTATION_BLOCK__, instrumentations.push(, or new NodeSDK(.`
+    );
+  }
+  const primaryFile = primary.file;
   const primaryPath = import_node_path43.default.join(ctx.scanPath, primaryFile);
   const filesTouched = [];
   const depsAdded = [];
@@ -7875,14 +7919,7 @@ async function applyExtension(ctx, args, options) {
     filesTouched.push("package.json");
     depsAdded.push(`${args.instrumentation_package}@${args.version}`);
   }
-  const hookContent = await import_node_fs26.promises.readFile(primaryPath, "utf8");
-  const patched = splicedContent(hookContent, args.registration_snippet);
-  if (!patched) {
-    throw new Error(
-      `Could not find instrumentation insertion point in ${primaryFile}. Expected __INSTRUMENTATION_BLOCK__, instrumentations.push(, or new NodeSDK(.`
-    );
-  }
-  await import_node_fs26.promises.writeFile(primaryPath, patched, "utf8");
+  await import_node_fs26.promises.writeFile(primaryPath, primary.patched, "utf8");
   filesTouched.push(primaryFile);
   const cmd = await detectPackageManager(ctx.scanPath);
   const installer = options?.runInstall ?? runPackageManagerInstall;
@@ -7924,7 +7961,7 @@ async function dryRunExtension(ctx, args) {
       };
     }
   }
-  const primaryFile = hookFiles[0];
+  const primary = await pickPrimaryHookFile(ctx.scanPath, hookFiles, args.registration_snippet);
   const filesTouched = [];
   const depsToAdd = [];
   let packageJsonPatch = {};
@@ -7935,10 +7972,8 @@ async function dryRunExtension(ctx, args) {
     depsToAdd.push(`${args.instrumentation_package}@${args.version}`);
     filesTouched.push("package.json");
   }
-  const hookContent = await import_node_fs26.promises.readFile(import_node_path43.default.join(ctx.scanPath, primaryFile), "utf8");
-  const patched = splicedContent(hookContent, args.registration_snippet);
-  if (patched) {
-    filesTouched.push(primaryFile);
+  if (primary.patched !== null) {
+    filesTouched.push(primary.file);
     templatePatch = `+ ${args.registration_snippet}`;
   } else {
     templatePatch = "Could not find insertion point in hook file.";
@@ -9586,7 +9621,7 @@ function registerRoutes(scope, ctx) {
   });
 }
 async function buildApi(opts) {
-  const app = (0, import_fastify.default)({ logger: false });
+  const app = (0, import_fastify.default)({ logger: false, routerOptions: { maxParamLength: 1024 } });
   await app.register(import_cors.default, { origin: true });
   const env = readAuthEnv();
   const authToken = opts.authToken ?? env.authToken;
@@ -9771,8 +9806,8 @@ async function runConnectorPoll(connector, ctx, graph, resolveTarget) {
       line: callSite.line
     } : void 0;
     const calls = Math.trunc(signal.callCount);
-    if (calls < 1) continue;
-    const errors = Math.min(Math.max(Math.trunc(signal.errorCount), 0), calls);
+    if (!Number.isFinite(calls) || calls < 1) continue;
+    const errors = Number.isFinite(signal.errorCount) ? Math.min(Math.max(Math.trunc(signal.errorCount), 0), calls) : 0;
     let created = false;
     let ok = true;
     for (let i = 0; i < calls; i++) {
@@ -10219,7 +10254,10 @@ function targetFromRestPath(path51) {
 var ERROR_STATUS_THRESHOLD = 500;
 function mapEdgeLogRowsToSignals(rows) {
   const buckets2 = /* @__PURE__ */ new Map();
+  if (!Array.isArray(rows)) return [];
   for (const row of rows) {
+    if (!row || typeof row !== "object") continue;
+    if (typeof row.path !== "string" || typeof row.timestamp !== "string") continue;
     const target = targetFromRestPath(row.path);
     if (!target) continue;
     const key = `${target.targetKind}:${target.name}`;
@@ -10260,9 +10298,12 @@ function tableNameFromQueryText(query) {
 function diffPgStatStatementsToSignals(rows, previous, nowIso2) {
   const signals = [];
   const seen = /* @__PURE__ */ new Set();
+  if (!Array.isArray(rows)) return signals;
   for (const row of rows) {
-    seen.add(row.queryid);
+    if (!row || typeof row !== "object" || typeof row.queryid !== "string") continue;
     const calls = Number(row.calls);
+    if (!Number.isFinite(calls)) continue;
+    seen.add(row.queryid);
     const prior = previous.get(row.queryid);
     previous.set(row.queryid, { calls });
     if (!prior || calls < prior.calls) continue;
@@ -10586,7 +10627,12 @@ function upsertBucket(buckets2, key, isError, timestamp, build) {
 }
 function mapRailwayHttpLogsToSignals(entries, routeIndex) {
   const buckets2 = /* @__PURE__ */ new Map();
+  if (!Array.isArray(entries)) return [];
   for (const entry of entries) {
+    if (!entry || typeof entry !== "object") continue;
+    if (typeof entry.method !== "string" || typeof entry.path !== "string" || typeof entry.timestamp !== "string") {
+      continue;
+    }
     const method = entry.method.toUpperCase();
     const normalizedPath = normalizePathTemplate(entry.path);
     const match = findRailwayRoute(routeIndex, method, normalizedPath);
@@ -10625,8 +10671,11 @@ function mapRailwayHttpLogsToSignals(entries, routeIndex) {
 }
 function mapRailwayNetworkFlowLogsToSignals(entries) {
   const buckets2 = /* @__PURE__ */ new Map();
+  if (!Array.isArray(entries)) return [];
   for (const entry of entries) {
-    if (!entry.peerServiceId) continue;
+    if (!entry || typeof entry !== "object") continue;
+    if (typeof entry.peerServiceId !== "string" || entry.peerServiceId.length === 0) continue;
+    if (typeof entry.timestamp !== "string") continue;
     const isError = entry.dropCause !== null && entry.dropCause !== "";
     upsertBucket(buckets2, entry.peerServiceId, isError, entry.timestamp, () => ({
       targetKind: PEER_SERVICE_TARGET_KIND,
@@ -10754,7 +10803,7 @@ async function fetchHttpRequestLogEntries(creds, sinceIso) {
       throw new Error(`Cloud Logging entries.list failed: ${res.status} ${res.statusText}`);
     }
     const json = await res.json();
-    out.push(...json.entries ?? []);
+    if (Array.isArray(json.entries)) out.push(...json.entries);
     if (!json.nextPageToken) break;
     pageToken = json.nextPageToken;
   }
@@ -10791,7 +10840,7 @@ function resourceNameFor(type, labels) {
   }
 }
 function pathFromRequestUrl(requestUrl) {
-  if (!requestUrl) return null;
+  if (typeof requestUrl !== "string" || requestUrl.length === 0) return null;
   if (requestUrl.startsWith("/")) {
     const withoutQuery = requestUrl.split("?")[0];
     return withoutQuery && withoutQuery.length > 0 ? withoutQuery : "/";
@@ -10806,18 +10855,19 @@ function pathFromRequestUrl(requestUrl) {
 }
 var ERROR_STATUS_THRESHOLD2 = 500;
 function mapLogEntryToSignal(entry) {
+  if (!entry || typeof entry !== "object") return null;
   const resourceType = entry.resource?.type;
   if (!resourceType || !isFirebaseResourceType(resourceType)) return null;
   const resourceName = resourceNameFor(resourceType, entry.resource?.labels);
   if (!resourceName) return null;
   const req = entry.httpRequest;
   if (!req) return null;
-  if (!req.requestMethod) return null;
+  if (typeof req.requestMethod !== "string" || req.requestMethod.length === 0) return null;
   const method = req.requestMethod.toUpperCase();
   const path51 = pathFromRequestUrl(req.requestUrl);
   if (path51 === null) return null;
   const timestamp = entry.timestamp;
-  if (!timestamp) return null;
+  if (typeof timestamp !== "string" || timestamp.length === 0) return null;
   const isError = typeof req.status === "number" && req.status >= ERROR_STATUS_THRESHOLD2;
   return {
     targetKind: resourceType,
@@ -10969,7 +11019,7 @@ async function queryWorkerInvocations(ctx, config, window, fetchImpl = fetch) {
     throw new Error(`cloudflare connector: telemetry query returned an error (${message})`);
   }
   const events = payload.result?.events?.events;
-  if (events === void 0) {
+  if (!Array.isArray(events)) {
     console.warn(
       "[neat connector] cloudflare: telemetry query returned success:true but no result.events.events array \u2014 the response shape may have changed; treating as zero events this tick"
     );
@@ -10999,7 +11049,7 @@ var HTTP_METHODS = /* @__PURE__ */ new Set([
 ]);
 var LEADING_TOKEN_RE = /^(\S+)\s+\S/;
 function parseHttpMethodFromTrigger(trigger) {
-  if (!trigger) return null;
+  if (typeof trigger !== "string") return null;
   const match = LEADING_TOKEN_RE.exec(trigger.trim());
   const token = match?.[1];
   if (!token) return null;
@@ -11015,14 +11065,17 @@ function parsePathFromTrigger(trigger) {
 }
 var ERROR_STATUS_THRESHOLD3 = 500;
 function mapEventToSignal(event) {
+  if (!event || typeof event !== "object") return null;
   const metadata = event.$metadata;
   const workers = event.$workers;
   const method = parseHttpMethodFromTrigger(metadata?.trigger);
   if (!method) return null;
   const scriptName = workers?.scriptName ?? metadata?.service;
-  if (!scriptName) return null;
+  if (typeof scriptName !== "string" || scriptName.length === 0) return null;
   const timestampMs = event.timestamp ?? metadata?.startTime;
   if (typeof timestampMs !== "number" || !Number.isFinite(timestampMs)) return null;
+  const observedAt = new Date(timestampMs);
+  if (Number.isNaN(observedAt.getTime())) return null;
   const statusCode = metadata?.statusCode;
   const isError = typeof statusCode === "number" && statusCode >= ERROR_STATUS_THRESHOLD3;
   const path51 = metadata?.trigger ? parsePathFromTrigger(metadata.trigger) : void 0;
@@ -11031,7 +11084,7 @@ function mapEventToSignal(event) {
     targetName: scriptName,
     callCount: 1,
     errorCount: isError ? 1 : 0,
-    lastObservedIso: new Date(timestampMs).toISOString(),
+    lastObservedIso: observedAt.toISOString(),
     method,
     ...path51 ? { path: path51 } : {},
     ...typeof statusCode === "number" ? { statusCode } : {},