drain-mcp 0.1.0

package/dist/services/wallet.js

@@ -0,0 +1,102 @@
+/**
+ * Wallet Service
+ *
+ * Handles wallet operations: balance, approvals, signing.
+ */
+import { formatUnits, parseUnits } from 'viem';
+import { USDC_DECIMALS, ERC20_ABI } from '../constants.js';
+export class WalletService {
+    publicClient;
+    walletClient;
+    account;
+    config;
+    constructor(publicClient, walletClient, account, config) {
+        this.publicClient = publicClient;
+        this.walletClient = walletClient;
+        this.account = account;
+        this.config = config;
+    }
+    /**
+     * Get the agent's wallet address
+     */
+    getAddress() {
+        return this.account.address;
+    }
+    /**
+     * Get USDC balance
+     */
+    async getUsdcBalance() {
+        const balance = await this.publicClient.readContract({
+            address: this.config.usdcAddress,
+            abi: ERC20_ABI,
+            functionName: 'balanceOf',
+            args: [this.account.address],
+        });
+        return {
+            raw: balance,
+            formatted: formatUnits(balance, USDC_DECIMALS),
+        };
+    }
+    /**
+     * Get current USDC allowance for DRAIN contract
+     */
+    async getAllowance() {
+        const allowance = await this.publicClient.readContract({
+            address: this.config.usdcAddress,
+            abi: ERC20_ABI,
+            functionName: 'allowance',
+            args: [this.account.address, this.config.drainAddress],
+        });
+        return {
+            raw: allowance,
+            formatted: formatUnits(allowance, USDC_DECIMALS),
+        };
+    }
+    /**
+     * Approve USDC spending for DRAIN contract
+     */
+    async approveUsdc(amount) {
+        const amountWei = parseUnits(amount, USDC_DECIMALS);
+        const hash = await this.walletClient.writeContract({
+            account: this.account,
+            address: this.config.usdcAddress,
+            abi: ERC20_ABI,
+            functionName: 'approve',
+            args: [this.config.drainAddress, amountWei],
+            chain: this.config.chain,
+        });
+        // Wait for confirmation
+        await this.publicClient.waitForTransactionReceipt({ hash });
+        return hash;
+    }
+    /**
+     * Approve maximum USDC spending (for convenience)
+     */
+    async approveMax() {
+        const maxAmount = BigInt('0xffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff');
+        const hash = await this.walletClient.writeContract({
+            account: this.account,
+            address: this.config.usdcAddress,
+            abi: ERC20_ABI,
+            functionName: 'approve',
+            args: [this.config.drainAddress, maxAmount],
+            chain: this.config.chain,
+        });
+        // Wait for confirmation
+        await this.publicClient.waitForTransactionReceipt({ hash });
+        return hash;
+    }
+    /**
+     * Get native token (POL/MATIC) balance for gas
+     */
+    async getNativeBalance() {
+        const balance = await this.publicClient.getBalance({
+            address: this.account.address,
+        });
+        return {
+            raw: balance,
+            formatted: formatUnits(balance, 18),
+        };
+    }
+}
+//# sourceMappingURL=wallet.js.map

