@metamask/snaps-controllers 1.0.2 → 2.0.0

package/dist/multichain/MultiChainController.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"MultiChainController.js","sourceRoot":"","sources":["../../src/multichain/MultiChainController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AACA,+DAGmC;AAOnC,uDAAmE;AACnE,uDAmB+B;AAC/B,2CAAsD;AACtD,mCAAgC;AAEhC,oCAOkB;AAClB,yDAAyE;AACzE,yCAAsD;AAEtD,MAAM,cAAc,GAAG,sBAAsB,CAAC;AAE9C,MAAM,YAAY,GAA8B;IAC9C,QAAQ,EAAE,EAAE;CACb,CAAC;AAyCF,uEAAuE;AACvE,MAAa,oBAAqB,SAAQ,kCAIzC;IAGC;;;;;;OAMG;IACH,YAAY,EAAE,SAAS,EAAE,MAAM,EAA4B;QACzD,KAAK,CAAC;YACJ,SAAS;YACT,QAAQ,EAAE;gBACR,QAAQ,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE;aAC/C;YACD,IAAI,EAAE,cAAc;YACpB,KAAK,EAAE,YAAY;SACpB,CAAC,CAAC;QAjBL,+CAAyB;QAmBvB,uBAAA,IAAI,gCAAW,MAAM,MAAA,CAAC;IACxB,CAAC;IAED;;;;;OAKG;IACH,UAAU,CAAC,MAAc;QACvB,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACxC,IAAA,cAAM,EAAC,OAAO,EAAE,sBAAsB,CAAC,CAAC;QAExC,MAAM,uBAAA,IAAI,oCAAQ,MAAZ,IAAI,EAAS,MAAM,EAAE;YACzB,MAAM,EAAE,oCAAoC;SAC7C,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,OAAO,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,CACf,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAClD,IAAI,CAAC,eAAe,CAAC,IAAI,CACvB,0CAA0C,EAC1C,MAAM,CACP,CACF,CACF,CAAC;IACJ,CAAC;IAED;;;;;;;;;OASG;IACH,KAAK,CAAC,SAAS,CACb,MAAc,EACd,UAA4B;QAE5B,MAAM,eAAe,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAChD,IAAI,eAAe,EAAE;YACnB,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;SACjC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QACjE,MAAM,aAAa,GAAG,IAAA,wBAAgB,EAAC,KAAK,CAAC,CAAC;QAE9C,mEAAmE;QACnE,MAAM,mBAAmB,GAAG,MAAM,CAAC,WAAW,CAC5C,MAAM,OAAO,CAAC,GAAG,CACf,aAAa,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC;YAChC,IAAI,CAAC,EAAE;YACP,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC;SAClC,CAAC,CACH,CACF,CAAC;QAEF,qDAAqD;QACrD,MAAM,gBAAgB,GAAG,IAAA,mCAAwB,EAC/C,UAAU,CAAC,kBAAkB,EAC7B,mBAAmB,CACpB,CAAC;QAEF,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3C,qCAAqC,EACrC,MAAM,CACP,CAAC;QAEF,IAAA,cAAM,EACJ,WAAW,KAAK,SAAS,EACzB,GAAG,MAAM,iCAAiC,CAC3C,CAAC;QAEF,sEAAsE;QACtE,MAAM,0BAA0B,GAAG,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAExE,CAAC,GAAG,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE;YAC9B,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAC9C,IAAA,6BAAe,EAAC,WAAW,EAAE,MAAM,CAAC,CACrC,CAAC;YAEF,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC5B,GAAG,CAAC,SAAS,CAAC,GAAG,aAAa,CAAC;aAChC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAAE,CAAC,CAAC;QAEP,6FAA6F;QAC7F,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,IAAI,CACrD,CAAC,SAAS,EAAE,EAAE,CACZ,CAAC,IAAA,mBAAW,EAAC,0BAA0B,EAAE,SAAS,CAAC;YACnD,0BAA0B,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,CAAC,CACpD,CAAC;QAEF,uEAAuE;QACvE,MAAM,0BAA0B,GAAG,YAAY;YAC7C,CAAC,CAAC,gBAAgB;YAClB,CAAC,CAAC,0BAA0B,CAAC;QAE/B,sCAAsC;QACtC,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,oBAAoB,CACtD,MAAM,EACN,0BAA0B,EAC1B,UAAU,CAAC,kBAAkB,CAC9B,CAAC;QAEF,oEAAoE;QACpE,gEAAgE;QAChE,4CAA4C;QAC5C,IAAA,cAAM,EACJ,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAClC,CAAC,eAAe,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAChD,EACD,uDAAuD,CACxD,CAAC;QAEF,yEAAyE;QACzE,gEAAgE;QAChE,MAAM,gBAAgB,GAAG,YAAY;YACnC,CAAC,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,gBAAgB,CAAC;YACvD,CAAC,CAAC,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAClC,CAAC,CAAC,SAAS,EAAE,eAAe,CAAC,EAAE,EAAE,CAAC;gBAChC,SAAS;gBACT,eAAe,CAAC,CAAC,CAAC,IAAI,IAAI;aAC3B,CACF,CACF,CAAC;QAEN,kDAAkD;QAClD,MAAM,kBAAkB,GAAG,MAAM,CAAC,OAAO,CACvC,UAAU,CAAC,kBAAkB,CAC9B,CAAC,MAAM,CACN,CAAC,GAAG,EAAE,CAAC,WAAW,EAAE,SAAS,CAAC,EAAE,EAAE;YAChC,MAAM,QAAQ,GAAG,gBAAgB,CAAC,WAAW,CAAC,EAAE,QAAQ,CAAC;YACzD,IAAI,QAAQ,EAAE;gBACZ,GAAG,CAAC,WAAW,CAAC,GAAG;oBACjB,QAAQ;oBACR,MAAM,EAAE,SAAS,CAAC,MAAM;oBACxB,MAAM,EAAE,SAAS,CAAC,MAAM;oBACxB,OAAO,EAAE,SAAS,CAAC,OAAO;iBAC3B,CAAC;aACH;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,CACH,CAAC;QAEF,8DAA8D;QAC9D,MAAM,aAAa,GAAG,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAE3D,CAAC,GAAG,EAAE,CAAC,WAAW,EAAE,eAAe,CAAC,EAAE,EAAE;YACxC,IAAI,eAAe,EAAE;gBACnB,GAAG,CAAC,WAAW,CAAC,GAAG,eAAe,CAAC,MAAM,CAAC;aAC3C;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAAE,CAAC,CAAC;QAEP,MAAM,OAAO,GAAgB;YAC3B,MAAM;YACN,mBAAmB,EAAE,UAAU,CAAC,kBAAkB;YAClD,kBAAkB;YAClB,aAAa;SACd,CAAC;QAEF,0EAA0E;QAC1E,MAAM,OAAO,CAAC,GAAG,CACf,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAClD,IAAI,CAAC,eAAe,CAAC,IAAI,CACvB,0CAA0C,EAC1C,MAAM,CACP,CACF,CACF,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,OAAO,EAAE,UAAU,EAAE,kBAAkB,EAAE,CAAC;IAC5C,CAAC;IAED;;;;;;;;;;;OAWG;IACH,KAAK,CAAC,SAAS,CACb,MAAc,EACd,IAAqD;QAErD,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACxC,IAAA,cAAM,EAAC,OAAO,EAAE,gBAAgB,MAAM,kBAAkB,CAAC,CAAC;QAE1D,MAAM,EAAE,SAAS,EAAE,GAAG,IAAA,0BAAY,EAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjD,MAAM,gBAAgB,GAAG,OAAO,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAC/D,IAAA,cAAM,EACJ,OAAO,CAAC,kBAAkB,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EACpE,gBAAgB,MAAM,0BAA0B,IAAI,CAAC,OAAO,UAAU,CACvE,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;QAChC,IAAA,cAAM,EACJ,gBAAgB,EAAE,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,EAC3C,gBAAgB,MAAM,sBAAsB,MAAM,EAAE,CACrD,CAAC;QAEF,MAAM,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QAChD,IAAA,cAAM,EAAC,MAAM,KAAK,SAAS,CAAC,CAAC;QAE7B,4EAA4E;QAC5E,oEAAoE;QACpE,mEAAmE;QACnE,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3C,qCAAqC,EACrC,MAAM,CACP,CAAC;QAEF,IAAA,cAAM,EACJ,WAAW,KAAK,SAAS,EACzB,GAAG,MAAM,iCAAiC,CAC3C,CAAC;QAEF,IAAA,cAAM,EACJ,IAAA,6BAAe,EAAC,WAAW,EAAE,MAAM,CAAC,EACpC,GAAG,MAAM,iDAAiD,MAAM,GAAG,CACpE,CAAC;QAEF,OAAO,IAAI,CAAC,WAAW,CAAC;YACtB,MAAM;YACN,MAAM;YACN,MAAM,EAAE,eAAe;YACvB,IAAI,EAAE,IAAI;SACX,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;;OAUG;IACK,KAAK,CAAC,WAAW,CAAC,EACxB,MAAM,EACN,MAAM,EACN,MAAM,EACN,IAAI,GAML;QACC,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,8BAA8B,EAAE;YAC/D,MAAM;YACN,MAAM;YACN,OAAO,EAAE,yBAAW,CAAC,WAAW;YAChC,OAAO,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;;OAUG;IACK,KAAK,CAAC,eAAe,CAC3B,MAAc,EACd,MAAc;QAEd,IAAI;YACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC;gBACpC,MAAM;gBACN,MAAM;gBACN,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;YAEH,IAAI,IAAA,8BAAgB,EAAC,MAAM,CAAC,EAAE;gBAC5B,OAAO,MAAM,CAAC;aACf;SACF;QAAC,OAAO,KAAK,EAAE;YACd,wBAAwB;YACxB,IAAA,sBAAQ,EAAC,KAAK,CAAC,CAAC;SACjB;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;OAMG;IACK,gBAAgB,CAAC,IAAmB;QAC1C,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3C,qCAAqC,EACrC,IAAI,CAAC,EAAE,CACR,CAAC;QAEF,MAAM,iBAAiB,GAAG,WAAW,EAAE,CAAC,sBAAc,CAAC,OAAO,CAAC,CAAC;QAChE,OAAO,IAAA,oCAA0B,EAAC,iBAAiB,CAAC,CAAC;IACvD,CAAC;IAED;;;;;;;;;;;;OAYG;IACK,KAAK,CAAC,oBAAoB,CAChC,MAAc,EACd,kBAAiD,EACjD,mBAA0D;QAE1D,MAAM,YAAY,GAAG;YACnB,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,CAAC;SACzC,CAAC;QAEd,MAAM,WAAW,GAAG,MAAM,YAAY,CAAC,MAAM,CAE3C,KAAK,EAAE,eAAe,EAAE,MAAM,EAAE,EAAE;YAClC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAC1D,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC;YAClC,IAAI,MAAM,EAAE;gBACV,GAAG,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC;aACtB;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;QAExB,OAAO,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,MAAM,CAE3C,CAAC,GAAG,EAAE,WAAW,EAAE,EAAE;YACrB,MAAM,EAAE,MAAM,EAAE,GAAG,mBAAmB,CAAC,WAAW,CAAC,CAAC;YAEpD,MAAM,0BAA0B,GAAG,CAAC,OAAkB,EAAE,EAAE;gBACxD,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,IAAA,4BAAc,EAAC,OAAO,CAAC,CAAC;gBAC3D,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,KAAK,aAAa,CAAC,CAAC;YAC7D,CAAC,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;iBACvC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC5B,MAAM;gBACN,QAAQ,EAAE,QAAQ,CAAC,MAAM,CAAC,0BAA0B,CAAC;aACtD,CAAC,CAAC;iBACF,MAAM,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAEjD,GAAG,CAAC,WAAW,CAAC,GAAG,MAAM,CAAC;YAE1B,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAAE,CAAC,CAAC;IACT,CAAC;IAED;;;;;;;;;OASG;IACK,KAAK,CAAC,gBAAgB,CAC5B,MAAc,EACd,gBAGC;QAID,oCAAoC;QACpC,MAAM,EAAE,GAAG,IAAA,eAAM,GAAE,CAAC;QACpB,MAAM,gBAAgB,GAAG,CAAC,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CACvD,+BAA+B,EAC/B;YACE,MAAM;YACN,EAAE;YACF,IAAI,EAAE,oBAAoB;YAC1B,WAAW,EAAE;gBACX,gBAAgB;aACjB;SACF,EACD,IAAI,CACL,CAA0E,CAAC;QAE5E,oEAAoE;QACpE,8BAA8B;QAC9B,4EAA4E;QAC5E,0EAA0E;QAC1E,4EAA4E;QAE5E,MAAM,mBAAmB,GAAG,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAGhE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;YACZ,IAAI,IAAI,KAAK,IAAI,EAAE;gBACjB,GAAG,CAAC,wCAA0B,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;aACnD;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,CAAC,wCAA0B,CAAC,EAAE,EAAE,EAAE,CACrC,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,uCAAuC,EAAE;YACjE,mBAAmB;YACnB,OAAO,EAAE,EAAE,MAAM,EAAE;SACpB,CAAC,CAAC;QAEH,OAAO,gBAAgB,CAAC;IAC1B,CAAC;CACF;AA9dD,oDA8dC","sourcesContent":["import { AddApprovalRequest } from '@metamask/approval-controller';\nimport {\n  BaseControllerV2 as BaseController,\n  RestrictedControllerMessenger,\n} from '@metamask/base-controller';\nimport {\n  GetPermissions,\n  GrantPermissions,\n  HasPermission,\n  PermissionConstraint,\n} from '@metamask/permission-controller';\nimport { WALLET_SNAP_PERMISSION_KEY } from '@metamask/rpc-methods';\nimport {\n  SnapKeyring,\n  parseAccountId,\n  AccountId,\n  parseChainId,\n  ChainId,\n  ConnectArguments,\n  isSnapPermitted,\n  HandlerType,\n  NamespaceId,\n  RequestArguments,\n  RequestNamespace,\n  Session,\n  TruncatedSnap,\n  SnapId,\n  SessionNamespace,\n  isAccountIdArray,\n  Namespaces,\n  logError,\n} from '@metamask/snaps-utils';\nimport { hasProperty, assert } from '@metamask/utils';\nimport { nanoid } from 'nanoid';\n\nimport {\n  GetAllSnaps,\n  HandleSnapRequest,\n  IncrementActiveReferences,\n  DecrementActiveReferences,\n  SnapEndowments,\n  getRunnableSnaps,\n} from '../snaps';\nimport { getKeyringCaveatNamespaces } from '../snaps/endowments/keyring';\nimport { findMatchingKeyringSnaps } from './matching';\n\nconst controllerName = 'MultiChainController';\n\nconst defaultState: MultiChainControllerState = {\n  sessions: {},\n};\n\ntype AllowedActions =\n  | GetAllSnaps\n  | IncrementActiveReferences\n  | DecrementActiveReferences\n  | HandleSnapRequest\n  | GetPermissions\n  | HasPermission\n  | AddApprovalRequest\n  | GrantPermissions;\n\ntype MultiChainControllerMessenger = RestrictedControllerMessenger<\n  typeof controllerName,\n  AllowedActions,\n  never,\n  AllowedActions['type'],\n  never\n>;\n\ntype SessionData = {\n  origin: string;\n  requestedNamespaces: Record<NamespaceId, RequestNamespace>;\n  providedNamespaces: Record<NamespaceId, RequestNamespace>;\n  handlingSnaps: Record<NamespaceId, SnapId>;\n};\n\ntype MultiChainControllerState = {\n  sessions: { [origin: string]: SessionData };\n};\n\ntype Notify = (\n  origin: string,\n  data: { method: string; params?: Record<string, unknown> },\n) => Promise<void>;\n\ntype MultiChainControllerArgs = {\n  notify: Notify;\n  messenger: MultiChainControllerMessenger;\n};\n\n// TODO(ritave): Support for legacy ethereum operations, not just snaps\nexport class MultiChainController extends BaseController<\n  typeof controllerName,\n  MultiChainControllerState,\n  MultiChainControllerMessenger\n> {\n  readonly #notify: Notify;\n\n  /**\n   * Construct a new {@link MultiChainController} instance.\n   *\n   * @param args - The arguments to construct the controller with.\n   * @param args.messenger - The controller messenger to use.\n   * @param args.notify - A function that should handle JSON-RPC notifications.\n   */\n  constructor({ messenger, notify }: MultiChainControllerArgs) {\n    super({\n      messenger,\n      metadata: {\n        sessions: { persist: false, anonymous: false },\n      },\n      name: controllerName,\n      state: defaultState,\n    });\n\n    this.#notify = notify;\n  }\n\n  /**\n   * Get an open session for the given origin.\n   *\n   * @param origin - The origin to get the session for.\n   * @returns The session, if it exists, or `undefined` otherwise.\n   */\n  getSession(origin: string): SessionData | undefined {\n    return this.state.sessions[origin];\n  }\n\n  /**\n   * Close a session for the given origin.\n   *\n   * @param origin - The origin to close the session for.\n   * @throws If the session does not exist.\n   */\n  async closeSession(origin: string): Promise<void> {\n    const session = this.getSession(origin);\n    assert(session, 'No session to close.');\n\n    await this.#notify(origin, {\n      method: 'multichainHack_metamask_disconnect',\n    });\n\n    this.update((state) => {\n      delete state.sessions[origin];\n    });\n\n    await Promise.all(\n      Object.values(session.handlingSnaps).map((snapId) =>\n        this.messagingSystem.call(\n          'SnapController:decrementActiveReferences',\n          snapId,\n        ),\n      ),\n    );\n  }\n\n  /**\n   * Handles a new connection from the given origin. This will create a new\n   * session, and close any existing session for the origin.\n   *\n   * @param origin - The origin to create the session for.\n   * @param connection - The connection arguments.\n   * @param connection.requiredNamespaces - The namespaces that the origin\n   * requires.\n   * @returns The session that was created.\n   */\n  async onConnect(\n    origin: string,\n    connection: ConnectArguments,\n  ): Promise<Session> {\n    const existingSession = this.getSession(origin);\n    if (existingSession) {\n      await this.closeSession(origin);\n    }\n\n    const snaps = this.messagingSystem.call('SnapController:getAll');\n    const filteredSnaps = getRunnableSnaps(snaps);\n\n    // Get available namespaces supported by currently installed Snaps.\n    const availableNamespaces = Object.fromEntries(\n      await Promise.all(\n        filteredSnaps.map(async (snap) => [\n          snap.id,\n          await this.snapToNamespaces(snap),\n        ]),\n      ),\n    );\n\n    // The magical matching algorithm specified in SIP-2.\n    const namespaceToSnaps = findMatchingKeyringSnaps(\n      connection.requiredNamespaces,\n      availableNamespaces,\n    );\n\n    const permissions = this.messagingSystem.call(\n      'PermissionController:getPermissions',\n      origin,\n    );\n\n    assert(\n      permissions !== undefined,\n      `${origin} does not have any permissions.`,\n    );\n\n    // Find namespaces that can be satisfied with existing approved Snaps.\n    const approvedNamespacesAndSnaps = Object.entries(namespaceToSnaps).reduce<\n      Record<NamespaceId, SnapId[]>\n    >((acc, [namespace, snapIds]) => {\n      const approvedSnaps = snapIds.filter((snapId) =>\n        isSnapPermitted(permissions, snapId),\n      );\n\n      if (approvedSnaps.length > 0) {\n        acc[namespace] = approvedSnaps;\n      }\n      return acc;\n    }, {});\n\n    // If we either don't have a snap to handle a namespace or we have multiple we have conflicts\n    const hasConflicts = Object.keys(namespaceToSnaps).some(\n      (namespace) =>\n        !hasProperty(approvedNamespacesAndSnaps, namespace) ||\n        approvedNamespacesAndSnaps[namespace]?.length > 1,\n    );\n\n    // Use already approved snaps if they satisfy the requested namespaces.\n    const filteredNamespacesAndSnaps = hasConflicts\n      ? namespaceToSnaps\n      : approvedNamespacesAndSnaps;\n\n    // Fetch possible accounts from snaps.\n    const possibleAccounts = await this.namespacesToAccounts(\n      origin,\n      filteredNamespacesAndSnaps,\n      connection.requiredNamespaces,\n    );\n\n    // For now we fail here if no namespaces could be matched to a snap.\n    // We don't fail if at least one namespace is matched to a snap.\n    // TODO: Decide whether this is what we want\n    assert(\n      Object.values(possibleAccounts).some(\n        (possibleAccount) => possibleAccount.length > 0,\n      ),\n      'No installed snaps found for any requested namespace.',\n    );\n\n    // If currently installed Snaps / configuration doesn't solve request, we\n    // need to show a prompt. This is handled by `resolveConflicts`.\n    const resolvedAccounts = hasConflicts\n      ? await this.resolveConflicts(origin, possibleAccounts)\n      : Object.fromEntries(\n          Object.entries(possibleAccounts).map(\n            ([namespace, snapAndAccounts]) => [\n              namespace,\n              snapAndAccounts[0] ?? null,\n            ],\n          ),\n        );\n\n    // Aggregate information about session namespaces.\n    const providedNamespaces = Object.entries(\n      connection.requiredNamespaces,\n    ).reduce<Record<NamespaceId, SessionNamespace>>(\n      (acc, [namespaceId, namespace]) => {\n        const accounts = resolvedAccounts[namespaceId]?.accounts;\n        if (accounts) {\n          acc[namespaceId] = {\n            accounts,\n            chains: namespace.chains,\n            events: namespace.events,\n            methods: namespace.methods,\n          };\n        }\n        return acc;\n      },\n      {},\n    );\n\n    // Collect information about handler Snaps for each namespace.\n    const handlingSnaps = Object.entries(resolvedAccounts).reduce<\n      Record<NamespaceId, SnapId>\n    >((acc, [namespaceId, accountsAndSnap]) => {\n      if (accountsAndSnap) {\n        acc[namespaceId] = accountsAndSnap.snapId;\n      }\n      return acc;\n    }, {});\n\n    const session: SessionData = {\n      origin,\n      requestedNamespaces: connection.requiredNamespaces,\n      providedNamespaces,\n      handlingSnaps,\n    };\n\n    // Makes sure used Snaps aren't killed while they are serving the session.\n    await Promise.all(\n      Object.values(session.handlingSnaps).map((snapId) =>\n        this.messagingSystem.call(\n          'SnapController:incrementActiveReferences',\n          snapId,\n        ),\n      ),\n    );\n\n    this.update((state) => {\n      state.sessions[origin] = session;\n    });\n\n    return { namespaces: providedNamespaces };\n  }\n\n  /**\n   * Handle an incoming multichain request from the given origin. This will\n   * forward the request to the appropriate Snap, and return the response.\n   *\n   * @param origin - The origin to handle the request for.\n   * @param data - The request data.\n   * @param data.chainId - The chain ID for the request.\n   * @param data.request - The request arguments, i.e., the method and params.\n   * @returns The response from the Snap.\n   * @throws If the session does not exist, or the session does not provide the\n   * requested namespace.\n   */\n  async onRequest(\n    origin: string,\n    data: { chainId: ChainId; request: RequestArguments },\n  ): Promise<unknown> {\n    const session = this.getSession(origin);\n    assert(session, `Session for \"${origin}\" doesn't exist.`);\n\n    const { namespace } = parseChainId(data.chainId);\n    const sessionNamespace = session.providedNamespaces[namespace];\n    assert(\n      session.providedNamespaces[namespace]?.chains.includes(data.chainId),\n      `Session for \"${origin}\" is not connected to \"${data.chainId}\" chain.`,\n    );\n\n    const { method } = data.request;\n    assert(\n      sessionNamespace?.methods?.includes(method),\n      `Session for \"${origin}\" does not support ${method}`,\n    );\n\n    const snapId = session.handlingSnaps[namespace];\n    assert(snapId !== undefined);\n\n    // TODO: Get permission for origin connecting to snap, or get user approval.\n    // In the future this is where we should prompt for this permission.\n    // In this iteration, we will grant this permission in `onConnect`.\n    const permissions = this.messagingSystem.call(\n      'PermissionController:getPermissions',\n      origin,\n    );\n\n    assert(\n      permissions !== undefined,\n      `${origin} does not have any permissions.`,\n    );\n\n    assert(\n      isSnapPermitted(permissions, snapId),\n      `${origin} does not have permission to communicate with ${snapId}.`,\n    );\n\n    return this.snapRequest({\n      snapId,\n      origin,\n      method: 'handleRequest',\n      args: data,\n    });\n  }\n\n  /**\n   * Send a request to the given Snap. This calls the given method with the\n   * given arguments on the keyring class in the given Snap.\n   *\n   * @param options - The request options.\n   * @param options.snapId - The ID of the Snap to send the request to.\n   * @param options.origin - The origin of the request.\n   * @param options.method - The request method.\n   * @param options.args - The request params.\n   * @returns The response from the Snap.\n   */\n  private async snapRequest({\n    snapId,\n    origin,\n    method,\n    args,\n  }: {\n    snapId: SnapId;\n    origin: string;\n    method: keyof SnapKeyring;\n    args?: unknown;\n  }) {\n    return this.messagingSystem.call('SnapController:handleRequest', {\n      snapId,\n      origin,\n      handler: HandlerType.SnapKeyring,\n      request: { method, params: args ? [args] : [] },\n    });\n  }\n\n  /**\n   * Get the accounts exposed by the Snap's keyring.\n   *\n   * This also verifies that the accounts returned by the snap are valid CAIP-10\n   * account IDs.\n   *\n   * @param origin - The origin of the request.\n   * @param snapId - The ID of the Snap to get the accounts from.\n   * @returns The accounts, or `null` if the Snap does not have any accounts, or\n   * the accounts are invalid (i.e., not valid CAIP-10 account IDs).\n   */\n  private async getSnapAccounts(\n    origin: string,\n    snapId: SnapId,\n  ): Promise<AccountId[] | null> {\n    try {\n      const result = await this.snapRequest({\n        snapId,\n        origin,\n        method: 'getAccounts',\n      });\n\n      if (isAccountIdArray(result)) {\n        return result;\n      }\n    } catch (error) {\n      // Ignore errors for now\n      logError(error);\n    }\n\n    return null;\n  }\n\n  /**\n   * Get the namespaces for the given Snap, as described in the Snap's manifest.\n   *\n   * @param snap - The Snap to get the namespaces for.\n   * @returns The namespaces, or `null` if the Snap does not have any\n   * namespaces.\n   */\n  private snapToNamespaces(snap: TruncatedSnap): Namespaces | null {\n    const permissions = this.messagingSystem.call(\n      'PermissionController:getPermissions',\n      snap.id,\n    );\n\n    const keyringPermission = permissions?.[SnapEndowments.Keyring];\n    return getKeyringCaveatNamespaces(keyringPermission);\n  }\n\n  /**\n   * Maps from an object of namespace IDs and Snap IDs, and an object of\n   * namespace IDs and requested namespaces, to an object of namespace IDs and\n   * resolved accounts, with the Snap ID providing the accounts.\n   *\n   * @param origin - The origin of the request.\n   * @param namespacesAndSnaps - An object of namespace IDs and Snap IDs\n   * providing the namespace.\n   * @param requestedNamespaces - An object of namespace IDs and requested\n   * namespaces.\n   * @returns An object of namespace IDs and resolved accounts, with the Snap ID\n   * providing the accounts.\n   */\n  private async namespacesToAccounts(\n    origin: string,\n    namespacesAndSnaps: Record<NamespaceId, SnapId[]>,\n    requestedNamespaces: Record<NamespaceId, RequestNamespace>,\n  ): Promise<Record<NamespaceId, { snapId: SnapId; accounts: AccountId[] }[]>> {\n    const dedupedSnaps = [\n      ...new Set(Object.values(namespacesAndSnaps).flat()),\n    ] as SnapId[];\n\n    const allAccounts = await dedupedSnaps.reduce<\n      Promise<Record<string, AccountId[]>>\n    >(async (previousPromise, snapId) => {\n      const result = await this.getSnapAccounts(origin, snapId);\n      const acc = await previousPromise;\n      if (result) {\n        acc[snapId] = result;\n      }\n      return acc;\n    }, Promise.resolve({}));\n\n    return Object.keys(namespacesAndSnaps).reduce<\n      Record<NamespaceId, { snapId: SnapId; accounts: AccountId[] }[]>\n    >((acc, namespaceId) => {\n      const { chains } = requestedNamespaces[namespaceId];\n\n      const accountInAnyRequestedChain = (account: AccountId) => {\n        const { chainId: parsedChainId } = parseAccountId(account);\n        return chains.some((chainId) => chainId === parsedChainId);\n      };\n\n      const result = Object.entries(allAccounts)\n        .map(([snapId, accounts]) => ({\n          snapId,\n          accounts: accounts.filter(accountInAnyRequestedChain),\n        }))\n        .filter(({ accounts }) => accounts.length > 0);\n\n      acc[namespaceId] = result;\n\n      return acc;\n    }, {});\n  }\n\n  /**\n   * If multiple Snap IDs are provided for a namespace, this method will\n   * determine which Snap ID to use for the namespace, by showing the user a\n   * prompt.\n   *\n   * @param origin - The origin of the request.\n   * @param possibleAccounts - An object containing the accounts provided by\n   * each Snap ID for each namespace.\n   * @returns An object containing the Snap ID to use for each namespace.\n   */\n  private async resolveConflicts(\n    origin: string,\n    possibleAccounts: Record<\n      NamespaceId,\n      { snapId: SnapId; accounts: AccountId[] }[]\n    >,\n  ): Promise<\n    Record<NamespaceId, { snapId: SnapId; accounts: AccountId[] } | null>\n  > {\n    // Get user approval for connection.\n    const id = nanoid();\n    const resolvedAccounts = (await this.messagingSystem.call(\n      'ApprovalController:addRequest',\n      {\n        origin,\n        id,\n        type: 'multichain_connect',\n        requestData: {\n          possibleAccounts,\n        },\n      },\n      true,\n    )) as Record<NamespaceId, { snapId: SnapId; accounts: AccountId[] } | null>;\n\n    // TODO: In the future, use another permission here to not give full\n    // permission after handshake.\n    // Instead we should give origin only a read-only access to list of accounts\n    // without allowing provider.request() talking to a snap before additional\n    // user approval. The additional approval would be requested in `onRequest`.\n\n    const approvedPermissions = Object.values(resolvedAccounts).reduce<\n      Record<string, Record<string, Partial<PermissionConstraint>>>\n    >(\n      (acc, curr) => {\n        if (curr !== null) {\n          acc[WALLET_SNAP_PERMISSION_KEY][curr.snapId] = {};\n        }\n        return acc;\n      },\n      { [WALLET_SNAP_PERMISSION_KEY]: {} },\n    );\n\n    this.messagingSystem.call('PermissionController:grantPermissions', {\n      approvedPermissions,\n      subject: { origin },\n    });\n\n    return resolvedAccounts;\n  }\n}\n"]}

