Restaking

src/commands/create_keys.py

@@ -4,7 +4,7 @@
 from pathlib import Path
 
 import click
-from eth_typing import HexAddress
+from eth_typing import ChecksumAddress, HexAddress
 
 from src.common.credentials import Credential, CredentialManager
 from src.common.password import generate_password, get_or_create_password_file
@@ -46,6 +46,12 @@
     type=str,
     callback=validate_eth_address,
 )
+@click.option(
+    '--withdrawal-address',
+    help='The address of the vaultk.',
+    type=str,
+    callback=validate_eth_address,
+)
 @click.option(
     '--pool-size',
     help='Number of processes in a pool.',
@@ -59,6 +65,7 @@ def create_keys(
     count: int,
     vault: HexAddress,
     data_dir: str,
+    withdrawal_address: ChecksumAddress | None,
     per_keystore_password: bool,
     pool_size: int | None,
 ) -> None:
@@ -75,6 +82,7 @@ def create_keys(
         mnemonic=mnemonic,
         count=count,
         start_index=config.mnemonic_next_index,
+        withdrawal_address=withdrawal_address,
         pool_size=pool_size,
     )
     deposit_data = _export_deposit_data_json(

src/commands/start_base.py

@@ -11,7 +11,10 @@
 from src.common.metrics import MetricsTask, metrics_server
 from src.common.startup_check import startup_checks
 from src.common.utils import get_build_version
+from src.common.vault import Vault
 from src.config.settings import settings
+from src.eigenlayer.database import CheckpointsCrud
+from src.eigenlayer.tasks import EigenlayerValidatorsTask, EigenlayerWithdrawalsTask
 from src.exits.tasks import ExitSignatureTask
 from src.harvest.tasks import HarvestTask
 from src.validators.database import NetworkValidatorCrud
@@ -35,6 +38,7 @@ async def start_base() -> None:
         await startup_checks()
 
     NetworkValidatorCrud().setup()
+    CheckpointsCrud().setup()
 
     # load network validators from ipfs dump
     await load_genesis_validators()
@@ -88,6 +92,9 @@ async def start_base() -> None:
         ]
         if settings.harvest_vault:
             tasks.append(HarvestTask().run(interrupt_handler))
+        if await Vault().is_restaking():
+            tasks.append(EigenlayerValidatorsTask().run(interrupt_handler))
+            tasks.append(EigenlayerWithdrawalsTask().run(interrupt_handler))
 
         await asyncio.gather(*tasks)