activation tx for stylus contracts

Author: kumaryash90

packages/thirdweb/scripts/generate/abis/stylus/IArbWasm.json

@@ -0,0 +1,3 @@
+[
+  "function activateProgram(address program)"
+]

packages/thirdweb/src/contract/deployment/deploy-with-abi.ts

@@ -1,5 +1,7 @@
 import type { Abi, AbiConstructor } from "abitype";
+import { activateStylusContract } from "../../extensions/stylus/write/activateStylusContract.js";
 import { sendAndConfirmTransaction } from "../../transaction/actions/send-and-confirm-transaction.js";
+import { sendTransaction } from "../../transaction/actions/send-transaction.js";
 import { prepareTransaction } from "../../transaction/prepare-transaction.js";
 import { encodeAbiParameters } from "../../utils/abi/encodeAbiParameters.js";
 import { normalizeFunctionParams } from "../../utils/abi/normalizeFunctionParams.js";
@@ -124,6 +126,7 @@ export async function deployContract(
     account: Account;
     salt?: string;
     extraDataWithUri?: Hex;
+    isStylus?: boolean;
   },
 ) {
   if (await isZkSyncChain(options.chain)) {
@@ -138,10 +141,11 @@ export async function deployContract(
     });
   }
 
+  let address: string | null | undefined;
   if (options.salt !== undefined) {
     // Deploy with CREATE2 if salt is provided
     const info = await computeDeploymentInfoFromBytecode(options);
-    const address = computeDeploymentAddress({
+    address = computeDeploymentAddress({
       bytecode: options.bytecode,
       encodedArgs: info.encodedArgs,
       create2FactoryAddress: info.create2FactoryAddress,
@@ -167,18 +171,32 @@ export async function deployContract(
         data: info.initCalldata,
       }),
     });
-    return address;
+  } else {
+    const deployTx = prepareDirectDeployTransaction(options);
+    const receipt = await sendAndConfirmTransaction({
+      account: options.account,
+      transaction: deployTx,
+    });
+    address = receipt.contractAddress;
+    if (!address) {
+      throw new Error(
+        `Could not find deployed contract address in transaction: ${receipt.transactionHash}`,
+      );
+    }
   }
 
-  const deployTx = prepareDirectDeployTransaction(options);
-  const receipt = await sendAndConfirmTransaction({
-    account: options.account,
-    transaction: deployTx,
-  });
-  if (!receipt.contractAddress) {
-    throw new Error(
-      `Could not find deployed contract address in transaction: ${receipt.transactionHash}`,
-    );
+  if (options.isStylus) {
+    const activationTransaction = activateStylusContract({
+      chain: options.chain,
+      client: options.client,
+      contractAddress: address,
+    });
+
+    await sendTransaction({
+      transaction: activationTransaction,
+      account: options.account,
+    });
   }
-  return receipt.contractAddress;
+
+  return address;
 }

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

@@ -1,4 +1,6 @@
+import { activateStylusContract } from "../../../extensions/stylus/write/activateStylusContract.js";
 import { sendAndConfirmTransaction } from "../../../transaction/actions/send-and-confirm-transaction.js";
+import { sendTransaction } from "../../../transaction/actions/send-transaction.js";
 import {
   type FetchDeployMetadataResult,
   fetchBytecodeFromCompilerMetadata,
@@ -255,5 +257,20 @@ export async function getOrDeployInfraContractFromMetadata(
   if (!deployedInfraContract) {
     throw new Error(`Failed to deploy ${options.contractMetadata.name}`);
   }
+
+  const isStylus = options.contractMetadata.metadata.language === "rust";
+  if (isStylus) {
+    const activationTransaction = activateStylusContract({
+      chain: options.chain,
+      client: options.client,
+      contractAddress: deployedInfraContract.address,
+    });
+
+    await sendTransaction({
+      transaction: activationTransaction,
+      account: options.account,
+    });
+  }
+
   return deployedInfraContract;
 }

packages/thirdweb/src/exports/extensions/stylus.ts

@@ -0,0 +1,7 @@
+/**
+ * Write
+ */
+export {
+  activateStylusContract,
+  type ActivateStylusContractOptions,
+} from "../../extensions/stylus/write/activateStylusContract.js";

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

@@ -363,6 +363,7 @@ async function directDeploy(options: {
           metadataUri: options.metadataUri,
         })
       : undefined,
+    isStylus: options.compilerMetadata.metadata.language === "rust",
   });
 }
 

packages/thirdweb/src/extensions/stylus/__generated__/IArbWasm/write/activateProgram.ts

@@ -0,0 +1,41 @@
+import type { Chain } from "../../../chains/types.js";
+import type { ThirdwebClient } from "../../../client/client.js";
+import { getContract } from "../../../contract/contract.js";
+import { activateProgram } from "../__generated__/IArbWasm/write/activateProgram.js";
+
+const ARB_WASM_ADDRESS = "0000000000000000000000000000000000000071";
+
+export type ActivateStylusContractOptions = {
+  chain: Chain;
+  client: ThirdwebClient;
+  contractAddress: string;
+};
+
+/**
+ * Activate a stylus contract by calling ArbWasm Precompile
+ * @param options - The options for activating the contract
+ * @returns A prepared transaction to send
+ * @example
+ * ```ts
+ * import { activateStylusContract } from "thirdweb/stylus";
+ * const transaction = activateStylusContract({
+ *  client,
+ *  chain,
+ *  contractAddress,
+ * });
+ * await sendTransaction({ transaction, account });
+ * ```
+ */
+export function activateStylusContract(options: ActivateStylusContractOptions) {
+  const { chain, client, contractAddress } = options;
+  const arbWasmPrecompile = getContract({
+    client,
+    chain,
+    address: ARB_WASM_ADDRESS,
+  });
+
+  return activateProgram({
+    program: contractAddress,
+    contract: arbWasmPrecompile,
+  });
+}