Wallet tools

tools/wallet-eth-balance/metadata.json

@@ -0,0 +1,62 @@
+{
+  "name": "Wallet Manager",
+  "description": "Manages wallet operations including fetching ETH and token balances.",
+  "author": "@@eduardosotomontaner.arb-sep-shinkai",
+  "keywords": [
+    "wallet",
+    "eth",
+    "token",
+    "balance",
+    "management"
+  ],
+  "configurations": {
+    "type": "object",
+    "properties": {
+      "rpcURL": {
+        "type": "string",
+        "description": "The RPC URL for the blockchain"
+      },
+      "chain": {
+        "type": "string",
+        "description": "The blockchain chain identifier"
+      },
+      "privateKey": {
+        "type": "string",
+        "description": "The private key for the wallet"
+      }
+    },
+    "required": [
+      "rpcURL",
+      "chain",
+      "privateKey"
+    ]
+  },
+  "parameters": {
+    "type": "object",
+    "properties": {
+      "address": {
+        "type": "string",
+        "description": "The wallet address to fetch the ETH balance for"
+      }
+    },
+    "required": [
+      "address"
+    ]
+  },
+  "result": {
+    "type": "object",
+    "properties": {
+      "ethBalance": {
+        "type": "string",
+        "description": "The ETH balance of the specified address"
+      }
+    },
+    "required": [
+      "ethBalance"
+    ]
+  },
+  "sqlTables": [],
+  "sqlQueries": [],
+  "tools": [],
+  "oauth": []
+}

tools/wallet-eth-balance/tool.ts

@@ -0,0 +1,173 @@
+import { 
+  createPublicClient, 
+  createWalletClient,
+  http,
+  formatEther,
+  encodeFunctionData,
+  type Address,
+  type PublicClient,
+  type WalletClient,
+  type Chain,
+  type Hex,
+  type TransactionReceipt,
+  erc20Abi
+} from 'npm:viem'
+import { privateKeyToAccount, type PrivateKeyAccount } from 'npm:viem/accounts'
+
+import { arbitrumSepolia, arbitrumNova, base, baseSepolia } from 'npm:viem/chains'
+
+interface WalletConfig {
+  rpcUrl: string;
+  chain: Chain;
+  privateKey: Hex;
+}
+
+const chainsDict = {
+  ARBITRUM_SEPOLIA : arbitrumSepolia,
+  ARBITRUM_NOVA : arbitrumNova,
+  BASE : base,
+  BASE_SEPOLIA : baseSepolia,
+}
+
+interface CONFIG {
+  chain: keyof typeof chainsDict;
+  privateKey: `0x${string}`;
+  rpcURL: string;
+}
+
+type INPUTS = {
+  address: `0x${string}`;
+}
+
+type OUTPUT = {
+  ethBalance: string;
+}
+
+export class WalletManager {
+  private publicClient: PublicClient;
+  private account: PrivateKeyAccount;
+  private chain: Chain;
+  private address: Address;
+
+  constructor({ rpcUrl, chain, privateKey }: WalletConfig) {
+    this.account = privateKeyToAccount(privateKey);
+    this.address = this.account.address;
+    this.chain = chain;
+    this.publicClient = createPublicClient({
+      chain,
+      transport: http(rpcUrl)
+    });
+  }
+
+  private createWalletClient(): WalletClient {
+    return createWalletClient({
+      chain: this.chain,
+      transport: http(this.publicClient.transport.url), // Use same RPC URL
+      account: this.account
+    });
+  }
+
+  async getETHBalance(address: Address=this.address): Promise<string> {
+    const balance = await this.publicClient.getBalance({ address });
+    return formatEther(balance);
+  }
+
+  async getTokenBalance(tokenAddress: Address, walletAddress: Address=this.address): Promise<string> {
+    const [balance, decimals] = await Promise.all([
+      this.publicClient.readContract({
+        address: tokenAddress,
+        abi: erc20Abi,
+        functionName: 'balanceOf',
+        args: [walletAddress]
+      }),
+      this.publicClient.readContract({
+        address: tokenAddress,
+        abi: erc20Abi,
+        functionName: 'decimals'
+      })
+    ]);
+
+    return (balance / 10n ** BigInt(decimals)).toString();
+  }
+
+  private async getTokenValue(tokenAddress: Address, amount: string | number | bigint): Promise<bigint> {
+    const decimals = await this.publicClient.readContract({
+      address: tokenAddress,
+      abi: erc20Abi,
+      functionName: 'decimals'
+    });
+    return BigInt(amount) * (10n ** BigInt(decimals));
+  }
+
+  async estimateGas(
+    tokenAddress: Address, 
+    toAddress: Address, 
+    amount: string | number | bigint
+  ): Promise<bigint> {
+    const value = await this.getTokenValue(tokenAddress, amount);
+    return await this.publicClient.estimateContractGas({
+      address: tokenAddress,
+      abi: erc20Abi,
+      functionName: 'transfer',
+      args: [toAddress, value],
+      account: this.account.address
+    });
+  }
+  async sendTokens(
+    tokenAddress: Address,
+    toAddress: Address,
+    amount: string | number | bigint
+  ): Promise<TransactionReceipt> {
+    const value = await this.getTokenValue(tokenAddress, amount);
+    const walletClient = this.createWalletClient();
+    const data = encodeFunctionData({
+      abi: erc20Abi,
+      functionName: 'transfer',
+      args: [toAddress, value]
+    });
+    const gasEstimate = await this.estimateGas(tokenAddress, toAddress, amount);
+    /*const request = await walletClient.prepareTransactionRequest({
+      to: tokenAddress,
+      account: this.account.address,
+      abi: erc20Abi,
+      chain: this.chain,
+      gas: gasEstimate,
+      data,
+    });*/
+    const tx = await walletClient.signTransaction({
+      to: tokenAddress,
+      account: this.account.address,
+      abi: erc20Abi,
+      chain: this.chain,
+      gas: gasEstimate,
+      data,
+    });
+    const hash = await this.publicClient.sendRawTransaction({
+      serializedTransaction: tx
+    });
+    return await this.publicClient.waitForTransactionReceipt({ hash });
+  }
+}
+
+const ETHBALANCE_ERROR = 'ETHBALANCE_ERROR';
+
+export async function run(
+  config: CONFIG,
+  inputs: INPUTS
+): Promise<OUTPUT> {
+  try {
+    const { address } = inputs;
+    if (!address) {
+      throw new Error('Address is required in inputs');
+    }
+    const walletManager = new WalletManager({
+      rpcUrl: config.rpcURL,
+      chain: chainsDict[config.chain],
+      privateKey: config.privateKey
+    });
+    const ethBalance = await walletManager.getETHBalance(address);
+    return { ethBalance };
+  } catch (error) {
+      throw new Error(ETHBALANCE_ERROR, { cause: error });
+  }
+}

