@kya-os/agentshield-nextjs 0.1.36 → 0.1.37

package/dist/index.mjs

@@ -373,6 +373,21 @@ var wasmInstance = null;
 var wasmExports = null;
 var initPromise = null;
 var WASM_PATH = "/wasm/agentshield_wasm_bg.wasm";
+var baseUrl = null;
+function setWasmBaseUrl(url) {
+  baseUrl = url;
+}
+function getWasmUrl() {
+  if (baseUrl) {
+    try {
+      const url = new URL(baseUrl);
+      return `${url.origin}${WASM_PATH}`;
+    } catch {
+      return WASM_PATH;
+    }
+  }
+  return WASM_PATH;
+}
 async function initWasm() {
   if (wasmExports) return true;
   if (initPromise) {
@@ -383,7 +398,8 @@ async function initWasm() {
     try {
       if (typeof WebAssembly.instantiateStreaming === "function") {
         try {
-          const response2 = await fetch(WASM_PATH);
+          const wasmUrl2 = getWasmUrl();
+          const response2 = await fetch(wasmUrl2);
           if (!response2.ok) {
             throw new Error(`Failed to fetch WASM: ${response2.status}`);
           }
@@ -409,7 +425,8 @@ async function initWasm() {
           console.log("[AgentShield] Streaming compilation failed, falling back to standard compilation");
         }
       }
-      const response = await fetch(WASM_PATH);
+      const wasmUrl = getWasmUrl();
+      const response = await fetch(wasmUrl);
       if (!response.ok) {
         throw new Error(`Failed to fetch WASM: ${response.status}`);
       }
@@ -508,6 +525,14 @@ var EdgeAgentDetectorWithWasm = class {
   }
   wasmEnabled = false;
   initPromise = null;
+  baseUrl = null;
+  /**
+   * Set the base URL for WASM loading in Edge Runtime
+   */
+  setBaseUrl(url) {
+    this.baseUrl = url;
+    setWasmBaseUrl(url);
+  }
   /**
    * Initialize the detector (including WASM if enabled)
    */
@@ -681,6 +706,12 @@ var EdgeAgentDetectorWrapperWithWasm = class {
   events = /* @__PURE__ */ new Map();
   constructor(config) {
     this.detector = new EdgeAgentDetectorWithWasm(config?.enableWasm ?? true);
+    if (config?.baseUrl) {
+      this.detector.setBaseUrl(config.baseUrl);
+    }
+  }
+  setBaseUrl(url) {
+    this.detector.setBaseUrl(url);
   }
   async analyze(input) {
     const result = await this.detector.analyze(input);
@@ -1371,10 +1402,18 @@ function createEnhancedAgentShieldMiddleware(config = {}) {
   };
   let detector = null;
   let detectorInitPromise = null;
-  const getDetector = async () => {
-    if (detector) return detector;
+  const getDetector = async (requestUrl) => {
+    if (detector) {
+      if (requestUrl && "setBaseUrl" in detector) {
+        detector.setBaseUrl(requestUrl);
+      }
+      return detector;
+    }
     if (detectorInitPromise) {
       await detectorInitPromise;
+      if (requestUrl && detector && "setBaseUrl" in detector) {
+        detector.setBaseUrl(requestUrl);
+      }
       return detector;
     }
     detectorInitPromise = (async () => {
@@ -1393,6 +1432,9 @@ function createEnhancedAgentShieldMiddleware(config = {}) {
       }
     })();
     await detectorInitPromise;
+    if (requestUrl && detector && "setBaseUrl" in detector) {
+      detector.setBaseUrl(requestUrl);
+    }
     return detector;
   };
   const sessionManager = new SessionManager();
@@ -1414,7 +1456,7 @@ function createEnhancedAgentShieldMiddleware(config = {}) {
       method: request.method,
       timestamp: /* @__PURE__ */ new Date()
     };
-    const activeDetector = await getDetector();
+    const activeDetector = await getDetector(request.url);
     const result = await activeDetector.analyze(context);
     let finalConfidence = result.confidence;
     let verificationMethod = result.verificationMethod || "pattern";