Added FeeStrategies

Author: rpanic

packages/sequencer/src/protocol/baselayer/MinaBaseLayer.ts

@@ -1,5 +1,6 @@
 import { DependencyFactory } from "@proto-kit/common";
 import { Mina } from "o1js";
+import { match } from "ts-pattern";
 
 import { MinaIncomingMessageAdapter } from "../../settlement/messages/MinaIncomingMessageAdapter";
 import { SequencerModule } from "../../sequencer/builder/SequencerModule";
@@ -9,11 +10,21 @@ import { WithdrawalQueue } from "../../settlement/messages/WithdrawalQueue";
 import { BaseLayer } from "./BaseLayer";
 
 export interface MinaBaseLayerConfig {
-  network: {
-    local: boolean;
-    graphql?: string;
-    archive?: string;
-  };
+  network:
+    | {
+        type: "local";
+      }
+    | {
+        type: "lightnet";
+        graphql: string;
+        archive: string;
+        accountManager?: string;
+      }
+    | {
+        type: "remote";
+        graphql: string;
+        archive: string;
+      };
 }
 
 export class MinaBaseLayer
@@ -22,6 +33,8 @@ export class MinaBaseLayer
 {
   public network?: Parameters<typeof Mina.setActiveInstance>[0];
 
+  public originalNetwork?: Parameters<typeof Mina.setActiveInstance>[0];
+
   public dependencies() {
     return {
       IncomingMessageAdapter: {
@@ -39,27 +52,36 @@ export class MinaBaseLayer
   }
 
   public isLocalBlockChain(): boolean {
-    return this.config.network.local;
+    return this.config.network.type === "local";
   }
 
   public async start(): Promise<void> {
     const { network } = this.config;
 
-    if (
-      !network.local &&
-      (network.graphql === undefined || network.archive === undefined)
-    ) {
-      throw new Error(
-        "The API endpoints have to be defined, if the network is remote"
-      );
-    }
+    this.originalNetwork = Mina.activeInstance;
 
-    const Network = this.config.network.local
-      ? await Mina.LocalBlockchain({ proofsEnabled: false })
-      : Mina.Network({
-          mina: network.graphql!,
-          archive: network.archive!,
+    const Network = await match(network)
+      .with(
+        { type: "local" },
+        async () => await Mina.LocalBlockchain({ proofsEnabled: false })
+      )
+      .with({ type: "lightnet" }, async (lightnet) => {
+        const net = Mina.Network({
+          mina: lightnet.graphql,
+          archive: lightnet.archive,
+          lightnetAccountManager: lightnet.accountManager,
         });
+        net.proofsEnabled = false;
+        return net;
+      })
+      .with({ type: "remote" }, async (remote) =>
+        Mina.Network({
+          mina: remote.graphql,
+          archive: remote.archive,
+        })
+      )
+      .exhaustive();
+
     Mina.setActiveInstance(Network);
     this.network = Network;
   }

packages/sequencer/src/protocol/baselayer/fees/ConstantFeeStrategy.ts

@@ -0,0 +1,26 @@
+import {
+  sequencerModule,
+  SequencerModule,
+} from "../../../sequencer/builder/SequencerModule";
+import { noop } from "@proto-kit/common";
+import { FeeStrategy } from "./FeeStrategy";
+
+export type ConstantFeeStrategyConfig = {
+  fee?: number;
+};
+
+const DEFAULT_FEE = 0.1 * 1e9;
+
+@sequencerModule()
+export class ConstantFeeStrategy
+  extends SequencerModule<ConstantFeeStrategyConfig>
+  implements FeeStrategy
+{
+  getFee(): number {
+    return this.config.fee ?? DEFAULT_FEE;
+  }
+
+  public async start() {
+    noop();
+  }
+}

packages/sequencer/src/protocol/baselayer/fees/FeeStrategy.ts

@@ -0,0 +1,3 @@
+export interface FeeStrategy {
+  getFee(): number;
+}