feat: epoch cache, do not attest if not in committee or from current proposer

l1-contracts/src/core/Leonidas.sol

@@ -329,6 +329,11 @@ contract Leonidas is Ownable, TimeFns, ILeonidas {
     validatorSet.add(_validator);
   }
 
+  // Public view function for get committee at
+  function getCommitteeAt(Timestamp _ts) external view returns (address[] memory) {
+    return _getCommitteeAt(_ts);
+  }
+
   function _getCommitteeAt(Timestamp _ts) internal view returns (address[] memory) {
     Epoch epochNumber = getEpochAt(_ts);
     EpochData storage epoch = epochs[epochNumber];

yarn-project/accounts/package.json

@@ -102,4 +102,4 @@
   "engines": {
     "node": ">=18"
   }
-}
+}

yarn-project/archiver/src/archiver/archiver.ts

@@ -1,7 +1,9 @@
 import {
+  EmptyL1RollupConstants,
   type GetUnencryptedLogsResponse,
   type InBlock,
   type InboxLeaf,
+  type L1RollupConstants,
   type L1ToL2MessageSource,
   type L2Block,
   type L2BlockId,
@@ -15,6 +17,10 @@ import {
   type TxReceipt,
   type TxScopedL2Log,
   type UnencryptedL2Log,
+  getEpochNumberAtTimestamp,
+  getSlotAtTimestamp,
+  getSlotRangeForEpoch,
+  getTimestampRangeForEpoch,
 } from '@aztec/circuit-types';
 import {
   type ContractClassPublic,
@@ -62,12 +68,6 @@ import {
 import { type ArchiverDataStore, type ArchiverL1SynchPoint } from './archiver_store.js';
 import { type ArchiverConfig } from './config.js';
 import { retrieveBlockFromRollup, retrieveL1ToL2Messages } from './data_retrieval.js';
-import {
-  getEpochNumberAtTimestamp,
-  getSlotAtTimestamp,
-  getSlotRangeForEpoch,
-  getTimestampRangeForEpoch,
-} from './epoch_helpers.js';
 import { ArchiverInstrumentation } from './instrumentation.js';
 import { type DataRetrieval } from './structs/data_retrieval.js';
 import { type L1Published } from './structs/published.js';
@@ -1061,19 +1061,3 @@ class ArchiverStoreHelper
     return this.store.estimateSize();
   }
 }
-
-type L1RollupConstants = {
-  l1StartBlock: bigint;
-  l1GenesisTime: bigint;
-  slotDuration: number;
-  epochDuration: number;
-  ethereumSlotDuration: number;
-};
-
-const EmptyL1RollupConstants: L1RollupConstants = {
-  l1StartBlock: 0n,
-  l1GenesisTime: 0n,
-  epochDuration: 0,
-  slotDuration: 0,
-  ethereumSlotDuration: 0,
-};

yarn-project/archiver/src/test/mock_l2_block_source.ts

@@ -7,12 +7,11 @@ import {
   TxReceipt,
   TxStatus,
 } from '@aztec/circuit-types';
+import { getSlotRangeForEpoch } from '@aztec/circuit-types';
 import { EthAddress, type Header } from '@aztec/circuits.js';
 import { DefaultL1ContractsConfig } from '@aztec/ethereum';
 import { createDebugLogger } from '@aztec/foundation/log';
 
-import { getSlotRangeForEpoch } from '../archiver/epoch_helpers.js';
-
 /**
  * A mocked implementation of L2BlockSource to be used in tests.
  */

yarn-project/aztec-node/src/aztec-node/server.ts

@@ -170,7 +170,7 @@ export class AztecNodeService implements AztecNode {
     // start both and wait for them to sync from the block source
     await Promise.all([p2pClient.start(), worldStateSynchronizer.start()]);
 
-    const validatorClient = createValidatorClient(config, p2pClient, telemetry);
+    const validatorClient = await createValidatorClient(config, config.l1Contracts.rollupAddress, p2pClient, telemetry);
 
     // now create the sequencer
     const sequencer = config.disableValidator

yarn-project/archiver/src/archiver/epoch_helpers.ts → yarn-project/circuit-types/src/epoch-helpers/index.ts

@@ -1,4 +1,18 @@
-// REFACTOR: This file should go in a package lower in the dependency graph.
+export type L1RollupConstants = {
+  l1StartBlock: bigint;
+  l1GenesisTime: bigint;
+  slotDuration: number;
+  epochDuration: number;
+  ethereumSlotDuration: number;
+};
+
+export const EmptyL1RollupConstants: L1RollupConstants = {
+  l1StartBlock: 0n,
+  l1GenesisTime: 0n,
+  epochDuration: 0,
+  slotDuration: 0,
+  ethereumSlotDuration: 0,
+};
 
 export type EpochConstants = {
   l1GenesisBlock: bigint;

yarn-project/circuit-types/src/index.ts

@@ -25,3 +25,4 @@ export * from './tx_execution_request.js';
 export * from './in_block.js';
 export * from './nullifier_with_block_source.js';
 export * from './proving_error.js';
+export * from './epoch-helpers/index.js';

yarn-project/circuits.js/package.json

@@ -99,4 +99,4 @@
       ]
     ]
   }