package/dist/multichain/index.d.ts

@@ -1,3 +0,0 @@
-export * from './matching';
-export * from './middleware';
-export * from './MultiChainController';

package/dist/multichain/index.js

@@ -1,20 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./matching"), exports);
-__exportStar(require("./middleware"), exports);
-__exportStar(require("./MultiChainController"), exports);
-//# sourceMappingURL=index.js.map

package/dist/multichain/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/multichain/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,6CAA2B;AAC3B,+CAA6B;AAC7B,yDAAuC","sourcesContent":["export * from './matching';\nexport * from './middleware';\nexport * from './MultiChainController';\n"]}

package/dist/multichain/matching.d.ts

@@ -1,9 +0,0 @@
-import { ConnectArguments, NamespaceId, SnapId, Namespace } from '@metamask/snaps-utils';
-/**
- * Finds a keyring snap for each namespace implements at a minimum the requested functionality.
- *
- * @param requestedNamespaces - The requested namespaces.
- * @param snaps - All snaps and their exposed keyring namespaces.
- * @returns A mapping between namespaces and snap ids.
- */
-export declare function findMatchingKeyringSnaps(requestedNamespaces: ConnectArguments['requiredNamespaces'], snaps: Record<SnapId, Record<NamespaceId, Namespace> | null>): Record<NamespaceId, SnapId[]>;

