@obelyzk/sdk 1.0.1 → 1.2.0

package/dist/chunk-BHUXKNDT.mjs

@@ -0,0 +1,206 @@
+// src/firewall/client.ts
+import { RpcProvider, Contract, CallData } from "starknet";
+var AgentFirewallSDK = class {
+  config;
+  provider;
+  constructor(config) {
+    this.config = config;
+    this.provider = new RpcProvider({ nodeUrl: config.rpcUrl });
+  }
+  // ── Classification ─────────────────────────────────────────────────
+  /**
+   * Classify a transaction through the ZKML classifier.
+   *
+   * Sends the transaction features to the prove-server, which runs
+   * the MLP classifier and generates a GKR+STARK proof. Returns the
+   * proven threat score and decision.
+   *
+   * Does NOT submit anything on-chain — use `evaluateAction()` for
+   * the full flow including on-chain submission.
+   */
+  async classify(tx) {
+    const headers = {
+      "Content-Type": "application/json"
+    };
+    if (this.config.apiKey) {
+      headers["Authorization"] = `Bearer ${this.config.apiKey}`;
+    }
+    const body = {
+      target: tx.target,
+      value: tx.value || "0",
+      selector: tx.selector || "0x0",
+      calldata: tx.calldata || "0x",
+      agent_trust_score: tx.agentTrustScore || 0,
+      agent_strikes: tx.agentStrikes || 0,
+      agent_age_blocks: tx.agentAgeBlocks || 0,
+      target_verified: tx.targetVerified || false,
+      target_is_proxy: tx.targetIsProxy || false,
+      target_has_source: tx.targetHasSource || false,
+      target_interaction_count: tx.targetInteractionCount || 0,
+      tx_frequency: tx.txFrequency || 0,
+      unique_targets_24h: tx.uniqueTargets24h || 0,
+      avg_value_24h: tx.avgValue24h || 0,
+      max_value_24h: tx.maxValue24h || 0
+    };
+    const response = await fetch(`${this.config.proverUrl}/api/v1/classify`, {
+      method: "POST",
+      headers,
+      body: JSON.stringify(body)
+    });
+    if (!response.ok) {
+      const error = await response.json().catch(() => ({ error: response.statusText }));
+      throw new Error(`Classification failed (${response.status}): ${error.error || response.statusText}`);
+    }
+    const data = await response.json();
+    return {
+      requestId: data.request_id,
+      decision: data.decision,
+      threatScore: data.threat_score,
+      scores: data.scores,
+      ioCommitment: data.io_commitment,
+      policyCommitment: data.policy_commitment,
+      proveTimeMs: data.prove_time_ms
+    };
+  }
+  // ── Agent Management ───────────────────────────────────────────────
+  /**
+   * Register a new agent on the firewall contract.
+   * The calling account becomes the agent owner.
+   */
+  async registerAgent(agentId) {
+    this.requireAccount();
+    const tx = await this.config.account.execute({
+      contractAddress: this.config.firewallContract,
+      entrypoint: "register_agent",
+      calldata: CallData.compile({ agent_id: agentId })
+    });
+    await this.provider.waitForTransaction(tx.transaction_hash);
+    return tx.transaction_hash;
+  }
+  /** Deactivate an agent (owner or contract admin). */
+  async deactivateAgent(agentId) {
+    this.requireAccount();
+    const tx = await this.config.account.execute({
+      contractAddress: this.config.firewallContract,
+      entrypoint: "deactivate_agent",
+      calldata: CallData.compile({ agent_id: agentId })
+    });
+    await this.provider.waitForTransaction(tx.transaction_hash);
+    return tx.transaction_hash;
+  }
+  /** Reactivate an agent and reset strikes (agent owner only). */
+  async reactivateAgent(agentId) {
+    this.requireAccount();
+    const tx = await this.config.account.execute({
+      contractAddress: this.config.firewallContract,
+      entrypoint: "reactivate_agent",
+      calldata: CallData.compile({ agent_id: agentId })
+    });
+    await this.provider.waitForTransaction(tx.transaction_hash);
+    return tx.transaction_hash;
+  }
+  // ── Queries ────────────────────────────────────────────────────────
+  /** Get the full status of an agent. */
+  async getAgentStatus(agentId) {
+    const contract = new Contract(
+      FIREWALL_ABI,
+      this.config.firewallContract,
+      this.provider
+    );
+    const [registered, active, trustScore, strikes, trusted] = await Promise.all([
+      contract.is_agent_registered(agentId),
+      contract.is_agent_active(agentId),
+      contract.get_trust_score(agentId),
+      contract.get_strikes(agentId),
+      contract.is_trusted(agentId)
+    ]);
+    return {
+      registered,
+      active,
+      trustScore: Number(trustScore),
+      strikes: Number(strikes),
+      trusted
+    };
+  }
+  /** Check if a specific action has been approved. */
+  async isActionApproved(actionId) {
+    const contract = new Contract(
+      FIREWALL_ABI,
+      this.config.firewallContract,
+      this.provider
+    );
+    return contract.is_action_approved(actionId);
+  }
+  /** Check if an agent is trusted. */
+  async isAgentTrusted(agentId) {
+    const contract = new Contract(
+      FIREWALL_ABI,
+      this.config.firewallContract,
+      this.provider
+    );
+    return contract.is_trusted(agentId);
+  }
+  // ── Helpers ────────────────────────────────────────────────────────
+  requireAccount() {
+    if (!this.config.account) {
+      throw new Error(
+        "Account required for write operations. Pass `account` in FirewallConfig."
+      );
+    }
+  }
+};
+var FIREWALL_ABI = [
+  {
+    name: "is_agent_registered",
+    type: "function",
+    inputs: [{ name: "agent_id", type: "felt" }],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "is_agent_active",
+    type: "function",
+    inputs: [{ name: "agent_id", type: "felt" }],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "get_trust_score",
+    type: "function",
+    inputs: [{ name: "agent_id", type: "felt" }],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "get_strikes",
+    type: "function",
+    inputs: [{ name: "agent_id", type: "felt" }],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "is_trusted",
+    type: "function",
+    inputs: [{ name: "agent_id", type: "felt" }],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "is_action_approved",
+    type: "function",
+    inputs: [{ name: "action_id", type: "felt" }],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "get_action_decision",
+    type: "function",
+    inputs: [{ name: "action_id", type: "felt" }],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  }
+];
+
+export {
+  AgentFirewallSDK
+};