Handle dynamic contract default extensions

Author: kumaryash90

packages/thirdweb/src/contract/deployment/utils/bootstrap.ts

@@ -142,7 +142,11 @@ export async function getOrDeployInfraForPublishedContract(
       version,
     });
   }
-  return { cloneFactoryContract, implementationContract };
+
+  return {
+    cloneFactoryContract,
+    implementationContract,
+  };
 }
 
 /**

packages/thirdweb/src/extensions/prebuilts/deploy-marketplace.ts

@@ -1,10 +1,4 @@
-import type {
-  Abi,
-  AbiFunction,
-  AbiParametersToPrimitiveTypes,
-  Address,
-} from "abitype";
-import { toFunctionSelector, toFunctionSignature } from "viem";
+import type { AbiParametersToPrimitiveTypes, Address } from "abitype";
 import type { ThirdwebClient } from "../../client/client.js";
 import { resolveContractAbi } from "../../contract/actions/resolve-abi.js";
 import type { ThirdwebContract } from "../../contract/contract.js";
@@ -19,6 +13,7 @@ import { getRoyaltyEngineV1ByChainId } from "../../utils/royalty-engine.js";
 import type { Prettify } from "../../utils/type-utils.js";
 import type { ClientAndChainAndAccount } from "../../utils/types.js";
 import { initialize as initMarketplace } from "./__generated__/Marketplace/write/initialize.js";
+import { generateExtensionFunctionsFromAbi } from "./get-required-transactions.js";
 
 export type MarketplaceContractParams = {
   name: string;
@@ -199,22 +194,6 @@ async function getInitializeTransaction(options: {
   });
 }
 
-// helperFns
-
-function generateExtensionFunctionsFromAbi(abi: Abi): Array<{
-  functionSelector: string;
-  functionSignature: string;
-}> {
-  const functions = abi.filter(
-    (item) => item.type === "function" && !item.name.startsWith("_"),
-  ) as AbiFunction[];
-
-  return functions.map((fn) => ({
-    functionSelector: toFunctionSelector(fn),
-    functionSignature: toFunctionSignature(fn),
-  }));
-}
-
 // let's just ... put this down here
 type MarketplaceConstructorParams = AbiParametersToPrimitiveTypes<
   [

packages/thirdweb/src/extensions/prebuilts/deploy-published.ts

@@ -204,6 +204,29 @@ export async function deployContractfromDeployMetadata(
         import("../../contract/deployment/deploy-via-autofactory.js"),
         import("../../contract/deployment/utils/bootstrap.js"),
       ]);
+
+      if (
+        deployMetadata.routerType === "dynamic" &&
+        deployMetadata.defaultExtensions
+      ) {
+        for (const e of deployMetadata.defaultExtensions) {
+          await getOrDeployInfraForPublishedContract({
+            chain,
+            client,
+            account,
+            contractId: e.extensionName,
+            version: e.extensionVersion || "latest",
+            publisher: e.publisherAddress,
+            constructorParams:
+              await getAllDefaultConstructorParamsForImplementation({
+                chain,
+                client,
+                contractId: e.extensionName,
+              }),
+          });
+        }
+      }
+
       const { cloneFactoryContract, implementationContract } =
         await getOrDeployInfraForPublishedContract({
           chain,
@@ -216,6 +239,7 @@ export async function deployContractfromDeployMetadata(
               chain,
               client,
               contractId: deployMetadata.name,
+              defaultExtensions: deployMetadata.defaultExtensions,
             })),
           publisher: deployMetadata.publisher,
           version: deployMetadata.version,

packages/thirdweb/src/extensions/prebuilts/get-required-transactions.ts

@@ -1,5 +1,8 @@
+import type { Abi, AbiFunction } from "abitype";
+import { toFunctionSelector, toFunctionSignature } from "viem";
 import type { Chain } from "../../chains/types.js";
 import type { ThirdwebClient } from "../../client/client.js";
+import { resolveContractAbi } from "../../contract/actions/resolve-abi.js";
 import { getDeployedCreate2Factory } from "../../contract/deployment/utils/create-2-factory.js";
 import { getDeployedInfraContract } from "../../contract/deployment/utils/infra.js";
 import { getDeployedInfraContractFromMetadata } from "../../contract/deployment/utils/infra.js";
@@ -19,6 +22,15 @@ type DeployTransactionType =
   | "extension"
   | "proxy";
 
+/**
+ * @internal
+ */
+export type DynamicContractExtension = {
+  extensionName: string;
+  extensionVersion: string;
+  publisherAddress: string;
+};
+
 /**
  * @internal
  */
@@ -226,6 +238,7 @@ export async function getAllDefaultConstructorParamsForImplementation(args: {
   chain: Chain;
   client: ThirdwebClient;
   contractId: string;
+  defaultExtensions?: DynamicContractExtension[];
 }) {
   const { chain, client } = args;
   const isZkSync = await isZkSyncChain(chain);
@@ -251,8 +264,81 @@ export async function getAllDefaultConstructorParamsForImplementation(args: {
       contractId: "WETH9",
     }),
   ]);
+
+  const defaultExtensionInput = args.defaultExtensions
+    ? await generateExtensionInput({
+        defaultExtensions: args.defaultExtensions,
+        chain,
+        client,
+        forwarder,
+        nativeTokenWrapper: weth,
+      })
+    : [];
+
   return {
     trustedForwarder: forwarder,
     nativeTokenWrapper: weth,
+    extensions: defaultExtensionInput,
   };
 }
+
+async function generateExtensionInput(args: {
+  defaultExtensions: DynamicContractExtension[];
+  chain: Chain;
+  client: ThirdwebClient;
+  forwarder: string;
+  nativeTokenWrapper: string;
+}) {
+  const { defaultExtensions, chain, client, forwarder, nativeTokenWrapper } =
+    args;
+
+  const deployedExtensions = await Promise.all(
+    defaultExtensions.map((e) =>
+      getDeployedInfraContract({
+        chain,
+        client,
+        contractId: e.extensionName,
+        publisher: e.publisherAddress,
+        version: e.extensionVersion || "latest",
+        constructorParams: { forwarder, nativeTokenWrapper },
+      }).then((c) => ({
+        name: e.extensionName,
+        metadataURI: "",
+        implementation: c,
+      })),
+    ),
+  );
+
+  const extensionInput = await Promise.all(
+    deployedExtensions.map(async (e) => {
+      if (!e.implementation) {
+        throw new Error("Extension not deployed");
+      }
+      return resolveContractAbi(e.implementation)
+        .then(generateExtensionFunctionsFromAbi)
+        .then((c) => ({
+          metadata: {
+            ...e,
+            implementation: e.implementation?.address,
+          },
+          functions: c,
+        }));
+    }),
+  );
+
+  return extensionInput;
+}
+
+export function generateExtensionFunctionsFromAbi(abi: Abi): Array<{
+  functionSelector: string;
+  functionSignature: string;
+}> {
+  const functions = abi.filter(
+    (item) => item.type === "function" && !item.name.startsWith("_"),
+  ) as AbiFunction[];
+
+  return functions.map((fn) => ({
+    functionSelector: toFunctionSelector(fn),
+    functionSignature: toFunctionSignature(fn),
+  }));
+}