estimate data fee for activation

Author: kumaryash90

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

@@ -1,3 +1,3 @@
 [
-  "function activateProgram(address program)"
+  "function activateProgram(address program) returns (uint16,uint256)"
 ]

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

@@ -186,16 +186,20 @@ export async function deployContract(
   }
 
   if (options.isStylus) {
-    const activationTransaction = activateStylusContract({
-      chain: options.chain,
-      client: options.client,
-      contractAddress: address,
-    });
+    try {
+      const activationTransaction = await activateStylusContract({
+        chain: options.chain,
+        client: options.client,
+        contractAddress: address,
+      });
 
-    await sendTransaction({
-      transaction: activationTransaction,
-      account: options.account,
-    });
+      await sendTransaction({
+        transaction: activationTransaction,
+        account: options.account,
+      });
+    } catch {
+      console.error("Error: Contract could not be activated.");
+    }
   }
 
   return address;

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

@@ -260,16 +260,20 @@ export async function getOrDeployInfraContractFromMetadata(
 
   const isStylus = options.contractMetadata.metadata.language === "rust";
   if (isStylus) {
-    const activationTransaction = activateStylusContract({
-      chain: options.chain,
-      client: options.client,
-      contractAddress: deployedInfraContract.address,
-    });
+    try {
+      const activationTransaction = await activateStylusContract({
+        chain: options.chain,
+        client: options.client,
+        contractAddress: deployedInfraContract.address,
+      });
 
-    await sendTransaction({
-      transaction: activationTransaction,
-      account: options.account,
-    });
+      await sendTransaction({
+        transaction: activationTransaction,
+        account: options.account,
+      });
+    } catch {
+      console.error("Error: Contract could not be activated.");
+    }
   }
 
   return deployedInfraContract;

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

@@ -1,6 +1,15 @@
+import {
+  decodeAbiParameters,
+  formatTransactionRequest,
+  parseEther,
+} from "viem";
 import type { Chain } from "../../../chains/types.js";
 import type { ThirdwebClient } from "../../../client/client.js";
 import { getContract } from "../../../contract/contract.js";
+import { eth_call } from "../../../rpc/actions/eth_call.js";
+import { getRpcClient } from "../../../rpc/rpc.js";
+import { encode } from "../../../transaction/actions/encode.js";
+import type { PreparedTransaction } from "../../../transaction/prepare-transaction.js";
 import { activateProgram } from "../__generated__/IArbWasm/write/activateProgram.js";
 
 const ARB_WASM_ADDRESS = "0x0000000000000000000000000000000000000071";
@@ -26,16 +35,61 @@ export type ActivateStylusContractOptions = {
  * await sendTransaction({ transaction, account });
  * ```
  */
-export function activateStylusContract(options: ActivateStylusContractOptions) {
+export async function activateStylusContract(
+  options: ActivateStylusContractOptions,
+) {
   const { chain, client, contractAddress } = options;
   const arbWasmPrecompile = getContract({
     client,
     chain,
     address: ARB_WASM_ADDRESS,
   });
 
+  const dataFee = await estimateDataFee({
+    transaction: activateProgram({
+      program: contractAddress,
+      contract: arbWasmPrecompile,
+    }),
+  });
+
   return activateProgram({
     program: contractAddress,
     contract: arbWasmPrecompile,
+    overrides: {
+      value: dataFee,
+    },
+  });
+}
+
+async function estimateDataFee(options: {
+  // biome-ignore lint/suspicious/noExplicitAny:
+  transaction: PreparedTransaction<any>;
+}) {
+  const data = await encode(options.transaction);
+
+  const serializedTx = formatTransactionRequest({
+    data,
+    to: ARB_WASM_ADDRESS,
+    value: parseEther("1"), // only for simulation. it will be replaced with estimated data fee.
   });
+
+  const rpcRequest = getRpcClient(options.transaction);
+  try {
+    const result = await eth_call(rpcRequest, serializedTx);
+    const [, dataFee] = decodeAbiParameters(
+      [
+        {
+          type: "uint16",
+        },
+        {
+          type: "uint256",
+        },
+      ],
+      result,
+    );
+
+    return (dataFee * BigInt(100 + 10)) / BigInt(100); // bump 10%
+  } catch {
+    return 0n;
+  }
 }