feat: get current configuration from snap

ChainSafe · Jul 2, 2024 · 050398f · 050398f
1 parent 259590d
commit 050398f
Show file tree

Hide file tree

Showing 8 changed files with 25 additions and 7 deletions.
diff --git a/packages/adapter/src/extension/index.ts b/packages/adapter/src/extension/index.ts
@@ -43,11 +43,13 @@ function injectPolkadotSnap({
         await enablePolkadotSnap(config, snapOrigin, snapInstallationParams)
       ).getMetamaskSnapApi();
 
+      const snapConfiguration = await snap.getConfiguration();
+
       return {
         accounts: {
           get: async (): Promise<InjectedAccount[]> => {
             const response = await snap.getAddress();
-            return transformAccounts([response], config);
+            return transformAccounts([response], snapConfiguration);
           },
           // Currently there is only available only one account, in that case this method will never return anything
           // eslint-disable-next-line @typescript-eslint/no-unused-vars

diff --git a/packages/adapter/src/methods.ts b/packages/adapter/src/methods.ts
@@ -80,6 +80,10 @@ export async function exportAccount(
   )) as string;
 }
 
+export async function getConfiguration(this: MetamaskPolkadotSnap): Promise<SnapConfig> {
+  return (await sendSnapMethod({ method: 'getConfiguration' }, this.snapId)) as SnapConfig;
+}
+
 export async function setConfiguration(
   this: MetamaskPolkadotSnap,
   config: SnapConfig

diff --git a/packages/adapter/src/snap.ts b/packages/adapter/src/snap.ts
@@ -6,6 +6,7 @@ import {
   getAddress,
   getAllTransactions,
   getBalance,
+  getConfiguration,
   getLatestBlock,
   getPublicKey,
   sendSignedData,
@@ -39,6 +40,7 @@ export class MetamaskPolkadotSnap {
       getLatestBlock: getLatestBlock.bind(this),
       getPublicKey: getPublicKey.bind(this),
       send: sendSignedData.bind(this),
+      getConfiguration: getConfiguration.bind(this),
       setConfiguration: setConfiguration.bind(this),
       signPayloadJSON: signPayloadJSON.bind(this),
       signPayloadRaw: signPayloadRaw.bind(this)

diff --git a/packages/adapter/src/types.ts b/packages/adapter/src/types.ts
@@ -22,6 +22,8 @@ export interface MetamaskSnapApi {
 
   getLatestBlock(): Promise<BlockInfo>;
 
+  getConfiguration(): Promise<SnapConfig>;
+
   setConfiguration(configuration: SnapConfig): Promise<void>;
 
   getAllTransactions(): Promise<Transaction[]>;

diff --git a/packages/example/src/containers/Dashboard/Dashboard.tsx b/packages/example/src/containers/Dashboard/Dashboard.tsx
@@ -15,7 +15,7 @@ import type {
   BlockInfo,
   SnapNetworks,
   Transaction,
-  WellKnownSnapNetworks
+  SupportedSnapNetworks
 } from '@chainsafe/metamask-polkadot-types';
 import type { MetamaskSnapApi } from '@chainsafe/metamask-polkadot-adapter/src/types';
 import { Transfer } from '../../components/Transfer/Transfer';
@@ -57,7 +57,7 @@ export const Dashboard = (): React.JSX.Element => {
       return;
     } else setCustomNetworkInputs(false);
 
-    const networkName = event.target.value as WellKnownSnapNetworks;
+    const networkName = event.target.value as SupportedSnapNetworks;
     if (networkName === network) return;
     if (!api) return;
     await api.setConfiguration({ networkName: networkName });

diff --git a/packages/snap/snap.manifest.json b/packages/snap/snap.manifest.json
@@ -7,7 +7,7 @@
     "url": "git+https://github.com/chainsafe/metamask-snap-polkadot.git"
   },
   "source": {
-    "shasum": "Sx91E/1tHSq5I0NqCXv1u+P93YJh2dvNzf89X+SaIz8=",
+    "shasum": "Rox1dBkfYddPGejHC+G8uQ/MZeUf+T36mtTFB5qnTy8=",
     "location": {
       "npm": {
         "filePath": "dist/bundle.js",

diff --git a/packages/snap/src/index.ts b/packages/snap/src/index.ts
@@ -24,6 +24,7 @@ import {
   validSignPayloadRawSchema
 } from './util/validation';
 import { exportAccount } from './rpc/exportAccount';
+import { getConfiguration } from './configuration';
 
 const apiDependentMethods = [
   'getBlock',
@@ -78,6 +79,8 @@ export const onRpcRequest: OnRpcRequestHandler = async ({ request }) => {
     case 'getBalance': {
       return await getBalance(api);
     }
+    case 'getConfiguration':
+      return await getConfiguration();
     case 'configure': {
       const state = (await snap.request({
         method: 'snap_manageState',

diff --git a/packages/types/index.d.ts b/packages/types/index.d.ts
@@ -34,6 +34,10 @@ export interface GetBalanceRequest {
   method: 'getBalance';
 }
 
+export interface GetConfigurationRequest {
+  method: 'getConfiguration';
+}
+
 export interface ConfigureSnapRequest {
   method: 'configure';
   params: {
@@ -87,6 +91,7 @@ export type MetamaskPolkadotRpcRequest =
   | GetTransactionsRequest
   | GetBlockRequest
   | GetBalanceRequest
+  | GetConfigurationRequest
   | ConfigureSnapRequest
   | AddPolkadotAssetRequest
   | GetChainHeadRequest
@@ -132,16 +137,16 @@ export interface UnitConfiguration {
   customViewUrl?: string;
 }
 
-export type WellKnownSnapNetworks = 'polkadot' | 'kusama' | 'westend';
+export type SupportedSnapNetworks = 'polkadot' | 'kusama' | 'westend';
 
-export type SnapNetworks = WellKnownSnapNetworks | string;
+export type SnapNetworks = SupportedSnapNetworks | string;
 
 export type SnapConfig = {
   wsRpcUrl?: string;
   addressPrefix?: number;
   unit?: UnitConfiguration;
 } & (
-  | { networkName: WellKnownSnapNetworks; genesisHash?: `0x${string}` }
+  | { networkName: SupportedSnapNetworks; genesisHash?: `0x${string}` }
   | { networkName: SnapNetworks; genesisHash: `0x${string}` }
 );