Fetch rollupAddress from sequencerInbox in setters

OffchainLabs · Jul 10, 2024 · f3b0075 · f3b0075
1 parent a5dc412
commit f3b0075
Show file tree

Hide file tree

Showing 6 changed files with 39 additions and 51 deletions.
diff --git a/src/actions/setConfirmPeriodBlocks.ts b/src/actions/setConfirmPeriodBlocks.ts
@@ -7,19 +7,12 @@ import {
   encodeFunctionData,
 } from 'viem';
 import { rollupAdminLogic } from '../contracts';
-import { ActionParameters, WithAccount } from '../types/Actions';
+import { WithAccount, WithContractAddress } from '../types/Actions';
 import { Prettify } from '../types/utils';
+import { getRollupAddress } from '../getRollupAddress';
 
 export type SetConfirmPeriodBlocksParameters<Curried extends boolean = false> = Prettify<
-  WithAccount<
-    ActionParameters<
-      {
-        newPeriod: bigint;
-      },
-      'rollupAdminLogic',
-      Curried
-    >
-  >
+  WithAccount<WithContractAddress<{ newPeriod: bigint }, 'rollupAdminLogic', Curried>>
 >;
 
 export type SetConfirmPeriodBlocksReturnType = PrepareTransactionRequestReturnType;
