@kya-os/agentshield-nextjs 0.1.18 → 0.1.19

package/dist/edge-runtime-loader.js

@@ -0,0 +1,198 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+// src/edge-runtime-loader.ts
+var EdgeRuntimeAgentShield = class {
+  wasmInstance = null;
+  wasmMemory = null;
+  config;
+  initialized = false;
+  constructor(config = {}) {
+    this.config = config;
+  }
+  async init(wasmModule) {
+    if (this.initialized) return;
+    const module = wasmModule || this.config.wasmModule;
+    if (module && this.config.enableWasm !== false) {
+      try {
+        this.wasmMemory = new WebAssembly.Memory({ initial: 17, maximum: 256 });
+        const imports = {
+          wbg: {
+            __wbindgen_throw: (ptr, len) => {
+              throw new Error(this.readString(ptr, len));
+            },
+            __wbg_log_: (ptr, len) => {
+              if (this.config.debug) {
+                console.log(this.readString(ptr, len));
+              }
+            }
+          },
+          env: {
+            memory: this.wasmMemory
+          }
+        };
+        this.wasmInstance = await WebAssembly.instantiate(module, imports);
+        this.initialized = true;
+        if (this.config.debug) {
+          console.log("\u2705 AgentShield WASM initialized in Edge Runtime");
+        }
+      } catch (error) {
+        console.warn(
+          "\u26A0\uFE0F WASM initialization failed, using pattern detection:",
+          error
+        );
+        this.initialized = true;
+      }
+    } else {
+      this.initialized = true;
+    }
+  }
+  readString(ptr, len) {
+    if (!this.wasmInstance || !this.wasmMemory) {
+      throw new Error("WASM not initialized");
+    }
+    const memory = new Uint8Array(this.wasmMemory.buffer);
+    const bytes = memory.slice(ptr, ptr + len);
+    return new TextDecoder().decode(bytes);
+  }
+  writeString(str) {
+    if (!this.wasmInstance) {
+      throw new Error("WASM not initialized");
+    }
+    const exports = this.wasmInstance.exports;
+    const encoded = new TextEncoder().encode(str);
+    const ptr = exports.__wbindgen_malloc(encoded.length);
+    const memory = new Uint8Array(exports.memory.buffer);
+    memory.set(encoded, ptr);
+    return [ptr, encoded.length];
+  }
+  async detect(request) {
+    if (!this.initialized) {
+      await this.init();
+    }
+    const metadata = {
+      userAgent: request.headers.get("user-agent") || "",
+      ipAddress: request.ip || request.headers.get("x-forwarded-for")?.split(",")[0] || "",
+      headers: Object.fromEntries(request.headers.entries())
+    };
+    if (this.wasmInstance && this.config.enableWasm !== false) {
+      try {
+        const exports = this.wasmInstance.exports;
+        const [ptr, len] = this.writeString(JSON.stringify(metadata));
+        const resultPtr = exports.detect_agent(ptr, len);
+        let resultStr = "";
+        let offset = 0;
+        const memory = new Uint8Array(
+          exports.memory.buffer
+        );
+        while (offset < 4096) {
+          const byte = memory[resultPtr + offset];
+          if (byte === 0 || byte === void 0) break;
+          offset++;
+        }
+        if (offset > 0) {
+          resultStr = this.readString(resultPtr, offset);
+        }
+        exports.__wbindgen_free(ptr, len);
+        if (offset > 0) {
+          exports.__wbindgen_free(resultPtr, offset);
+        }
+        const result = JSON.parse(resultStr);
+        const detection = {
+          ...result,
+          verificationMethod: "cryptographic",
+          timestamp: (/* @__PURE__ */ new Date()).toISOString()
+        };
+        if (this.config.onAgentDetected && detection.isAgent) {
+          this.config.onAgentDetected(detection);
+        }
+        return detection;
+      } catch (error) {
+        if (this.config.debug) {
+          console.error("WASM detection failed:", error);
+        }
+      }
+    }
+    return this.patternDetection(metadata);
+  }
+  patternDetection(metadata) {
+    const userAgent = metadata.userAgent.toLowerCase();
+    const patterns = [
+      // High confidence - explicit AI identifiers
+      { pattern: /chatgpt-user/i, name: "ChatGPT", confidence: 0.95 },
+      { pattern: /claude-web/i, name: "Claude", confidence: 0.95 },
+      { pattern: /gpt-crawler/i, name: "GPT Crawler", confidence: 0.95 },
+      // Medium-high confidence - company identifiers
+      { pattern: /anthropic/i, name: "Anthropic", confidence: 0.9 },
+      { pattern: /openai/i, name: "OpenAI", confidence: 0.9 },
+      // Medium confidence - product names
+      { pattern: /copilot/i, name: "GitHub Copilot", confidence: 0.85 },
+      { pattern: /bard/i, name: "Google Bard", confidence: 0.85 },
+      { pattern: /gemini/i, name: "Google Gemini", confidence: 0.85 },
+      { pattern: /perplexity/i, name: "Perplexity", confidence: 0.85 },
+      { pattern: /you\.com/i, name: "You.com", confidence: 0.8 },
+      { pattern: /phind/i, name: "Phind", confidence: 0.8 }
+    ];
+    const suspiciousHeaders = [
+      "x-openai-",
+      "x-anthropic-",
+      "x-ai-",
+      "x-llm-",
+      "x-gpt-"
+    ];
+    let headerBoost = 0;
+    for (const [key] of Object.entries(metadata.headers)) {
+      if (suspiciousHeaders.some((prefix) => key.toLowerCase().startsWith(prefix))) {
+        headerBoost = 0.1;
+        break;
+      }
+    }
+    for (const { pattern, name, confidence } of patterns) {
+      if (pattern.test(userAgent)) {
+        const finalConfidence = Math.min(confidence + headerBoost, 1);
+        const result = {
+          isAgent: true,
+          confidence: finalConfidence,
+          agent: name,
+          verificationMethod: "pattern",
+          riskLevel: finalConfidence > 0.9 ? "high" : "medium",
+          timestamp: (/* @__PURE__ */ new Date()).toISOString()
+        };
+        if (this.config.onAgentDetected) {
+          this.config.onAgentDetected(result);
+        }
+        return result;
+      }
+    }
+    return {
+      isAgent: false,
+      confidence: 0.85,
+      verificationMethod: "pattern",
+      timestamp: (/* @__PURE__ */ new Date()).toISOString()
+    };
+  }
+  isInitialized() {
+    return this.initialized;
+  }
+  getVerificationMethod() {
+    return this.wasmInstance ? "cryptographic" : "pattern";
+  }
+};
+function createEdgeAgentShield(config) {
+  return new EdgeRuntimeAgentShield(config);
+}
+var defaultInstance = null;
+function getDefaultAgentShield(config) {
+  if (!defaultInstance) {
+    defaultInstance = new EdgeRuntimeAgentShield(config);
+  }
+  return defaultInstance;
+}
+var edge_runtime_loader_default = createEdgeAgentShield;
+
+exports.createEdgeAgentShield = createEdgeAgentShield;
+exports.default = edge_runtime_loader_default;
+exports.getDefaultAgentShield = getDefaultAgentShield;
+//# sourceMappingURL=edge-runtime-loader.js.map
+//# sourceMappingURL=edge-runtime-loader.js.map

