@deserialize/multi-vm-wallet 1.2.2 → 1.2.4

package/utils/evm/evm.ts

@@ -6,7 +6,7 @@
 
 import { EVMDeriveChildPrivateKey } from "../walletBip32";
 import { ChainWallet } from "../IChainWallet";
-import { Balance, ChainWalletConfig, NFTInfo, UserTokenBalance, TokenInfo, TransactionResult } from "../types";
+import { Balance, ChainWalletConfig, NFTInfo, UserTokenBalance, TokenInfo, TransactionResult, NFT } from "../types";
 import { VM } from "../vm";
 import { ethers, JsonRpcProvider, Wallet, formatUnits } from "ethers";
 import BN from "bn.js";
@@ -30,7 +30,8 @@ import {
     executeContractMethod,
     getTokenInfo,
     DESERIALIZED_SUPPORTED_CHAINS,
-    discoverTokens
+    discoverTokens,
+    discoverNFTs
 } from "./utils";
 import { EVMTransactionHistoryItem, getEVMTransactionHistory } from "./transactionParsing";
 import { createPublicClient, Hex, http, PublicClient } from "viem";
@@ -79,6 +80,7 @@ export class EVMVM extends VM<string, string, JsonRpcProvider> {
     }
 
     getTokenInfo = getTokenInfo
+    static getTokenInfo = getTokenInfo
     generatePrivateKey(index: number, seed?: string, mnemonic?: string, derivationPath = this.derivationPath) {
         let _seed: string
 
@@ -114,12 +116,14 @@ export class EVMVM extends VM<string, string, JsonRpcProvider> {
 
 export class EVMChainWallet extends ChainWallet<string, string, JsonRpcProvider> {
     client: PublicClient
+    wallet: Wallet
     constructor(config: ChainWalletConfig, privateKey: string, index: number) {
         super(config, privateKey, index);
-        const wallet = new Wallet(privateKey);
+        this.connection = new JsonRpcProvider(config.rpcUrl)
+        const wallet = new Wallet(privateKey, this.connection);
+        this.wallet = wallet
         this.address = wallet.address;
         this.privateKey = privateKey;
-        this.connection = new JsonRpcProvider(config.rpcUrl)
 
         //client for viem
         this.client = createPublicClient(
@@ -172,11 +176,20 @@ export class EVMChainWallet extends ChainWallet<string, string, JsonRpcProvider>
         return await EVMVM.getTokenBalance(this.address, tokenAddress, this.connection!)
     }
 
+    async getTokenInfo(tokenAddress: string) {
+        return await EVMVM.getTokenInfo(tokenAddress, this.connection!)
+    }
+
     async discoverToken(): Promise<UserTokenBalance<string>[]> {
         // Implement token discovery logic here
         return await discoverTokens(this.address, this.config)
     }
 
+    async discoverNFT(): Promise<NFT[]> {
+        // Implement NFT discovery logic here
+        return await discoverNFTs(this.address, this.config)
+    }
+
     async transferNative(to: string, amount: number): Promise<TransactionResult> {
         // Implement native transfer logic here
         const wallet = this.getWallet()