WIP: implement local configuration for mithril network configuration and adapt dependencies

Author: turmelclem

mithril-aggregator/src/dependency_injection/builder/enablers/mithril_network_configuration.rs

@@ -1,10 +1,19 @@
-use std::collections::BTreeSet;
+use std::{collections::BTreeSet, sync::Arc};
 
-use crate::entities::AggregatorEpochSettings;
+use crate::{
+    ConfigurationSource,
+    dependency_injection::{DependenciesBuilder, Result},
+    entities::AggregatorEpochSettings,
+    get_dependency,
+};
 use async_trait::async_trait;
-use mithril_common::{StdResult, entities::SignedEntityTypeDiscriminants, test::double::Dummy};
+use mithril_common::{
+    StdResult,
+    entities::{Epoch, SignedEntityTypeDiscriminants},
+};
 use mithril_protocol_config::{
-    interface::MithrilNetworkConfigurationProvider, model::MithrilNetworkConfiguration,
+    interface::MithrilNetworkConfigurationProvider,
+    model::{MithrilNetworkConfiguration, SignedEntityTypeConfiguration},
 };
 
 pub struct LocalMithrilNetworkConfigurationProvider {
@@ -13,10 +22,12 @@ pub struct LocalMithrilNetworkConfigurationProvider {
 }
 
 impl LocalMithrilNetworkConfigurationProvider {
-    pub fn new(
-        epoch_settings: AggregatorEpochSettings,
-        allowed_discriminants: BTreeSet<SignedEntityTypeDiscriminants>,
-    ) -> Self {
+    pub fn new(configuration: Arc<dyn ConfigurationSource>) -> Self {
+        let epoch_settings = configuration.get_epoch_settings_configuration();
+        let allowed_discriminants = configuration
+            .compute_allowed_signed_entity_types_discriminants()
+            .expect("Failed to compute allowed signed entity types discriminants");
+
         Self {
             epoch_settings,
             allowed_discriminants,
@@ -26,7 +37,42 @@ impl LocalMithrilNetworkConfigurationProvider {
 
 #[async_trait]
 impl MithrilNetworkConfigurationProvider for LocalMithrilNetworkConfigurationProvider {
-    async fn get(&self) -> StdResult<MithrilNetworkConfiguration> {
-        Ok(MithrilNetworkConfiguration::dummy())
+    async fn get_network_configuration(&self) -> StdResult<MithrilNetworkConfiguration> {
+        let epoch = Epoch(42); // TODO implement proper epoch retrieval
+        let signer_registration_protocol_parameters =
+            self.epoch_settings.protocol_parameters.clone();
+        let available_signed_entity_types = self.allowed_discriminants.clone();
+        let signed_entity_types_config = SignedEntityTypeConfiguration {
+            cardano_transactions: Some(
+                self.epoch_settings.cardano_transactions_signing_config.clone(),
+            ),
+        };
+
+        let config = MithrilNetworkConfiguration {
+            epoch, //TODO implement
+            signer_registration_protocol_parameters,
+            available_signed_entity_types,
+            signed_entity_types_config,
+        };
+        Ok(config)
+    }
+}
+
+impl DependenciesBuilder {
+    async fn build_mithril_network_configuration_provider(
+        &mut self,
+    ) -> Result<Arc<dyn MithrilNetworkConfigurationProvider>> {
+        let network_configuration_provider = Arc::new(
+            LocalMithrilNetworkConfigurationProvider::new(self.configuration.clone()),
+        );
+
+        Ok(network_configuration_provider)
+    }
+
+    /// [MithrilNetworkConfigurationProvider][mithril_protocol_config::interface::MithrilNetworkConfigurationProvider] service
+    pub async fn get_mithril_network_configuration_provider(
+        &mut self,
+    ) -> Result<Arc<dyn MithrilNetworkConfigurationProvider>> {
+        get_dependency!(self.mithril_network_configuration_provider)
     }
 }

mithril-aggregator/src/dependency_injection/builder/mod.rs

@@ -35,6 +35,7 @@ use mithril_persistence::{
     database::repository::CardanoTransactionRepository,
     sqlite::{SqliteConnection, SqliteConnectionPool},
 };
+use mithril_protocol_config::interface::MithrilNetworkConfigurationProvider;
 use mithril_signed_entity_lock::SignedEntityTypeLock;
 use mithril_ticker::TickerService;
 
@@ -251,6 +252,10 @@ pub struct DependenciesBuilder {
     /// Epoch service.
     pub epoch_service: Option<EpochServiceWrapper>,
 
+    /// Mithril network configuration provider
+    pub mithril_network_configuration_provider:
+        Option<Arc<dyn MithrilNetworkConfigurationProvider>>,
+
     /// Signed Entity storer
     pub signed_entity_storer: Option<Arc<dyn SignedEntityStorer>>,
 
@@ -336,6 +341,7 @@ impl DependenciesBuilder {
             signed_entity_service: None,
             certifier_service: None,
             epoch_service: None,
+            mithril_network_configuration_provider: None,
             signed_entity_storer: None,
             message_service: None,
             prover_service: None,
@@ -392,6 +398,9 @@ impl DependenciesBuilder {
             signed_entity_service: self.get_signed_entity_service().await?,
             certifier_service: self.get_certifier_service().await?,
             epoch_service: self.get_epoch_service().await?,
+            mithril_network_configuration_provider: self
+                .get_mithril_network_configuration_provider()
+                .await?,
             ticker_service: self.get_ticker_service().await?,
             signed_entity_storer: self.get_signed_entity_storer().await?,
             signer_getter: self.get_signer_store().await?,

mithril-aggregator/src/dependency_injection/containers/serve.rs

@@ -1,3 +1,4 @@
+use mithril_protocol_config::interface::MithrilNetworkConfigurationProvider;
 use slog::Logger;
 use std::sync::Arc;
 use tokio::sync::RwLock;
@@ -100,6 +101,9 @@ pub struct ServeCommandDependenciesContainer {
     /// Epoch service
     pub(crate) epoch_service: EpochServiceWrapper,
 
+    /// Mithril network configuration provider
+    pub(crate) mithril_network_configuration_provider: Arc<dyn MithrilNetworkConfigurationProvider>,
+
     /// Ticker Service
     pub(crate) ticker_service: Arc<dyn TickerService>,