basileus-agentkit-plugin 0.1.0

package/src/types.ts

@@ -0,0 +1,17 @@
+export interface ToolExecution {
+  name: string;
+  args?: Record<string, string>;
+  result?: string;
+  txHash?: string;
+  meta?: Record<string, string>;
+}
+
+export type ActivityType = "inventory" | "survival" | "strategy" | "error";
+
+export interface AgentActivity {
+  summary: string;
+  model: string;
+  cycleId: string;
+  tools?: ToolExecution[];
+  txHashes?: string[];
+}

package/src/wallet.ts

@@ -0,0 +1,73 @@
+import {
+  createWalletClient,
+  createPublicClient,
+  http,
+  formatUnits,
+  type Hex,
+  type Chain,
+} from "viem";
+import { privateKeyToAccount } from "viem/accounts";
+import { erc20Abi } from "viem";
+import { ViemWalletProvider } from "@coinbase/agentkit";
+import { Attribution } from "ox/erc8021";
+
+const BUILDER_CODE = "bc_kj26kx76";
+
+// USDC on Base mainnet
+const USDC_BASE = "0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913" as const;
+// USDC on Base Sepolia
+const USDC_BASE_SEPOLIA = "0x036CbD53842c5426634e7929541eC2318f3dCF7e" as const;
+
+export interface WalletInfo {
+  address: string;
+  ethBalance: string;
+  usdcBalance: string;
+  chainName: string;
+}
+
+export async function createAgentWallet(privateKey: Hex, chain: Chain, rpcUrl?: string) {
+  const account = privateKeyToAccount(privateKey);
+  const dataSuffix = Attribution.toDataSuffix({ codes: [BUILDER_CODE] });
+
+  const walletClient = createWalletClient({
+    account,
+    chain,
+    transport: http(rpcUrl),
+    dataSuffix,
+  });
+
+  const publicClient = createPublicClient({
+    chain,
+    transport: http(rpcUrl),
+  });
+
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  const provider = new ViemWalletProvider(walletClient as any);
+
+  const usdcAddress = chain.id === 8453 ? USDC_BASE : USDC_BASE_SEPOLIA;
+
+  return { walletClient, publicClient, provider, account, usdcAddress };
+}
+
+export async function getBalances(
+  wallet: Awaited<ReturnType<typeof createAgentWallet>>,
+): Promise<WalletInfo> {
+  const { publicClient, account, usdcAddress, walletClient } = wallet;
+
+  const [ethBal, usdcBal] = await Promise.all([
+    publicClient.getBalance({ address: account.address }),
+    publicClient.readContract({
+      address: usdcAddress,
+      abi: erc20Abi,
+      functionName: "balanceOf",
+      args: [account.address],
+    }),
+  ]);
+
+  return {
+    address: account.address,
+    ethBalance: formatUnits(ethBal, 18),
+    usdcBalance: formatUnits(usdcBal as bigint, 6),
+    chainName: walletClient.chain?.name ?? "unknown",
+  };
+}

package/src/x402-tracker.ts

@@ -0,0 +1,43 @@
+let paymentTxHashes: string[] = [];
+let installed = false;
+
+export function installX402Tracker(): void {
+  if (installed) return;
+  const originalFetch = globalThis.fetch;
+
+  globalThis.fetch = async (...args: Parameters<typeof fetch>): Promise<Response> => {
+    const response = await originalFetch(...args);
+
+    const receipt = response.headers.get("x-payment-receipt");
+    if (receipt && /^0x[a-fA-F0-9]{64}$/.test(receipt)) {
+      paymentTxHashes.push(receipt);
+      console.log(`[x402] Captured payment tx: ${receipt.slice(0, 14)}...`);
+      return response;
+    }
+
+    const paymentHeader = response.headers.get("payment-response");
+    if (paymentHeader) {
+      try {
+        const decoded = JSON.parse(atob(paymentHeader));
+        const txHash = decoded.transaction ?? decoded.txHash;
+        if (txHash && typeof txHash === "string") {
+          paymentTxHashes.push(txHash);
+          console.log(`[x402] Captured payment tx: ${txHash.slice(0, 14)}...`);
+        }
+      } catch {
+        console.warn("[x402] Failed to decode payment-response header");
+      }
+    }
+
+    return response;
+  };
+
+  installed = true;
+  console.log("[x402] Fetch interceptor installed");
+}
+
+export function drainX402TxHashes(): string[] {
+  const hashes = [...paymentTxHashes];
+  paymentTxHashes = [];
+  return hashes;
+}

package/tsconfig.json

@@ -0,0 +1,15 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "NodeNext",
+    "moduleResolution": "NodeNext",
+    "strict": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "outDir": "dist",
+    "rootDir": "src",
+    "declaration": true,
+    "sourceMap": true
+  },
+  "include": ["src/**/*"]
+}