package/dist/services/wallet.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wallet.js","sourceRoot":"","sources":["../../src/services/wallet.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,WAAW,EAAE,UAAU,EAA+E,MAAM,MAAM,CAAC;AAC5H,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAG3D,MAAM,OAAO,aAAa;IAEd;IACA;IACA;IACA;IAJV,YACU,YAA0B,EAC1B,YAA0B,EAC1B,OAAgB,EAChB,MAAmB;QAHnB,iBAAY,GAAZ,YAAY,CAAc;QAC1B,iBAAY,GAAZ,YAAY,CAAc;QAC1B,YAAO,GAAP,OAAO,CAAS;QAChB,WAAM,GAAN,MAAM,CAAa;IAC1B,CAAC;IAEJ;;OAEG;IACH,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc;QAClB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC;YACnD,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;YAChC,GAAG,EAAE,SAAS;YACd,YAAY,EAAE,WAAW;YACzB,IAAI,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC;SAC7B,CAAW,CAAC;QAEb,OAAO;YACL,GAAG,EAAE,OAAO;YACZ,SAAS,EAAE,WAAW,CAAC,OAAO,EAAE,aAAa,CAAC;SAC/C,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY;QAChB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC;YACrD,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;YAChC,GAAG,EAAE,SAAS;YACd,YAAY,EAAE,WAAW;YACzB,IAAI,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;SACvD,CAAW,CAAC;QAEb,OAAO;YACL,GAAG,EAAE,SAAS;YACd,SAAS,EAAE,WAAW,CAAC,SAAS,EAAE,aAAa,CAAC;SACjD,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,MAAc;QAC9B,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAEpD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC;YACjD,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;YAChC,GAAG,EAAE,SAAS;YACd,YAAY,EAAE,SAAS;YACvB,IAAI,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,SAAS,CAAC;YAC3C,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;SACzB,CAAC,CAAC;QAEH,wBAAwB;QACxB,MAAM,IAAI,CAAC,YAAY,CAAC,yBAAyB,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QAE5D,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,MAAM,SAAS,GAAG,MAAM,CAAC,oEAAoE,CAAC,CAAC;QAE/F,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC;YACjD,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;YAChC,GAAG,EAAE,SAAS;YACd,YAAY,EAAE,SAAS;YACvB,IAAI,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,SAAS,CAAC;YAC3C,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;SACzB,CAAC,CAAC;QAEH,wBAAwB;QACxB,MAAM,IAAI,CAAC,YAAY,CAAC,yBAAyB,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QAE5D,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB;QACpB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC;YACjD,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;SAC9B,CAAC,CAAC;QAEH,OAAO;YACL,GAAG,EAAE,OAAO;YACZ,SAAS,EAAE,WAAW,CAAC,OAAO,EAAE,EAAE,CAAC;SACpC,CAAC;IACJ,CAAC;CACF"}

package/dist/tools/balance.d.ts

@@ -0,0 +1,40 @@
+/**
+ * Balance Tools
+ *
+ * Check wallet balance and status.
+ */
+import type { WalletService } from '../services/wallet.js';
+import type { DrainConfig } from '../config.js';
+/**
+ * Get wallet balance and status
+ */
+export declare function getBalance(walletService: WalletService, config: DrainConfig): Promise<string>;
+/**
+ * Approve USDC spending
+ */
+export declare function approveUsdc(walletService: WalletService, args: {
+    amount?: string;
+}): Promise<string>;
+export declare const balanceTools: ({
+    name: string;
+    description: string;
+    inputSchema: {
+        type: string;
+        properties: {
+            amount?: undefined;
+        };
+    };
+} | {
+    name: string;
+    description: string;
+    inputSchema: {
+        type: string;
+        properties: {
+            amount: {
+                type: string;
+                description: string;
+            };
+        };
+    };
+})[];
+//# sourceMappingURL=balance.d.ts.map

package/dist/tools/balance.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"balance.d.ts","sourceRoot":"","sources":["../../src/tools/balance.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAEhD;;GAEG;AACH,wBAAsB,UAAU,CAC9B,aAAa,EAAE,aAAa,EAC5B,MAAM,EAAE,WAAW,GAClB,OAAO,CAAC,MAAM,CAAC,CA4BjB;AAED;;GAEG;AACH,wBAAsB,WAAW,CAC/B,aAAa,EAAE,aAAa,EAC5B,IAAI,EAAE;IAAE,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,GACxB,OAAO,CAAC,MAAM,CAAC,CAUjB;AAGD,eAAO,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;IAkCxB,CAAC"}

package/dist/tools/balance.js