-}
+}

yarn-project/epoch-cache/.eslintrc.cjs

@@ -0,0 +1 @@
+module.exports = require('@aztec/foundation/eslint');

yarn-project/epoch-cache/READMD.md

@@ -0,0 +1,3 @@
+## Epoch Cache
+
+Stores the current validator set.

yarn-project/epoch-cache/package.json

@@ -0,0 +1,93 @@
+{
+  "name": "@aztec/epoch-cache",
+  "version": "0.1.0",
+  "type": "module",
+  "exports": {
+    ".": "./dest/index.js",
+    "./test": "./dest/test/index.js",
+    "./contracts": "./dest/contracts/index.js"
+  },
+  "typedocOptions": {
+    "entryPoints": [
+      "./src/index.ts"
+    ],
+    "name": "Epoch Cache",
+    "tsconfig": "./tsconfig.json"
+  },
+  "scripts": {
+    "build": "yarn clean && tsc -b",
+    "build:dev": "tsc -b --watch",
+    "clean": "rm -rf ./dest .tsbuildinfo",
+    "formatting": "run -T prettier --check ./src && run -T eslint ./src",
+    "formatting:fix": "run -T eslint --fix ./src && run -T prettier -w ./src",
+    "start:dev": "tsc-watch -p tsconfig.json --onSuccess 'yarn start'",
+    "start": "node ./dest/index.js",
+    "test": "NODE_NO_WARNINGS=1 node --experimental-vm-modules ../node_modules/.bin/jest --passWithNoTests"
+  },
+  "inherits": [
+    "../package.common.json"
+  ],
+  "dependencies": {
+    "@aztec/circuit-types": "workspace:*",
+    "@aztec/ethereum": "workspace:*",
+    "@aztec/foundation": "workspace:^",
+    "@aztec/l1-artifacts": "workspace:^",
+    "@viem/anvil": "^0.0.10",
+    "dotenv": "^16.0.3",
+    "get-port": "^7.1.0",
+    "tslib": "^2.4.0",
+    "viem": "^2.7.15",
+    "zod": "^3.23.8"
+  },
+  "devDependencies": {
+    "@jest/globals": "^29.5.0",
+    "@types/jest": "^29.5.0",
+    "@types/node": "^18.14.6",
+    "jest": "^29.5.0",
+    "ts-node": "^10.9.1",
+    "typescript": "^5.0.4"
+  },
+  "files": [
+    "dest",
+    "src",
+    "!*.test.*"
+  ],
+  "types": "./dest/index.d.ts",
+  "jest": {
+    "moduleNameMapper": {
+      "^(\\.{1,2}/.*)\\.[cm]?js$": "$1"
+    },
+    "testRegex": "./src/.*\\.test\\.(js|mjs|ts)$",
+    "rootDir": "./src",
+    "transform": {
+      "^.+\\.tsx?$": [
+        "@swc/jest",
+        {
+          "jsc": {
+            "parser": {
+              "syntax": "typescript",
+              "decorators": true
+            },
+            "transform": {
+              "decoratorVersion": "2022-03"
+            }
+          }
+        }
+      ]
+    },
+    "extensionsToTreatAsEsm": [
+      ".ts"
+    ],
+    "reporters": [
+      [
+        "default",
+        {
+          "summaryThreshold": 9999
+        }
+      ]
+    ]
+  },
+  "engines": {
+    "node": ">=18"
+  }
+}

yarn-project/epoch-cache/src/epoch_cache.test.ts

@@ -0,0 +1,10 @@
+// import { EpochCache } from "./epoch_cache.js";
+
+describe('EpochCache', () => {
+  it('Should return the current validator', () => {
+    expect(true).toBe(true);
+    // const epochCache = await EpochCache.create(EthAddress.random());
+    // const currentValidator = await epochCache.getCurrentValidator();
+    // expect(currentValidator).toBeDefined();
+  });
+});

yarn-project/epoch-cache/src/epoch_cache.ts