tools/wallet-send-token/metadata.json

@@ -0,0 +1,104 @@
+{
+  "name": "WalletManager",
+  "description": "Manages operations related to a cryptocurrency wallet including sending tokens and checking balances.",
+  "author": "@@eduardosotomontaner.arb-sep-shinkai",
+  "keywords": [
+    "wallet",
+    "cryptocurrency",
+    "tokens",
+    "ethereum"
+  ],
+  "configurations": {
+    "type": "object",
+    "properties": {
+      "rpcUrl": {
+        "type": "string",
+        "description": "The RPC URL to connect to the blockchain"
+      },
+      "chain": {
+        "type": "string",
+        "description": "The blockchain chain to interact with"
+      },
+      "privateKey": {
+        "type": "string",
+        "description": "The private key associated with the wallet"
+      }
+    },
+    "required": [
+      "rpcUrl",
+      "chain",
+      "privateKey"
+    ]
+  },
+  "parameters": {
+    "type": "object",
+    "properties": {
+      "to_address": {
+        "type": "string",
+        "description": "The address to which tokens are sent"
+      },
+      "tokenAddress": {
+        "type": "string",
+        "description": "The address of the token contract"
+      },
+      "amount": {
+        "type": "string",
+        "description": "The amount of tokens to send"
+      }
+    },
+    "required": [
+      "to_address",
+      "tokenAddress",
+      "amount"
+    ]
+  },
+  "result": {
+    "type": "object",
+    "properties": {
+      "receipt": {
+        "type": "object",
+        "properties": {
+          "transactionHash": {
+            "type": "string",
+            "description": "The hash of the transaction"
+          },
+          "status": {
+            "type": "string",
+            "description": "The status of the transaction"
+          },
+          "gasUsed": {
+            "type": "integer",
+            "description": "The amount of gas used for the transaction"
+          },
+          "gasPrice": {
+            "type": "integer",
+            "description": "The gas price at which the transaction was executed"
+          },
+          "from": {
+            "type": "string",
+            "description": "The address of the sender"
+          },
+          "to": {
+            "type": "string",
+            "description": "The address of the receiver"
+          }
+        },
+        "required": [
+          "transactionHash",
+          "status",
+          "gasUsed",
+          "gasPrice",
+          "from",
+          "to"
+        ]
+      }
+    },
+    "required": [
+      "receipt"
+    ]
+  },
+  "sqlTables": [],
+  "sqlQueries": [],
+  "tools": [],
+  "oauth": []
+}