@@ -37,9 +30,9 @@ export async function setConfirmPeriodBlocks<TChain extends Chain | undefined>(
   args: SetConfirmPeriodBlocksParameters,
 ): Promise<SetConfirmPeriodBlocksReturnType> {
   const data = rollupAdminLogicFunctionData(args);
-
+  const rollupAdminLogicAddresss = await getRollupAddress(client, args);
   return client.prepareTransactionRequest({
-    to: args.rollupAdminLogic,
+    to: rollupAdminLogicAddresss,
     value: BigInt(0),
     chain: client.chain,
     data,

diff --git a/src/actions/setExtraChallengeTimeBlocks.ts b/src/actions/setExtraChallengeTimeBlocks.ts
@@ -7,19 +7,12 @@ import {
   encodeFunctionData,
 } from 'viem';
 import { rollupAdminLogic } from '../contracts';
-import { ActionParameters, WithAccount } from '../types/Actions';
+import { WithAccount, WithContractAddress } from '../types/Actions';
 import { Prettify } from '../types/utils';
+import { getRollupAddress } from '../getRollupAddress';
 
 export type SetExtraChallengeTimeBlocksParameters<Curried extends boolean = false> = Prettify<
-  WithAccount<
-    ActionParameters<
-      {
-        newExtraTimeBlocks: bigint;
-      },
-      'rollupAdminLogic',
-      Curried
-    >
-  >
+  WithAccount<WithContractAddress<{ newExtraTimeBlocks: bigint }, 'rollupAdminLogic', Curried>>
 >;
 
 export type SetExtraChallengeTimeBlocksReturnType = PrepareTransactionRequestReturnType;
@@ -39,9 +32,9 @@ export async function setExtraChallengeTimeBlocks<TChain extends Chain | undefin
   args: SetExtraChallengeTimeBlocksParameters,
 ): Promise<SetExtraChallengeTimeBlocksReturnType> {
   const data = rollupAdminLogicFunctionData(args);
-
+  const rollupAdminLogicAddresss = await getRollupAddress(client, args);
   return client.prepareTransactionRequest({
-    to: args.rollupAdminLogic,
+    to: rollupAdminLogicAddresss,
     value: BigInt(0),
     chain: client.chain,
     data,

diff --git a/src/actions/setMinimumAssertionPeriod.ts b/src/actions/setMinimumAssertionPeriod.ts
@@ -7,19 +7,12 @@ import {
   encodeFunctionData,
 } from 'viem';
 import { rollupAdminLogic } from '../contracts';
-import { ActionParameters, WithAccount } from '../types/Actions';
+import { WithAccount, WithContractAddress } from '../types/Actions';
 import { Prettify } from '../types/utils';
+import { getRollupAddress } from '../getRollupAddress';
 
 export type SetMinimumAssertionPeriodParameters<Curried extends boolean = false> = Prettify<
-  WithAccount<
-    ActionParameters<
-      {
-        newPeriod: bigint;
-      },
-      'rollupAdminLogic',
-      Curried
-    >
-  >
+  WithAccount<WithContractAddress<{ newPeriod: bigint }, 'rollupAdminLogic', Curried>>
 >;
 
 export type SetMinimumAssertionPeriodReturnType = PrepareTransactionRequestReturnType;
@@ -37,9 +30,9 @@ export async function setMinimumAssertionPeriod<TChain extends Chain | undefined
   args: SetMinimumAssertionPeriodParameters,
 ): Promise<SetMinimumAssertionPeriodReturnType> {
   const data = rollupAdminLogicFunctionData(args);
-
+  const rollupAdminLogicAddresss = await getRollupAddress(client, args);
   return client.prepareTransactionRequest({
-    to: args.rollupAdminLogic,
+    to: rollupAdminLogicAddresss,
     value: BigInt(0),
     chain: client.chain,
     data,

diff --git a/src/actions/setValidator.ts b/src/actions/setValidator.ts
@@ -8,16 +8,21 @@ import {
   encodeFunctionData,
 } from 'viem';
 import { rollupAdminLogic } from '../contracts';
-import { ActionParameters, WithAccount } from '../types/Actions';
+import { WithAccount, WithContractAddress } from '../types/Actions';
 import { Prettify } from '../types/utils';
-
-type Args = {
-  add: Address[];
-  remove: Address[];
-};
+import { getRollupAddress } from '../getRollupAddress';
 
 export type SetIsValidatorParameters<Curried extends boolean = false> = Prettify<
-  WithAccount<ActionParameters<Args, 'rollupAdminLogic', Curried>>
+  WithAccount<
+    WithContractAddress<
+      {
+        add: Address[];
+        remove: Address[];
+      },
+      'rollupAdminLogic',
+      Curried
+    >
+  >
 >;
 
 export type SetIsValidatorReturnType = PrepareTransactionRequestReturnType;
@@ -37,8 +42,9 @@ export async function setValidators<TChain extends Chain | undefined>(
   args: SetIsValidatorParameters,
 ): Promise<SetIsValidatorReturnType> {
   const data = rollupAdminLogicFunctionData(args);
+  const rollupAdminLogicAddresss = await getRollupAddress(client, args);
   return client.prepareTransactionRequest({
-    to: args.rollupAdminLogic,
+    to: rollupAdminLogicAddresss,
     value: BigInt(0),
     chain: client.chain,
     data,

diff --git a/src/actions/setValidatorWhitelistDisabled.ts b/src/actions/setValidatorWhitelistDisabled.ts
@@ -7,11 +7,12 @@ import {
   encodeFunctionData,
 } from 'viem';
 import { rollupAdminLogic } from '../contracts';
-import { ActionParameters, WithAccount } from '../types/Actions';
+import { WithAccount, WithContractAddress } from '../types/Actions';
 import { Prettify } from '../types/utils';
+import { getRollupAddress } from '../getRollupAddress';
 
 export type SetValidatorWhitelistDisabledParameters<Curried extends boolean = false> = Prettify<
-  WithAccount<ActionParameters<{}, 'rollupAdminLogic', Curried>>
+  WithAccount<WithContractAddress<{}, 'rollupAdminLogic', Curried>>
 >;
 
 export type SetValidatorWhitelistDisabledReturnType = PrepareTransactionRequestReturnType;
@@ -31,8 +32,9 @@ async function setValidatorWhitelistDisabled<TChain extends Chain | undefined>(
   args: SetValidatorWhitelistDisabledParameters & { enable: boolean },
 ): Promise<SetValidatorWhitelistDisabledReturnType> {
   const data = rollupAdminLogicFunctionData(args);
+  const rollupAdminLogicAddresss = await getRollupAddress(client, args);
   return client.prepareTransactionRequest({
-    to: args.rollupAdminLogic,
+    to: rollupAdminLogicAddresss,
     value: BigInt(0),
     chain: client.chain,
     data,

diff --git a/src/actions/setWasmModuleRoot.ts b/src/actions/setWasmModuleRoot.ts
@@ -8,12 +8,13 @@ import {
   encodeFunctionData,
 } from 'viem';
 import { rollupAdminLogic } from '../contracts';
-import { ActionParameters, WithAccount } from '../types/Actions';
+import { WithAccount, WithContractAddress } from '../types/Actions';
 import { Prettify } from '../types/utils';
+import { getRollupAddress } from '../getRollupAddress';
 
 export type SetWasmModuleRootParameters<Curried extends boolean = false> = Prettify<
   WithAccount<
-    ActionParameters<
+    WithContractAddress<
       {
         newWasmModuleRoot: Hex;
       },
@@ -38,9 +39,9 @@ export async function setWasmModuleRoot<TChain extends Chain | undefined>(
   args: SetWasmModuleRootParameters,
 ): Promise<SetWasmModuleRootReturnType> {
   const data = rollupAdminLogicFunctionData(args);
-
+  const rollupAdminLogicAddresss = await getRollupAddress(client, args);
   return client.prepareTransactionRequest({
-    to: args.rollupAdminLogic,
+    to: rollupAdminLogicAddresss,
     value: BigInt(0),
     chain: client.chain,
     data,