@@ -0,0 +1,86 @@
+/**
+ * Balance Tools
+ *
+ * Check wallet balance and status.
+ */
+/**
+ * Get wallet balance and status
+ */
+export async function getBalance(walletService, config) {
+    const address = walletService.getAddress();
+    const usdcBalance = await walletService.getUsdcBalance();
+    const nativeBalance = await walletService.getNativeBalance();
+    const allowance = await walletService.getAllowance();
+    const network = config.chainId === 137 ? 'Polygon Mainnet' : 'Polygon Amoy Testnet';
+    const hasGas = parseFloat(nativeBalance.formatted) > 0.01;
+    const hasAllowance = parseFloat(allowance.formatted) > 0;
+    return `# Wallet Status
+
+**Address:** \`${address}\`
+**Network:** ${network}
+
+## Balances
+- **USDC:** $${usdcBalance.formatted}
+- **POL (for gas):** ${nativeBalance.formatted} POL
+
+## DRAIN Contract
+- **Allowance:** $${allowance.formatted} USDC
+- **Contract:** \`${config.drainAddress}\`
+
+## Status
+${parseFloat(usdcBalance.formatted) > 0 ? '✅' : '⚠️'} USDC Balance: ${parseFloat(usdcBalance.formatted) > 0 ? 'OK' : 'LOW - need USDC to open channels'}
+${hasGas ? '✅' : '⚠️'} Gas: ${hasGas ? 'OK' : 'LOW - need POL for transactions'}
+${hasAllowance ? '✅' : 'ℹ️'} Allowance: ${hasAllowance ? 'Approved' : 'Not yet approved - will need to approve before opening channel'}
+`;
+}
+/**
+ * Approve USDC spending
+ */
+export async function approveUsdc(walletService, args) {
+    let txHash;
+    if (args.amount) {
+        txHash = await walletService.approveUsdc(args.amount);
+        return `✅ Approved ${args.amount} USDC for DRAIN contract.\n\nTransaction: \`${txHash}\``;
+    }
+    else {
+        txHash = await walletService.approveMax();
+        return `✅ Approved unlimited USDC for DRAIN contract.\n\nTransaction: \`${txHash}\``;
+    }
+}
+// Tool definitions for MCP
+export const balanceTools = [
+    {
+        name: 'drain_balance',
+        description: `Check wallet balance, USDC allowance, and readiness for DRAIN payments.
+
+Use this to verify:
+- You have enough USDC to open channels
+- You have enough POL for gas fees  
+- USDC is approved for the DRAIN contract
+
+Returns: Wallet address, balances, and status indicators.`,
+        inputSchema: {
+            type: 'object',
+            properties: {},
+        },
+    },
+    {
+        name: 'drain_approve',
+        description: `Approve USDC spending for the DRAIN contract.
+
+This is required before opening payment channels. 
+If no amount is specified, approves unlimited spending (recommended for convenience).
+
+Returns: Transaction hash of the approval.`,
+        inputSchema: {
+            type: 'object',
+            properties: {
+                amount: {
+                    type: 'string',
+                    description: 'Amount of USDC to approve (e.g., "100"). If omitted, approves unlimited.',
+                },
+            },
+        },
+    },
+];
+//# sourceMappingURL=balance.js.map

package/dist/tools/balance.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"balance.js","sourceRoot":"","sources":["../../src/tools/balance.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,aAA4B,EAC5B,MAAmB;IAEnB,MAAM,OAAO,GAAG,aAAa,CAAC,UAAU,EAAE,CAAC;IAC3C,MAAM,WAAW,GAAG,MAAM,aAAa,CAAC,cAAc,EAAE,CAAC;IACzD,MAAM,aAAa,GAAG,MAAM,aAAa,CAAC,gBAAgB,EAAE,CAAC;IAC7D,MAAM,SAAS,GAAG,MAAM,aAAa,CAAC,YAAY,EAAE,CAAC;IAErD,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,KAAK,GAAG,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,sBAAsB,CAAC;IACpF,MAAM,MAAM,GAAG,UAAU,CAAC,aAAa,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;IAC1D,MAAM,YAAY,GAAG,UAAU,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAEzD,OAAO;;iBAEQ,OAAO;eACT,OAAO;;;eAGP,WAAW,CAAC,SAAS;uBACb,aAAa,CAAC,SAAS;;;oBAG1B,SAAS,CAAC,SAAS;oBACnB,MAAM,CAAC,YAAY;;;EAGrC,UAAU,CAAC,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,kBAAkB,UAAU,CAAC,WAAW,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,kCAAkC;EACrJ,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,SAAS,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,iCAAiC;EAC7E,YAAY,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,eAAe,YAAY,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,gEAAgE;CACrI,CAAC;AACF,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,aAA4B,EAC5B,IAAyB;IAEzB,IAAI,MAAc,CAAC;IAEnB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QAChB,MAAM,GAAG,MAAM,aAAa,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtD,OAAO,cAAc,IAAI,CAAC,MAAM,+CAA+C,MAAM,IAAI,CAAC;IAC5F,CAAC;SAAM,CAAC;QACN,MAAM,GAAG,MAAM,aAAa,CAAC,UAAU,EAAE,CAAC;QAC1C,OAAO,mEAAmE,MAAM,IAAI,CAAC;IACvF,CAAC;AACH,CAAC;AAED,2BAA2B;AAC3B,MAAM,CAAC,MAAM,YAAY,GAAG;IAC1B;QACE,IAAI,EAAE,eAAe;QACrB,WAAW,EAAE;;;;;;;0DAOyC;QACtD,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACf;KACF;IACD;QACE,IAAI,EAAE,eAAe;QACrB,WAAW,EAAE;;;;;2CAK0B;QACvC,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,0EAA0E;iBACxF;aACF;SACF;KACF;CACF,CAAC"}

