@kya-os/agentshield-nextjs 0.1.34 → 0.1.36

package/dist/index.js

@@ -1063,18 +1063,9 @@ async function checkWasmAvailability() {
     if (typeof WebAssembly === "undefined") {
       return false;
     }
-    const wasmCode = new Uint8Array([
-      0,
-      97,
-      115,
-      109,
-      1,
-      0,
-      0,
-      0
-    ]);
-    const module = await WebAssembly.compile(wasmCode);
-    await WebAssembly.instantiate(module);
+    if (!WebAssembly.instantiate || !WebAssembly.Module) {
+      return false;
+    }
     return true;
   } catch {
     return false;
@@ -1380,14 +1371,32 @@ function createEnhancedAgentShieldMiddleware(config = {}) {
     });
     return storageInitPromise;
   };
-  let wasmAvailable = false;
-  checkWasmAvailability().then((available) => {
-    wasmAvailable = available;
-    if (available) {
-      console.log("[AgentShield] \u2705 WASM support detected - enhanced detection enabled");
+  let detector = null;
+  let detectorInitPromise = null;
+  const getDetector = async () => {
+    if (detector) return detector;
+    if (detectorInitPromise) {
+      await detectorInitPromise;
+      return detector;
     }
-  });
-  const detector = wasmAvailable ? new EdgeAgentDetectorWrapperWithWasm({ enableWasm: true }) : new EdgeAgentDetectorWrapper({});
+    detectorInitPromise = (async () => {
+      try {
+        const wasmAvailable = await checkWasmAvailability();
+        if (wasmAvailable) {
+          console.log("[AgentShield] \u2705 WASM support detected - enhanced detection enabled");
+          detector = new EdgeAgentDetectorWrapperWithWasm({ enableWasm: true });
+        } else {
+          console.log("[AgentShield] \u2139\uFE0F Using pattern-based detection (WASM not available)");
+          detector = new EdgeAgentDetectorWrapper({});
+        }
+      } catch (error) {
+        console.warn("[AgentShield] Failed to initialize WASM, using fallback:", error);
+        detector = new EdgeAgentDetectorWrapper({});
+      }
+    })();
+    await detectorInitPromise;
+    return detector;
+  };
   const sessionManager = new SessionManager();
   const sessionTrackingEnabled = config.sessionTracking?.enabled !== false;
   return async (request) => {
@@ -1407,7 +1416,8 @@ function createEnhancedAgentShieldMiddleware(config = {}) {
       method: request.method,
       timestamp: /* @__PURE__ */ new Date()
     };
-    const result = await detector.analyze(context);
+    const activeDetector = await getDetector();
+    const result = await activeDetector.analyze(context);
     let finalConfidence = result.confidence;
     let verificationMethod = result.verificationMethod || "pattern";
     if (result.isAgent) {
@@ -1492,12 +1502,15 @@ function createEnhancedAgentShieldMiddleware(config = {}) {
           return response2;
         }
         case "log":
-          console.log(`[AgentShield] \u{1F916} AI Agent detected (${verificationMethod}):`, {
-            agent: result.detectedAgent?.name,
-            confidence: `${(finalConfidence * 100).toFixed(0)}%`,
-            path: pathWithQuery,
-            verification: verificationMethod
-          });
+          const isInteresting = finalConfidence >= 0.9 || result.detectedAgent?.name?.toLowerCase().includes("chatgpt") || result.detectedAgent?.name?.toLowerCase().includes("perplexity") || verificationMethod === "signature";
+          if (isInteresting) {
+            console.log(`[AgentShield] \u{1F916} AI Agent detected (${verificationMethod}):`, {
+              agent: result.detectedAgent?.name,
+              confidence: `${(finalConfidence * 100).toFixed(0)}%`,
+              path: pathWithQuery,
+              verification: verificationMethod
+            });
+          }
           break;
       }
     }