package/dist/edge-runtime-loader.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/edge-runtime-loader.ts"],"names":[],"mappings":";;;;;AA0CA,IAAM,yBAAN,MAA6B;AAAA,EACnB,YAAA,GAA4C,IAAA;AAAA,EAC5C,UAAA,GAAwC,IAAA;AAAA,EACxC,MAAA;AAAA,EACA,WAAA,GAAc,KAAA;AAAA,EAEtB,WAAA,CAAY,MAAA,GAA4B,EAAC,EAAG;AAC1C,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,EAChB;AAAA,EAEA,MAAM,KAAK,UAAA,EAAgD;AACzD,IAAA,IAAI,KAAK,WAAA,EAAa;AAEtB,IAAA,MAAM,MAAA,GAAS,UAAA,IAAc,IAAA,CAAK,MAAA,CAAO,UAAA;AAEzC,IAAA,IAAI,MAAA,IAAU,IAAA,CAAK,MAAA,CAAO,UAAA,KAAe,KAAA,EAAO;AAC9C,MAAA,IAAI;AACF,QAAA,IAAA,CAAK,UAAA,GAAa,IAAI,WAAA,CAAY,MAAA,CAAO,EAAE,OAAA,EAAS,EAAA,EAAI,OAAA,EAAS,GAAA,EAAK,CAAA;AAEtE,QAAA,MAAM,OAAA,GAAU;AAAA,UACd,GAAA,EAAK;AAAA,YACH,gBAAA,EAAkB,CAAC,GAAA,EAAa,GAAA,KAAgB;AAC9C,cAAA,MAAM,IAAI,KAAA,CAAM,IAAA,CAAK,UAAA,CAAW,GAAA,EAAK,GAAG,CAAC,CAAA;AAAA,YAC3C,CAAA;AAAA,YACA,UAAA,EAAY,CAAC,GAAA,EAAa,GAAA,KAAgB;AACxC,cAAA,IAAI,IAAA,CAAK,OAAO,KAAA,EAAO;AACrB,gBAAA,OAAA,CAAQ,GAAA,CAAI,IAAA,CAAK,UAAA,CAAW,GAAA,EAAK,GAAG,CAAC,CAAA;AAAA,cACvC;AAAA,YACF;AAAA,WACF;AAAA,UACA,GAAA,EAAK;AAAA,YACH,QAAQ,IAAA,CAAK;AAAA;AACf,SACF;AAEA,QAAA,IAAA,CAAK,YAAA,GAAe,MAAM,WAAA,CAAY,WAAA,CAAY,QAAQ,OAAO,CAAA;AACjE,QAAA,IAAA,CAAK,WAAA,GAAc,IAAA;AAEnB,QAAA,IAAI,IAAA,CAAK,OAAO,KAAA,EAAO;AACrB,UAAA,OAAA,CAAQ,IAAI,qDAAgD,CAAA;AAAA,QAC9D;AAAA,MACF,SAAS,KAAA,EAAO;AACd,QAAA,OAAA,CAAQ,IAAA;AAAA,UACN,mEAAA;AAAA,UACA;AAAA,SACF;AACA,QAAA,IAAA,CAAK,WAAA,GAAc,IAAA;AAAA,MACrB;AAAA,IACF,CAAA,MAAO;AACL,MAAA,IAAA,CAAK,WAAA,GAAc,IAAA;AAAA,IACrB;AAAA,EACF;AAAA,EAEQ,UAAA,CAAW,KAAa,GAAA,EAAqB;AACnD,IAAA,IAAI,CAAC,IAAA,CAAK,YAAA,IAAgB,CAAC,KAAK,UAAA,EAAY;AAC1C,MAAA,MAAM,IAAI,MAAM,sBAAsB,CAAA;AAAA,IACxC;AACA,IAAA,MAAM,MAAA,GAAS,IAAI,UAAA,CAAW,IAAA,CAAK,WAAW,MAAM,CAAA;AACpD,IAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,KAAA,CAAM,GAAA,EAAK,MAAM,GAAG,CAAA;AACzC,IAAA,OAAO,IAAI,WAAA,EAAY,CAAE,MAAA,CAAO,KAAK,CAAA;AAAA,EACvC;AAAA,EAEQ,YAAY,GAAA,EAA+B;AACjD,IAAA,IAAI,CAAC,KAAK,YAAA,EAAc;AACtB,MAAA,MAAM,IAAI,MAAM,sBAAsB,CAAA;AAAA,IACxC;AACA,IAAA,MAAM,OAAA,GAAU,KAAK,YAAA,CAAa,OAAA;AAElC,IAAA,MAAM,OAAA,GAAU,IAAI,WAAA,EAAY,CAAE,OAAO,GAAG,CAAA;AAC5C,IAAA,MAAM,GAAA,GAAM,OAAA,CAAQ,iBAAA,CAAkB,OAAA,CAAQ,MAAM,CAAA;AAEpD,IAAA,MAAM,MAAA,GAAS,IAAI,UAAA,CAAW,OAAA,CAAQ,OAAO,MAAM,CAAA;AACnD,IAAA,MAAA,CAAO,GAAA,CAAI,SAAS,GAAG,CAAA;AAEvB,IAAA,OAAO,CAAC,GAAA,EAAK,OAAA,CAAQ,MAAM,CAAA;AAAA,EAC7B;AAAA,EAEA,MAAM,OAAO,OAAA,EAAgD;AAC3D,IAAA,IAAI,CAAC,KAAK,WAAA,EAAa;AACrB,MAAA,MAAM,KAAK,IAAA,EAAK;AAAA,IAClB;AAEA,IAAA,MAAM,QAAA,GAAW;AAAA,MACf,SAAA,EAAW,OAAA,CAAQ,OAAA,CAAQ,GAAA,CAAI,YAAY,CAAA,IAAK,EAAA;AAAA,MAChD,SAAA,EACE,OAAA,CAAQ,EAAA,IACR,OAAA,CAAQ,OAAA,CAAQ,GAAA,CAAI,iBAAiB,CAAA,EAAG,KAAA,CAAM,GAAG,CAAA,CAAE,CAAC,CAAA,IACpD,EAAA;AAAA,MACF,SAAS,MAAA,CAAO,WAAA,CAAY,OAAA,CAAQ,OAAA,CAAQ,SAAS;AAAA,KACvD;AAGA,IAAA,IAAI,IAAA,CAAK,YAAA,IAAgB,IAAA,CAAK,MAAA,CAAO,eAAe,KAAA,EAAO;AACzD,MAAA,IAAI;AACF,QAAA,MAAM,OAAA,GAAU,KAAK,YAAA,CAAa,OAAA;AAClC,QAAA,MAAM,CAAC,KAAK,GAAG,CAAA,GAAI,KAAK,WAAA,CAAY,IAAA,CAAK,SAAA,CAAU,QAAQ,CAAC,CAAA;AAE5D,QAAA,MAAM,SAAA,GAAY,OAAA,CAAQ,YAAA,CAAa,GAAA,EAAK,GAAG,CAAA;AAE/C,QAAA,IAAI,SAAA,GAAY,EAAA;AAChB,QAAA,IAAI,MAAA,GAAS,CAAA;AACb,QAAA,MAAM,SAAS,IAAI,UAAA;AAAA,UAChB,QAAQ,MAAA,CAA8B;AAAA,SACzC;AAGA,QAAA,OAAO,SAAS,IAAA,EAAM;AAEpB,UAAA,MAAM,IAAA,GAAO,MAAA,CAAO,SAAA,GAAY,MAAM,CAAA;AACtC,UAAA,IAAI,IAAA,KAAS,CAAA,IAAK,IAAA,KAAS,KAAA,CAAA,EAAW;AACtC,UAAA,MAAA,EAAA;AAAA,QACF;AAEA,QAAA,IAAI,SAAS,CAAA,EAAG;AACd,UAAA,SAAA,GAAY,IAAA,CAAK,UAAA,CAAW,SAAA,EAAW,MAAM,CAAA;AAAA,QAC/C;AAEA,QAAA,OAAA,CAAQ,eAAA,CAAgB,KAAK,GAAG,CAAA;AAChC,QAAA,IAAI,SAAS,CAAA,EAAG;AACd,UAAA,OAAA,CAAQ,eAAA,CAAgB,WAAW,MAAM,CAAA;AAAA,QAC3C;AAEA,QAAA,MAAM,MAAA,GAAS,IAAA,CAAK,KAAA,CAAM,SAAS,CAAA;AAEnC,QAAA,MAAM,SAAA,GAA6B;AAAA,UACjC,GAAG,MAAA;AAAA,UACH,kBAAA,EAAoB,eAAA;AAAA,UACpB,SAAA,EAAA,iBAAW,IAAI,IAAA,EAAK,EAAE,WAAA;AAAY,SACpC;AAEA,QAAA,IAAI,IAAA,CAAK,MAAA,CAAO,eAAA,IAAmB,SAAA,CAAU,OAAA,EAAS;AACpD,UAAA,IAAA,CAAK,MAAA,CAAO,gBAAgB,SAAS,CAAA;AAAA,QACvC;AAEA,QAAA,OAAO,SAAA;AAAA,MACT,SAAS,KAAA,EAAO;AACd,QAAA,IAAI,IAAA,CAAK,OAAO,KAAA,EAAO;AACrB,UAAA,OAAA,CAAQ,KAAA,CAAM,0BAA0B,KAAK,CAAA;AAAA,QAC/C;AAAA,MAEF;AAAA,IACF;AAGA,IAAA,OAAO,IAAA,CAAK,iBAAiB,QAAQ,CAAA;AAAA,EACvC;AAAA,EAEQ,iBAAiB,QAAA,EAIL;AAClB,IAAA,MAAM,SAAA,GAAY,QAAA,CAAS,SAAA,CAAU,WAAA,EAAY;AAGjD,IAAA,MAAM,QAAA,GAAW;AAAA;AAAA,MAEf,EAAE,OAAA,EAAS,eAAA,EAAiB,IAAA,EAAM,SAAA,EAAW,YAAY,IAAA,EAAK;AAAA,MAC9D,EAAE,OAAA,EAAS,aAAA,EAAe,IAAA,EAAM,QAAA,EAAU,YAAY,IAAA,EAAK;AAAA,MAC3D,EAAE,OAAA,EAAS,cAAA,EAAgB,IAAA,EAAM,aAAA,EAAe,YAAY,IAAA,EAAK;AAAA;AAAA,MAGjE,EAAE,OAAA,EAAS,YAAA,EAAc,IAAA,EAAM,WAAA,EAAa,YAAY,GAAA,EAAI;AAAA,MAC5D,EAAE,OAAA,EAAS,SAAA,EAAW,IAAA,EAAM,QAAA,EAAU,YAAY,GAAA,EAAI;AAAA;AAAA,MAGtD,EAAE,OAAA,EAAS,UAAA,EAAY,IAAA,EAAM,gBAAA,EAAkB,YAAY,IAAA,EAAK;AAAA,MAChE,EAAE,OAAA,EAAS,OAAA,EAAS,IAAA,EAAM,aAAA,EAAe,YAAY,IAAA,EAAK;AAAA,MAC1D,EAAE,OAAA,EAAS,SAAA,EAAW,IAAA,EAAM,eAAA,EAAiB,YAAY,IAAA,EAAK;AAAA,MAC9D,EAAE,OAAA,EAAS,aAAA,EAAe,IAAA,EAAM,YAAA,EAAc,YAAY,IAAA,EAAK;AAAA,MAC/D,EAAE,OAAA,EAAS,WAAA,EAAa,IAAA,EAAM,SAAA,EAAW,YAAY,GAAA,EAAI;AAAA,MACzD,EAAE,OAAA,EAAS,QAAA,EAAU,IAAA,EAAM,OAAA,EAAS,YAAY,GAAA;AAAI,KACtD;AAGA,IAAA,MAAM,iBAAA,GAAoB;AAAA,MACxB,WAAA;AAAA,MACA,cAAA;AAAA,MACA,OAAA;AAAA,MACA,QAAA;AAAA,MACA;AAAA,KACF;AAEA,IAAA,IAAI,WAAA,GAAc,CAAA;AAClB,IAAA,KAAA,MAAW,CAAC,GAAG,CAAA,IAAK,OAAO,OAAA,CAAQ,QAAA,CAAS,OAAO,CAAA,EAAG;AACpD,MAAA,IACE,iBAAA,CAAkB,KAAK,CAAA,MAAA,KAAU,GAAA,CAAI,aAAY,CAAE,UAAA,CAAW,MAAM,CAAC,CAAA,EACrE;AACA,QAAA,WAAA,GAAc,GAAA;AACd,QAAA;AAAA,MACF;AAAA,IACF;AAGA,IAAA,KAAA,MAAW,EAAE,OAAA,EAAS,IAAA,EAAM,UAAA,MAAgB,QAAA,EAAU;AACpD,MAAA,IAAI,OAAA,CAAQ,IAAA,CAAK,SAAS,CAAA,EAAG;AAC3B,QAAA,MAAM,eAAA,GAAkB,IAAA,CAAK,GAAA,CAAI,UAAA,GAAa,aAAa,CAAG,CAAA;AAE9D,QAAA,MAAM,MAAA,GAA0B;AAAA,UAC9B,OAAA,EAAS,IAAA;AAAA,UACT,UAAA,EAAY,eAAA;AAAA,UACZ,KAAA,EAAO,IAAA;AAAA,UACP,kBAAA,EAAoB,SAAA;AAAA,UACpB,SAAA,EAAW,eAAA,GAAkB,GAAA,GAAM,MAAA,GAAS,QAAA;AAAA,UAC5C,SAAA,EAAA,iBAAW,IAAI,IAAA,EAAK,EAAE,WAAA;AAAY,SACpC;AAEA,QAAA,IAAI,IAAA,CAAK,OAAO,eAAA,EAAiB;AAC/B,UAAA,IAAA,CAAK,MAAA,CAAO,gBAAgB,MAAM,CAAA;AAAA,QACpC;AAEA,QAAA,OAAO,MAAA;AAAA,MACT;AAAA,IACF;AAGA,IAAA,OAAO;AAAA,MACL,OAAA,EAAS,KAAA;AAAA,MACT,UAAA,EAAY,IAAA;AAAA,MACZ,kBAAA,EAAoB,SAAA;AAAA,MACpB,SAAA,EAAA,iBAAW,IAAI,IAAA,EAAK,EAAE,WAAA;AAAY,KACpC;AAAA,EACF;AAAA,EAEA,aAAA,GAAyB;AACvB,IAAA,OAAO,IAAA,CAAK,WAAA;AAAA,EACd;AAAA,EAEA,qBAAA,GAAqD;AACnD,IAAA,OAAO,IAAA,CAAK,eAAe,eAAA,GAAkB,SAAA;AAAA,EAC/C;AACF,CAAA;AAKO,SAAS,sBAAsB,MAAA,EAA4B;AAChE,EAAA,OAAO,IAAI,uBAAuB,MAAM,CAAA;AAC1C;AAKA,IAAI,eAAA,GAAiD,IAAA;AAE9C,SAAS,sBACd,MAAA,EACwB;AACxB,EAAA,IAAI,CAAC,eAAA,EAAiB;AACpB,IAAA,eAAA,GAAkB,IAAI,uBAAuB,MAAM,CAAA;AAAA,EACrD;AACA,EAAA,OAAO,eAAA;AACT;AAEA,IAAO,2BAAA,GAAQ","file":"edge-runtime-loader.js","sourcesContent":["/**\n * Edge Runtime Compatible WASM Loader for AgentShield\n *\n * This module provides a pre-built solution for loading WASM in Edge Runtime.\n * It requires the WASM file to be manually placed in the project.\n */\n\nimport type { NextRequest } from 'next/server';\n\n// User must provide the WASM module via import\n// This is intentionally not imported here to allow user control\nexport interface WasmModule {\n  default: WebAssembly.Module;\n}\n\nexport interface DetectionResult {\n  isAgent: boolean;\n  confidence: number;\n  agent?: string;\n  verificationMethod: 'cryptographic' | 'pattern';\n  riskLevel?: 'low' | 'medium' | 'high' | 'critical';\n  timestamp: string;\n}\n\nexport interface AgentShieldConfig {\n  wasmModule?: WebAssembly.Module;\n  enableWasm?: boolean;\n  onAgentDetected?: (result: DetectionResult) => void;\n  blockAgents?: boolean;\n  allowedAgents?: string[];\n  debug?: boolean;\n}\n\ninterface WasmExports {\n  detect_agent(metadataPtr: number, metadataLen: number): number;\n  get_version(): number;\n  __wbindgen_malloc(size: number): number;\n  __wbindgen_free(ptr: number, size: number): void;\n  __wbindgen_realloc(ptr: number, oldSize: number, newSize: number): number;\n  memory: WebAssembly.Memory;\n}\n\nclass EdgeRuntimeAgentShield {\n  private wasmInstance: WebAssembly.Instance | null = null;\n  private wasmMemory: WebAssembly.Memory | null = null;\n  private config: AgentShieldConfig;\n  private initialized = false;\n\n  constructor(config: AgentShieldConfig = {}) {\n    this.config = config;\n  }\n\n  async init(wasmModule?: WebAssembly.Module): Promise<void> {\n    if (this.initialized) return;\n\n    const module = wasmModule || this.config.wasmModule;\n\n    if (module && this.config.enableWasm !== false) {\n      try {\n        this.wasmMemory = new WebAssembly.Memory({ initial: 17, maximum: 256 });\n\n        const imports = {\n          wbg: {\n            __wbindgen_throw: (ptr: number, len: number) => {\n              throw new Error(this.readString(ptr, len));\n            },\n            __wbg_log_: (ptr: number, len: number) => {\n              if (this.config.debug) {\n                console.log(this.readString(ptr, len));\n              }\n            },\n          },\n          env: {\n            memory: this.wasmMemory,\n          },\n        };\n\n        this.wasmInstance = await WebAssembly.instantiate(module, imports);\n        this.initialized = true;\n\n        if (this.config.debug) {\n          console.log('✅ AgentShield WASM initialized in Edge Runtime');\n        }\n      } catch (error) {\n        console.warn(\n          '⚠️ WASM initialization failed, using pattern detection:',\n          error\n        );\n        this.initialized = true;\n      }\n    } else {\n      this.initialized = true;\n    }\n  }\n\n  private readString(ptr: number, len: number): string {\n    if (!this.wasmInstance || !this.wasmMemory) {\n      throw new Error('WASM not initialized');\n    }\n    const memory = new Uint8Array(this.wasmMemory.buffer);\n    const bytes = memory.slice(ptr, ptr + len);\n    return new TextDecoder().decode(bytes);\n  }\n\n  private writeString(str: string): [number, number] {\n    if (!this.wasmInstance) {\n      throw new Error('WASM not initialized');\n    }\n    const exports = this.wasmInstance.exports as unknown as WasmExports;\n\n    const encoded = new TextEncoder().encode(str);\n    const ptr = exports.__wbindgen_malloc(encoded.length);\n\n    const memory = new Uint8Array(exports.memory.buffer);\n    memory.set(encoded, ptr);\n\n    return [ptr, encoded.length];\n  }\n\n  async detect(request: NextRequest): Promise<DetectionResult> {\n    if (!this.initialized) {\n      await this.init();\n    }\n\n    const metadata = {\n      userAgent: request.headers.get('user-agent') || '',\n      ipAddress:\n        request.ip ||\n        request.headers.get('x-forwarded-for')?.split(',')[0] ||\n        '',\n      headers: Object.fromEntries(request.headers.entries()),\n    };\n\n    // Try WASM detection first\n    if (this.wasmInstance && this.config.enableWasm !== false) {\n      try {\n        const exports = this.wasmInstance.exports as unknown as WasmExports;\n        const [ptr, len] = this.writeString(JSON.stringify(metadata));\n\n        const resultPtr = exports.detect_agent(ptr, len);\n        // Read result with dynamic size\n        let resultStr = '';\n        let offset = 0;\n        const memory = new Uint8Array(\n          (exports.memory as WebAssembly.Memory).buffer\n        );\n\n        // Read until we find the end of the JSON string\n        while (offset < 4096) {\n          // Max 4KB result\n          const byte = memory[resultPtr + offset];\n          if (byte === 0 || byte === undefined) break;\n          offset++;\n        }\n\n        if (offset > 0) {\n          resultStr = this.readString(resultPtr, offset);\n        }\n\n        exports.__wbindgen_free(ptr, len);\n        if (offset > 0) {\n          exports.__wbindgen_free(resultPtr, offset);\n        }\n\n        const result = JSON.parse(resultStr);\n\n        const detection: DetectionResult = {\n          ...result,\n          verificationMethod: 'cryptographic',\n          timestamp: new Date().toISOString(),\n        };\n\n        if (this.config.onAgentDetected && detection.isAgent) {\n          this.config.onAgentDetected(detection);\n        }\n\n        return detection;\n      } catch (error) {\n        if (this.config.debug) {\n          console.error('WASM detection failed:', error);\n        }\n        // Fall through to pattern detection\n      }\n    }\n\n    // Fallback: Pattern-based detection\n    return this.patternDetection(metadata);\n  }\n\n  private patternDetection(metadata: {\n    userAgent: string;\n    ipAddress: string;\n    headers: Record<string, string>;\n  }): DetectionResult {\n    const userAgent = metadata.userAgent.toLowerCase();\n\n    // Enhanced AI agent patterns with confidence scores\n    const patterns = [\n      // High confidence - explicit AI identifiers\n      { pattern: /chatgpt-user/i, name: 'ChatGPT', confidence: 0.95 },\n      { pattern: /claude-web/i, name: 'Claude', confidence: 0.95 },\n      { pattern: /gpt-crawler/i, name: 'GPT Crawler', confidence: 0.95 },\n\n      // Medium-high confidence - company identifiers\n      { pattern: /anthropic/i, name: 'Anthropic', confidence: 0.9 },\n      { pattern: /openai/i, name: 'OpenAI', confidence: 0.9 },\n\n      // Medium confidence - product names\n      { pattern: /copilot/i, name: 'GitHub Copilot', confidence: 0.85 },\n      { pattern: /bard/i, name: 'Google Bard', confidence: 0.85 },\n      { pattern: /gemini/i, name: 'Google Gemini', confidence: 0.85 },\n      { pattern: /perplexity/i, name: 'Perplexity', confidence: 0.85 },\n      { pattern: /you\\.com/i, name: 'You.com', confidence: 0.8 },\n      { pattern: /phind/i, name: 'Phind', confidence: 0.8 },\n    ];\n\n    // Check for suspicious headers\n    const suspiciousHeaders = [\n      'x-openai-',\n      'x-anthropic-',\n      'x-ai-',\n      'x-llm-',\n      'x-gpt-',\n    ];\n\n    let headerBoost = 0;\n    for (const [key] of Object.entries(metadata.headers)) {\n      if (\n        suspiciousHeaders.some(prefix => key.toLowerCase().startsWith(prefix))\n      ) {\n        headerBoost = 0.1;\n        break;\n      }\n    }\n\n    // Check patterns\n    for (const { pattern, name, confidence } of patterns) {\n      if (pattern.test(userAgent)) {\n        const finalConfidence = Math.min(confidence + headerBoost, 1.0);\n\n        const result: DetectionResult = {\n          isAgent: true,\n          confidence: finalConfidence,\n          agent: name,\n          verificationMethod: 'pattern',\n          riskLevel: finalConfidence > 0.9 ? 'high' : 'medium',\n          timestamp: new Date().toISOString(),\n        };\n\n        if (this.config.onAgentDetected) {\n          this.config.onAgentDetected(result);\n        }\n\n        return result;\n      }\n    }\n\n    // Not detected\n    return {\n      isAgent: false,\n      confidence: 0.85,\n      verificationMethod: 'pattern',\n      timestamp: new Date().toISOString(),\n    };\n  }\n\n  isInitialized(): boolean {\n    return this.initialized;\n  }\n\n  getVerificationMethod(): 'cryptographic' | 'pattern' {\n    return this.wasmInstance ? 'cryptographic' : 'pattern';\n  }\n}\n\n/**\n * Factory function to create an AgentShield instance for Edge Runtime\n */\nexport function createEdgeAgentShield(config?: AgentShieldConfig) {\n  return new EdgeRuntimeAgentShield(config);\n}\n\n/**\n * Singleton instance for simple usage\n */\nlet defaultInstance: EdgeRuntimeAgentShield | null = null;\n\nexport function getDefaultAgentShield(\n  config?: AgentShieldConfig\n): EdgeRuntimeAgentShield {\n  if (!defaultInstance) {\n    defaultInstance = new EdgeRuntimeAgentShield(config);\n  }\n  return defaultInstance;\n}\n\nexport default createEdgeAgentShield;\n"]}