package/dist/tools/channel.d.ts

@@ -0,0 +1,67 @@
+/**
+ * Channel Tools
+ *
+ * Manage payment channels: open, close, status.
+ */
+import type { ChannelService } from '../services/channel.js';
+import type { ProviderService } from '../services/provider.js';
+/**
+ * Open a payment channel
+ */
+export declare function openChannel(channelService: ChannelService, providerService: ProviderService, args: {
+    provider: string;
+    amount: string;
+    duration: string;
+}): Promise<string>;
+/**
+ * Close an expired channel
+ */
+export declare function closeChannel(channelService: ChannelService, args: {
+    channelId: string;
+}): Promise<string>;
+/**
+ * Get channel status
+ */
+export declare function getChannelStatus(channelService: ChannelService, args: {
+    channelId: string;
+}): Promise<string>;
+export declare const channelTools: ({
+    name: string;
+    description: string;
+    inputSchema: {
+        type: string;
+        properties: {
+            provider: {
+                type: string;
+                description: string;
+            };
+            amount: {
+                type: string;
+                description: string;
+            };
+            duration: {
+                type: string;
+                description: string;
+            };
+            channelId?: undefined;
+        };
+        required: string[];
+    };
+} | {
+    name: string;
+    description: string;
+    inputSchema: {
+        type: string;
+        properties: {
+            channelId: {
+                type: string;
+                description: string;
+            };
+            provider?: undefined;
+            amount?: undefined;
+            duration?: undefined;
+        };
+        required: string[];
+    };
+})[];
+//# sourceMappingURL=channel.d.ts.map

package/dist/tools/channel.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"channel.d.ts","sourceRoot":"","sources":["../../src/tools/channel.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AA0B/D;;GAEG;AACH,wBAAsB,WAAW,CAC/B,cAAc,EAAE,cAAc,EAC9B,eAAe,EAAE,eAAe,EAChC,IAAI,EAAE;IACJ,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;CAClB,GACA,OAAO,CAAC,MAAM,CAAC,CAwCjB;AAED;;GAEG;AACH,wBAAsB,YAAY,CAChC,cAAc,EAAE,cAAc,EAC9B,IAAI,EAAE;IAAE,SAAS,EAAE,MAAM,CAAA;CAAE,GAC1B,OAAO,CAAC,MAAM,CAAC,CAYjB;AAED;;GAEG;AACH,wBAAsB,gBAAgB,CACpC,cAAc,EAAE,cAAc,EAC9B,IAAI,EAAE;IAAE,SAAS,EAAE,MAAM,CAAA;CAAE,GAC1B,OAAO,CAAC,MAAM,CAAC,CA0BjB;AAGD,eAAO,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAkExB,CAAC"}

package/dist/tools/channel.js

