Merge pull request #7 from blockydevs/ELIZAAI-16-wallet-provider

[ELIZAAI-16](feat): cosmos wallet provider with balances
elizaOS · Jan 2, 2025 · ab56e1a · ab56e1a
2 parents 331c643 + 00cfb42
commit ab56e1a
Show file tree

Hide file tree

Showing 5 changed files with 2,619 additions and 2,450 deletions.
diff --git a/packages/plugin-cosmos/package.json b/packages/plugin-cosmos/package.json
@@ -13,12 +13,16 @@
         "bignumber.js": "9.1.2",
         "chain-registry": "^1.69.68",
         "tsup": "8.3.5",
-        "zod": "3.23.8"
+        "zod": "3.23.8",
+        "interchain": "^1.10.4"
     },
     "scripts": {
         "build": "tsup --format esm --dts",
         "dev": "tsup --format esm --dts --watch",
         "lint": "eslint --fix  --cache .",
         "test": "vitest run"
+    },
+    "devDependencies": {
+        "@chain-registry/types": "^0.50.44"
     }
 }
diff --git a/packages/plugin-cosmos/src/actions/transfer.ts b/packages/plugin-cosmos/src/actions/transfer.ts
@@ -5,10 +5,8 @@ import { FeeEstimator } from "../services/fee-estimator";
 import { getNativeAssetByChainName } from "@chain-registry/utils";
 import { assets } from "chain-registry";
 import {
-    CosmosWalletProvider,
-    fetchChainDetails,
-    genCosmosChainsFromRuntime,
-    initWalletProvider,
+    CosmosWalletChainsData,
+    cosmosWalletProvider,
 } from "../providers/wallet.ts";
 import {
     composeContext,
@@ -24,17 +22,17 @@ import { AssetList } from "@chain-registry/types";
 import { Coin } from "@cosmjs/stargate";
 
 export class TransferAction {
-    constructor(private cosmosWalletProvider: CosmosWalletProvider) {
-        this.cosmosWalletProvider = cosmosWalletProvider;
+    constructor(private cosmosChainsData: CosmosWalletChainsData) {
+        this.cosmosChainsData = cosmosChainsData;
     }
 
     async transfer(params: CosmosTransferParams): Promise<Transaction> {
         const signingCosmWasmClient =
-            await this.cosmosWalletProvider.getSigningCosmWasmClient();
+            this.cosmosChainsData.getSigningCosmWasmClient(params.fromChain);
 
-        const wallet = await this.cosmosWalletProvider.getWallet();
-        const accounts = await wallet.getAccounts();
-        const senderAddress = accounts[0]?.address;
+        const senderAddress = await this.cosmosChainsData.getWalletAddress(
+            params.fromChain
+        );
 
         if (!senderAddress) {
             throw new Error("No sender address");
@@ -44,9 +42,9 @@ export class TransferAction {
             throw new Error("No receiver address");
         }
 
-        const chainAssets: AssetList = fetchChainDetails(
+        const chainAssets: AssetList = this.cosmosChainsData.getAssetsList(
             params.fromChain
-        ).chainAssets;
+        );
 
         const formatedDenom = params.denomOrIbc.toString();
 
@@ -145,10 +143,8 @@ export const transferAction = {
         };
 
         try {
-            const walletProvider = await initWalletProvider(
-                _runtime,
-                paramOptions.fromChain
-            );
+            const walletProvider: CosmosWalletChainsData =
+                await cosmosWalletProvider.initWalletChainsData(_runtime);
 
             const action = new TransferAction(walletProvider);
 
@@ -202,10 +198,11 @@ export const transferAction = {
     },
     template: transferTemplate,
     validate: async (runtime: IAgentRuntime) => {
-        const recoveryPhrase = runtime.getSetting("COSMOS_RECOVERY_PHRASE");
-        const chains = genCosmosChainsFromRuntime(runtime);
+        const mnemonic = runtime.getSetting("COSMOS_RECOVERY_PHRASE");
+        const availableChains = runtime.getSetting("COSMOS_AVAILABLE_CHAINS");
+        const availableChainsArray = availableChains?.split(",");
 
-        return recoveryPhrase !== undefined && Object.keys(chains).length > 0;
+        return !(mnemonic && availableChains && availableChainsArray.length);
     },
     examples: [
         [

diff --git a/packages/plugin-cosmos/src/index.ts b/packages/plugin-cosmos/src/index.ts
@@ -1,15 +1,14 @@
-import { cosmosWalletProvider } from "./providers/wallet.ts";
+import {cosmosWalletProvider} from "./providers/wallet.ts";
 import type { Plugin } from "@ai16z/eliza";
-import { balanceAction } from "./actions/walletProviderTestAction.ts";
-import {transferAction} from "./actions/transfer.ts";
+import { transferAction } from "./actions/transfer.ts";
 
 export const cosmosPlugin: Plugin = {
     name: "cosmos",
     description: "Cosmos blockchain integration plugin",
     providers: [cosmosWalletProvider],
     evaluators: [],
     services: [],
-    actions: [transferAction, balanceAction],
+    actions: [transferAction],
 };
 
 export default cosmosPlugin;