package/dist/edge-runtime-loader.mjs

@@ -0,0 +1,192 @@
+// src/edge-runtime-loader.ts
+var EdgeRuntimeAgentShield = class {
+  wasmInstance = null;
+  wasmMemory = null;
+  config;
+  initialized = false;
+  constructor(config = {}) {
+    this.config = config;
+  }
+  async init(wasmModule) {
+    if (this.initialized) return;
+    const module = wasmModule || this.config.wasmModule;
+    if (module && this.config.enableWasm !== false) {
+      try {
+        this.wasmMemory = new WebAssembly.Memory({ initial: 17, maximum: 256 });
+        const imports = {
+          wbg: {
+            __wbindgen_throw: (ptr, len) => {
+              throw new Error(this.readString(ptr, len));
+            },
+            __wbg_log_: (ptr, len) => {
+              if (this.config.debug) {
+                console.log(this.readString(ptr, len));
+              }
+            }
+          },
+          env: {
+            memory: this.wasmMemory
+          }
+        };
+        this.wasmInstance = await WebAssembly.instantiate(module, imports);
+        this.initialized = true;
+        if (this.config.debug) {
+          console.log("\u2705 AgentShield WASM initialized in Edge Runtime");
+        }
+      } catch (error) {
+        console.warn(
+          "\u26A0\uFE0F WASM initialization failed, using pattern detection:",
+          error
+        );
+        this.initialized = true;
+      }
+    } else {
+      this.initialized = true;
+    }
+  }
+  readString(ptr, len) {
+    if (!this.wasmInstance || !this.wasmMemory) {
+      throw new Error("WASM not initialized");
+    }
+    const memory = new Uint8Array(this.wasmMemory.buffer);
+    const bytes = memory.slice(ptr, ptr + len);
+    return new TextDecoder().decode(bytes);
+  }
+  writeString(str) {
+    if (!this.wasmInstance) {
+      throw new Error("WASM not initialized");
+    }
+    const exports = this.wasmInstance.exports;
+    const encoded = new TextEncoder().encode(str);
+    const ptr = exports.__wbindgen_malloc(encoded.length);
+    const memory = new Uint8Array(exports.memory.buffer);
+    memory.set(encoded, ptr);
+    return [ptr, encoded.length];
+  }
+  async detect(request) {
+    if (!this.initialized) {
+      await this.init();
+    }
+    const metadata = {
+      userAgent: request.headers.get("user-agent") || "",
+      ipAddress: request.ip || request.headers.get("x-forwarded-for")?.split(",")[0] || "",
+      headers: Object.fromEntries(request.headers.entries())
+    };
+    if (this.wasmInstance && this.config.enableWasm !== false) {
+      try {
+        const exports = this.wasmInstance.exports;
+        const [ptr, len] = this.writeString(JSON.stringify(metadata));
+        const resultPtr = exports.detect_agent(ptr, len);
+        let resultStr = "";
+        let offset = 0;
+        const memory = new Uint8Array(
+          exports.memory.buffer
+        );
+        while (offset < 4096) {
+          const byte = memory[resultPtr + offset];
+          if (byte === 0 || byte === void 0) break;
+          offset++;
+        }
+        if (offset > 0) {
+          resultStr = this.readString(resultPtr, offset);
+        }
+        exports.__wbindgen_free(ptr, len);
+        if (offset > 0) {
+          exports.__wbindgen_free(resultPtr, offset);
+        }
+        const result = JSON.parse(resultStr);
+        const detection = {
+          ...result,
+          verificationMethod: "cryptographic",
+          timestamp: (/* @__PURE__ */ new Date()).toISOString()
+        };
+        if (this.config.onAgentDetected && detection.isAgent) {
+          this.config.onAgentDetected(detection);
+        }
+        return detection;
+      } catch (error) {
+        if (this.config.debug) {
+          console.error("WASM detection failed:", error);
+        }
+      }
+    }
+    return this.patternDetection(metadata);
+  }
+  patternDetection(metadata) {
+    const userAgent = metadata.userAgent.toLowerCase();
+    const patterns = [
+      // High confidence - explicit AI identifiers
+      { pattern: /chatgpt-user/i, name: "ChatGPT", confidence: 0.95 },
+      { pattern: /claude-web/i, name: "Claude", confidence: 0.95 },
+      { pattern: /gpt-crawler/i, name: "GPT Crawler", confidence: 0.95 },
+      // Medium-high confidence - company identifiers
+      { pattern: /anthropic/i, name: "Anthropic", confidence: 0.9 },
+      { pattern: /openai/i, name: "OpenAI", confidence: 0.9 },
+      // Medium confidence - product names
+      { pattern: /copilot/i, name: "GitHub Copilot", confidence: 0.85 },
+      { pattern: /bard/i, name: "Google Bard", confidence: 0.85 },
+      { pattern: /gemini/i, name: "Google Gemini", confidence: 0.85 },
+      { pattern: /perplexity/i, name: "Perplexity", confidence: 0.85 },
+      { pattern: /you\.com/i, name: "You.com", confidence: 0.8 },
+      { pattern: /phind/i, name: "Phind", confidence: 0.8 }
+    ];
+    const suspiciousHeaders = [
+      "x-openai-",
+      "x-anthropic-",
+      "x-ai-",
+      "x-llm-",
+      "x-gpt-"
+    ];
+    let headerBoost = 0;
+    for (const [key] of Object.entries(metadata.headers)) {
+      if (suspiciousHeaders.some((prefix) => key.toLowerCase().startsWith(prefix))) {
+        headerBoost = 0.1;
+        break;
+      }
+    }
+    for (const { pattern, name, confidence } of patterns) {
+      if (pattern.test(userAgent)) {
+        const finalConfidence = Math.min(confidence + headerBoost, 1);
+        const result = {
+          isAgent: true,
+          confidence: finalConfidence,
+          agent: name,
+          verificationMethod: "pattern",
+          riskLevel: finalConfidence > 0.9 ? "high" : "medium",
+          timestamp: (/* @__PURE__ */ new Date()).toISOString()
+        };
+        if (this.config.onAgentDetected) {
+          this.config.onAgentDetected(result);
+        }
+        return result;
+      }
+    }
+    return {
+      isAgent: false,
+      confidence: 0.85,
+      verificationMethod: "pattern",
+      timestamp: (/* @__PURE__ */ new Date()).toISOString()
+    };
+  }
+  isInitialized() {
+    return this.initialized;
+  }
+  getVerificationMethod() {
+    return this.wasmInstance ? "cryptographic" : "pattern";
+  }
+};
+function createEdgeAgentShield(config) {
+  return new EdgeRuntimeAgentShield(config);
+}
+var defaultInstance = null;
+function getDefaultAgentShield(config) {
+  if (!defaultInstance) {
+    defaultInstance = new EdgeRuntimeAgentShield(config);
+  }
+  return defaultInstance;
+}
+var edge_runtime_loader_default = createEdgeAgentShield;
+
+export { createEdgeAgentShield, edge_runtime_loader_default as default, getDefaultAgentShield };
+//# sourceMappingURL=edge-runtime-loader.mjs.map
+//# sourceMappingURL=edge-runtime-loader.mjs.map