@@ -0,0 +1,177 @@
+/**
+ * Channel Tools
+ *
+ * Manage payment channels: open, close, status.
+ */
+/**
+ * Parse duration string to seconds
+ */
+function parseDuration(duration) {
+    const match = duration.match(/^(\d+)(s|m|h|d)$/);
+    if (!match) {
+        // Try parsing as raw seconds
+        const seconds = parseInt(duration);
+        if (!isNaN(seconds))
+            return seconds;
+        throw new Error(`Invalid duration format: ${duration}. Use formats like "1h", "24h", "7d", or seconds.`);
+    }
+    const value = parseInt(match[1]);
+    const unit = match[2];
+    switch (unit) {
+        case 's': return value;
+        case 'm': return value * 60;
+        case 'h': return value * 3600;
+        case 'd': return value * 86400;
+        default: throw new Error(`Unknown duration unit: ${unit}`);
+    }
+}
+/**
+ * Open a payment channel
+ */
+export async function openChannel(channelService, providerService, args) {
+    // Resolve provider address
+    let providerAddress;
+    let providerName;
+    if (args.provider.startsWith('0x')) {
+        providerAddress = args.provider;
+        providerName = args.provider;
+    }
+    else {
+        const provider = await providerService.getProvider(args.provider);
+        if (!provider) {
+            throw new Error(`Provider "${args.provider}" not found. Use drain_providers to list available providers.`);
+        }
+        providerAddress = provider.providerAddress;
+        providerName = provider.name;
+    }
+    const durationSeconds = parseDuration(args.duration);
+    const result = await channelService.openChannel(providerAddress, args.amount, durationSeconds);
+    const expiryDate = result.channel.expiry.toISOString();
+    const hours = Math.floor(durationSeconds / 3600);
+    return `# ✅ Channel Opened
+
+**Channel ID:** \`${result.channelId}\`
+**Transaction:** \`${result.txHash}\`
+
+## Details
+- **Provider:** ${providerName} (\`${providerAddress}\`)
+- **Deposit:** $${args.amount} USDC
+- **Duration:** ${hours} hours
+- **Expires:** ${expiryDate}
+
+## Next Steps
+Use \`drain_chat\` to make AI requests through this channel.
+The channel ID will be used automatically, or you can specify it explicitly.
+
+When finished, wait for expiry and use \`drain_close_channel\` to reclaim unused funds.`;
+}
+/**
+ * Close an expired channel
+ */
+export async function closeChannel(channelService, args) {
+    const channelId = args.channelId;
+    const result = await channelService.closeChannel(channelId);
+    return `# ✅ Channel Closed
+
+**Channel ID:** \`${channelId}\`
+**Transaction:** \`${result.txHash}\`
+**Refunded:** $${result.refundAmount} USDC
+
+The remaining balance has been returned to your wallet.`;
+}
+/**
+ * Get channel status
+ */
+export async function getChannelStatus(channelService, args) {
+    const channelId = args.channelId;
+    const channel = await channelService.getChannel(channelId);
+    const localSpending = channelService.getSpending(channelId);
+    const hoursRemaining = Math.floor(channel.secondsRemaining / 3600);
+    const minutesRemaining = Math.floor((channel.secondsRemaining % 3600) / 60);
+    const statusEmoji = channel.isExpired ? '⏰ EXPIRED' : '🟢 ACTIVE';
+    return `# Channel Status
+
+**Channel ID:** \`${channel.id}\`
+**Status:** ${statusEmoji}
+
+## Balances
+- **Deposit:** $${channel.deposit} USDC
+- **Claimed by Provider:** $${channel.claimed} USDC
+- **Remaining:** $${channel.remaining} USDC
+- **Local Spending (this session):** $${localSpending} USDC
+
+## Timing
+- **Expires:** ${channel.expiry.toISOString()}
+- **Time Remaining:** ${channel.isExpired ? 'EXPIRED' : `${hoursRemaining}h ${minutesRemaining}m`}
+
+${channel.isExpired ? '⚠️ Channel has expired. You can now close it to reclaim remaining funds using `drain_close_channel`.' : ''}`;
+}
+// Tool definitions for MCP
+export const channelTools = [
+    {
+        name: 'drain_open_channel',
+        description: `Open a DRAIN payment channel with an AI provider.
+
+This deposits USDC into a smart contract, creating a payment channel.
+You can then use the channel to pay for AI inference requests.
+
+IMPORTANT: You must have:
+1. Sufficient USDC balance
+2. USDC approved for the DRAIN contract (use drain_approve if needed)
+3. POL for gas fees
+
+The channel will expire after the specified duration. After expiry, you can close it to reclaim unused funds.`,
+        inputSchema: {
+            type: 'object',
+            properties: {
+                provider: {
+                    type: 'string',
+                    description: 'Provider ID (from drain_providers) or provider wallet address (0x...)',
+                },
+                amount: {
+                    type: 'string',
+                    description: 'Amount of USDC to deposit (e.g., "5.00" for $5)',
+                },
+                duration: {
+                    type: 'string',
+                    description: 'Channel duration. Examples: "1h", "24h", "7d", or seconds like "3600"',
+                },
+            },
+            required: ['provider', 'amount', 'duration'],
+        },
+    },
+    {
+        name: 'drain_close_channel',
+        description: `Close an expired DRAIN payment channel and reclaim remaining funds.
+
+Can only be called after the channel has expired.
+Any unused deposit will be returned to your wallet.`,
+        inputSchema: {
+            type: 'object',
+            properties: {
+                channelId: {
+                    type: 'string',
+                    description: 'The channel ID to close (0x...)',
+                },
+            },
+            required: ['channelId'],
+        },
+    },
+    {
+        name: 'drain_channel_status',
+        description: `Get the current status of a DRAIN payment channel.
+
+Shows deposit, spending, remaining balance, and expiry time.`,
+        inputSchema: {
+            type: 'object',
+            properties: {
+                channelId: {
+                    type: 'string',
+                    description: 'The channel ID to check (0x...)',
+                },
+            },
+            required: ['channelId'],
+        },
+    },
+];
+//# sourceMappingURL=channel.js.map

