Fix/mantle preverification (#337)

* mantle calculations * refactor * rename to TimedQueue * cleanup gasPriceManager * split into different files * cleanup --------- Co-authored-by: mouseless <[email protected]>
pimlicolabs · Oct 21, 2024 · da149ac · da149ac
1 parent 1900e42
commit da149ac
Show file tree

Hide file tree

Showing 11 changed files with 613 additions and 383 deletions.
diff --git a/src/cli/config/bundler.ts b/src/cli/config/bundler.ts
@@ -123,7 +123,13 @@ export const bundlerArgsSchema = z.object({
 })
 
 export const compatibilityArgsSchema = z.object({
-    "chain-type": z.enum(["default", "op-stack", "arbitrum", "hedera"]),
+    "chain-type": z.enum([
+        "default",
+        "op-stack",
+        "arbitrum",
+        "hedera",
+        "mantle"
+    ]),
     "legacy-transactions": z.boolean(),
     "api-version": z
         .string()

diff --git a/src/cli/config/options.ts b/src/cli/config/options.ts
@@ -219,7 +219,7 @@ export const compatibilityOptions: CliCommandOptions<ICompatibilityArgsInput> =
             description:
                 "Indicates weather the chain is a OP stack chain, arbitrum chain, or default EVM chain",
             type: "string",
-            choices: ["default", "op-stack", "arbitrum", "hedera"],
+            choices: ["default", "op-stack", "arbitrum", "hedera", "mantle"],
             default: "default"
         },
         "legacy-transactions": {

diff --git a/src/handlers/arbitrumGasPriceManager.ts b/src/handlers/arbitrumGasPriceManager.ts
@@ -0,0 +1,36 @@
+import { maxUint128 } from "viem"
+import { TimedQueue } from "../utils/timedQueue"
+
+export class ArbitrumManager {
+    private l1BaseFeeQueue: TimedQueue
+    private l2BaseFeeQueue: TimedQueue
+
+    constructor(maxQueueSize: number) {
+        const queueValidity = 15_000
+        this.l1BaseFeeQueue = new TimedQueue(maxQueueSize, queueValidity)
+        this.l2BaseFeeQueue = new TimedQueue(maxQueueSize, queueValidity)
+    }
+
+    public saveL1BaseFee(baseFee: bigint) {
+        this.l1BaseFeeQueue.saveValue(baseFee)
+    }
+
+    public saveL2BaseFee(baseFee: bigint) {
+        this.l2BaseFeeQueue.saveValue(baseFee)
+    }
+
+    public getMinL1BaseFee() {
+        let minL1BaseFee = this.l1BaseFeeQueue.getMinValue() || 1n
+        return minL1BaseFee
+    }
+
+    public getMaxL1BaseFee() {
+        let maxL1BaseFee = this.l1BaseFeeQueue.getMaxValue() || maxUint128
+        return maxL1BaseFee
+    }
+
+    public getMaxL2BaseFee() {
+        let maxL2BaseFee = this.l2BaseFeeQueue.getMaxValue() || maxUint128
+        return maxL2BaseFee
+    }
+}