feat: add setWeb3Provider, remove chain from defineWeb3Provider

lidofinance · Aug 24, 2023 · 960fdcd · 960fdcd
1 parent 7bc4201
commit 960fdcd
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -61,7 +61,7 @@ const lidoSDK = new LidoSDK({
 });
 
 // Define default web3 provider in sdk (window.ethereum) if web3Provider is not defined in constructor
-lidoSDK.core.defineWeb3Provider(lidoSDK.core.chain);
+lidoSDK.core.defineWeb3Provider();
 
 // Views
 const balanceETH = await lidoSDK.core.balanceETH(address);

diff --git a/packages/sdk/README.md b/packages/sdk/README.md
@@ -150,7 +150,7 @@ const lidoSDK = new LidoSDK({
 });
 
 // Define default web3 provider in sdk (window.ethereum) if web3Provider is not defined in constructor
-lidoSDK.core.defineWeb3Provider(lidoSDK.core.chain);
+lidoSDK.core.defineWeb3Provider();
 
 const callback: StakeStageCallback = ({ stage, payload }) => {
   switch (stage) {

diff --git a/packages/sdk/src/core/core.ts b/packages/sdk/src/core/core.ts
@@ -103,14 +103,20 @@ export default class LidoSDKCore {
   }
 
   @Logger("Provider:")
-  public defineWeb3Provider(chain: Chain): WalletClient {
+  public defineWeb3Provider(): WalletClient {
     invariant(!this.web3Provider, "Web3 provider is already defined");
 
-    this.web3Provider = this.createWeb3Provider(chain);
+    this.web3Provider = this.createWeb3Provider(this.chain);
 
     return this.web3Provider;
   }
 
+  @Logger("Provider:")
+  public setWeb3Provider(web3Provider: WalletClient): void {
+    invariant(web3Provider.chain === this.chain, "Wrong chain");
+
+    this.web3Provider = web3Provider;
+  }
   // Balances
 
   @Logger("Balances:")