package/dist/tools/channel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"channel.js","sourceRoot":"","sources":["../../src/tools/channel.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAMH;;GAEG;AACH,SAAS,aAAa,CAAC,QAAgB;IACrC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;IACjD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,6BAA6B;QAC7B,MAAM,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACnC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;YAAE,OAAO,OAAO,CAAC;QACpC,MAAM,IAAI,KAAK,CAAC,4BAA4B,QAAQ,mDAAmD,CAAC,CAAC;IAC3G,CAAC;IAED,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACjC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IAEtB,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,GAAG,CAAC,CAAC,OAAO,KAAK,CAAC;QACvB,KAAK,GAAG,CAAC,CAAC,OAAO,KAAK,GAAG,EAAE,CAAC;QAC5B,KAAK,GAAG,CAAC,CAAC,OAAO,KAAK,GAAG,IAAI,CAAC;QAC9B,KAAK,GAAG,CAAC,CAAC,OAAO,KAAK,GAAG,KAAK,CAAC;QAC/B,OAAO,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC,0BAA0B,IAAI,EAAE,CAAC,CAAC;IAC7D,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,cAA8B,EAC9B,eAAgC,EAChC,IAIC;IAED,2BAA2B;IAC3B,IAAI,eAAwB,CAAC;IAC7B,IAAI,YAAoB,CAAC;IAEzB,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QACnC,eAAe,GAAG,IAAI,CAAC,QAAmB,CAAC;QAC3C,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC;IAC/B,CAAC;SAAM,CAAC;QACN,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,aAAa,IAAI,CAAC,QAAQ,+DAA+D,CAAC,CAAC;QAC7G,CAAC;QACD,eAAe,GAAG,QAAQ,CAAC,eAA0B,CAAC;QACtD,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC;IAC/B,CAAC;IAED,MAAM,eAAe,GAAG,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAErD,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,WAAW,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;IAE/F,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;IACvD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,CAAC;IAEjD,OAAO;;oBAEW,MAAM,CAAC,SAAS;qBACf,MAAM,CAAC,MAAM;;;kBAGhB,YAAY,OAAO,eAAe;kBAClC,IAAI,CAAC,MAAM;kBACX,KAAK;iBACN,UAAU;;;;;;wFAM6D,CAAC;AACzF,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,cAA8B,EAC9B,IAA2B;IAE3B,MAAM,SAAS,GAAG,IAAI,CAAC,SAAiB,CAAC;IAEzC,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IAE5D,OAAO;;oBAEW,SAAS;qBACR,MAAM,CAAC,MAAM;iBACjB,MAAM,CAAC,YAAY;;wDAEoB,CAAC;AACzD,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,cAA8B,EAC9B,IAA2B;IAE3B,MAAM,SAAS,GAAG,IAAI,CAAC,SAAiB,CAAC;IACzC,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;IAC3D,MAAM,aAAa,GAAG,cAAc,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;IAE5D,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,CAAC;IACnE,MAAM,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IAE5E,MAAM,WAAW,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC;IAElE,OAAO;;oBAEW,OAAO,CAAC,EAAE;cAChB,WAAW;;;kBAGP,OAAO,CAAC,OAAO;8BACH,OAAO,CAAC,OAAO;oBACzB,OAAO,CAAC,SAAS;wCACG,aAAa;;;iBAGpC,OAAO,CAAC,MAAM,CAAC,WAAW,EAAE;wBACrB,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,cAAc,KAAK,gBAAgB,GAAG;;EAE/F,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,sGAAsG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;AACpI,CAAC;AAED,2BAA2B;AAC3B,MAAM,CAAC,MAAM,YAAY,GAAG;IAC1B;QACE,IAAI,EAAE,oBAAoB;QAC1B,WAAW,EAAE;;;;;;;;;;8GAU6F;QAC1G,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,uEAAuE;iBACrF;gBACD,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iDAAiD;iBAC/D;gBACD,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,uEAAuE;iBACrF;aACF;YACD,QAAQ,EAAE,CAAC,UAAU,EAAE,QAAQ,EAAE,UAAU,CAAC;SAC7C;KACF;IACD;QACE,IAAI,EAAE,qBAAqB;QAC3B,WAAW,EAAE;;;oDAGmC;QAChD,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,SAAS,EAAE;oBACT,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iCAAiC;iBAC/C;aACF;YACD,QAAQ,EAAE,CAAC,WAAW,CAAC;SACxB;KACF;IACD;QACE,IAAI,EAAE,sBAAsB;QAC5B,WAAW,EAAE;;6DAE4C;QACzD,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,SAAS,EAAE;oBACT,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iCAAiC;iBAC/C;aACF;YACD,QAAQ,EAAE,CAAC,WAAW,CAAC;SACxB;KACF;CACF,CAAC"}

