From 3db265fa0a245c94102d5d367c91e25e73ec05a8 Mon Sep 17 00:00:00 2001
From: Christophe <contact@christophedeveaux.com>
Date: Thu, 27 Jun 2024 19:08:07 +0000
Subject: [PATCH] Feat: Add v1 SequencerInbox setters

Closes FS-535
---
 src/actions/invalidateKeysetHash.ts | 49 +++++++++++++++++++++
 src/actions/setIsbatchPoster.ts     | 67 +++++++++++++++++++++++++++++
 src/actions/setKeyset.ts            | 49 +++++++++++++++++++++
 src/actions/setMaxTimeVariation.ts  | 46 ++++++++++++++++++++
 src/types/Actions.ts                |  4 ++
 5 files changed, 215 insertions(+)
 create mode 100644 src/actions/invalidateKeysetHash.ts
 create mode 100644 src/actions/setIsbatchPoster.ts
 create mode 100644 src/actions/setKeyset.ts
 create mode 100644 src/actions/setMaxTimeVariation.ts

diff --git a/src/actions/invalidateKeysetHash.ts b/src/actions/invalidateKeysetHash.ts
new file mode 100644
index 00000000..1d0bd681
--- /dev/null
+++ b/src/actions/invalidateKeysetHash.ts
@@ -0,0 +1,49 @@
+import {
+  Chain,
+  Hex,
+  PrepareTransactionRequestParameters,
+  PrepareTransactionRequestReturnType,
+  PublicClient,
+  Transport,
+  encodeFunctionData,
+} from 'viem';
+import { sequencerInbox } from '../contracts';
+import { ActionParameters, WithAccount } from '../types/Actions';
+import { Prettify } from '../types/utils';
+
+export type InvalidateKeysetHashParameters<Curried extends boolean = false> = Prettify<
+  WithAccount<
+    ActionParameters<
+      {
+        keysetHash: Hex;
+      },
+      'sequencerInbox',
+      Curried
+    >
+  >
+>;
+
+export type InvalidateKeysetHashReturnType = PrepareTransactionRequestReturnType;
+
+function sequencerInboxFunctionData({ keysetHash }: InvalidateKeysetHashParameters) {
+  return encodeFunctionData({
+    abi: sequencerInbox.abi,
+    functionName: 'invalidateKeysetHash',
+    args: [keysetHash],
+  });
+}
+
+export async function invalidateKeysetHash<TChain extends Chain | undefined>(
+  client: PublicClient<Transport, TChain>,
+  args: InvalidateKeysetHashParameters,
+): Promise<InvalidateKeysetHashReturnType> {
+  const data = sequencerInboxFunctionData(args);
+
+  return client.prepareTransactionRequest({
+    to: args.sequencerInbox,
+    value: BigInt(0),
+    chain: client.chain,
+    data,
+    account: args.account,
+  } satisfies PrepareTransactionRequestParameters);
+}
diff --git a/src/actions/setIsbatchPoster.ts b/src/actions/setIsbatchPoster.ts
new file mode 100644
index 00000000..eef471d3
--- /dev/null
+++ b/src/actions/setIsbatchPoster.ts
@@ -0,0 +1,67 @@
+import {
+  Address,
+  Chain,
+  PrepareTransactionRequestParameters,
+  PrepareTransactionRequestReturnType,
+  PublicClient,
+  Transport,
+  encodeFunctionData,
+} from 'viem';
+import { sequencerInbox } from '../contracts';
+import { ActionParameters, WithAccount } from '../types/Actions';
+import { Prettify } from '../types/utils';
+
+type Args = {
+  batchPoster: Address;
+};
+
+export type SetIsBatchPosterParameters<Curried extends boolean = false> = Prettify<
+  WithAccount<ActionParameters<Args, 'sequencerInbox', Curried>>
+>;
+
+export type SetIsBatchPosterReturnType = PrepareTransactionRequestReturnType;
+
+function sequencerInboxFunctionData({
+  batchPoster,
+  enable,
+}: SetIsBatchPosterParameters & { enable: boolean }) {
+  return encodeFunctionData({
+    abi: sequencerInbox.abi,
+    functionName: 'setIsBatchPoster',
+    args: [batchPoster, enable],
+  });
+}
+
+async function setIsBatchPoster<TChain extends Chain | undefined>(
+  client: PublicClient<Transport, TChain>,
+  args: SetIsBatchPosterParameters & { enable: boolean },
+): Promise<SetIsBatchPosterReturnType> {
+  const data = sequencerInboxFunctionData(args);
+  return client.prepareTransactionRequest({
+    to: args.sequencerInbox,
+    value: BigInt(0),
+    chain: client.chain,
+    data,
+    account: args.account,
+  } satisfies PrepareTransactionRequestParameters);
+}
+
+export async function enableBatchPoster<TChain extends Chain | undefined>(
+  client: PublicClient<Transport, TChain>,
+  args: SetIsBatchPosterParameters,
+): Promise<SetIsBatchPosterReturnType> {
+  return setIsBatchPoster(client, {
+    ...args,
+    enable: true,
+  });
+}
+
+export async function disableBatchPoster<TChain extends Chain | undefined>(
+  client: PublicClient<Transport, TChain>,
+  args: SetIsBatchPosterParameters,
+): Promise<SetIsBatchPosterReturnType> {
+  return setIsBatchPoster(client, {
+    ...args,
+    enable: false,
+  });
+}
diff --git a/src/actions/setKeyset.ts b/src/actions/setKeyset.ts
new file mode 100644
index 00000000..d71c1212
--- /dev/null
+++ b/src/actions/setKeyset.ts
@@ -0,0 +1,49 @@
+import {
+  Chain,
+  Hex,
+  PrepareTransactionRequestParameters,
+  PrepareTransactionRequestReturnType,
+  PublicClient,
+  Transport,
+  encodeFunctionData,
+} from 'viem';
+import { sequencerInbox } from '../contracts';
+import { ActionParameters, WithAccount } from '../types/Actions';
+import { Prettify } from '../types/utils';
+
+export type SetKeysetParameters<Curried extends boolean = false> = Prettify<
+  WithAccount<
+    ActionParameters<
+      {
+        keyset: Hex;
+      },
+      'sequencerInbox',
+      Curried
+    >
+  >
+>;
+
+export type SetKeysetReturnType = PrepareTransactionRequestReturnType;
+
+function sequencerInboxFunctionData({ keyset }: SetKeysetParameters) {
+  return encodeFunctionData({
+    abi: sequencerInbox.abi,
+    functionName: 'setValidKeyset',
+    args: [keyset],
+  });
+}
+
+export async function setKeyset<TChain extends Chain | undefined>(
+  client: PublicClient<Transport, TChain>,
+  args: SetKeysetParameters,
+): Promise<SetKeysetReturnType> {
+  const data = sequencerInboxFunctionData(args);
+
+  return client.prepareTransactionRequest({
+    to: args.sequencerInbox,
+    value: BigInt(0),
+    chain: client.chain,
+    data,
+    account: args.account,
+  } satisfies PrepareTransactionRequestParameters);
+}
diff --git a/src/actions/setMaxTimeVariation.ts b/src/actions/setMaxTimeVariation.ts
new file mode 100644
index 00000000..7f763d84
--- /dev/null
+++ b/src/actions/setMaxTimeVariation.ts
@@ -0,0 +1,46 @@
+import {
+  Chain,
+  PrepareTransactionRequestParameters,
+  PrepareTransactionRequestReturnType,
+  PublicClient,
+  Transport,
+  encodeFunctionData,
+} from 'viem';
+import { sequencerInbox } from '../contracts';
+import { ActionParameters, WithAccount } from '../types/Actions';
+import { Prettify } from '../types/utils';
+
+type Args = {
+  delayBlocks: bigint;
+  futureBlocks: bigint;
+  delaySeconds: bigint;
+  futureSeconds: bigint;
+};
+export type SetMaxTimeVariationParameters<Curried extends boolean = false> = Prettify<
+  WithAccount<ActionParameters<Args, 'sequencerInbox', Curried>>
+>;
+
+export type SetMaxTimeVariationReturnType = PrepareTransactionRequestReturnType;
+
+function sequencerInboxFunctionData(args: SetMaxTimeVariationParameters) {
+  return encodeFunctionData({
+    abi: sequencerInbox.abi,
+    functionName: 'setMaxTimeVariation',
+    args: [args],
+  });
+}
+
+export async function setMaxTimeVariation<TChain extends Chain | undefined>(
+  client: PublicClient<Transport, TChain>,
+  args: SetMaxTimeVariationParameters,
+): Promise<SetMaxTimeVariationReturnType> {
+  const data = sequencerInboxFunctionData(args);
+
+  return client.prepareTransactionRequest({
+    to: args.sequencerInbox,
+    value: BigInt(0),
+    chain: client.chain,
+    data,
+    account: args.account,
+  } satisfies PrepareTransactionRequestParameters);
+}
diff --git a/src/types/Actions.ts b/src/types/Actions.ts
index 9072c656..68daf108 100644
--- a/src/types/Actions.ts
+++ b/src/types/Actions.ts
@@ -22,3 +22,7 @@ export type ActionParameters<Args, ContractName extends string, Curried extends
         [key in ContractName]?: Address;
       }
 >;
+
+export type WithAccount<Args> = Args & {
+  account: Address;
+};