hybrid 1.2.2 → 1.2.3

package/dist/tools/index.js

@@ -0,0 +1,681 @@
+// src/tools/blockchain.ts
+import {
+  createPublicClient,
+  createWalletClient,
+  formatEther,
+  http,
+  parseEther
+} from "viem";
+import { privateKeyToAccount } from "viem/accounts";
+import {
+  arbitrum,
+  base,
+  mainnet,
+  optimism,
+  polygon,
+  sepolia
+} from "viem/chains";
+import { z } from "zod";
+
+// src/core/tool.ts
+function toolFactory() {
+  return (config) => {
+    return {
+      description: config.description,
+      inputSchema: config.inputSchema,
+      outputSchema: config.outputSchema,
+      execute: async (args) => {
+        const input = config.inputSchema.parse(args.input);
+        const result = await config.execute({
+          input,
+          runtime: args.runtime,
+          messages: args.messages
+        });
+        if (config.outputSchema) {
+          return config.outputSchema.parse(result);
+        }
+        return result;
+      }
+    };
+  };
+}
+var createTool = toolFactory();
+
+// src/tools/blockchain.ts
+var SUPPORTED_CHAINS = {
+  mainnet,
+  sepolia,
+  polygon,
+  arbitrum,
+  optimism,
+  base
+};
+var getBalanceTool = createTool({
+  id: "getBalance",
+  description: "Get the native token balance for a wallet address on a blockchain",
+  inputSchema: z.object({
+    address: z.string().describe("The wallet address to check balance for"),
+    chain: z.enum(["mainnet", "sepolia", "polygon", "arbitrum", "optimism", "base"]).default("mainnet").describe("The blockchain network to check on")
+  }),
+  outputSchema: z.object({
+    success: z.boolean(),
+    balance: z.string().describe("Balance in human readable format (ETH, MATIC, etc.)"),
+    balanceWei: z.string().describe("Balance in wei (smallest unit)"),
+    address: z.string(),
+    chain: z.string(),
+    error: z.string().optional()
+  }),
+  execute: async ({ input, runtime }) => {
+    try {
+      const { address, chain } = input;
+      const chainConfig = SUPPORTED_CHAINS[chain];
+      const rpcUrl = runtime.rpcUrl || chainConfig.rpcUrls.default.http[0];
+      const client = createPublicClient({
+        chain: chainConfig,
+        transport: http(rpcUrl)
+      });
+      console.log(`\u{1F50D} [getBalance] Checking balance for ${address} on ${chain}`);
+      const balanceWei = await client.getBalance({
+        address
+      });
+      const balance = formatEther(balanceWei);
+      console.log(
+        `\u2705 [getBalance] Balance: ${balance} ${chainConfig.nativeCurrency.symbol}`
+      );
+      return {
+        success: true,
+        balance: `${balance} ${chainConfig.nativeCurrency.symbol}`,
+        balanceWei: balanceWei.toString(),
+        address,
+        chain
+      };
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      console.error("\u274C [getBalance] Error:", errorMessage);
+      return {
+        success: false,
+        balance: "0",
+        balanceWei: "0",
+        address: input.address,
+        chain: input.chain,
+        error: errorMessage
+      };
+    }
+  }
+});
+var getTransactionTool = createTool({
+  id: "getTransaction",
+  description: "Get transaction details by transaction hash",
+  inputSchema: z.object({
+    hash: z.string().describe("The transaction hash to look up"),
+    chain: z.enum(["mainnet", "sepolia", "polygon", "arbitrum", "optimism", "base"]).default("mainnet").describe("The blockchain network to check on")
+  }),
+  outputSchema: z.object({
+    success: z.boolean(),
+    transaction: z.object({
+      hash: z.string(),
+      from: z.string(),
+      to: z.string().nullable(),
+      value: z.string(),
+      gasUsed: z.string().optional(),
+      gasPrice: z.string().optional(),
+      blockNumber: z.string().optional(),
+      status: z.string().optional()
+    }).optional(),
+    error: z.string().optional()
+  }),
+  execute: async ({ input, runtime }) => {
+    try {
+      const { hash, chain } = input;
+      const chainConfig = SUPPORTED_CHAINS[chain];
+      const rpcUrl = runtime.rpcUrl || chainConfig.rpcUrls.default.http[0];
+      const client = createPublicClient({
+        chain: chainConfig,
+        transport: http(rpcUrl)
+      });
+      console.log(
+        `\u{1F50D} [getTransaction] Looking up transaction ${hash} on ${chain}`
+      );
+      const transaction = await client.getTransaction({
+        hash
+      });
+      const receipt = await client.getTransactionReceipt({
+        hash
+      }).catch(() => null);
+      console.log(
+        `\u2705 [getTransaction] Found transaction from ${transaction.from} to ${transaction.to}`
+      );
+      return {
+        success: true,
+        transaction: {
+          hash: transaction.hash,
+          from: transaction.from,
+          to: transaction.to,
+          value: formatEther(transaction.value),
+          gasUsed: receipt?.gasUsed.toString(),
+          gasPrice: transaction.gasPrice?.toString(),
+          blockNumber: transaction.blockNumber?.toString(),
+          status: receipt?.status === "success" ? "success" : receipt?.status === "reverted" ? "failed" : "pending"
+        }
+      };
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      console.error("\u274C [getTransaction] Error:", errorMessage);
+      return {
+        success: false,
+        error: errorMessage
+      };
+    }
+  }
+});
+var sendTransactionTool = createTool({
+  id: "sendTransaction",
+  description: "Send native tokens to another address",
+  inputSchema: z.object({
+    to: z.string().describe("The recipient address"),
+    amount: z.string().describe("The amount to send (in ETH, MATIC, etc.)"),
+    chain: z.enum(["mainnet", "sepolia", "polygon", "arbitrum", "optimism", "base"]).default("mainnet").describe("The blockchain network to send on")
+  }),
+  outputSchema: z.object({
+    success: z.boolean(),
+    hash: z.string().optional(),
+    from: z.string().optional(),
+    to: z.string(),
+    amount: z.string(),
+    chain: z.string(),
+    error: z.string().optional()
+  }),
+  execute: async ({ input, runtime }) => {
+    try {
+      const { to, amount, chain } = input;
+      const chainConfig = SUPPORTED_CHAINS[chain];
+      const privateKey = runtime.privateKey;
+      if (!privateKey) {
+        return {
+          success: false,
+          to,
+          amount,
+          chain,
+          error: "Private key not configured in runtime"
+        };
+      }
+      const rpcUrl = runtime.rpcUrl || chainConfig.rpcUrls.default.http[0];
+      const account = privateKeyToAccount(privateKey);
+      const client = createWalletClient({
+        account,
+        chain: chainConfig,
+        transport: http(rpcUrl)
+      });
+      console.log(
+        `\u{1F4B8} [sendTransaction] Sending ${amount} ${chainConfig.nativeCurrency.symbol} to ${to} on ${chain}`
+      );
+      const hash = await client.sendTransaction({
+        to,
+        value: parseEther(amount)
+      });
+      console.log(`\u2705 [sendTransaction] Transaction sent: ${hash}`);
+      return {
+        success: true,
+        hash,
+        from: account.address,
+        to,
+        amount,
+        chain
+      };
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      console.error("\u274C [sendTransaction] Error:", errorMessage);
+      return {
+        success: false,
+        to: input.to,
+        amount: input.amount,
+        chain: input.chain,
+        error: errorMessage
+      };
+    }
+  }
+});
+var getBlockTool = createTool({
+  id: "getBlock",
+  description: "Get information about a blockchain block",
+  inputSchema: z.object({
+    blockNumber: z.string().optional().describe("Block number (defaults to latest)"),
+    chain: z.enum(["mainnet", "sepolia", "polygon", "arbitrum", "optimism", "base"]).default("mainnet").describe("The blockchain network to check on")
+  }),
+  outputSchema: z.object({
+    success: z.boolean(),
+    block: z.object({
+      number: z.string(),
+      hash: z.string(),
+      timestamp: z.string(),
+      transactionCount: z.number(),
+      gasUsed: z.string(),
+      gasLimit: z.string()
+    }).optional(),
+    error: z.string().optional()
+  }),
+  execute: async ({ input, runtime }) => {
+    try {
+      const { blockNumber, chain } = input;
+      const chainConfig = SUPPORTED_CHAINS[chain];
+      const rpcUrl = runtime.rpcUrl || chainConfig.rpcUrls.default.http[0];
+      const client = createPublicClient({
+        chain: chainConfig,
+        transport: http(rpcUrl)
+      });
+      console.log(
+        `\u{1F50D} [getBlock] Getting block ${blockNumber || "latest"} on ${chain}`
+      );
+      const block = await client.getBlock({
+        blockNumber: blockNumber ? BigInt(blockNumber) : void 0
+      });
+      console.log(
+        `\u2705 [getBlock] Found block ${block.number} with ${block.transactions.length} transactions`
+      );
+      return {
+        success: true,
+        block: {
+          number: block.number.toString(),
+          hash: block.hash,
+          timestamp: block.timestamp.toString(),
+          transactionCount: block.transactions.length,
+          gasUsed: block.gasUsed.toString(),
+          gasLimit: block.gasLimit.toString()
+        }
+      };
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      console.error("\u274C [getBlock] Error:", errorMessage);
+      return {
+        success: false,
+        error: errorMessage
+      };
+    }
+  }
+});
+var getGasPriceTool = createTool({
+  id: "getGasPrice",
+  description: "Get current gas price for a blockchain",
+  inputSchema: z.object({
+    chain: z.enum(["mainnet", "sepolia", "polygon", "arbitrum", "optimism", "base"]).default("mainnet").describe("The blockchain network to check on")
+  }),
+  outputSchema: z.object({
+    success: z.boolean(),
+    gasPrice: z.string().optional().describe("Gas price in gwei"),
+    gasPriceWei: z.string().optional().describe("Gas price in wei"),
+    chain: z.string(),
+    error: z.string().optional()
+  }),
+  execute: async ({ input, runtime }) => {
+    try {
+      const { chain } = input;
+      const chainConfig = SUPPORTED_CHAINS[chain];
+      const rpcUrl = runtime.rpcUrl || chainConfig.rpcUrls.default.http[0];
+      const client = createPublicClient({
+        chain: chainConfig,
+        transport: http(rpcUrl)
+      });
+      console.log(`\u26FD [getGasPrice] Getting gas price for ${chain}`);
+      const gasPrice = await client.getGasPrice();
+      const gasPriceGwei = formatEther(gasPrice * BigInt(1e9));
+      console.log(`\u2705 [getGasPrice] Current gas price: ${gasPriceGwei} gwei`);
+      return {
+        success: true,
+        gasPrice: `${gasPriceGwei} gwei`,
+        gasPriceWei: gasPrice.toString(),
+        chain
+      };
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      console.error("\u274C [getGasPrice] Error:", errorMessage);
+      return {
+        success: false,
+        chain: input.chain,
+        error: errorMessage
+      };
+    }
+  }
+});
+var estimateGasTool = createTool({
+  id: "estimateGas",
+  description: "Estimate gas required for a transaction",
+  inputSchema: z.object({
+    to: z.string().describe("The recipient address"),
+    amount: z.string().default("0").describe("The amount to send (defaults to 0)"),
+    data: z.string().optional().describe("Transaction data (for contract calls)"),
+    chain: z.enum(["mainnet", "sepolia", "polygon", "arbitrum", "optimism", "base"]).default("mainnet").describe("The blockchain network to estimate on")
+  }),
+  outputSchema: z.object({
+    success: z.boolean(),
+    gasEstimate: z.string().optional(),
+    to: z.string(),
+    amount: z.string(),
+    chain: z.string(),
+    error: z.string().optional()
+  }),
+  execute: async ({ input, runtime }) => {
+    try {
+      const { to, amount, data, chain } = input;
+      const chainConfig = SUPPORTED_CHAINS[chain];
+      const privateKey = runtime.privateKey;
+      if (!privateKey) {
+        return {
+          success: false,
+          to,
+          amount,
+          chain,
+          error: "Private key not configured in runtime"
+        };
+      }
+      const rpcUrl = runtime.rpcUrl || chainConfig.rpcUrls.default.http[0];
+      const account = privateKeyToAccount(privateKey);
+      const client = createPublicClient({
+        chain: chainConfig,
+        transport: http(rpcUrl)
+      });
+      console.log(
+        `\u26FD [estimateGas] Estimating gas for transaction to ${to} on ${chain}`
+      );
+      const gasEstimate = await client.estimateGas({
+        account: account.address,
+        to,
+        value: parseEther(amount),
+        data
+      });
+      console.log(`\u2705 [estimateGas] Estimated gas: ${gasEstimate.toString()}`);
+      return {
+        success: true,
+        gasEstimate: gasEstimate.toString(),
+        to,
+        amount,
+        chain
+      };
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      console.error("\u274C [estimateGas] Error:", errorMessage);
+      return {
+        success: false,
+        to: input.to,
+        amount: input.amount,
+        chain: input.chain,
+        error: errorMessage
+      };
+    }
+  }
+});
+var blockchainTools = {
+  getBalance: getBalanceTool,
+  getTransaction: getTransactionTool,
+  sendTransaction: sendTransactionTool,
+  getBlock: getBlockTool,
+  getGasPrice: getGasPriceTool,
+  estimateGas: estimateGasTool
+};
+
+// src/tools/xmtp.ts
+import { z as z2 } from "zod";
+var sendReactionTool = createTool({
+  id: "sendReaction",
+  description: "Send an emoji reaction to a message to indicate it has been seen",
+  inputSchema: z2.object({
+    emoji: z2.string().default("\u{1F440}").describe(
+      "The emoji to send as a reaction (supports common emoji like \u{1F44D}, \u2764\uFE0F, \u{1F525}, etc.)"
+    ),
+    referenceMessageId: z2.string().optional().describe(
+      "The message ID to react to (uses current message if not provided)"
+    )
+  }),
+  outputSchema: z2.object({
+    success: z2.boolean(),
+    emoji: z2.string(),
+    error: z2.string().optional()
+  }),
+  execute: async ({ input, runtime }) => {
+    try {
+      const xmtpClient = runtime.xmtpClient;
+      const currentMessage = runtime.message;
+      if (!xmtpClient) {
+        const errorMsg = "\u274C XMTP service not available";
+        return { success: false, emoji: input.emoji, error: errorMsg };
+      }
+      if (!currentMessage) {
+        const errorMsg = "\u274C No message to react to";
+        return { success: false, emoji: input.emoji, error: errorMsg };
+      }
+      const messageIdToReactTo = input.referenceMessageId || currentMessage.id;
+      console.log(
+        `\u{1F440} [sendReaction] Sending ${input.emoji} reaction to message ${messageIdToReactTo}`
+      );
+      const reactionResult = await xmtpClient.sendReaction({
+        messageId: messageIdToReactTo,
+        emoji: input.emoji,
+        action: "added"
+      });
+      if (!reactionResult.success) {
+        const errorMsg = `\u274C Failed to send reaction: ${reactionResult.error || "Unknown error"}`;
+        return { success: false, emoji: input.emoji, error: errorMsg };
+      }
+      console.log(`\u2705 [sendReaction] Successfully sent ${input.emoji} reaction`);
+      return { success: true, emoji: input.emoji };
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      console.error("\u274C [sendReaction] Error:", errorMessage);
+      return { success: false, emoji: input.emoji, error: errorMessage };
+    }
+  }
+});
+var sendMessageTool = createTool({
+  id: "sendMessage",
+  description: "Send a message to an XMTP conversation",
+  inputSchema: z2.object({
+    content: z2.string().describe("The message content to send"),
+    recipientAddress: z2.string().optional().describe("Recipient address for new conversations"),
+    conversationId: z2.string().optional().describe("Existing conversation ID to send to")
+  }).refine((data) => data.recipientAddress || data.conversationId, {
+    message: "Either recipientAddress or conversationId must be provided"
+  }),
+  outputSchema: z2.object({
+    success: z2.boolean(),
+    messageId: z2.string().optional(),
+    conversationId: z2.string().optional(),
+    content: z2.string(),
+    error: z2.string().optional()
+  }),
+  execute: async ({ input, runtime }) => {
+    try {
+      const xmtpClient = runtime.xmtpClient;
+      const { content, recipientAddress, conversationId } = input;
+      if (!xmtpClient) {
+        return {
+          success: false,
+          content,
+          error: "XMTP service not available"
+        };
+      }
+      console.log(
+        `\u{1F4AC} [sendMessage] Sending message: "${content.substring(0, 50)}${content.length > 50 ? "..." : ""}"`
+      );
+      let targetConversationId = conversationId;
+      if (!targetConversationId && recipientAddress) {
+        console.log(
+          `\u{1F50D} [sendMessage] Creating/finding conversation with ${recipientAddress}`
+        );
+        targetConversationId = recipientAddress;
+      }
+      const messageResult = await xmtpClient.sendMessage({
+        content
+      });
+      if (!messageResult.success) {
+        return {
+          success: false,
+          content,
+          error: messageResult.error || "Failed to send message"
+        };
+      }
+      console.log(`\u2705 [sendMessage] Message sent successfully`);
+      return {
+        success: true,
+        messageId: messageResult.data?.conversationId,
+        conversationId: messageResult.data?.conversationId,
+        content
+      };
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      console.error("\u274C [sendMessage] Error:", errorMessage);
+      return {
+        success: false,
+        content: input.content,
+        error: errorMessage
+      };
+    }
+  }
+});
+var sendReplyTool = createTool({
+  id: "sendReply",
+  description: "Send a reply to a specific message in an XMTP conversation",
+  inputSchema: z2.object({
+    content: z2.string().describe("The reply content to send"),
+    replyToMessageId: z2.string().optional().describe("Message ID to reply to (uses current message if not provided)")
+  }),
+  outputSchema: z2.object({
+    success: z2.boolean(),
+    messageId: z2.string().optional(),
+    replyToMessageId: z2.string().optional(),
+    content: z2.string(),
+    error: z2.string().optional()
+  }),
+  execute: async ({ input, runtime }) => {
+    try {
+      const xmtpClient = runtime.xmtpClient;
+      const currentMessage = runtime.message;
+      const { content, replyToMessageId } = input;
+      if (!xmtpClient) {
+        return {
+          success: false,
+          content,
+          error: "XMTP service not available"
+        };
+      }
+      if (!currentMessage && !replyToMessageId) {
+        return {
+          success: false,
+          content,
+          error: "No message to reply to"
+        };
+      }
+      const targetMessageId = replyToMessageId || currentMessage?.id;
+      console.log(
+        `\u21A9\uFE0F [sendReply] Sending reply to message ${targetMessageId}: "${content.substring(0, 50)}${content.length > 50 ? "..." : ""}"`
+      );
+      const replyResult = await xmtpClient.sendReply({
+        content,
+        messageId: targetMessageId
+      });
+      if (!replyResult.success) {
+        return {
+          success: false,
+          content,
+          replyToMessageId: targetMessageId,
+          error: replyResult.error || "Failed to send reply"
+        };
+      }
+      console.log(`\u2705 [sendReply] Reply sent successfully`);
+      return {
+        success: true,
+        messageId: replyResult.data?.conversationId,
+        replyToMessageId: targetMessageId,
+        content
+      };
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      console.error("\u274C [sendReply] Error:", errorMessage);
+      return {
+        success: false,
+        content: input.content,
+        replyToMessageId: input.replyToMessageId,
+        error: errorMessage
+      };
+    }
+  }
+});
+var getMessageTool = createTool({
+  id: "getMessage",
+  description: "Get a specific message by ID from XMTP",
+  inputSchema: z2.object({
+    messageId: z2.string().describe("The message ID to retrieve")
+  }),
+  outputSchema: z2.object({
+    success: z2.boolean(),
+    message: z2.object({
+      id: z2.string(),
+      conversationId: z2.string(),
+      content: z2.union([z2.string(), z2.record(z2.unknown())]),
+      senderInboxId: z2.string(),
+      sentAt: z2.string(),
+      contentType: z2.object({
+        typeId: z2.string(),
+        authorityId: z2.string().optional(),
+        versionMajor: z2.number().optional(),
+        versionMinor: z2.number().optional()
+      }).optional()
+    }).optional(),
+    error: z2.string().optional()
+  }),
+  execute: async ({ input, runtime }) => {
+    try {
+      const xmtpClient = runtime.xmtpClient;
+      const { messageId } = input;
+      if (!xmtpClient) {
+        return {
+          success: false,
+          error: "XMTP service not available"
+        };
+      }
+      console.log(`\u{1F4DC} [getMessage] Retrieving message ${messageId}`);
+      const messageResult = await xmtpClient.getMessage({
+        messageId
+      });
+      if (!messageResult.success) {
+        return {
+          success: false,
+          error: messageResult.error || "Failed to get message"
+        };
+      }
+      console.log(
+        `\u2705 [getMessage] Retrieved message from ${messageResult.data?.senderInboxId}`
+      );
+      return {
+        success: true,
+        message: messageResult.data
+      };
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      console.error("\u274C [getMessage] Error:", errorMessage);
+      return {
+        success: false,
+        error: errorMessage
+      };
+    }
+  }
+});
+var xmtpTools = {
+  sendMessage: sendMessageTool,
+  sendReply: sendReplyTool,
+  sendReaction: sendReactionTool,
+  getMessage: getMessageTool
+};
+export {
+  blockchainTools,
+  estimateGasTool,
+  getBalanceTool,
+  getBlockTool,
+  getGasPriceTool,
+  getMessageTool,
+  getTransactionTool,
+  sendMessageTool,
+  sendReactionTool,
+  sendReplyTool,
+  sendTransactionTool,
+  xmtpTools
+};
+//# sourceMappingURL=index.js.map