Merge pull request #105 from unstoppabledomains/qrtp/new-rpc-methods

Add Ethereum provider methods: eth_getBalance, eth_getCode, eth_gasPrice

CHANGELOG.md

@@ -1,3 +1,9 @@
+## 3.1.46
+* Add Ethereum RPC methods: eth_getBalance, eth_getCode, eth_gasPrice
+
+## 3.1.45
+* Bug fixes
+
 ## 3.1.44
 * Enable Solana wallet provider in the browser
 

package.json

@@ -1,6 +1,6 @@
 {
   "name": "ud-extension",
-  "version": "3.1.45",
+  "version": "3.1.46",
   "license": "MIT",
   "author": {
     "email": "eng@unstoppabledomains.com",

src/scripts/liteWalletProvider/background.ts

@@ -452,6 +452,24 @@ const handleRpcRequest = async (
           await web3.eth.estimateGas(rpcParams[0]),
         );
         break;
+      case "gasPrice":
+        result = web3utils.numberToHex(await web3.eth.getGasPrice());
+        break;
+      case "getBalance":
+        result = await web3.eth.getBalance(
+          rpcParams[0],
+          rpcParams.length > 1 ? rpcParams[1] : undefined,
+        );
+        if (typeof result !== "string" || !result.startsWith("0x")) {
+          result = web3utils.numberToHex(result);
+        }
+        break;
+      case "getCode":
+        result = web3.eth.getCode(
+          rpcParams[0],
+          rpcParams.length > 1 ? rpcParams[1] : undefined,
+        );
+        break;
       case "getTransaction":
         result = await web3.eth.getTransaction(rpcParams[0]);
         break;

src/scripts/liteWalletProvider/main.ts

@@ -180,6 +180,15 @@ class LiteWalletProvider extends EventEmitter {
         case "eth_estimateGas":
           result = await this.handleRpcEstimateGas(clone(request.params));
           break;
+        case "eth_gasPrice":
+          result = await this.handleRpcGasPrice();
+          break;
+        case "eth_getBalance":
+          result = await this.handleRpcGetBalance(clone(request.params));
+          break;
+        case "eth_getCode":
+          result = await this.handleRpcGetCode(clone(request.params));
+          break;
         case "eth_getTransactionByHash":
           result = await this.handleRpcGetTransaction(clone(request.params));
           break;
@@ -536,6 +545,27 @@ class LiteWalletProvider extends EventEmitter {
     return await this.handleRpcMethod(chainId, "estimateGas", params);
   }
 
+  private async handleRpcGasPrice() {
+    // retrieve the web3 provider for connected chain
+    const chainIdHex = await this.handleGetConnectedChainIds();
+    const chainId = web3utils.hexToNumber(chainIdHex) as number;
+    return await this.handleRpcMethod(chainId, "gasPrice", []);
+  }
+
+  private async handleRpcGetBalance(params: any[]) {
+    // retrieve the web3 provider for connected chain
+    const chainIdHex = await this.handleGetConnectedChainIds();
+    const chainId = web3utils.hexToNumber(chainIdHex) as number;
+    return await this.handleRpcMethod(chainId, "getBalance", params);
+  }
+
+  private async handleRpcGetCode(params: any[]) {
+    // retrieve the web3 provider for connected chain
+    const chainIdHex = await this.handleGetConnectedChainIds();
+    const chainId = web3utils.hexToNumber(chainIdHex) as number;
+    return await this.handleRpcMethod(chainId, "getCode", params);
+  }
+
   private async handleRpcGetTransaction(params: any[]) {
     const chainIdHex = await this.handleGetConnectedChainIds();
     const chainId = web3utils.hexToNumber(chainIdHex) as number;

src/types/wallet/provider.ts

@@ -15,6 +15,9 @@ export type ProviderMethod =
   | "eth_sendTransaction"
   | "eth_signTypedData_v4"
   | "eth_blockNumber"
+  | "eth_getBalance"
+  | "eth_getCode"
+  | "eth_gasPrice"
   | "eth_getTransactionReceipt"
   | "eth_call"
   | "eth_estimateGas"
@@ -39,6 +42,9 @@ export const ProviderMethodsWithPrompt: ProviderMethod[] = [
 export type RpcRequest =
   | "call"
   | "estimateGas"
+  | "gasPrice"
+  | "getCode"
+  | "getBalance"
   | "getBlockNumber"
   | "getTransaction"
   | "getTransactionReceipt";