package/dist/multichain/matching.js

@@ -1,57 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.findMatchingKeyringSnaps = void 0;
-/**
- * Finds a keyring snap for each namespace implements at a minimum the requested functionality.
- *
- * @param requestedNamespaces - The requested namespaces.
- * @param snaps - All snaps and their exposed keyring namespaces.
- * @returns A mapping between namespaces and snap ids.
- */
-function findMatchingKeyringSnaps(requestedNamespaces, snaps) {
-    const snapEntries = Object.entries(snaps);
-    return Object.entries(requestedNamespaces).reduce((acc, [requestedNamespaceId, currentRequestedNamespace]) => {
-        const matchedSnaps = snapEntries
-            .filter(([, namespaces]) => {
-            const potentialMatch = namespaces?.[requestedNamespaceId];
-            return (potentialMatch !== undefined &&
-                matchNamespace(currentRequestedNamespace, potentialMatch));
-        })
-            .map(([snapId]) => snapId);
-        acc[requestedNamespaceId] = matchedSnaps;
-        return acc;
-    }, {});
-}
-exports.findMatchingKeyringSnaps = findMatchingKeyringSnaps;
-/**
- * Determines whether a keyring namespace is a match given a potential match and the requested namespace.
- *
- * This function assumes that the namespace ID has already been matched.
- *
- * @param requestedNamespace - The requested namespace information.
- * @param potentialMatchNamespace - The potential match.
- * @returns True if the potentially matching namespace is a match.
- */
-function matchNamespace(requestedNamespace, potentialMatchNamespace) {
-    if (!requestedNamespace.chains.every((requestedChain) => potentialMatchNamespace.chains.some((potentialMatchChain) => potentialMatchChain.id === requestedChain))) {
-        return false;
-    }
-    if (!isSubset(requestedNamespace.events ?? [], potentialMatchNamespace.events ?? [])) {
-        return false;
-    }
-    if (!isSubset(requestedNamespace.methods ?? [], potentialMatchNamespace.methods ?? [])) {
-        return false;
-    }
-    return true;
-}
-/**
- * Determines whether an array is a subset of another array.
- *
- * @param potentialSubset - The potential subset.
- * @param array - The other array.
- * @returns True if the first argument is a subset of second argument.
- */
-function isSubset(potentialSubset, array) {
-    return potentialSubset.every((item) => array.includes(item));
-}
-//# sourceMappingURL=matching.js.map