package/dist/edge-runtime-loader.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/edge-runtime-loader.ts"],"names":[],"mappings":";AA0CA,IAAM,yBAAN,MAA6B;AAAA,EACnB,YAAA,GAA4C,IAAA;AAAA,EAC5C,UAAA,GAAwC,IAAA;AAAA,EACxC,MAAA;AAAA,EACA,WAAA,GAAc,KAAA;AAAA,EAEtB,WAAA,CAAY,MAAA,GAA4B,EAAC,EAAG;AAC1C,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAAA,EAChB;AAAA,EAEA,MAAM,KAAK,UAAA,EAAgD;AACzD,IAAA,IAAI,KAAK,WAAA,EAAa;AAEtB,IAAA,MAAM,MAAA,GAAS,UAAA,IAAc,IAAA,CAAK,MAAA,CAAO,UAAA;AAEzC,IAAA,IAAI,MAAA,IAAU,IAAA,CAAK,MAAA,CAAO,UAAA,KAAe,KAAA,EAAO;AAC9C,MAAA,IAAI;AACF,QAAA,IAAA,CAAK,UAAA,GAAa,IAAI,WAAA,CAAY,MAAA,CAAO,EAAE,OAAA,EAAS,EAAA,EAAI,OAAA,EAAS,GAAA,EAAK,CAAA;AAEtE,QAAA,MAAM,OAAA,GAAU;AAAA,UACd,GAAA,EAAK;AAAA,YACH,gBAAA,EAAkB,CAAC,GAAA,EAAa,GAAA,KAAgB;AAC9C,cAAA,MAAM,IAAI,KAAA,CAAM,IAAA,CAAK,UAAA,CAAW,GAAA,EAAK,GAAG,CAAC,CAAA;AAAA,YAC3C,CAAA;AAAA,YACA,UAAA,EAAY,CAAC,GAAA,EAAa,GAAA,KAAgB;AACxC,cAAA,IAAI,IAAA,CAAK,OAAO,KAAA,EAAO;AACrB,gBAAA,OAAA,CAAQ,GAAA,CAAI,IAAA,CAAK,UAAA,CAAW,GAAA,EAAK,GAAG,CAAC,CAAA;AAAA,cACvC;AAAA,YACF;AAAA,WACF;AAAA,UACA,GAAA,EAAK;AAAA,YACH,QAAQ,IAAA,CAAK;AAAA;AACf,SACF;AAEA,QAAA,IAAA,CAAK,YAAA,GAAe,MAAM,WAAA,CAAY,WAAA,CAAY,QAAQ,OAAO,CAAA;AACjE,QAAA,IAAA,CAAK,WAAA,GAAc,IAAA;AAEnB,QAAA,IAAI,IAAA,CAAK,OAAO,KAAA,EAAO;AACrB,UAAA,OAAA,CAAQ,IAAI,qDAAgD,CAAA;AAAA,QAC9D;AAAA,MACF,SAAS,KAAA,EAAO;AACd,QAAA,OAAA,CAAQ,IAAA;AAAA,UACN,mEAAA;AAAA,UACA;AAAA,SACF;AACA,QAAA,IAAA,CAAK,WAAA,GAAc,IAAA;AAAA,MACrB;AAAA,IACF,CAAA,MAAO;AACL,MAAA,IAAA,CAAK,WAAA,GAAc,IAAA;AAAA,IACrB;AAAA,EACF;AAAA,EAEQ,UAAA,CAAW,KAAa,GAAA,EAAqB;AACnD,IAAA,IAAI,CAAC,IAAA,CAAK,YAAA,IAAgB,CAAC,KAAK,UAAA,EAAY;AAC1C,MAAA,MAAM,IAAI,MAAM,sBAAsB,CAAA;AAAA,IACxC;AACA,IAAA,MAAM,MAAA,GAAS,IAAI,UAAA,CAAW,IAAA,CAAK,WAAW,MAAM,CAAA;AACpD,IAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,KAAA,CAAM,GAAA,EAAK,MAAM,GAAG,CAAA;AACzC,IAAA,OAAO,IAAI,WAAA,EAAY,CAAE,MAAA,CAAO,KAAK,CAAA;AAAA,EACvC;AAAA,EAEQ,YAAY,GAAA,EAA+B;AACjD,IAAA,IAAI,CAAC,KAAK,YAAA,EAAc;AACtB,MAAA,MAAM,IAAI,MAAM,sBAAsB,CAAA;AAAA,IACxC;AACA,IAAA,MAAM,OAAA,GAAU,KAAK,YAAA,CAAa,OAAA;AAElC,IAAA,MAAM,OAAA,GAAU,IAAI,WAAA,EAAY,CAAE,OAAO,GAAG,CAAA;AAC5C,IAAA,MAAM,GAAA,GAAM,OAAA,CAAQ,iBAAA,CAAkB,OAAA,CAAQ,MAAM,CAAA;AAEpD,IAAA,MAAM,MAAA,GAAS,IAAI,UAAA,CAAW,OAAA,CAAQ,OAAO,MAAM,CAAA;AACnD,IAAA,MAAA,CAAO,GAAA,CAAI,SAAS,GAAG,CAAA;AAEvB,IAAA,OAAO,CAAC,GAAA,EAAK,OAAA,CAAQ,MAAM,CAAA;AAAA,EAC7B;AAAA,EAEA,MAAM,OAAO,OAAA,EAAgD;AAC3D,IAAA,IAAI,CAAC,KAAK,WAAA,EAAa;AACrB,MAAA,MAAM,KAAK,IAAA,EAAK;AAAA,IAClB;AAEA,IAAA,MAAM,QAAA,GAAW;AAAA,MACf,SAAA,EAAW,OAAA,CAAQ,OAAA,CAAQ,GAAA,CAAI,YAAY,CAAA,IAAK,EAAA;AAAA,MAChD,SAAA,EACE,OAAA,CAAQ,EAAA,IACR,OAAA,CAAQ,OAAA,CAAQ,GAAA,CAAI,iBAAiB,CAAA,EAAG,KAAA,CAAM,GAAG,CAAA,CAAE,CAAC,CAAA,IACpD,EAAA;AAAA,MACF,SAAS,MAAA,CAAO,WAAA,CAAY,OAAA,CAAQ,OAAA,CAAQ,SAAS;AAAA,KACvD;AAGA,IAAA,IAAI,IAAA,CAAK,YAAA,IAAgB,IAAA,CAAK,MAAA,CAAO,eAAe,KAAA,EAAO;AACzD,MAAA,IAAI;AACF,QAAA,MAAM,OAAA,GAAU,KAAK,YAAA,CAAa,OAAA;AAClC,QAAA,MAAM,CAAC,KAAK,GAAG,CAAA,GAAI,KAAK,WAAA,CAAY,IAAA,CAAK,SAAA,CAAU,QAAQ,CAAC,CAAA;AAE5D,QAAA,MAAM,SAAA,GAAY,OAAA,CAAQ,YAAA,CAAa,GAAA,EAAK,GAAG,CAAA;AAE/C,QAAA,IAAI,SAAA,GAAY,EAAA;AAChB,QAAA,IAAI,MAAA,GAAS,CAAA;AACb,QAAA,MAAM,SAAS,IAAI,UAAA;AAAA,UAChB,QAAQ,MAAA,CAA8B;AAAA,SACzC;AAGA,QAAA,OAAO,SAAS,IAAA,EAAM;AAEpB,UAAA,MAAM,IAAA,GAAO,MAAA,CAAO,SAAA,GAAY,MAAM,CAAA;AACtC,UAAA,IAAI,IAAA,KAAS,CAAA,IAAK,IAAA,KAAS,KAAA,CAAA,EAAW;AACtC,UAAA,MAAA,EAAA;AAAA,QACF;AAEA,QAAA,IAAI,SAAS,CAAA,EAAG;AACd,UAAA,SAAA,GAAY,IAAA,CAAK,UAAA,CAAW,SAAA,EAAW,MAAM,CAAA;AAAA,QAC/C;AAEA,QAAA,OAAA,CAAQ,eAAA,CAAgB,KAAK,GAAG,CAAA;AAChC,QAAA,IAAI,SAAS,CAAA,EAAG;AACd,UAAA,OAAA,CAAQ,eAAA,CAAgB,WAAW,MAAM,CAAA;AAAA,QAC3C;AAEA,QAAA,MAAM,MAAA,GAAS,IAAA,CAAK,KAAA,CAAM,SAAS,CAAA;AAEnC,QAAA,MAAM,SAAA,GAA6B;AAAA,UACjC,GAAG,MAAA;AAAA,UACH,kBAAA,EAAoB,eAAA;AAAA,UACpB,SAAA,EAAA,iBAAW,IAAI,IAAA,EAAK,EAAE,WAAA;AAAY,SACpC;AAEA,QAAA,IAAI,IAAA,CAAK,MAAA,CAAO,eAAA,IAAmB,SAAA,CAAU,OAAA,EAAS;AACpD,UAAA,IAAA,CAAK,MAAA,CAAO,gBAAgB,SAAS,CAAA;AAAA,QACvC;AAEA,QAAA,OAAO,SAAA;AAAA,MACT,SAAS,KAAA,EAAO;AACd,QAAA,IAAI,IAAA,CAAK,OAAO,KAAA,EAAO;AACrB,UAAA,OAAA,CAAQ,KAAA,CAAM,0BAA0B,KAAK,CAAA;AAAA,QAC/C;AAAA,MAEF;AAAA,IACF;AAGA,IAAA,OAAO,IAAA,CAAK,iBAAiB,QAAQ,CAAA;AAAA,EACvC;AAAA,EAEQ,iBAAiB,QAAA,EAIL;AAClB,IAAA,MAAM,SAAA,GAAY,QAAA,CAAS,SAAA,CAAU,WAAA,EAAY;AAGjD,IAAA,MAAM,QAAA,GAAW;AAAA;AAAA,MAEf,EAAE,OAAA,EAAS,eAAA,EAAiB,IAAA,EAAM,SAAA,EAAW,YAAY,IAAA,EAAK;AAAA,MAC9D,EAAE,OAAA,EAAS,aAAA,EAAe,IAAA,EAAM,QAAA,EAAU,YAAY,IAAA,EAAK;AAAA,MAC3D,EAAE,OAAA,EAAS,cAAA,EAAgB,IAAA,EAAM,aAAA,EAAe,YAAY,IAAA,EAAK;AAAA;AAAA,MAGjE,EAAE,OAAA,EAAS,YAAA,EAAc,IAAA,EAAM,WAAA,EAAa,YAAY,GAAA,EAAI;AAAA,MAC5D,EAAE,OAAA,EAAS,SAAA,EAAW,IAAA,EAAM,QAAA,EAAU,YAAY,GAAA,EAAI;AAAA;AAAA,MAGtD,EAAE,OAAA,EAAS,UAAA,EAAY,IAAA,EAAM,gBAAA,EAAkB,YAAY,IAAA,EAAK;AAAA,MAChE,EAAE,OAAA,EAAS,OAAA,EAAS,IAAA,EAAM,aAAA,EAAe,YAAY,IAAA,EAAK;AAAA,MAC1D,EAAE,OAAA,EAAS,SAAA,EAAW,IAAA,EAAM,eAAA,EAAiB,YAAY,IAAA,EAAK;AAAA,MAC9D,EAAE,OAAA,EAAS,aAAA,EAAe,IAAA,EAAM,YAAA,EAAc,YAAY,IAAA,EAAK;AAAA,MAC/D,EAAE,OAAA,EAAS,WAAA,EAAa,IAAA,EAAM,SAAA,EAAW,YAAY,GAAA,EAAI;AAAA,MACzD,EAAE,OAAA,EAAS,QAAA,EAAU,IAAA,EAAM,OAAA,EAAS,YAAY,GAAA;AAAI,KACtD;AAGA,IAAA,MAAM,iBAAA,GAAoB;AAAA,MACxB,WAAA;AAAA,MACA,cAAA;AAAA,MACA,OAAA;AAAA,MACA,QAAA;AAAA,MACA;AAAA,KACF;AAEA,IAAA,IAAI,WAAA,GAAc,CAAA;AAClB,IAAA,KAAA,MAAW,CAAC,GAAG,CAAA,IAAK,OAAO,OAAA,CAAQ,QAAA,CAAS,OAAO,CAAA,EAAG;AACpD,MAAA,IACE,iBAAA,CAAkB,KAAK,CAAA,MAAA,KAAU,GAAA,CAAI,aAAY,CAAE,UAAA,CAAW,MAAM,CAAC,CAAA,EACrE;AACA,QAAA,WAAA,GAAc,GAAA;AACd,QAAA;AAAA,MACF;AAAA,IACF;AAGA,IAAA,KAAA,MAAW,EAAE,OAAA,EAAS,IAAA,EAAM,UAAA,MAAgB,QAAA,EAAU;AACpD,MAAA,IAAI,OAAA,CAAQ,IAAA,CAAK,SAAS,CAAA,EAAG;AAC3B,QAAA,MAAM,eAAA,GAAkB,IAAA,CAAK,GAAA,CAAI,UAAA,GAAa,aAAa,CAAG,CAAA;AAE9D,QAAA,MAAM,MAAA,GAA0B;AAAA,UAC9B,OAAA,EAAS,IAAA;AAAA,UACT,UAAA,EAAY,eAAA;AAAA,UACZ,KAAA,EAAO,IAAA;AAAA,UACP,kBAAA,EAAoB,SAAA;AAAA,UACpB,SAAA,EAAW,eAAA,GAAkB,GAAA,GAAM,MAAA,GAAS,QAAA;AAAA,UAC5C,SAAA,EAAA,iBAAW,IAAI,IAAA,EAAK,EAAE,WAAA;AAAY,SACpC;AAEA,QAAA,IAAI,IAAA,CAAK,OAAO,eAAA,EAAiB;AAC/B,UAAA,IAAA,CAAK,MAAA,CAAO,gBAAgB,MAAM,CAAA;AAAA,QACpC;AAEA,QAAA,OAAO,MAAA;AAAA,MACT;AAAA,IACF;AAGA,IAAA,OAAO;AAAA,MACL,OAAA,EAAS,KAAA;AAAA,MACT,UAAA,EAAY,IAAA;AAAA,MACZ,kBAAA,EAAoB,SAAA;AAAA,MACpB,SAAA,EAAA,iBAAW,IAAI,IAAA,EAAK,EAAE,WAAA;AAAY,KACpC;AAAA,EACF;AAAA,EAEA,aAAA,GAAyB;AACvB,IAAA,OAAO,IAAA,CAAK,WAAA;AAAA,EACd;AAAA,EAEA,qBAAA,GAAqD;AACnD,IAAA,OAAO,IAAA,CAAK,eAAe,eAAA,GAAkB,SAAA;AAAA,EAC/C;AACF,CAAA;AAKO,SAAS,sBAAsB,MAAA,EAA4B;AAChE,EAAA,OAAO,IAAI,uBAAuB,MAAM,CAAA;AAC1C;AAKA,IAAI,eAAA,GAAiD,IAAA;AAE9C,SAAS,sBACd,MAAA,EACwB;AACxB,EAAA,IAAI,CAAC,eAAA,EAAiB;AACpB,IAAA,eAAA,GAAkB,IAAI,uBAAuB,MAAM,CAAA;AAAA,EACrD;AACA,EAAA,OAAO,eAAA;AACT;AAEA,IAAO,2BAAA,GAAQ","file":"edge-runtime-loader.mjs","sourcesContent":["/**\n * Edge Runtime Compatible WASM Loader for AgentShield\n *\n * This module provides a pre-built solution for loading WASM in Edge Runtime.\n * It requires the WASM file to be manually placed in the project.\n */\n\nimport type { NextRequest } from 'next/server';\n\n// User must provide the WASM module via import\n// This is intentionally not imported here to allow user control\nexport interface WasmModule {\n  default: WebAssembly.Module;\n}\n\nexport interface DetectionResult {\n  isAgent: boolean;\n  confidence: number;\n  agent?: string;\n  verificationMethod: 'cryptographic' | 'pattern';\n  riskLevel?: 'low' | 'medium' | 'high' | 'critical';\n  timestamp: string;\n}\n\nexport interface AgentShieldConfig {\n  wasmModule?: WebAssembly.Module;\n  enableWasm?: boolean;\n  onAgentDetected?: (result: DetectionResult) => void;\n  blockAgents?: boolean;\n  allowedAgents?: string[];\n  debug?: boolean;\n}\n\ninterface WasmExports {\n  detect_agent(metadataPtr: number, metadataLen: number): number;\n  get_version(): number;\n  __wbindgen_malloc(size: number): number;\n  __wbindgen_free(ptr: number, size: number): void;\n  __wbindgen_realloc(ptr: number, oldSize: number, newSize: number): number;\n  memory: WebAssembly.Memory;\n}\n\nclass EdgeRuntimeAgentShield {\n  private wasmInstance: WebAssembly.Instance | null = null;\n  private wasmMemory: WebAssembly.Memory | null = null;\n  private config: AgentShieldConfig;\n  private initialized = false;\n\n  constructor(config: AgentShieldConfig = {}) {\n    this.config = config;\n  }\n\n  async init(wasmModule?: WebAssembly.Module): Promise<void> {\n    if (this.initialized) return;\n\n    const module = wasmModule || this.config.wasmModule;\n\n    if (module && this.config.enableWasm !== false) {\n      try {\n        this.wasmMemory = new WebAssembly.Memory({ initial: 17, maximum: 256 });\n\n        const imports = {\n          wbg: {\n            __wbindgen_throw: (ptr: number, len: number) => {\n              throw new Error(this.readString(ptr, len));\n            },\n            __wbg_log_: (ptr: number, len: number) => {\n              if (this.config.debug) {\n                console.log(this.readString(ptr, len));\n              }\n            },\n          },\n          env: {\n            memory: this.wasmMemory,\n          },\n        };\n\n        this.wasmInstance = await WebAssembly.instantiate(module, imports);\n        this.initialized = true;\n\n        if (this.config.debug) {\n          console.log('✅ AgentShield WASM initialized in Edge Runtime');\n        }\n      } catch (error) {\n        console.warn(\n          '⚠️ WASM initialization failed, using pattern detection:',\n          error\n        );\n        this.initialized = true;\n      }\n    } else {\n      this.initialized = true;\n    }\n  }\n\n  private readString(ptr: number, len: number): string {\n    if (!this.wasmInstance || !this.wasmMemory) {\n      throw new Error('WASM not initialized');\n    }\n    const memory = new Uint8Array(this.wasmMemory.buffer);\n    const bytes = memory.slice(ptr, ptr + len);\n    return new TextDecoder().decode(bytes);\n  }\n\n  private writeString(str: string): [number, number] {\n    if (!this.wasmInstance) {\n      throw new Error('WASM not initialized');\n    }\n    const exports = this.wasmInstance.exports as unknown as WasmExports;\n\n    const encoded = new TextEncoder().encode(str);\n    const ptr = exports.__wbindgen_malloc(encoded.length);\n\n    const memory = new Uint8Array(exports.memory.buffer);\n    memory.set(encoded, ptr);\n\n    return [ptr, encoded.length];\n  }\n\n  async detect(request: NextRequest): Promise<DetectionResult> {\n    if (!this.initialized) {\n      await this.init();\n    }\n\n    const metadata = {\n      userAgent: request.headers.get('user-agent') || '',\n      ipAddress:\n        request.ip ||\n        request.headers.get('x-forwarded-for')?.split(',')[0] ||\n        '',\n      headers: Object.fromEntries(request.headers.entries()),\n    };\n\n    // Try WASM detection first\n    if (this.wasmInstance && this.config.enableWasm !== false) {\n      try {\n        const exports = this.wasmInstance.exports as unknown as WasmExports;\n        const [ptr, len] = this.writeString(JSON.stringify(metadata));\n\n        const resultPtr = exports.detect_agent(ptr, len);\n        // Read result with dynamic size\n        let resultStr = '';\n        let offset = 0;\n        const memory = new Uint8Array(\n          (exports.memory as WebAssembly.Memory).buffer\n        );\n\n        // Read until we find the end of the JSON string\n        while (offset < 4096) {\n          // Max 4KB result\n          const byte = memory[resultPtr + offset];\n          if (byte === 0 || byte === undefined) break;\n          offset++;\n        }\n\n        if (offset > 0) {\n          resultStr = this.readString(resultPtr, offset);\n        }\n\n        exports.__wbindgen_free(ptr, len);\n        if (offset > 0) {\n          exports.__wbindgen_free(resultPtr, offset);\n        }\n\n        const result = JSON.parse(resultStr);\n\n        const detection: DetectionResult = {\n          ...result,\n          verificationMethod: 'cryptographic',\n          timestamp: new Date().toISOString(),\n        };\n\n        if (this.config.onAgentDetected && detection.isAgent) {\n          this.config.onAgentDetected(detection);\n        }\n\n        return detection;\n      } catch (error) {\n        if (this.config.debug) {\n          console.error('WASM detection failed:', error);\n        }\n        // Fall through to pattern detection\n      }\n    }\n\n    // Fallback: Pattern-based detection\n    return this.patternDetection(metadata);\n  }\n\n  private patternDetection(metadata: {\n    userAgent: string;\n    ipAddress: string;\n    headers: Record<string, string>;\n  }): DetectionResult {\n    const userAgent = metadata.userAgent.toLowerCase();\n\n    // Enhanced AI agent patterns with confidence scores\n    const patterns = [\n      // High confidence - explicit AI identifiers\n      { pattern: /chatgpt-user/i, name: 'ChatGPT', confidence: 0.95 },\n      { pattern: /claude-web/i, name: 'Claude', confidence: 0.95 },\n      { pattern: /gpt-crawler/i, name: 'GPT Crawler', confidence: 0.95 },\n\n      // Medium-high confidence - company identifiers\n      { pattern: /anthropic/i, name: 'Anthropic', confidence: 0.9 },\n      { pattern: /openai/i, name: 'OpenAI', confidence: 0.9 },\n\n      // Medium confidence - product names\n      { pattern: /copilot/i, name: 'GitHub Copilot', confidence: 0.85 },\n      { pattern: /bard/i, name: 'Google Bard', confidence: 0.85 },\n      { pattern: /gemini/i, name: 'Google Gemini', confidence: 0.85 },\n      { pattern: /perplexity/i, name: 'Perplexity', confidence: 0.85 },\n      { pattern: /you\\.com/i, name: 'You.com', confidence: 0.8 },\n      { pattern: /phind/i, name: 'Phind', confidence: 0.8 },\n    ];\n\n    // Check for suspicious headers\n    const suspiciousHeaders = [\n      'x-openai-',\n      'x-anthropic-',\n      'x-ai-',\n      'x-llm-',\n      'x-gpt-',\n    ];\n\n    let headerBoost = 0;\n    for (const [key] of Object.entries(metadata.headers)) {\n      if (\n        suspiciousHeaders.some(prefix => key.toLowerCase().startsWith(prefix))\n      ) {\n        headerBoost = 0.1;\n        break;\n      }\n    }\n\n    // Check patterns\n    for (const { pattern, name, confidence } of patterns) {\n      if (pattern.test(userAgent)) {\n        const finalConfidence = Math.min(confidence + headerBoost, 1.0);\n\n        const result: DetectionResult = {\n          isAgent: true,\n          confidence: finalConfidence,\n          agent: name,\n          verificationMethod: 'pattern',\n          riskLevel: finalConfidence > 0.9 ? 'high' : 'medium',\n          timestamp: new Date().toISOString(),\n        };\n\n        if (this.config.onAgentDetected) {\n          this.config.onAgentDetected(result);\n        }\n\n        return result;\n      }\n    }\n\n    // Not detected\n    return {\n      isAgent: false,\n      confidence: 0.85,\n      verificationMethod: 'pattern',\n      timestamp: new Date().toISOString(),\n    };\n  }\n\n  isInitialized(): boolean {\n    return this.initialized;\n  }\n\n  getVerificationMethod(): 'cryptographic' | 'pattern' {\n    return this.wasmInstance ? 'cryptographic' : 'pattern';\n  }\n}\n\n/**\n * Factory function to create an AgentShield instance for Edge Runtime\n */\nexport function createEdgeAgentShield(config?: AgentShieldConfig) {\n  return new EdgeRuntimeAgentShield(config);\n}\n\n/**\n * Singleton instance for simple usage\n */\nlet defaultInstance: EdgeRuntimeAgentShield | null = null;\n\nexport function getDefaultAgentShield(\n  config?: AgentShieldConfig\n): EdgeRuntimeAgentShield {\n  if (!defaultInstance) {\n    defaultInstance = new EdgeRuntimeAgentShield(config);\n  }\n  return defaultInstance;\n}\n\nexport default createEdgeAgentShield;\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kya-os/agentshield-nextjs",
-  "version": "0.1.18",
+  "version": "0.1.19",
   "description": "Next.js middleware for AgentShield AI agent detection",
   "keywords": [
     "nextjs",
@@ -25,12 +25,20 @@
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
   "types": "./dist/index.d.ts",
+  "bin": {
+    "agentshield-setup-edge": "./bin/setup-edge-wasm.js"
+  },
   "exports": {
     ".": {
       "types": "./dist/index.d.ts",
       "import": "./dist/index.mjs",
       "require": "./dist/index.js"
     },
+    "./edge": {
+      "types": "./dist/edge-runtime-loader.d.ts",
+      "import": "./dist/edge-runtime-loader.mjs",
+      "require": "./dist/edge-runtime-loader.js"
+    },
     "./middleware": {
       "types": "./dist/middleware.d.ts",
       "import": "./dist/middleware.mjs",
@@ -45,9 +53,11 @@
   },
   "files": [
     "dist",
+    "bin",
     "wasm.d.ts",
     "README.md",
-    "CHANGELOG.md"
+    "CHANGELOG.md",
+    "EDGE_RUNTIME_WASM_SETUP.md"
   ],
   "engines": {
     "node": ">=18.0.0"
@@ -91,6 +101,6 @@
   },
   "sideEffects": false,
   "dependencies": {
-    "@kya-os/agentshield": "^0.1.18"
+    "@kya-os/agentshield": "^0.1.19"
   }
-}
+}