[SDK] Improve contract-roles methods (#1798)

Author: kien-ngo

.changeset/clever-buttons-bake.md

@@ -0,0 +1,5 @@
+---
+"@thirdweb-dev/sdk": patch
+---
+
+Improve performance for contract-roles methods

packages/sdk/src/evm/core/classes/contract-roles.ts

@@ -68,9 +68,10 @@ export class ContractRoles<TContract extends IPermissions, TRole extends Role>
   public async getAll(): Promise<Record<TRole, string[]>> {
     invariant(this.roles.length, "this contract has no support for roles");
     const roles = {} as Record<TRole, string[]>;
-    for (const role of this.roles) {
-      roles[role] = await this.get(role);
-    }
+    const entries = Object.entries(this.roles);
+    (await Promise.all(entries.map(([, role]) => this.get(role)))).forEach(
+      (item, index) => (roles[entries[index][1]] = item),
+    );
     return roles;
   }
 
@@ -151,17 +152,18 @@ export class ContractRoles<TContract extends IPermissions, TRole extends Role>
       const sortedRoles = roles.sort((role) => (role === "admin" ? 1 : -1));
       for (let i = 0; i < sortedRoles.length; i++) {
         const role = sortedRoles[i];
-        const addresses: Address[] = await Promise.all(
-          rolesWithAddresses[role]?.map(
-            async (addressOrEns) => await resolveAddress(addressOrEns),
-          ) || [],
-        );
-        const currentAddresses: Address[] = await Promise.all(
-          currentRoles[role]?.map(
-            async (addressOrEns) =>
-              await resolveAddress(addressOrEns as AddressOrEns),
-          ) || [],
-        );
+        const [addresses, currentAddresses] = await Promise.all([
+          Promise.all(
+            rolesWithAddresses[role]?.map((addressOrEns) =>
+              resolveAddress(addressOrEns),
+            ) || [],
+          ),
+          Promise.all(
+            currentRoles[role]?.map((addressOrEns) =>
+              resolveAddress(addressOrEns as AddressOrEns),
+            ) || [],
+          ),
+        ]);
         const toAdd = addresses.filter(
           (address) => !currentAddresses.includes(address),
         );
@@ -176,18 +178,17 @@ export class ContractRoles<TContract extends IPermissions, TRole extends Role>
           });
         }
         if (toRemove.length) {
-          for (let j = 0; j < toRemove.length; j++) {
-            const address = toRemove[j];
-            const revokeFunctionName = (await this.getRevokeRoleFunctionName(
-              address,
-            )) as any;
+          const revokeFunctionNames = await Promise.all(
+            toRemove.map((address) => this.getRevokeRoleFunctionName(address)),
+          );
+          revokeFunctionNames.forEach((revokeFunctionName, index) =>
             encoded.push(
               contractEncoder.encode(revokeFunctionName, [
                 getRoleHash(role),
-                address,
+                toRemove[index],
               ]),
-            );
-          }
+            ),
+          );
         }
       }
 
@@ -211,8 +212,10 @@ export class ContractRoles<TContract extends IPermissions, TRole extends Role>
   public async verify(roles: TRole[], address: AddressOrEns): Promise<void> {
     await Promise.all(
       roles.map(async (role) => {
-        const members = await this.get(role);
-        const resolvedAddress: Address = await resolveAddress(address);
+        const [members, resolvedAddress] = await Promise.all([
+          this.get(role),
+          resolveAddress(address),
+        ]);
         if (
           !members
             .map((a) => a.toLowerCase())
@@ -310,8 +313,10 @@ export class ContractRoles<TContract extends IPermissions, TRole extends Role>
    ****************************/
 
   private async getRevokeRoleFunctionName(address: AddressOrEns) {
-    const resolvedAddress = await resolveAddress(address);
-    const signerAddress = await this.contractWrapper.getSignerAddress();
+    const [resolvedAddress, signerAddress] = await Promise.all([
+      resolveAddress(address),
+      this.contractWrapper.getSignerAddress(),
+    ]);
     if (signerAddress.toLowerCase() === resolvedAddress.toLowerCase()) {
       return "renounceRole";
     }