package/dist/multichain/matching.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"matching.js","sourceRoot":"","sources":["../../src/multichain/matching.ts"],"names":[],"mappings":";;;AAOA;;;;;;GAMG;AACH,SAAgB,wBAAwB,CACtC,mBAA2D,EAC3D,KAA4D;IAE5D,MAAM,WAAW,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC1C,OAAO,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,MAAM,CAE/C,CAAC,GAAG,EAAE,CAAC,oBAAoB,EAAE,yBAAyB,CAAC,EAAE,EAAE;QAC3D,MAAM,YAAY,GAAG,WAAW;aAC7B,MAAM,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,EAAE,EAAE;YACzB,MAAM,cAAc,GAAG,UAAU,EAAE,CAAC,oBAAoB,CAAC,CAAC;YAC1D,OAAO,CACL,cAAc,KAAK,SAAS;gBAC5B,cAAc,CAAC,yBAAyB,EAAE,cAAc,CAAC,CAC1D,CAAC;QACJ,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC;QAC7B,GAAG,CAAC,oBAAoB,CAAC,GAAG,YAAY,CAAC;QACzC,OAAO,GAAG,CAAC;IACb,CAAC,EAAE,EAAE,CAAC,CAAC;AACT,CAAC;AApBD,4DAoBC;AAED;;;;;;;;GAQG;AACH,SAAS,cAAc,CACrB,kBAAuE,EACvE,uBAAkC;IAElC,IACE,CAAC,kBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,EAAE,EAAE,CAClD,uBAAuB,CAAC,MAAM,CAAC,IAAI,CACjC,CAAC,mBAAmB,EAAE,EAAE,CAAC,mBAAmB,CAAC,EAAE,KAAK,cAAc,CACnE,CACF,EACD;QACA,OAAO,KAAK,CAAC;KACd;IAED,IACE,CAAC,QAAQ,CACP,kBAAkB,CAAC,MAAM,IAAI,EAAE,EAC/B,uBAAuB,CAAC,MAAM,IAAI,EAAE,CACrC,EACD;QACA,OAAO,KAAK,CAAC;KACd;IAED,IACE,CAAC,QAAQ,CACP,kBAAkB,CAAC,OAAO,IAAI,EAAE,EAChC,uBAAuB,CAAC,OAAO,IAAI,EAAE,CACtC,EACD;QACA,OAAO,KAAK,CAAC;KACd;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;;;;GAMG;AACH,SAAS,QAAQ,CAAI,eAAoB,EAAE,KAAU;IACnD,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;AAC/D,CAAC","sourcesContent":["import {\n  ConnectArguments,\n  NamespaceId,\n  SnapId,\n  Namespace,\n} from '@metamask/snaps-utils';\n\n/**\n * Finds a keyring snap for each namespace implements at a minimum the requested functionality.\n *\n * @param requestedNamespaces - The requested namespaces.\n * @param snaps - All snaps and their exposed keyring namespaces.\n * @returns A mapping between namespaces and snap ids.\n */\nexport function findMatchingKeyringSnaps(\n  requestedNamespaces: ConnectArguments['requiredNamespaces'],\n  snaps: Record<SnapId, Record<NamespaceId, Namespace> | null>,\n): Record<NamespaceId, SnapId[]> {\n  const snapEntries = Object.entries(snaps);\n  return Object.entries(requestedNamespaces).reduce<\n    Record<NamespaceId, SnapId[]>\n  >((acc, [requestedNamespaceId, currentRequestedNamespace]) => {\n    const matchedSnaps = snapEntries\n      .filter(([, namespaces]) => {\n        const potentialMatch = namespaces?.[requestedNamespaceId];\n        return (\n          potentialMatch !== undefined &&\n          matchNamespace(currentRequestedNamespace, potentialMatch)\n        );\n      })\n      .map(([snapId]) => snapId);\n    acc[requestedNamespaceId] = matchedSnaps;\n    return acc;\n  }, {});\n}\n\n/**\n * Determines whether a keyring namespace is a match given a potential match and the requested namespace.\n *\n * This function assumes that the namespace ID has already been matched.\n *\n * @param requestedNamespace - The requested namespace information.\n * @param potentialMatchNamespace - The potential match.\n * @returns True if the potentially matching namespace is a match.\n */\nfunction matchNamespace(\n  requestedNamespace: ConnectArguments['requiredNamespaces'][NamespaceId],\n  potentialMatchNamespace: Namespace,\n) {\n  if (\n    !requestedNamespace.chains.every((requestedChain) =>\n      potentialMatchNamespace.chains.some(\n        (potentialMatchChain) => potentialMatchChain.id === requestedChain,\n      ),\n    )\n  ) {\n    return false;\n  }\n\n  if (\n    !isSubset(\n      requestedNamespace.events ?? [],\n      potentialMatchNamespace.events ?? [],\n    )\n  ) {\n    return false;\n  }\n\n  if (\n    !isSubset(\n      requestedNamespace.methods ?? [],\n      potentialMatchNamespace.methods ?? [],\n    )\n  ) {\n    return false;\n  }\n\n  return true;\n}\n\n/**\n * Determines whether an array is a subset of another array.\n *\n * @param potentialSubset - The potential subset.\n * @param array - The other array.\n * @returns True if the first argument is a subset of second argument.\n */\nfunction isSubset<T>(potentialSubset: T[], array: T[]) {\n  return potentialSubset.every((item) => array.includes(item));\n}\n"]}

package/dist/multichain/middleware.d.ts

@@ -1,14 +0,0 @@
-import { ConnectArguments, Session, MultiChainRequest } from '@metamask/snaps-utils';
-import { JsonRpcMiddleware, JsonRpcRequest } from 'json-rpc-engine';
-/**
- * Creates a middleware that handles requests to the multichain controller.
- *
- * @param hooks - The hooks required by the middleware.
- * @param hooks.onConnect - The onConnect hook.
- * @param hooks.onRequest - The onRequest hook.
- * @returns The middleware.
- */
-export declare function createMultiChainMiddleware({ onConnect, onRequest, }: {
-    onConnect: (origin: string, requestedNamespaces: ConnectArguments) => Promise<Session>;
-    onRequest: (origin: string, data: MultiChainRequest) => Promise<unknown>;
-}): JsonRpcMiddleware<Omit<JsonRpcRequest<unknown>, 'id' | 'jsonrpc'>, any>;

package/dist/multichain/middleware.js

@@ -1,42 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createMultiChainMiddleware = void 0;
-const snaps_utils_1 = require("@metamask/snaps-utils");
-const utils_1 = require("@metamask/utils");
-const json_rpc_engine_1 = require("json-rpc-engine");
-/**
- * Creates a middleware that handles requests to the multichain controller.
- *
- * @param hooks - The hooks required by the middleware.
- * @param hooks.onConnect - The onConnect hook.
- * @param hooks.onRequest - The onRequest hook.
- * @returns The middleware.
- */
-function createMultiChainMiddleware({ onConnect, onRequest, }) {
-    return (0, json_rpc_engine_1.createAsyncMiddleware)(async function middleware(req, res, next) {
-        // This is added by other middleware
-        const { origin, params: unwrapped } = req;
-        if (req.method !== 'wallet_multiChainRequestHack') {
-            await next();
-            return;
-        }
-        (0, utils_1.assert)(unwrapped !== undefined, `Invalid params for ${req.method}`);
-        switch (unwrapped.method) {
-            case 'caip_request': {
-                (0, snaps_utils_1.assertIsMultiChainRequest)(unwrapped.params);
-                res.result = await onRequest(origin, unwrapped.params);
-                return;
-            }
-            case 'metamask_handshake': {
-                (0, snaps_utils_1.assertIsConnectArguments)(unwrapped.params);
-                res.result = await onConnect(origin, unwrapped.params);
-                return;
-            }
-            default: {
-                await next();
-            }
-        }
-    });
-}
-exports.createMultiChainMiddleware = createMultiChainMiddleware;
-//# sourceMappingURL=middleware.js.map

