Merge pull request #24 from BuildOnBeam/feat/OB-870-improved-switch-chain-support

feat: improved switch chain support

Author: RobbyUitbeijerse

packages/sdk/src/BeamClient.ts

@@ -1,15 +1,15 @@
 import { AXIOS_INSTANCE } from './lib/api/beam-axios-client';
+import { getPlayerAPI } from './lib/api/beam.player-api.generated';
 import { BeamConfiguration } from './lib/config';
+import { beamIcon } from './lib/icon';
 import {
   BeamProvider,
   announceProvider,
   beamProviderInfo,
 } from './lib/provider';
 import { StorageKeys, StorageService } from './lib/storage';
 import { SessionManager } from './sessionManager';
-import { ClientConfig } from './types';
-import { getPlayerAPI } from './lib/api/beam.player-api.generated';
-import { beamIcon } from './lib/icon';
+import { ChainId, ClientConfig } from './types';
 
 export class BeamClient {
   readonly #config: BeamConfiguration;
@@ -42,14 +42,7 @@ export class BeamClient {
       storage: this.#storage,
     });
 
-    AXIOS_INSTANCE.interceptors.request.use((config) => {
-      config.baseURL = this.#config.getChainConfig().apiUrl;
-      config.headers.set(
-        'x-api-key',
-        this.#config.getChainConfig().publishableKey,
-      );
-      return config;
-    });
+    this.#setAxiosInterceptors();
   }
 
   /**
@@ -61,6 +54,25 @@ export class BeamClient {
     this.#storage = new StorageService<StorageKeys>(storage);
   }
 
+  /**
+   * Get the current chain configuration
+   */
+  public get chain() {
+    return this.#config.getChainConfig();
+  }
+
+  /**
+   * Set the chainId for the current client
+   * @param chainId
+   */
+  public switchChain(chainId: ChainId) {
+    if (this.#config.chainId === chainId) return;
+
+    this.#config.setChainId(chainId);
+
+    this.#setAxiosInterceptors();
+  }
+
   /**
    * Instantiate a new provider
    * @param options
@@ -156,6 +168,22 @@ export class BeamClient {
     );
   }
 
+  /**
+   * Set the axios interceptors for the current chain
+   */
+  #setAxiosInterceptors() {
+    if (!this.#config.chainId) return;
+
+    AXIOS_INSTANCE.interceptors.request.use((config) => {
+      config.baseURL = this.#config.getChainConfig().apiUrl;
+      config.headers.set(
+        'x-api-key',
+        this.#config.getChainConfig().publishableKey,
+      );
+      return config;
+    });
+  }
+
   /**
    * Base64 encoded SVG Beam logo
    */

packages/sdk/src/lib/config.ts

@@ -4,7 +4,7 @@ import { ChainId } from '../types';
 export class BeamConfiguration {
   readonly chains: ClientConfig['chains'];
 
-  #chainId: ChainId;
+  #chainId?: ChainId;
 
   readonly debug?: boolean;
 
@@ -22,7 +22,7 @@ export class BeamConfiguration {
       throw new Error(`Chain ${config.chainId} not found in configuration`);
     }
 
-    this.#chainId = config.chainId ?? config.chains[0].id;
+    if (config.chainId) this.#chainId = config.chainId;
 
     this.debug = config.debug || false;
   }
@@ -41,6 +41,11 @@ export class BeamConfiguration {
 
   getChainConfig() {
     const chainId = this.chainId;
+
+    if (!chainId) {
+      throw new Error('ChainId not configured, call setChainId first');
+    }
+
     const chain = this.chains.find((chain) => chain.id === chainId);
 
     if (!chain) {