chore: adding the ability to pass the snapOrigin as a wallet plugin a…

…rgument
wharfkit · Aug 16, 2024 · 1b8266c · 1b8266c
1 parent cc90d78
commit 1b8266c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -22,11 +22,17 @@ export type GetSnapsResponse = Record<string, Snap>
 
 export * from './metamask'
 
+interface WalletPluginMetaMaskConfig {
+    snapOrigin?: string
+}
+
 const defaultSnapOrigin = 'local:http://localhost:8080'
 // const defaultSnapOrigin = 'npm:@greymass/test-snap'
 
 export class WalletPluginMetaMask extends AbstractWalletPlugin implements WalletPlugin {
     public id = 'wallet-plugin-metamask'
+    public snapOrigin: string
+
     readonly config: WalletPluginConfig = {
         requiresChainSelect: true,
         requiresPermissionSelect: true,
@@ -39,6 +45,12 @@ export class WalletPluginMetaMask extends AbstractWalletPlugin implements Wallet
         download: '',
     })
 
+    constructor(walletPluginMetaMaskConfig?: WalletPluginMetaMaskConfig) {
+        super()
+
+        this.snapOrigin = walletPluginMetaMaskConfig?.snapOrigin || defaultSnapOrigin
+    }
+
     login(context: LoginContext): Cancelable<WalletPluginLoginResponse> {
         const promise = this.metamaskLogin(context)
         return cancelable(promise, (canceled) => {