package/dist/multichain/middleware.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../../src/multichain/middleware.ts"],"names":[],"mappings":";;;AAAA,uDAM+B;AAC/B,2CAAyC;AACzC,qDAIyB;AAEzB;;;;;;;GAOG;AACH,SAAgB,0BAA0B,CAAC,EACzC,SAAS,EACT,SAAS,GAOV;IACC,OAAO,IAAA,uCAAqB,EAAC,KAAK,UAAU,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI;QACnE,oCAAoC;QACpC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,GAEhB,CAAC;QACvB,IAAI,GAAG,CAAC,MAAM,KAAK,8BAA8B,EAAE;YACjD,MAAM,IAAI,EAAE,CAAC;YACb,OAAO;SACR;QAED,IAAA,cAAM,EAAC,SAAS,KAAK,SAAS,EAAE,sBAAsB,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;QAEpE,QAAQ,SAAS,CAAC,MAAM,EAAE;YACxB,KAAK,cAAc,CAAC,CAAC;gBACnB,IAAA,uCAAyB,EAAC,SAAS,CAAC,MAAM,CAAC,CAAC;gBAC5C,GAAG,CAAC,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;gBACvD,OAAO;aACR;YAED,KAAK,oBAAoB,CAAC,CAAC;gBACzB,IAAA,sCAAwB,EAAC,SAAS,CAAC,MAAM,CAAC,CAAC;gBAC3C,GAAG,CAAC,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;gBACvD,OAAO;aACR;YAED,OAAO,CAAC,CAAC;gBACP,MAAM,IAAI,EAAE,CAAC;aACd;SACF;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAxCD,gEAwCC","sourcesContent":["import {\n  assertIsConnectArguments,\n  assertIsMultiChainRequest,\n  ConnectArguments,\n  Session,\n  MultiChainRequest,\n} from '@metamask/snaps-utils';\nimport { assert } from '@metamask/utils';\nimport {\n  createAsyncMiddleware,\n  JsonRpcMiddleware,\n  JsonRpcRequest,\n} from 'json-rpc-engine';\n\n/**\n * Creates a middleware that handles requests to the multichain controller.\n *\n * @param hooks - The hooks required by the middleware.\n * @param hooks.onConnect - The onConnect hook.\n * @param hooks.onRequest - The onRequest hook.\n * @returns The middleware.\n */\nexport function createMultiChainMiddleware({\n  onConnect,\n  onRequest,\n}: {\n  onConnect: (\n    origin: string,\n    requestedNamespaces: ConnectArguments,\n  ) => Promise<Session>;\n  onRequest: (origin: string, data: MultiChainRequest) => Promise<unknown>;\n}): JsonRpcMiddleware<Omit<JsonRpcRequest<unknown>, 'id' | 'jsonrpc'>, any> {\n  return createAsyncMiddleware(async function middleware(req, res, next) {\n    // This is added by other middleware\n    const { origin, params: unwrapped } = req as JsonRpcRequest<\n      JsonRpcRequest<unknown>\n    > & { origin: string };\n    if (req.method !== 'wallet_multiChainRequestHack') {\n      await next();\n      return;\n    }\n\n    assert(unwrapped !== undefined, `Invalid params for ${req.method}`);\n\n    switch (unwrapped.method) {\n      case 'caip_request': {\n        assertIsMultiChainRequest(unwrapped.params);\n        res.result = await onRequest(origin, unwrapped.params);\n        return;\n      }\n\n      case 'metamask_handshake': {\n        assertIsConnectArguments(unwrapped.params);\n        res.result = await onConnect(origin, unwrapped.params);\n        return;\n      }\n\n      default: {\n        await next();\n      }\n    }\n  });\n}\n"]}

package/dist/services/AbstractExecutionService.js

@@ -1,318 +0,0 @@
-"use strict";
-var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
-    if (kind === "m") throw new TypeError("Private method is not writable");
-    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
-    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
-    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
-};
-var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
-    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
-    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
-    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-var _AbstractExecutionService_instances, _AbstractExecutionService_snapRpcHooks, _AbstractExecutionService_snapToJobMap, _AbstractExecutionService_jobToSnapMap, _AbstractExecutionService_messenger, _AbstractExecutionService_terminationTimeout, _AbstractExecutionService_removeSnapHooks, _AbstractExecutionService_createSnapHooks, _AbstractExecutionService_getJobForSnap, _AbstractExecutionService_getSnapForJob, _AbstractExecutionService_mapSnapAndJob, _AbstractExecutionService_removeSnapAndJobMapping;
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.setupMultiplex = exports.AbstractExecutionService = void 0;
-const object_multiplex_1 = __importDefault(require("@metamask/object-multiplex"));
-const snaps_utils_1 = require("@metamask/snaps-utils");
-const utils_1 = require("@metamask/utils");
-const json_rpc_engine_1 = require("json-rpc-engine");
-const json_rpc_middleware_stream_1 = require("json-rpc-middleware-stream");
-const nanoid_1 = require("nanoid");
-const pump_1 = __importDefault(require("pump"));
-const logging_1 = require("../logging");
-const utils_2 = require("../utils");
-const controllerName = 'ExecutionService';
-class AbstractExecutionService {
-    constructor({ setupSnapProvider, messenger, terminationTimeout = utils_1.Duration.Second, }) {
-        _AbstractExecutionService_instances.add(this);
-        _AbstractExecutionService_snapRpcHooks.set(this, void 0);
-        _AbstractExecutionService_snapToJobMap.set(this, void 0);
-        _AbstractExecutionService_jobToSnapMap.set(this, void 0);
-        _AbstractExecutionService_messenger.set(this, void 0);
-        _AbstractExecutionService_terminationTimeout.set(this, void 0);
-        __classPrivateFieldSet(this, _AbstractExecutionService_snapRpcHooks, new Map(), "f");
-        this.jobs = new Map();
-        this.setupSnapProvider = setupSnapProvider;
-        __classPrivateFieldSet(this, _AbstractExecutionService_snapToJobMap, new Map(), "f");
-        __classPrivateFieldSet(this, _AbstractExecutionService_jobToSnapMap, new Map(), "f");
-        __classPrivateFieldSet(this, _AbstractExecutionService_messenger, messenger, "f");
-        __classPrivateFieldSet(this, _AbstractExecutionService_terminationTimeout, terminationTimeout, "f");
-        this.registerMessageHandlers();
-    }
-    /**
-     * Constructor helper for registering the controller's messaging system
-     * actions.
-     */
-    registerMessageHandlers() {
-        __classPrivateFieldGet(this, _AbstractExecutionService_messenger, "f").registerActionHandler(`${controllerName}:handleRpcRequest`, async (snapId, options) => this.handleRpcRequest(snapId, options));
-        __classPrivateFieldGet(this, _AbstractExecutionService_messenger, "f").registerActionHandler(`${controllerName}:executeSnap`, async (snapData) => this.executeSnap(snapData));
-        __classPrivateFieldGet(this, _AbstractExecutionService_messenger, "f").registerActionHandler(`${controllerName}:terminateSnap`, async (snapId) => this.terminateSnap(snapId));
-        __classPrivateFieldGet(this, _AbstractExecutionService_messenger, "f").registerActionHandler(`${controllerName}:terminateAllSnaps`, async () => this.terminateAllSnaps());
-    }
-    /**
-     * Terminates the job with the specified ID and deletes all its associated
-     * data. Any subsequent messages targeting the job will fail with an error.
-     * Throws an error if the specified job does not exist, or if termination
-     * fails unexpectedly.
-     *
-     * @param jobId - The id of the job to be terminated.
-     */
-    async terminate(jobId) {
-        const jobWrapper = this.jobs.get(jobId);
-        if (!jobWrapper) {
-            throw new Error(`Job with id "${jobId}" not found.`);
-        }
-        // Ping worker and tell it to run teardown, continue with termination if it takes too long
-        const result = await (0, utils_2.withTimeout)(this.command(jobId, {
-            jsonrpc: '2.0',
-            method: 'terminate',
-            params: [],
-            id: (0, nanoid_1.nanoid)(),
-        }), __classPrivateFieldGet(this, _AbstractExecutionService_terminationTimeout, "f"));
-        if (result === utils_2.hasTimedOut || result !== 'OK') {
-            // We tried to shutdown gracefully but failed. This probably means the Snap is in infinite loop and
-            // hogging down the whole JS process.
-            // TODO(ritave): It might be doing weird things such as posting a lot of setTimeouts. Add a test to ensure that this behaviour
-            //               doesn't leak into other workers. Especially important in IframeExecutionEnvironment since they all share the same
-            //               JS process.
-            (0, snaps_utils_1.logError)(`Job "${jobId}" failed to terminate gracefully.`, result);
-        }
-        Object.values(jobWrapper.streams).forEach((stream) => {
-            try {
-                !stream.destroyed && stream.destroy();
-                stream.removeAllListeners();
-            }
-            catch (error) {
-                (0, snaps_utils_1.logError)('Error while destroying stream', error);
-            }
-        });
-        this.terminateJob(jobWrapper);
-        __classPrivateFieldGet(this, _AbstractExecutionService_instances, "m", _AbstractExecutionService_removeSnapAndJobMapping).call(this, jobId);
-        this.jobs.delete(jobId);
-        (0, logging_1.log)(`Job "${jobId}" terminated.`);
-    }
-    /**
-     * Initiates a job for a snap.
-     *
-     * Depending on the execution environment, this may run forever if the Snap fails to start up properly, therefore any call to this function should be wrapped in a timeout.
-     *
-     * @returns Information regarding the created job.
-     */
-    async initJob() {
-        const jobId = (0, nanoid_1.nanoid)();
-        const { streams, worker } = await this.initStreams(jobId);
-        const rpcEngine = new json_rpc_engine_1.JsonRpcEngine();
-        const jsonRpcConnection = (0, json_rpc_middleware_stream_1.createStreamMiddleware)();
-        (0, pump_1.default)(jsonRpcConnection.stream, streams.command, jsonRpcConnection.stream);
-        rpcEngine.push(jsonRpcConnection.middleware);
-        const envMetadata = {
-            id: jobId,
-            streams,
-            rpcEngine,
-            worker,
-        };
-        this.jobs.set(jobId, envMetadata);
-        return envMetadata;
-    }
-    /**
-     * Sets up the streams for an initiated job.
-     *
-     * Depending on the execution environment, this may run forever if the Snap fails to start up properly, therefore any call to this function should be wrapped in a timeout.
-     *
-     * @param jobId - The id of the job.
-     * @returns The streams to communicate with the worker and the worker itself.
-     */
-    async initStreams(jobId) {
-        const { worker, stream: envStream } = await this.initEnvStream(jobId);
-        // Typecast justification: stream type mismatch
-        const mux = setupMultiplex(envStream, `Job: "${jobId}"`);
-        const commandStream = mux.createStream(snaps_utils_1.SNAP_STREAM_NAMES.COMMAND);
-        // Handle out-of-band errors, i.e. errors thrown from the snap outside of the req/res cycle.
-        // Also keep track of outbound request/responses
-        const notificationHandler = (message) => {
-            if (!(0, utils_1.isJsonRpcNotification)(message)) {
-                return;
-            }
-            // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
-            const snapId = __classPrivateFieldGet(this, _AbstractExecutionService_jobToSnapMap, "f").get(jobId);
-            if (message.method === 'OutboundRequest') {
-                __classPrivateFieldGet(this, _AbstractExecutionService_messenger, "f").publish('ExecutionService:outboundRequest', snapId);
-            }
-            else if (message.method === 'OutboundResponse') {
-                __classPrivateFieldGet(this, _AbstractExecutionService_messenger, "f").publish('ExecutionService:outboundResponse', snapId);
-            }
-            else if (message.method === 'UnhandledError') {
-                if ((0, utils_1.isObject)(message.params) && message.params.error) {
-                    __classPrivateFieldGet(this, _AbstractExecutionService_messenger, "f").publish('ExecutionService:unhandledError', snapId, message.params.error);
-                    commandStream.removeListener('data', notificationHandler);
-                }
-                else {
-                    (0, snaps_utils_1.logError)(new Error(`Received malformed "${message.method}" command stream notification.`));
-                }
-            }
-            else {
-                (0, snaps_utils_1.logError)(new Error(`Received unexpected command stream notification "${message.method}".`));
-            }
-        };
-        commandStream.on('data', notificationHandler);
-        const rpcStream = mux.createStream(snaps_utils_1.SNAP_STREAM_NAMES.JSON_RPC);
-        // Typecast: stream type mismatch
-        return {
-            streams: {
-                command: commandStream,
-                rpc: rpcStream,
-                // eslint-disable-next-line @typescript-eslint/naming-convention
-                _connection: envStream,
-            },
-            worker,
-        };
-    }
-    /**
-     * Terminates the Snap with the specified ID. May throw an error if
-     * termination unexpectedly fails, but will not fail if no job for the snap
-     * with the specified ID is found.
-     *
-     * @param snapId - The ID of the snap to terminate.
-     */
-    async terminateSnap(snapId) {
-        const jobId = __classPrivateFieldGet(this, _AbstractExecutionService_snapToJobMap, "f").get(snapId);
-        if (jobId) {
-            await this.terminate(jobId);
-        }
-    }
-    async terminateAllSnaps() {
-        await Promise.all([...this.jobs.keys()].map(async (jobId) => this.terminate(jobId)));
-        __classPrivateFieldGet(this, _AbstractExecutionService_snapRpcHooks, "f").clear();
-    }
-    /**
-     * Gets the RPC request handler for the given snap.
-     *
-     * @param snapId - The id of the Snap whose message handler to get.
-     * @returns The RPC request handler for the snap.
-     */
-    getRpcRequestHandler(snapId) {
-        return __classPrivateFieldGet(this, _AbstractExecutionService_snapRpcHooks, "f").get(snapId);
-    }
-    /**
-     * Initializes and executes a snap, setting up the communication channels to the snap etc.
-     *
-     * Depending on the execution environment, this may run forever if the Snap fails to start up properly, therefore any call to this function should be wrapped in a timeout.
-     *
-     * @param snapData - Data needed for Snap execution.
-     * @returns A string `OK` if execution succeeded.
-     * @throws If the execution service returns an error.
-     */
-    async executeSnap(snapData) {
-        if (__classPrivateFieldGet(this, _AbstractExecutionService_snapToJobMap, "f").has(snapData.snapId)) {
-            throw new Error(`Snap "${snapData.snapId}" is already being executed.`);
-        }
-        const job = await this.initJob();
-        __classPrivateFieldGet(this, _AbstractExecutionService_instances, "m", _AbstractExecutionService_mapSnapAndJob).call(this, snapData.snapId, job.id);
-        // Ping the worker to ensure that it started up
-        await this.command(job.id, {
-            jsonrpc: '2.0',
-            method: 'ping',
-            id: (0, nanoid_1.nanoid)(),
-        });
-        const rpcStream = job.streams.rpc;
-        this.setupSnapProvider(snapData.snapId, rpcStream);
-        const result = await this.command(job.id, {
-            jsonrpc: '2.0',
-            method: 'executeSnap',
-            params: snapData,
-            id: (0, nanoid_1.nanoid)(),
-        });
-        __classPrivateFieldGet(this, _AbstractExecutionService_instances, "m", _AbstractExecutionService_createSnapHooks).call(this, snapData.snapId, job.id);
-        return result;
-    }
-    // Cannot be hash private yet because of tests.
-    async command(jobId, message) {
-        if (typeof message !== 'object') {
-            throw new Error('Must send object.');
-        }
-        const job = this.jobs.get(jobId);
-        if (!job) {
-            throw new Error(`Job with id "${jobId}" not found.`);
-        }
-        (0, logging_1.log)('Parent: Sending Command', message);
-        const response = await job.rpcEngine.handle(message);
-        if (response.error) {
-            throw new Error(response.error.message);
-        }
-        return response.result;
-    }
-    /**
-     * Handle RPC request.
-     *
-     * @param snapId - The ID of the recipient snap.
-     * @param options - Bag of options to pass to the RPC handler.
-     * @returns Promise that can handle the request.
-     */
-    async handleRpcRequest(snapId, options) {
-        const rpcRequestHandler = await this.getRpcRequestHandler(snapId);
-        if (!rpcRequestHandler) {
-            throw new Error(`Snap execution service returned no RPC handler for running snap "${snapId}".`);
-        }
-        return rpcRequestHandler(options);
-    }
-}
-exports.AbstractExecutionService = AbstractExecutionService;
-_AbstractExecutionService_snapRpcHooks = new WeakMap(), _AbstractExecutionService_snapToJobMap = new WeakMap(), _AbstractExecutionService_jobToSnapMap = new WeakMap(), _AbstractExecutionService_messenger = new WeakMap(), _AbstractExecutionService_terminationTimeout = new WeakMap(), _AbstractExecutionService_instances = new WeakSet(), _AbstractExecutionService_removeSnapHooks = function _AbstractExecutionService_removeSnapHooks(snapId) {
-    __classPrivateFieldGet(this, _AbstractExecutionService_snapRpcHooks, "f").delete(snapId);
-}, _AbstractExecutionService_createSnapHooks = function _AbstractExecutionService_createSnapHooks(snapId, workerId) {
-    const rpcHook = async ({ origin, handler, request }) => {
-        return await this.command(workerId, {
-            id: (0, nanoid_1.nanoid)(),
-            jsonrpc: '2.0',
-            method: 'snapRpc',
-            params: {
-                origin,
-                handler,
-                request,
-                target: snapId,
-            },
-        });
-    };
-    __classPrivateFieldGet(this, _AbstractExecutionService_snapRpcHooks, "f").set(snapId, rpcHook);
-}, _AbstractExecutionService_getJobForSnap = function _AbstractExecutionService_getJobForSnap(snapId) {
-    return __classPrivateFieldGet(this, _AbstractExecutionService_snapToJobMap, "f").get(snapId);
-}, _AbstractExecutionService_getSnapForJob = function _AbstractExecutionService_getSnapForJob(jobId) {
-    return __classPrivateFieldGet(this, _AbstractExecutionService_jobToSnapMap, "f").get(jobId);
-}, _AbstractExecutionService_mapSnapAndJob = function _AbstractExecutionService_mapSnapAndJob(snapId, jobId) {
-    __classPrivateFieldGet(this, _AbstractExecutionService_snapToJobMap, "f").set(snapId, jobId);
-    __classPrivateFieldGet(this, _AbstractExecutionService_jobToSnapMap, "f").set(jobId, snapId);
-}, _AbstractExecutionService_removeSnapAndJobMapping = function _AbstractExecutionService_removeSnapAndJobMapping(jobId) {
-    const snapId = __classPrivateFieldGet(this, _AbstractExecutionService_jobToSnapMap, "f").get(jobId);
-    if (!snapId) {
-        throw new Error(`job: "${jobId}" has no mapped snap.`);
-    }
-    __classPrivateFieldGet(this, _AbstractExecutionService_jobToSnapMap, "f").delete(jobId);
-    __classPrivateFieldGet(this, _AbstractExecutionService_snapToJobMap, "f").delete(snapId);
-    __classPrivateFieldGet(this, _AbstractExecutionService_instances, "m", _AbstractExecutionService_removeSnapHooks).call(this, snapId);
-};
-/**
- * Sets up stream multiplexing for the given stream.
- *
- * @param connectionStream - The stream to mux.
- * @param streamName - The name of the stream, for identification in errors.
- * @returns The multiplexed stream.
- */
-function setupMultiplex(connectionStream, streamName) {
-    const mux = new object_multiplex_1.default();
-    (0, pump_1.default)(connectionStream, 
-    // Typecast: stream type mismatch
-    mux, connectionStream, (error) => {
-        if (error) {
-            streamName
-                ? (0, snaps_utils_1.logError)(`"${streamName}" stream failure.`, error)
-                : (0, snaps_utils_1.logError)(error);
-        }
-    });
-    return mux;
-}
-exports.setupMultiplex = setupMultiplex;
-//# sourceMappingURL=AbstractExecutionService.js.map

package/dist/services/AbstractExecutionService.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"AbstractExecutionService.js","sourceRoot":"","sources":["../../src/services/AbstractExecutionService.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;AAAA,kFAAyD;AAEzD,uDAK+B;AAC/B,2CAMyB;AACzB,qDAKyB;AACzB,2EAAoE;AACpE,mCAAgC;AAChC,gDAAwB;AAGxB,wCAAiC;AACjC,oCAAoD;AAQpD,MAAM,cAAc,GAAG,kBAAkB,CAAC;AAuB1C,MAAsB,wBAAwB;IAmB5C,YAAY,EACV,iBAAiB,EACjB,SAAS,EACT,kBAAkB,GAAG,gBAAQ,CAAC,MAAM,GACf;;QApBvB,yDAAwC;QAQxC,yDAAmC;QAEnC,yDAAmC;QAEnC,sDAAsC;QAEtC,+DAA4B;QAO1B,uBAAA,IAAI,0CAAiB,IAAI,GAAG,EAAE,MAAA,CAAC;QAC/B,IAAI,CAAC,IAAI,GAAG,IAAI,GAAG,EAAE,CAAC;QACtB,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,uBAAA,IAAI,0CAAiB,IAAI,GAAG,EAAE,MAAA,CAAC;QAC/B,uBAAA,IAAI,0CAAiB,IAAI,GAAG,EAAE,MAAA,CAAC;QAC/B,uBAAA,IAAI,uCAAc,SAAS,MAAA,CAAC;QAC5B,uBAAA,IAAI,gDAAuB,kBAAkB,MAAA,CAAC;QAE9C,IAAI,CAAC,uBAAuB,EAAE,CAAC;IACjC,CAAC;IAED;;;OAGG;IACK,uBAAuB;QAC7B,uBAAA,IAAI,2CAAW,CAAC,qBAAqB,CACnC,GAAG,cAAc,mBAAmB,EACpC,KAAK,EAAE,MAAc,EAAE,OAAwB,EAAE,EAAE,CACjD,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,OAAO,CAAC,CACzC,CAAC;QAEF,uBAAA,IAAI,2CAAW,CAAC,qBAAqB,CACnC,GAAG,cAAc,cAAc,EAC/B,KAAK,EAAE,QAA2B,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAClE,CAAC;QAEF,uBAAA,IAAI,2CAAW,CAAC,qBAAqB,CACnC,GAAG,cAAc,gBAAgB,EACjC,KAAK,EAAE,MAAc,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CACrD,CAAC;QAEF,uBAAA,IAAI,2CAAW,CAAC,qBAAqB,CACnC,GAAG,cAAc,oBAAoB,EACrC,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,iBAAiB,EAAE,CACrC,CAAC;IACJ,CAAC;IAWD;;;;;;;OAOG;IACI,KAAK,CAAC,SAAS,CAAC,KAAa;QAClC,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACxC,IAAI,CAAC,UAAU,EAAE;YACf,MAAM,IAAI,KAAK,CAAC,gBAAgB,KAAK,cAAc,CAAC,CAAC;SACtD;QAED,0FAA0F;QAC1F,MAAM,MAAM,GAAG,MAAM,IAAA,mBAAW,EAC9B,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE;YAClB,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,WAAW;YACnB,MAAM,EAAE,EAAE;YACV,EAAE,EAAE,IAAA,eAAM,GAAE;SACb,CAAC,EACF,uBAAA,IAAI,oDAAoB,CACzB,CAAC;QAEF,IAAI,MAAM,KAAK,mBAAW,IAAI,MAAM,KAAK,IAAI,EAAE;YAC7C,mGAAmG;YACnG,qCAAqC;YACrC,8HAA8H;YAC9H,kIAAkI;YAClI,4BAA4B;YAC5B,IAAA,sBAAQ,EAAC,QAAQ,KAAK,mCAAmC,EAAE,MAAM,CAAC,CAAC;SACpE;QAED,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YACnD,IAAI;gBACF,CAAC,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACtC,MAAM,CAAC,kBAAkB,EAAE,CAAC;aAC7B;YAAC,OAAO,KAAK,EAAE;gBACd,IAAA,sBAAQ,EAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;aAClD;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QAE9B,uBAAA,IAAI,8FAAyB,MAA7B,IAAI,EAA0B,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACxB,IAAA,aAAG,EAAC,QAAQ,KAAK,eAAe,CAAC,CAAC;IACpC,CAAC;IAED;;;;;;OAMG;IACO,KAAK,CAAC,OAAO;QACrB,MAAM,KAAK,GAAG,IAAA,eAAM,GAAE,CAAC;QACvB,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC1D,MAAM,SAAS,GAAG,IAAI,+BAAa,EAAE,CAAC;QAEtC,MAAM,iBAAiB,GAAG,IAAA,mDAAsB,GAAE,CAAC;QAEnD,IAAA,cAAI,EAAC,iBAAiB,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,EAAE,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAE1E,SAAS,CAAC,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QAE7C,MAAM,WAAW,GAAG;YAClB,EAAE,EAAE,KAAK;YACT,OAAO;YACP,SAAS;YACT,MAAM;SACP,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QAElC,OAAO,WAAW,CAAC;IACrB,CAAC;IAED;;;;;;;OAOG;IACO,KAAK,CAAC,WAAW,CACzB,KAAa;QAEb,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QACtE,+CAA+C;QAC/C,MAAM,GAAG,GAAG,cAAc,CACxB,SAA8B,EAC9B,SAAS,KAAK,GAAG,CAClB,CAAC;QAEF,MAAM,aAAa,GAAG,GAAG,CAAC,YAAY,CAAC,+BAAiB,CAAC,OAAO,CAAC,CAAC;QAElE,4FAA4F;QAC5F,gDAAgD;QAChD,MAAM,mBAAmB,GAAG,CAC1B,OAEsD,EACtD,EAAE;YACF,IAAI,CAAC,IAAA,6BAAqB,EAAC,OAAO,CAAC,EAAE;gBACnC,OAAO;aACR;YAED,oEAAoE;YACpE,MAAM,MAAM,GAAG,uBAAA,IAAI,8CAAc,CAAC,GAAG,CAAC,KAAK,CAAE,CAAC;YAC9C,IAAI,OAAO,CAAC,MAAM,KAAK,iBAAiB,EAAE;gBACxC,uBAAA,IAAI,2CAAW,CAAC,OAAO,CAAC,kCAAkC,EAAE,MAAM,CAAC,CAAC;aACrE;iBAAM,IAAI,OAAO,CAAC,MAAM,KAAK,kBAAkB,EAAE;gBAChD,uBAAA,IAAI,2CAAW,CAAC,OAAO,CAAC,mCAAmC,EAAE,MAAM,CAAC,CAAC;aACtE;iBAAM,IAAI,OAAO,CAAC,MAAM,KAAK,gBAAgB,EAAE;gBAC9C,IAAI,IAAA,gBAAQ,EAAC,OAAO,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE;oBACpD,uBAAA,IAAI,2CAAW,CAAC,OAAO,CACrB,iCAAiC,EACjC,MAAM,EACN,OAAO,CAAC,MAAM,CAAC,KAAsB,CACtC,CAAC;oBACF,aAAa,CAAC,cAAc,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC;iBAC3D;qBAAM;oBACL,IAAA,sBAAQ,EACN,IAAI,KAAK,CACP,uBAAuB,OAAO,CAAC,MAAM,gCAAgC,CACtE,CACF,CAAC;iBACH;aACF;iBAAM;gBACL,IAAA,sBAAQ,EACN,IAAI,KAAK,CACP,oDAAoD,OAAO,CAAC,MAAM,IAAI,CACvE,CACF,CAAC;aACH;QACH,CAAC,CAAC;QAEF,aAAa,CAAC,EAAE,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC;QAC9C,MAAM,SAAS,GAAG,GAAG,CAAC,YAAY,CAAC,+BAAiB,CAAC,QAAQ,CAAC,CAAC;QAE/D,iCAAiC;QACjC,OAAO;YACL,OAAO,EAAE;gBACP,OAAO,EAAE,aAAkC;gBAC3C,GAAG,EAAE,SAAS;gBACd,gEAAgE;gBAChE,WAAW,EAAE,SAAS;aACvB;YACD,MAAM;SACP,CAAC;IACJ,CAAC;IAYD;;;;;;OAMG;IACH,KAAK,CAAC,aAAa,CAAC,MAAc;QAChC,MAAM,KAAK,GAAG,uBAAA,IAAI,8CAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,KAAK,EAAE;YACT,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;SAC7B;IACH,CAAC;IAED,KAAK,CAAC,iBAAiB;QACrB,MAAM,OAAO,CAAC,GAAG,CACf,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAClE,CAAC;QACF,uBAAA,IAAI,8CAAc,CAAC,KAAK,EAAE,CAAC;IAC7B,CAAC;IAED;;;;;OAKG;IACK,oBAAoB,CAAC,MAAc;QACzC,OAAO,uBAAA,IAAI,8CAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACxC,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,WAAW,CAAC,QAA2B;QAC3C,IAAI,uBAAA,IAAI,8CAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE;YAC3C,MAAM,IAAI,KAAK,CAAC,SAAS,QAAQ,CAAC,MAAM,8BAA8B,CAAC,CAAC;SACzE;QAED,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACjC,uBAAA,IAAI,oFAAe,MAAnB,IAAI,EAAgB,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QAE7C,+CAA+C;QAC/C,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE;YACzB,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,MAAM;YACd,EAAE,EAAE,IAAA,eAAM,GAAE;SACb,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,GAAwB,CAAC;QAEvD,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAEnD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE;YACxC,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,aAAa;YACrB,MAAM,EAAE,QAAQ;YAChB,EAAE,EAAE,IAAA,eAAM,GAAE;SACb,CAAC,CAAC;QACH,uBAAA,IAAI,sFAAiB,MAArB,IAAI,EAAkB,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QAC/C,OAAO,MAAgB,CAAC;IAC1B,CAAC;IAED,+CAA+C;IACvC,KAAK,CAAC,OAAO,CACnB,KAAa,EACb,OAAgC;QAEhC,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE;YAC/B,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;SACtC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,IAAI,CAAC,GAAG,EAAE;YACR,MAAM,IAAI,KAAK,CAAC,gBAAgB,KAAK,cAAc,CAAC,CAAC;SACtD;QAED,IAAA,aAAG,EAAC,yBAAyB,EAAE,OAAO,CAAC,CAAC;QACxC,MAAM,QAAQ,GACZ,MAAM,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACtC,IAAI,QAAQ,CAAC,KAAK,EAAE;YAClB,MAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;SACzC;QACD,OAAO,QAAQ,CAAC,MAAM,CAAC;IACzB,CAAC;IA4DD;;;;;;OAMG;IACI,KAAK,CAAC,gBAAgB,CAC3B,MAAc,EACd,OAAwB;QAExB,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QAElE,IAAI,CAAC,iBAAiB,EAAE;YACtB,MAAM,IAAI,KAAK,CACb,oEAAoE,MAAM,IAAI,CAC/E,CAAC;SACH;QAED,OAAO,iBAAiB,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;CACF;AAvZD,4DAuZC;+aA/EkB,MAAc;IAC7B,uBAAA,IAAI,8CAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;AACpC,CAAC,iGAEgB,MAAc,EAAE,QAAgB;IAC/C,MAAM,OAAO,GAAG,KAAK,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAmB,EAAE,EAAE;QACtE,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE;YAClC,EAAE,EAAE,IAAA,eAAM,GAAE;YACZ,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,SAAS;YACjB,MAAM,EAAE;gBACN,MAAM;gBACN,OAAO;gBACP,OAAO;gBACP,MAAM,EAAE,MAAM;aACf;SACF,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,uBAAA,IAAI,8CAAc,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAC1C,CAAC,6FAQc,MAAc;IAC3B,OAAO,uBAAA,IAAI,8CAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AACxC,CAAC,6FAQc,KAAa;IAC1B,OAAO,uBAAA,IAAI,8CAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;AACvC,CAAC,6FAEc,MAAc,EAAE,KAAa;IAC1C,uBAAA,IAAI,8CAAc,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACtC,uBAAA,IAAI,8CAAc,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;AACxC,CAAC,iHAEwB,KAAa;IACpC,MAAM,MAAM,GAAG,uBAAA,IAAI,8CAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC7C,IAAI,CAAC,MAAM,EAAE;QACX,MAAM,IAAI,KAAK,CAAC,SAAS,KAAK,uBAAuB,CAAC,CAAC;KACxD;IAED,uBAAA,IAAI,8CAAc,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACjC,uBAAA,IAAI,8CAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAClC,uBAAA,IAAI,sFAAiB,MAArB,IAAI,EAAkB,MAAM,CAAC,CAAC;AAChC,CAAC;AAyBH;;;;;;GAMG;AACH,SAAgB,cAAc,CAC5B,gBAAwB,EACxB,UAAkB;IAElB,MAAM,GAAG,GAAG,IAAI,0BAAe,EAAE,CAAC;IAClC,IAAA,cAAI,EACF,gBAAgB;IAChB,iCAAiC;IACjC,GAAwB,EACxB,gBAAgB,EAChB,CAAC,KAAK,EAAE,EAAE;QACR,IAAI,KAAK,EAAE;YACT,UAAU;gBACR,CAAC,CAAC,IAAA,sBAAQ,EAAC,IAAI,UAAU,mBAAmB,EAAE,KAAK,CAAC;gBACpD,CAAC,CAAC,IAAA,sBAAQ,EAAC,KAAK,CAAC,CAAC;SACrB;IACH,CAAC,CACF,CAAC;IACF,OAAO,GAAG,CAAC;AACb,CAAC;AAnBD,wCAmBC","sourcesContent":["import ObjectMultiplex from '@metamask/object-multiplex';\nimport { BasePostMessageStream } from '@metamask/post-message-stream';\nimport {\n  SnapRpcHook,\n  SnapRpcHookArgs,\n  SNAP_STREAM_NAMES,\n  logError,\n} from '@metamask/snaps-utils';\nimport {\n  Duration,\n  isJsonRpcNotification,\n  isObject,\n  Json,\n  JsonRpcNotification,\n} from '@metamask/utils';\nimport {\n  JsonRpcEngine,\n  // TODO: Replace with @metamask/utils version after bumping json-rpc-engine\n  JsonRpcRequest,\n  PendingJsonRpcResponse,\n} from 'json-rpc-engine';\nimport { createStreamMiddleware } from 'json-rpc-middleware-stream';\nimport { nanoid } from 'nanoid';\nimport pump from 'pump';\nimport { Duplex } from 'stream';\n\nimport { log } from '../logging';\nimport { hasTimedOut, withTimeout } from '../utils';\nimport {\n  ExecutionService,\n  ExecutionServiceMessenger,\n  SnapErrorJson,\n  SnapExecutionData,\n} from './ExecutionService';\n\nconst controllerName = 'ExecutionService';\n\nexport type SetupSnapProvider = (snapId: string, stream: Duplex) => void;\n\nexport type ExecutionServiceArgs = {\n  setupSnapProvider: SetupSnapProvider;\n  messenger: ExecutionServiceMessenger;\n  terminationTimeout?: number;\n};\n\nexport type JobStreams = {\n  command: Duplex;\n  rpc: Duplex;\n  _connection: BasePostMessageStream;\n};\n\nexport type Job<WorkerType> = {\n  id: string;\n  streams: JobStreams;\n  rpcEngine: JsonRpcEngine;\n  worker: WorkerType;\n};\n\nexport abstract class AbstractExecutionService<WorkerType>\n  implements ExecutionService\n{\n  #snapRpcHooks: Map<string, SnapRpcHook>;\n\n  // Cannot be hash private yet because of tests.\n  protected jobs: Map<string, Job<WorkerType>>;\n\n  // Cannot be hash private yet because of tests.\n  private readonly setupSnapProvider: SetupSnapProvider;\n\n  #snapToJobMap: Map<string, string>;\n\n  #jobToSnapMap: Map<string, string>;\n\n  #messenger: ExecutionServiceMessenger;\n\n  #terminationTimeout: number;\n\n  constructor({\n    setupSnapProvider,\n    messenger,\n    terminationTimeout = Duration.Second,\n  }: ExecutionServiceArgs) {\n    this.#snapRpcHooks = new Map();\n    this.jobs = new Map();\n    this.setupSnapProvider = setupSnapProvider;\n    this.#snapToJobMap = new Map();\n    this.#jobToSnapMap = new Map();\n    this.#messenger = messenger;\n    this.#terminationTimeout = terminationTimeout;\n\n    this.registerMessageHandlers();\n  }\n\n  /**\n   * Constructor helper for registering the controller's messaging system\n   * actions.\n   */\n  private registerMessageHandlers(): void {\n    this.#messenger.registerActionHandler(\n      `${controllerName}:handleRpcRequest`,\n      async (snapId: string, options: SnapRpcHookArgs) =>\n        this.handleRpcRequest(snapId, options),\n    );\n\n    this.#messenger.registerActionHandler(\n      `${controllerName}:executeSnap`,\n      async (snapData: SnapExecutionData) => this.executeSnap(snapData),\n    );\n\n    this.#messenger.registerActionHandler(\n      `${controllerName}:terminateSnap`,\n      async (snapId: string) => this.terminateSnap(snapId),\n    );\n\n    this.#messenger.registerActionHandler(\n      `${controllerName}:terminateAllSnaps`,\n      async () => this.terminateAllSnaps(),\n    );\n  }\n\n  /**\n   * Performs additional necessary work during job termination. **MUST** be\n   * implemented by concrete implementations. See\n   * {@link AbstractExecutionService.terminate} for details.\n   *\n   * @param job - The object corresponding to the job to be terminated.\n   */\n  protected abstract terminateJob(job: Job<WorkerType>): void;\n\n  /**\n   * Terminates the job with the specified ID and deletes all its associated\n   * data. Any subsequent messages targeting the job will fail with an error.\n   * Throws an error if the specified job does not exist, or if termination\n   * fails unexpectedly.\n   *\n   * @param jobId - The id of the job to be terminated.\n   */\n  public async terminate(jobId: string): Promise<void> {\n    const jobWrapper = this.jobs.get(jobId);\n    if (!jobWrapper) {\n      throw new Error(`Job with id \"${jobId}\" not found.`);\n    }\n\n    // Ping worker and tell it to run teardown, continue with termination if it takes too long\n    const result = await withTimeout(\n      this.command(jobId, {\n        jsonrpc: '2.0',\n        method: 'terminate',\n        params: [],\n        id: nanoid(),\n      }),\n      this.#terminationTimeout,\n    );\n\n    if (result === hasTimedOut || result !== 'OK') {\n      // We tried to shutdown gracefully but failed. This probably means the Snap is in infinite loop and\n      // hogging down the whole JS process.\n      // TODO(ritave): It might be doing weird things such as posting a lot of setTimeouts. Add a test to ensure that this behaviour\n      //               doesn't leak into other workers. Especially important in IframeExecutionEnvironment since they all share the same\n      //               JS process.\n      logError(`Job \"${jobId}\" failed to terminate gracefully.`, result);\n    }\n\n    Object.values(jobWrapper.streams).forEach((stream) => {\n      try {\n        !stream.destroyed && stream.destroy();\n        stream.removeAllListeners();\n      } catch (error) {\n        logError('Error while destroying stream', error);\n      }\n    });\n\n    this.terminateJob(jobWrapper);\n\n    this.#removeSnapAndJobMapping(jobId);\n    this.jobs.delete(jobId);\n    log(`Job \"${jobId}\" terminated.`);\n  }\n\n  /**\n   * Initiates a job for a snap.\n   *\n   * Depending on the execution environment, this may run forever if the Snap fails to start up properly, therefore any call to this function should be wrapped in a timeout.\n   *\n   * @returns Information regarding the created job.\n   */\n  protected async initJob(): Promise<Job<WorkerType>> {\n    const jobId = nanoid();\n    const { streams, worker } = await this.initStreams(jobId);\n    const rpcEngine = new JsonRpcEngine();\n\n    const jsonRpcConnection = createStreamMiddleware();\n\n    pump(jsonRpcConnection.stream, streams.command, jsonRpcConnection.stream);\n\n    rpcEngine.push(jsonRpcConnection.middleware);\n\n    const envMetadata = {\n      id: jobId,\n      streams,\n      rpcEngine,\n      worker,\n    };\n    this.jobs.set(jobId, envMetadata);\n\n    return envMetadata;\n  }\n\n  /**\n   * Sets up the streams for an initiated job.\n   *\n   * Depending on the execution environment, this may run forever if the Snap fails to start up properly, therefore any call to this function should be wrapped in a timeout.\n   *\n   * @param jobId - The id of the job.\n   * @returns The streams to communicate with the worker and the worker itself.\n   */\n  protected async initStreams(\n    jobId: string,\n  ): Promise<{ streams: JobStreams; worker: WorkerType }> {\n    const { worker, stream: envStream } = await this.initEnvStream(jobId);\n    // Typecast justification: stream type mismatch\n    const mux = setupMultiplex(\n      envStream as unknown as Duplex,\n      `Job: \"${jobId}\"`,\n    );\n\n    const commandStream = mux.createStream(SNAP_STREAM_NAMES.COMMAND);\n\n    // Handle out-of-band errors, i.e. errors thrown from the snap outside of the req/res cycle.\n    // Also keep track of outbound request/responses\n    const notificationHandler = (\n      message:\n        | JsonRpcRequest<unknown>\n        | JsonRpcNotification<Json[] | Record<string, Json>>,\n    ) => {\n      if (!isJsonRpcNotification(message)) {\n        return;\n      }\n\n      // eslint-disable-next-line @typescript-eslint/no-non-null-assertion\n      const snapId = this.#jobToSnapMap.get(jobId)!;\n      if (message.method === 'OutboundRequest') {\n        this.#messenger.publish('ExecutionService:outboundRequest', snapId);\n      } else if (message.method === 'OutboundResponse') {\n        this.#messenger.publish('ExecutionService:outboundResponse', snapId);\n      } else if (message.method === 'UnhandledError') {\n        if (isObject(message.params) && message.params.error) {\n          this.#messenger.publish(\n            'ExecutionService:unhandledError',\n            snapId,\n            message.params.error as SnapErrorJson,\n          );\n          commandStream.removeListener('data', notificationHandler);\n        } else {\n          logError(\n            new Error(\n              `Received malformed \"${message.method}\" command stream notification.`,\n            ),\n          );\n        }\n      } else {\n        logError(\n          new Error(\n            `Received unexpected command stream notification \"${message.method}\".`,\n          ),\n        );\n      }\n    };\n\n    commandStream.on('data', notificationHandler);\n    const rpcStream = mux.createStream(SNAP_STREAM_NAMES.JSON_RPC);\n\n    // Typecast: stream type mismatch\n    return {\n      streams: {\n        command: commandStream as unknown as Duplex,\n        rpc: rpcStream,\n        // eslint-disable-next-line @typescript-eslint/naming-convention\n        _connection: envStream,\n      },\n      worker,\n    };\n  }\n\n  /**\n   * Abstract function implemented by implementing class that spins up a new worker for a job.\n   *\n   * Depending on the execution environment, this may run forever if the Snap fails to start up properly, therefore any call to this function should be wrapped in a timeout.\n   */\n  protected abstract initEnvStream(jobId: string): Promise<{\n    worker: WorkerType;\n    stream: BasePostMessageStream;\n  }>;\n\n  /**\n   * Terminates the Snap with the specified ID. May throw an error if\n   * termination unexpectedly fails, but will not fail if no job for the snap\n   * with the specified ID is found.\n   *\n   * @param snapId - The ID of the snap to terminate.\n   */\n  async terminateSnap(snapId: string) {\n    const jobId = this.#snapToJobMap.get(snapId);\n    if (jobId) {\n      await this.terminate(jobId);\n    }\n  }\n\n  async terminateAllSnaps() {\n    await Promise.all(\n      [...this.jobs.keys()].map(async (jobId) => this.terminate(jobId)),\n    );\n    this.#snapRpcHooks.clear();\n  }\n\n  /**\n   * Gets the RPC request handler for the given snap.\n   *\n   * @param snapId - The id of the Snap whose message handler to get.\n   * @returns The RPC request handler for the snap.\n   */\n  private getRpcRequestHandler(snapId: string) {\n    return this.#snapRpcHooks.get(snapId);\n  }\n\n  /**\n   * Initializes and executes a snap, setting up the communication channels to the snap etc.\n   *\n   * Depending on the execution environment, this may run forever if the Snap fails to start up properly, therefore any call to this function should be wrapped in a timeout.\n   *\n   * @param snapData - Data needed for Snap execution.\n   * @returns A string `OK` if execution succeeded.\n   * @throws If the execution service returns an error.\n   */\n  async executeSnap(snapData: SnapExecutionData): Promise<string> {\n    if (this.#snapToJobMap.has(snapData.snapId)) {\n      throw new Error(`Snap \"${snapData.snapId}\" is already being executed.`);\n    }\n\n    const job = await this.initJob();\n    this.#mapSnapAndJob(snapData.snapId, job.id);\n\n    // Ping the worker to ensure that it started up\n    await this.command(job.id, {\n      jsonrpc: '2.0',\n      method: 'ping',\n      id: nanoid(),\n    });\n\n    const rpcStream = job.streams.rpc as unknown as Duplex;\n\n    this.setupSnapProvider(snapData.snapId, rpcStream);\n\n    const result = await this.command(job.id, {\n      jsonrpc: '2.0',\n      method: 'executeSnap',\n      params: snapData,\n      id: nanoid(),\n    });\n    this.#createSnapHooks(snapData.snapId, job.id);\n    return result as string;\n  }\n\n  // Cannot be hash private yet because of tests.\n  private async command(\n    jobId: string,\n    message: JsonRpcRequest<unknown>,\n  ): Promise<unknown> {\n    if (typeof message !== 'object') {\n      throw new Error('Must send object.');\n    }\n\n    const job = this.jobs.get(jobId);\n    if (!job) {\n      throw new Error(`Job with id \"${jobId}\" not found.`);\n    }\n\n    log('Parent: Sending Command', message);\n    const response: PendingJsonRpcResponse<unknown> =\n      await job.rpcEngine.handle(message);\n    if (response.error) {\n      throw new Error(response.error.message);\n    }\n    return response.result;\n  }\n\n  #removeSnapHooks(snapId: string) {\n    this.#snapRpcHooks.delete(snapId);\n  }\n\n  #createSnapHooks(snapId: string, workerId: string) {\n    const rpcHook = async ({ origin, handler, request }: SnapRpcHookArgs) => {\n      return await this.command(workerId, {\n        id: nanoid(),\n        jsonrpc: '2.0',\n        method: 'snapRpc',\n        params: {\n          origin,\n          handler,\n          request,\n          target: snapId,\n        },\n      });\n    };\n\n    this.#snapRpcHooks.set(snapId, rpcHook);\n  }\n\n  /**\n   * Gets the job id for a given snap.\n   *\n   * @param snapId - A given snap id.\n   * @returns The ID of the snap's job.\n   */\n  #getJobForSnap(snapId: string): string | undefined {\n    return this.#snapToJobMap.get(snapId);\n  }\n\n  /**\n   * Gets the snap id for a given job.\n   *\n   * @param jobId - A given job id.\n   * @returns The ID of the snap that is running the job.\n   */\n  #getSnapForJob(jobId: string): string | undefined {\n    return this.#jobToSnapMap.get(jobId);\n  }\n\n  #mapSnapAndJob(snapId: string, jobId: string): void {\n    this.#snapToJobMap.set(snapId, jobId);\n    this.#jobToSnapMap.set(jobId, snapId);\n  }\n\n  #removeSnapAndJobMapping(jobId: string): void {\n    const snapId = this.#jobToSnapMap.get(jobId);\n    if (!snapId) {\n      throw new Error(`job: \"${jobId}\" has no mapped snap.`);\n    }\n\n    this.#jobToSnapMap.delete(jobId);\n    this.#snapToJobMap.delete(snapId);\n    this.#removeSnapHooks(snapId);\n  }\n\n  /**\n   * Handle RPC request.\n   *\n   * @param snapId - The ID of the recipient snap.\n   * @param options - Bag of options to pass to the RPC handler.\n   * @returns Promise that can handle the request.\n   */\n  public async handleRpcRequest(\n    snapId: string,\n    options: SnapRpcHookArgs,\n  ): Promise<unknown> {\n    const rpcRequestHandler = await this.getRpcRequestHandler(snapId);\n\n    if (!rpcRequestHandler) {\n      throw new Error(\n        `Snap execution service returned no RPC handler for running snap \"${snapId}\".`,\n      );\n    }\n\n    return rpcRequestHandler(options);\n  }\n}\n\n/**\n * Sets up stream multiplexing for the given stream.\n *\n * @param connectionStream - The stream to mux.\n * @param streamName - The name of the stream, for identification in errors.\n * @returns The multiplexed stream.\n */\nexport function setupMultiplex(\n  connectionStream: Duplex,\n  streamName: string,\n): ObjectMultiplex {\n  const mux = new ObjectMultiplex();\n  pump(\n    connectionStream,\n    // Typecast: stream type mismatch\n    mux as unknown as Duplex,\n    connectionStream,\n    (error) => {\n      if (error) {\n        streamName\n          ? logError(`\"${streamName}\" stream failure.`, error)\n          : logError(error);\n      }\n    },\n  );\n  return mux;\n}\n"]}

package/dist/services/ExecutionService.js

@@ -1,4 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const controllerName = 'ExecutionService';
-//# sourceMappingURL=ExecutionService.js.map

package/dist/services/ExecutionService.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"ExecutionService.js","sourceRoot":"","sources":["../../src/services/ExecutionService.ts"],"names":[],"mappings":";;AAgCA,MAAM,cAAc,GAAG,kBAAkB,CAAC","sourcesContent":["import { RestrictedControllerMessenger } from '@metamask/base-controller';\nimport { SnapId, SnapRpcHookArgs } from '@metamask/snaps-utils';\nimport { Json } from '@metamask/types';\n\ntype TerminateSnap = (snapId: string) => Promise<void>;\ntype TerminateAll = () => Promise<void>;\ntype ExecuteSnap = (snapData: SnapExecutionData) => Promise<unknown>;\n\ntype HandleRpcRequest = (\n  snapId: string,\n  options: SnapRpcHookArgs,\n) => Promise<unknown>;\n\nexport interface ExecutionService {\n  terminateSnap: TerminateSnap;\n  terminateAllSnaps: TerminateAll;\n  executeSnap: ExecuteSnap;\n  handleRpcRequest: HandleRpcRequest;\n}\n\nexport type SnapExecutionData = {\n  snapId: string;\n  sourceCode: string;\n  endowments?: Json;\n};\n\nexport type SnapErrorJson = {\n  message: string;\n  code: number;\n  data?: Json;\n};\n\nconst controllerName = 'ExecutionService';\n\nexport type ErrorMessageEvent = {\n  type: 'ExecutionService:unhandledError';\n  payload: [SnapId, SnapErrorJson];\n};\n\nexport type OutboundRequest = {\n  type: 'ExecutionService:outboundRequest';\n  payload: [SnapId];\n};\n\nexport type OutboundResponse = {\n  type: 'ExecutionService:outboundResponse';\n  payload: [SnapId];\n};\n\nexport type ExecutionServiceEvents =\n  | ErrorMessageEvent\n  | OutboundRequest\n  | OutboundResponse;\n\n/**\n * Handles RPC request.\n */\nexport type HandleRpcRequestAction = {\n  type: `${typeof controllerName}:handleRpcRequest`;\n  handler: ExecutionService['handleRpcRequest'];\n};\n\n/**\n * Executes a given snap.\n */\nexport type ExecuteSnapAction = {\n  type: `${typeof controllerName}:executeSnap`;\n  handler: ExecutionService['executeSnap'];\n};\n\n/**\n * Terminates a given snap.\n */\nexport type TerminateSnapAction = {\n  type: `${typeof controllerName}:terminateSnap`;\n  handler: ExecutionService['terminateSnap'];\n};\n\n/**\n * Terminates all snaps.\n */\nexport type TerminateAllSnapsAction = {\n  type: `${typeof controllerName}:terminateAllSnaps`;\n  handler: ExecutionService['terminateAllSnaps'];\n};\n\nexport type ExecutionServiceActions =\n  | HandleRpcRequestAction\n  | ExecuteSnapAction\n  | TerminateSnapAction\n  | TerminateAllSnapsAction;\n\nexport type ExecutionServiceMessenger = RestrictedControllerMessenger<\n  'ExecutionService',\n  ExecutionServiceActions,\n  ExecutionServiceEvents,\n  ExecutionServiceActions['type'],\n  ExecutionServiceEvents['type']\n>;\n"]}