Wire Mithril Stake Distribution signable builder in signer

Author: jpraynaud

mithril-common/src/signable_builder/mithril_stake_distribution.rs

@@ -6,7 +6,7 @@ use crate::{
     StdResult,
 };
 
-/// A [MithrilStakeDistributionSignable] builder
+/// A [MithrilStakeDistributionSignableBuilder] builder
 pub struct MithrilStakeDistributionSignableBuilder {}
 
 impl MithrilStakeDistributionSignableBuilder {

mithril-signer/src/runtime/runner.rs

@@ -485,7 +485,9 @@ mod tests {
             adapters::{EraReaderAdapterType, EraReaderBootstrapAdapter},
             EraChecker, EraReader,
         },
-        signable_builder::CardanoImmutableFilesFullSignableBuilder,
+        signable_builder::{
+            CardanoImmutableFilesFullSignableBuilder, MithrilStakeDistributionSignableBuilder,
+        },
         store::{
             adapter::{DumbStoreAdapter, MemoryAdapter},
             StakeStore, StakeStorer,
@@ -534,10 +536,14 @@ mod tests {
 
         let api_version_provider = Arc::new(APIVersionProvider::new(era_checker.clone()));
         let digester = Arc::new(DumbImmutableDigester::new(DIGESTER_RESULT, true));
-        let signable_builder =
+        let immutable_signable_builder =
             CardanoImmutableFilesFullSignableBuilder::new(digester.clone(), slog_scope::logger());
-
-        let signable_builder_service = Arc::new(SignableBuilderService::new(signable_builder));
+        let mithril_stake_distribution_signable_builder =
+            MithrilStakeDistributionSignableBuilder::new();
+        let signable_builder_service = Arc::new(SignableBuilderService::new(
+            immutable_signable_builder,
+            mithril_stake_distribution_signable_builder,
+        ));
 
         SignerServices {
             stake_store: Arc::new(StakeStore::new(Box::new(DumbStoreAdapter::new()), None)),

mithril-signer/src/runtime/signer_services.rs

@@ -14,7 +14,9 @@ use mithril_common::{
     },
     digesters::{CardanoImmutableDigester, ImmutableDigester, ImmutableFileSystemObserver},
     era::{EraChecker, EraReader},
-    signable_builder::CardanoImmutableFilesFullSignableBuilder,
+    signable_builder::{
+        CardanoImmutableFilesFullSignableBuilder, MithrilStakeDistributionSignableBuilder,
+    },
     store::{adapter::SQLiteAdapter, StakeStore},
     BeaconProvider, BeaconProviderImpl, StdError,
 };
@@ -201,10 +203,14 @@ impl<'a> ServiceBuilder for ProductionServiceBuilder<'a> {
             api_version_provider.clone(),
         ));
 
-        let dummy_signable_builder =
+        let immutable_snapshot_builder =
             CardanoImmutableFilesFullSignableBuilder::new(digester.clone(), slog_scope::logger());
-        let signable_builder_service =
-            Arc::new(SignableBuilderService::new(dummy_signable_builder));
+        let mithril_stake_distribution_signable_builder =
+            MithrilStakeDistributionSignableBuilder::new();
+        let signable_builder_service = Arc::new(SignableBuilderService::new(
+            immutable_snapshot_builder,
+            mithril_stake_distribution_signable_builder,
+        ));
 
         let services = SignerServices {
             beacon_provider,

mithril-signer/src/signable_builder/signable_builder_service.rs

@@ -2,21 +2,28 @@ use std::sync::Arc;
 
 use mithril_common::{
     entities::SignedEntityType,
-    signable_builder::{CardanoImmutableFilesFullSignableBuilder, Signable, SignableBuilder},
+    signable_builder::{
+        CardanoImmutableFilesFullSignableBuilder, MithrilStakeDistributionSignableBuilder,
+        Signable, SignableBuilder,
+    },
     StdResult,
 };
 
 /// SignableBuilder Service
-// TODO: temporary implementation
 pub struct SignableBuilderService {
     immutable_signable_builder: CardanoImmutableFilesFullSignableBuilder,
+    mithril_stake_distribution_builder: MithrilStakeDistributionSignableBuilder,
 }
 
 impl SignableBuilderService {
     /// SignableBuilderService factory
-    pub fn new(immutable_signable_builder: CardanoImmutableFilesFullSignableBuilder) -> Self {
+    pub fn new(
+        immutable_signable_builder: CardanoImmutableFilesFullSignableBuilder,
+        mithril_stake_distribution_builder: MithrilStakeDistributionSignableBuilder,
+    ) -> Self {
         Self {
             immutable_signable_builder,
+            mithril_stake_distribution_builder,
         }
     }
 }
@@ -33,6 +40,11 @@ impl SignableBuilderService {
                     .compute_signable(beacon)
                     .await?,
             ),
+            SignedEntityType::MithrilStakeDistribution(epoch) => Arc::new(
+                self.mithril_stake_distribution_builder
+                    .compute_signable(epoch)
+                    .await?,
+            ),
             _ => todo!(),
         };
 

mithril-signer/tests/test_extensions/state_machine_tester.rs

@@ -7,7 +7,9 @@ use mithril_common::era::{
     EraChecker, EraReader,
 };
 use mithril_common::era::{EraMarker, SupportedEra};
-use mithril_common::signable_builder::CardanoImmutableFilesFullSignableBuilder;
+use mithril_common::signable_builder::{
+    CardanoImmutableFilesFullSignableBuilder, MithrilStakeDistributionSignableBuilder,
+};
 use mithril_common::BeaconProvider;
 use slog::Drain;
 use slog_scope::debug;
@@ -152,7 +154,12 @@ impl StateMachineTester {
 
         let signable_builder =
             CardanoImmutableFilesFullSignableBuilder::new(digester.clone(), slog_scope::logger());
-        let signable_builder_service = Arc::new(SignableBuilderService::new(signable_builder));
+        let mithril_stake_distribution_signable_builder =
+            MithrilStakeDistributionSignableBuilder::new();
+        let signable_builder_service = Arc::new(SignableBuilderService::new(
+            signable_builder,
+            mithril_stake_distribution_signable_builder,
+        ));
 
         let services = SignerServices {
             certificate_handler: certificate_handler.clone(),