Add simulation support

Author: FrederikBolding

packages/snaps-rpc-methods/src/permitted/handlers.ts

@@ -4,6 +4,7 @@ import { getAllSnapsHandler } from './getAllSnaps';
 import { getClientStatusHandler } from './getClientStatus';
 import { getCurrencyRateHandler } from './getCurrencyRate';
 import { getFileHandler } from './getFile';
+import { getInterfaceContextHandler } from './getInterfaceContext';
 import { getInterfaceStateHandler } from './getInterfaceState';
 import { getSnapsHandler } from './getSnaps';
 import { invokeKeyringHandler } from './invokeKeyring';
@@ -24,6 +25,7 @@ export const methodHandlers = {
   snap_createInterface: createInterfaceHandler,
   snap_updateInterface: updateInterfaceHandler,
   snap_getInterfaceState: getInterfaceStateHandler,
+  snap_getInterfaceContext: getInterfaceContextHandler,
   snap_resolveInterface: resolveInterfaceHandler,
   snap_getCurrencyRate: getCurrencyRateHandler,
   snap_experimentalProviderRequest: providerRequestHandler,

packages/snaps-sdk/src/types/methods/methods.ts

@@ -26,6 +26,10 @@ import type {
 } from './get-currency-rate';
 import type { GetEntropyParams, GetEntropyResult } from './get-entropy';
 import type { GetFileParams, GetFileResult } from './get-file';
+import type {
+  GetInterfaceContextParams,
+  GetInterfaceContextResult,
+} from './get-interface-context';
 import type {
   GetInterfaceStateParams,
   GetInterfaceStateResult,
@@ -79,6 +83,10 @@ export type SnapMethods = {
   snap_createInterface: [CreateInterfaceParams, CreateInterfaceResult];
   snap_updateInterface: [UpdateInterfaceParams, UpdateInterfaceResult];
   snap_getInterfaceState: [GetInterfaceStateParams, GetInterfaceStateResult];
+  snap_getInterfaceContext: [
+    GetInterfaceContextParams,
+    GetInterfaceContextResult,
+  ];
   snap_resolveInterface: [ResolveInterfaceParams, ResolveInterfaceResult];
   wallet_getSnaps: [GetSnapsParams, GetSnapsResult];
   wallet_invokeKeyring: [InvokeKeyringParams, InvokeKeyringResult];

packages/snaps-simulation/src/simulation.test.ts

@@ -376,6 +376,46 @@ describe('getHooks', () => {
     await close();
   });
 
+  it('returns the `getInterfaceContext` hook', async () => {
+    // eslint-disable-next-line no-new
+    new SnapInterfaceController({
+      messenger:
+        getRestrictedSnapInterfaceControllerMessenger(controllerMessenger),
+    });
+
+    jest.spyOn(controllerMessenger, 'call');
+
+    const { snapId, close } = await getMockServer({
+      manifest: getSnapManifest(),
+    });
+
+    const location = detectSnapLocation(snapId, {
+      allowLocal: true,
+    });
+    const snapFiles = await fetchSnap(snapId, location);
+
+    const { createInterface, getInterfaceContext } = getHooks(
+      getMockOptions(),
+      snapFiles,
+      snapId,
+      controllerMessenger,
+    );
+
+    const id = await createInterface(text('foo'), { bar: 'baz' });
+
+    const result = getInterfaceContext(id);
+
+    expect(controllerMessenger.call).toHaveBeenNthCalledWith(
+      3,
+      'SnapInterfaceController:getInterface',
+      snapId,
+      id,
+    );
+
+    expect(result).toStrictEqual({ bar: 'baz' });
+    await close();
+  });
+
   it('returns the `resolveInterface` hook', async () => {
     // eslint-disable-next-line no-new
     const snapInterfaceController = new SnapInterfaceController({

packages/snaps-simulation/src/simulation.ts

@@ -18,6 +18,7 @@ import type {
   AuxiliaryFileEncoding,
   Component,
   InterfaceState,
+  InterfaceContext,
   SnapId,
 } from '@metamask/snaps-sdk';
 import type { FetchedSnapFiles } from '@metamask/snaps-utils';
@@ -115,9 +116,13 @@ export type MiddlewareHooks = {
    * @returns A boolean flag signaling whether the client is locked.
    */
   getIsLocked: () => boolean;
-  createInterface: (content: Component) => Promise<string>;
+  createInterface: (
+    content: Component,
+    context?: InterfaceContext,
+  ) => Promise<string>;
   updateInterface: (id: string, content: Component) => Promise<void>;
   getInterfaceState: (id: string) => InterfaceState;
+  getInterfaceContext: (id: string) => InterfaceContext | null;
   resolveInterface: (id: string, value: Json) => Promise<void>;
 };
 
@@ -278,6 +283,12 @@ export function getHooks(
         snapId,
         ...args,
       ).state,
+    getInterfaceContext: (...args) =>
+      controllerMessenger.call(
+        'SnapInterfaceController:getInterface',
+        snapId,
+        ...args,
+      ).context,
     resolveInterface: async (...args) =>
       controllerMessenger.call(
         'SnapInterfaceController:resolveInterface',

packages/snaps-simulator/src/features/simulation/sagas.ts

@@ -213,6 +213,8 @@ export function* initSaga({ payload }: PayloadAction<string>) {
         await runSaga(createInterface, payload, content).toPromise(),
       getInterfaceState: (id: string) =>
         runSaga(getInterfaceState, payload, id).result(),
+      getInterfaceContext: (id: string) =>
+        runSaga(getInterface, payload, id).result().context,
       updateInterface: async (id: string, content: Component) =>
         await runSaga(updateInterface, payload, id, content).toPromise(),
     }),