add back light client to stake table contract

EspressoSystems · Mar 8, 2025 · 9a1adfa · 9a1adfa
1 parent ed50a96
commit 9a1adfa
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 6 deletions.
diff --git a/contracts/src/StakeTable.sol b/contracts/src/StakeTable.sol
@@ -47,6 +47,9 @@ contract StakeTable is AbstractStakeTable, Ownable, InitializedAt {
     /// The commission is invalid
     error InvalidCommission();
 
+    /// Error raised when the light client address is invalid
+    error InvalidAddress();
+
     struct Validator {
         bool isRegistered;
         ValidatorStatus status;
@@ -63,6 +66,9 @@ contract StakeTable is AbstractStakeTable, Ownable, InitializedAt {
         Exited
     }
 
+    /// Reference to the light client contract.
+    LightClient public lightClient;
+
     /// Currently active validators
     mapping(address validator => Validator) public validators;
 
@@ -96,11 +102,16 @@ contract StakeTable is AbstractStakeTable, Ownable, InitializedAt {
     address public admin;
 
     /// TODO change constructor to initialize function when we make the contract upgradeable
-    constructor(address _tokenAddress, uint256 _exitEscrowPeriod, address _initialOwner)
-        Ownable(_initialOwner)
-        InitializedAt()
-    {
+    constructor(
+        address _tokenAddress,
+        address _lightClientAddress,
+        uint256 _exitEscrowPeriod,
+        address _initialOwner
+    ) Ownable(_initialOwner) InitializedAt() {
+        // TODO ensure address not zero
         tokenAddress = _tokenAddress;
+        // TODO ensure address not zero
+        lightClient = LightClient(_lightClientAddress);
         exitEscrowPeriod = _exitEscrowPeriod;
         admin = msg.sender;
     }

diff --git a/contracts/test/mocks/StakeTableMock.sol b/contracts/test/mocks/StakeTableMock.sol
@@ -4,8 +4,8 @@ pragma solidity ^0.8.0;
 import { StakeTable } from "../../src/StakeTable.sol";
 
 contract StakeTableMock is StakeTable {
-    constructor(address token, uint256 escrowPeriod, address initialOwner)
-        StakeTable(token, escrowPeriod, initialOwner)
+    constructor(address token, address lightClient, uint256 escrowPeriod, address initialOwner)
+        StakeTable(token, lightClient, escrowPeriod, initialOwner)
     // solhint-disable-next-line no-empty-blocks
     { }
 }