@kya-os/agentshield-nextjs 0.1.38 → 0.1.40

package/dist/index.js

@@ -398,55 +398,71 @@ async function initWasm() {
   }
   initPromise = (async () => {
     try {
-      if (typeof WebAssembly.instantiateStreaming === "function") {
-        try {
-          const wasmUrl2 = getWasmUrl();
-          const response2 = await fetch(wasmUrl2);
-          if (!response2.ok) {
-            throw new Error(`Failed to fetch WASM: ${response2.status}`);
-          }
-          const streamResponse = response2.clone();
-          const { instance } = await WebAssembly.instantiateStreaming(
-            streamResponse,
-            {
-              wbg: {
-                __wbg_log_1d3ae13c3d5e6b8e: (ptr, len) => {
-                  console.log("WASM:", ptr, len);
-                },
-                __wbindgen_throw: (ptr, len) => {
-                  throw new Error(`WASM Error at ${ptr}, length ${len}`);
+      const controller = new AbortController();
+      const timeout = setTimeout(() => controller.abort(), 3e3);
+      try {
+        const wasmUrl = getWasmUrl();
+        if (typeof WebAssembly.instantiateStreaming === "function") {
+          try {
+            const response2 = await fetch(wasmUrl, { signal: controller.signal });
+            clearTimeout(timeout);
+            if (!response2.ok) {
+              throw new Error(`Failed to fetch WASM: ${response2.status}`);
+            }
+            const streamResponse = response2.clone();
+            const { instance } = await WebAssembly.instantiateStreaming(
+              streamResponse,
+              {
+                wbg: {
+                  __wbg_log_1d3ae13c3d5e6b8e: (ptr, len) => {
+                    console.log("WASM:", ptr, len);
+                  },
+                  __wbindgen_throw: (ptr, len) => {
+                    throw new Error(`WASM Error at ${ptr}, length ${len}`);
+                  }
                 }
               }
+            );
+            wasmInstance = instance;
+            wasmExports = instance.exports;
+            console.log("[AgentShield] \u2705 WASM module initialized with streaming");
+            return;
+          } catch (streamError) {
+            if (!controller.signal.aborted) {
+              console.log("[AgentShield] Streaming compilation failed, falling back to standard compilation");
+            } else {
+              throw streamError;
             }
-          );
-          wasmInstance = instance;
-          wasmExports = instance.exports;
-          console.log("[AgentShield] \u2705 WASM module initialized with streaming");
-          return;
-        } catch (streamError) {
-          console.log("[AgentShield] Streaming compilation failed, falling back to standard compilation");
+          }
         }
-      }
-      const wasmUrl = getWasmUrl();
-      const response = await fetch(wasmUrl);
-      if (!response.ok) {
-        throw new Error(`Failed to fetch WASM: ${response.status}`);
-      }
-      const wasmArrayBuffer = await response.arrayBuffer();
-      const compiledModule = await WebAssembly.compile(wasmArrayBuffer);
-      const imports = {
-        wbg: {
-          __wbg_log_1d3ae13c3d5e6b8e: (ptr, len) => {
-            console.log("WASM:", ptr, len);
-          },
-          __wbindgen_throw: (ptr, len) => {
-            throw new Error(`WASM Error at ${ptr}, length ${len}`);
+        const response = await fetch(wasmUrl, { signal: controller.signal });
+        clearTimeout(timeout);
+        if (!response.ok) {
+          throw new Error(`Failed to fetch WASM: ${response.status}`);
+        }
+        const wasmArrayBuffer = await response.arrayBuffer();
+        const compiledModule = await WebAssembly.compile(wasmArrayBuffer);
+        const imports = {
+          wbg: {
+            __wbg_log_1d3ae13c3d5e6b8e: (ptr, len) => {
+              console.log("WASM:", ptr, len);
+            },
+            __wbindgen_throw: (ptr, len) => {
+              throw new Error(`WASM Error at ${ptr}, length ${len}`);
+            }
           }
+        };
+        wasmInstance = await WebAssembly.instantiate(compiledModule, imports);
+        wasmExports = wasmInstance.exports;
+        console.log("[AgentShield] \u2705 WASM module initialized via fallback");
+      } catch (fetchError) {
+        if (fetchError.name === "AbortError") {
+          console.warn("[AgentShield] WASM fetch timed out after 3 seconds - using pattern detection");
+        } else {
+          console.warn("[AgentShield] Failed to fetch WASM file:", fetchError.message);
         }
-      };
-      wasmInstance = await WebAssembly.instantiate(compiledModule, imports);
-      wasmExports = wasmInstance.exports;
-      console.log("[AgentShield] \u2705 WASM module initialized via fallback");
+        wasmExports = null;
+      }
     } catch (error) {
       console.error("[AgentShield] Failed to initialize WASM:", error);
       wasmExports = null;