Add anyhow context for signable_builder

Author: Damien LACHAUME / PALO-IT

mithril-aggregator/src/runtime/runner.rs

@@ -363,7 +363,8 @@ impl AggregatorRunnerTrait for AggregatorRunner {
             .dependencies
             .signable_builder_service
             .compute_protocol_message(signed_entity_type.to_owned())
-            .await?;
+            .await
+            .with_context(|| format!("Runner can not compute protocol message for signed entity type: '{signed_entity_type}'"))?;
 
         let multi_signer = self.dependencies.multi_signer.write().await;
         protocol_message.set_message_part(

mithril-common/src/signable_builder/signable_builder_service.rs

@@ -1,3 +1,4 @@
+use anyhow::Context;
 use async_trait::async_trait;
 use std::sync::Arc;
 
@@ -47,16 +48,20 @@ impl SignableBuilderService for MithrilSignableBuilderService {
         signed_entity_type: SignedEntityType,
     ) -> StdResult<ProtocolMessage> {
         let protocol_message = match signed_entity_type {
-            SignedEntityType::MithrilStakeDistribution(e) => {
-                self.mithril_stake_distribution_builder
-                    .compute_protocol_message(e)
-                    .await?
-            }
-            SignedEntityType::CardanoImmutableFilesFull(beacon) => {
-                self.immutable_signable_builder
-                    .compute_protocol_message(beacon)
-                    .await?
-            }
+            SignedEntityType::MithrilStakeDistribution(e) => self
+                .mithril_stake_distribution_builder
+                .compute_protocol_message(e)
+                .await
+                .with_context(|| format!(
+                    "Signable builder service can not compute protocol message with epoch: '{e}'"
+                ))?,
+            SignedEntityType::CardanoImmutableFilesFull(beacon) => self
+                .immutable_signable_builder
+                .compute_protocol_message(beacon.clone())
+                .await
+                .with_context(|| format!(
+                    "Signable builder service can not compute protocol message with beacon: '{beacon}'"
+                ))?,
             SignedEntityType::CardanoStakeDistribution(_) => todo!(),
         };
 

mithril-signer/src/runtime/runner.rs

@@ -332,7 +332,8 @@ impl Runner for SignerRunner {
             .services
             .signable_builder_service
             .compute_protocol_message(signed_entity_type.to_owned())
-            .await?;
+            .await
+            .with_context(|| format!("Runner can not compute protocol message for signed entity type: '{signed_entity_type}'"))?;
 
         // 2 set the next signers keys and stakes in the message
         let epoch = signed_entity_type.get_epoch();