@@ -0,0 +1,104 @@
+import {
+  EmptyL1RollupConstants,
+  type L1RollupConstants,
+  getEpochNumberAtTimestamp,
+  getSlotAtTimestamp,
+} from '@aztec/circuit-types';
+import {
+  RollupContract,
+  createEthereumChain,
+  getL1ContractsConfigEnvVars,
+  getL1ReaderConfigFromEnv,
+} from '@aztec/ethereum';
+import { EthAddress } from '@aztec/foundation/eth-address';
+
+import { createPublicClient, http } from 'viem';
+
+type EpochAndSlot = {
+  epoch: bigint;
+  slot: bigint;
+  ts: bigint;
+};
+
+export class EpochCache {
+  private validators: Map<EthAddress, boolean>;
+  private currentEpoch: bigint;
+
+  constructor(
+    private rollup: RollupContract,
+    private readonly l1constants: L1RollupConstants = EmptyL1RollupConstants,
+  ) {
+    this.validators = new Map<EthAddress, boolean>();
+
+    this.currentEpoch = getEpochNumberAtTimestamp(BigInt(Math.floor(Date.now() / 1000)), this.l1constants);
+  }
+
+  // TODO: cleanup and merge rollup getters with l1 createAndSync in the archiver
+  static async create(rollupAddress: EthAddress) {
+    const l1ReaderConfig = getL1ReaderConfigFromEnv();
+    const l1constants = getL1ContractsConfigEnvVars();
+
+    const chain = createEthereumChain(l1ReaderConfig.l1RpcUrl, l1ReaderConfig.l1ChainId);
+    const publicClient = createPublicClient({
+      chain: chain.chainInfo,
+      transport: http(chain.rpcUrl),
+      pollingInterval: l1ReaderConfig.viemPollingIntervalMS,
+    });
+
+    const rollup = new RollupContract(publicClient, rollupAddress.toString());
+    const [l1StartBlock, l1GenesisTime] = await Promise.all([
+      rollup.getL1StartBlock(),
+      rollup.getL1GenesisTime(),
+    ] as const);
+
+    const l1RollupConstants: L1RollupConstants = {
+      l1StartBlock,
+      l1GenesisTime,
+      slotDuration: l1constants.aztecSlotDuration,
+      epochDuration: l1constants.aztecEpochDuration,
+      ethereumSlotDuration: l1constants.ethereumSlotDuration,
+    };
+
+    return new EpochCache(rollup, l1RollupConstants);
+  }
+
+  getEpochAndSlotNow(): EpochAndSlot {
+    const now = BigInt(Math.floor(Date.now() / 1000));
+    return this.getEpochAndSlotAtTimestamp(now);
+  }
+
+  getEpochAndSlotAtTimestamp(ts: bigint): EpochAndSlot {
+    return {
+      epoch: getEpochNumberAtTimestamp(ts, this.l1constants),
+      slot: getSlotAtTimestamp(ts, this.l1constants),
+      ts,
+    };
+  }
+
+  async getValidatorSet(): Promise<EthAddress[]> {
+    // If the current epoch has changed, then we need to make a request to update the validator set
+    const { epoch: currentEpoch, ts } = this.getEpochAndSlotNow();
+
+    if (currentEpoch !== this.currentEpoch) {
+      this.currentEpoch = currentEpoch;
+      const validatorSet = await this.rollup.getCommitteeAt(ts);
+      this.validators = new Map(validatorSet.map((v: `0x${string}`) => [EthAddress.fromString(v), true]));
+    }
+
+    return Array.from(this.validators.keys());
+  }
+
+  async getCurrentValidator(): Promise<EthAddress> {
+    // Validators are sorted by their index in the committee, and getValidatorSet will cache
+    // TODO: should we get the timestamp from the underlying l1 node?
+    const { slot: currentSlot } = this.getEpochAndSlotNow();
+    const validators = await this.getValidatorSet();
+
+    return validators[Number(currentSlot) % validators.length];
+  }
+
+  async isInCommittee(validator: EthAddress): Promise<boolean> {
+    const validators = await this.getValidatorSet();
+    return validators.includes(validator);
+  }
+}

yarn-project/epoch-cache/src/index.ts

@@ -0,0 +1 @@
+export * from './epoch_cache.js';

yarn-project/epoch-cache/tsconfig.json

@@ -0,0 +1,23 @@
+{
+  "extends": "..",
+  "compilerOptions": {
+    "outDir": "dest",
+    "rootDir": "src",
+    "tsBuildInfoFile": ".tsbuildinfo"
+  },
+  "include": ["src"],
+  "references": [
+    {
+      "path": "../circuit-types"
+    },
+    {
+      "path": "../ethereum"
+    },
+    {
+      "path": "../foundation"
+    },
+    {
+      "path": "../l1-artifacts"
+    }
+  ]
+}

yarn-project/ethereum/src/contracts/rollup.ts

@@ -44,6 +44,10 @@ export class RollupContract {
     return this.rollup.read.getCurrentSlot();
   }
 
+  async getCommitteeAt(timestamp: bigint) {
+    return this.rollup.read.getCommitteeAt([timestamp]);
+  }
+
   async getEpochNumber(blockNumber?: bigint) {
     blockNumber ??= await this.getBlockNumber();
     return this.rollup.read.getEpochForBlock([BigInt(blockNumber)]);

yarn-project/ethereum/src/index.ts

@@ -6,3 +6,4 @@ export * from './ethereum_chain.js';
 export * from './utils.js';
 export * from './config.js';
 export * from './types.js';
+export * from './contracts/index.js';