package/dist/tools/chat.d.ts

@@ -0,0 +1,64 @@
+/**
+ * Chat Tool
+ *
+ * The main tool: Make AI chat requests with DRAIN payment.
+ */
+import type { ChannelService } from '../services/channel.js';
+import type { ProviderService } from '../services/provider.js';
+import type { InferenceService, ChatMessage } from '../services/inference.js';
+/**
+ * Make a chat completion request
+ */
+export declare function chat(channelService: ChannelService, providerService: ProviderService, inferenceService: InferenceService, args: {
+    channelId: string;
+    model: string;
+    messages: ChatMessage[];
+    maxTokens?: number;
+    temperature?: number;
+}): Promise<string>;
+export declare const chatTools: {
+    name: string;
+    description: string;
+    inputSchema: {
+        type: string;
+        properties: {
+            channelId: {
+                type: string;
+                description: string;
+            };
+            model: {
+                type: string;
+                description: string;
+            };
+            messages: {
+                type: string;
+                description: string;
+                items: {
+                    type: string;
+                    properties: {
+                        role: {
+                            type: string;
+                            enum: string[];
+                            description: string;
+                        };
+                        content: {
+                            type: string;
+                            description: string;
+                        };
+                    };
+                    required: string[];
+                };
+            };
+            maxTokens: {
+                type: string;
+                description: string;
+            };
+            temperature: {
+                type: string;
+                description: string;
+            };
+        };
+        required: string[];
+    };
+}[];
+//# sourceMappingURL=chat.d.ts.map

package/dist/tools/chat.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat.d.ts","sourceRoot":"","sources":["../../src/tools/chat.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAE9E;;GAEG;AACH,wBAAsB,IAAI,CACxB,cAAc,EAAE,cAAc,EAC9B,eAAe,EAAE,eAAe,EAChC,gBAAgB,EAAE,gBAAgB,EAClC,IAAI,EAAE;IACJ,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,WAAW,EAAE,CAAC;IACxB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,GACA,OAAO,CAAC,MAAM,CAAC,CA4CjB;AAcD,eAAO,MAAM,SAAS;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2DrB,CAAC"}