feat: Editing flow

app/scripts/controllers/permissions/background-api.js

@@ -3,13 +3,10 @@ import {
   CaveatTypes,
   RestrictedMethods,
 } from '../../../../shared/constants/permissions';
-import { CaveatFactories } from './specifications';
+import { CaveatFactories, PermissionNames } from './specifications';
 
 export function getPermissionBackgroundApiMethods(permissionController) {
-  const addMoreAccounts = (origin, accountOrAccounts) => {
-    const accounts = Array.isArray(accountOrAccounts)
-      ? accountOrAccounts
-      : [accountOrAccounts];
+  const addMoreAccounts = (origin, accounts) => {
     const caveat = CaveatFactories.restrictReturnedAccounts(accounts);
 
     permissionController.grantPermissionsIncremental({
@@ -20,11 +17,21 @@ export function getPermissionBackgroundApiMethods(permissionController) {
     });
   };
 
-  return {
-    addPermittedAccount: (origin, account) => addMoreAccounts(origin, account),
+  const addMoreChains = (origin, chainIds) => {
+    const caveat = CaveatFactories.restrictNetworkSwitching(chainIds);
+
+    permissionController.grantPermissionsIncremental({
+      subject: { origin },
+      approvedPermissions: {
+        [PermissionNames.permittedChains]: { caveats: [caveat] },
+      },
+    });
+  };
 
-    // To add more than one account when already connected to the dapp
-    addMorePermittedAccounts: (origin, accounts) =>
+  return {
+    addPermittedAccount: (origin, account) =>
+      addMoreAccounts(origin, [account]),
+    addPermittedAccounts: (origin, accounts) =>
       addMoreAccounts(origin, accounts),
 
     removePermittedAccount: (origin, account) => {
@@ -57,6 +64,52 @@ export function getPermissionBackgroundApiMethods(permissionController) {
       }
     },
 
+    addPermittedChain: (origin, chainId) => addMoreChains(origin, [chainId]),
+    addPermittedChains: (origin, chainIds) => addMoreChains(origin, chainIds),
+
+    removePermittedChain: (origin, chainId) => {
+      const { value: existingChains } = permissionController.getCaveat(
+        origin,
+        PermissionNames.permittedChains,
+        CaveatTypes.restrictNetworkSwitching,
+      );
+
+      const remainingChains = existingChains.filter(
+        (existingChain) => existingChain !== chainId,
+      );
+
+      if (remainingChains.length === existingChains.length) {
+        return;
+      }
+
+      if (remainingChains.length === 0) {
+        permissionController.revokePermission(
+          origin,
+          PermissionNames.permittedChains,
+        );
+      } else {
+        permissionController.updateCaveat(
+          origin,
+          PermissionNames.permittedChains,
+          CaveatTypes.restrictNetworkSwitching,
+          remainingChains,
+        );
+      }
+    },
+
+    requestAccountsAndChainPermissionsWithId: async (origin) => {
+      const id = nanoid();
+      permissionController.requestPermissions(
+        { origin },
+        {
+          [PermissionNames.eth_accounts]: {},
+          [PermissionNames.permittedChains]: {},
+        },
+        { id },
+      );
+      return id;
+    },
+
     requestAccountsPermissionWithId: async (origin) => {
       const id = nanoid();
       permissionController.requestPermissions(