WIP: adapt call of compute_cardano_database_message to handle the new error

Author: turmelclem

mithril-client-cli/src/commands/cardano_db/verify.rs

@@ -4,10 +4,10 @@ use std::{
     sync::Arc,
 };
 
-use anyhow::Context;
+use anyhow::{Context, anyhow};
 use chrono::Utc;
 use clap::Parser;
-use mithril_client::MithrilResult;
+use mithril_client::{ComputeCardanoDatabaseMessageError, MithrilResult};
 
 use crate::{
     CommandContext,
@@ -122,26 +122,38 @@ impl CardanoDbVerifyCommand {
             db_dir,
             &verified_digests,
         )
-        .await?;
-
-        shared_steps::verify_message_matches_certificate(
-            &context.logger().clone(),
-            4,
-            &progress_printer,
-            &certificate,
-            &message,
-            &cardano_db_message,
-            db_dir,
-        )
-        .await?;
-
-        Self::log_verified_information(
-            db_dir,
-            &cardano_db_message.hash,
-            context.is_json_output_enabled(),
-        )?;
-
-        Ok(())
+        .await;
+
+        //TODO due to wait spinner, we need to cast error to access the inner error's lists, could it be improved to avoid downcasting?
+        match message {
+            Err(e) => match e.downcast_ref::<ComputeCardanoDatabaseMessageError>() {
+                Some(ComputeCardanoDatabaseMessageError::ImmutableFilesVerification(lists)) => {
+                    // let missing_files = lists.missing;
+                    Ok(())
+                }
+                _ => Err(anyhow!(e)),
+            },
+            Ok(message) => {
+                shared_steps::verify_message_matches_certificate(
+                    &context.logger().clone(),
+                    4,
+                    &progress_printer,
+                    &certificate,
+                    &message,
+                    &cardano_db_message,
+                    db_dir,
+                )
+                .await?;
+
+                Self::log_verified_information(
+                    db_dir,
+                    &cardano_db_message.hash,
+                    context.is_json_output_enabled(),
+                )?;
+
+                Ok(())
+            }
+        }
     }
 
     fn log_verified_information(

mithril-client-cli/src/utils/cardano_db.rs

@@ -4,7 +4,7 @@ use indicatif::{MultiProgress, ProgressBar};
 use std::time::Duration;
 
 use super::CardanoDbDownloadCheckerError;
-use mithril_client::{MithrilError, MithrilResult};
+use mithril_client::{ComputeCardanoDatabaseMessageError, MithrilError, MithrilResult};
 
 /// Utility functions for to the CardanoDb commands
 pub struct CardanoDbUtils;
@@ -22,10 +22,13 @@ impl CardanoDbUtils {
     }
 
     /// Display a spinner while waiting for the result of a future
-    pub async fn wait_spinner<T>(
+    pub async fn wait_spinner<T, E>(
         progress_bar: &MultiProgress,
-        future: impl Future<Output = MithrilResult<T>>,
-    ) -> MithrilResult<T> {
+        future: impl Future<Output = Result<T, E>>,
+    ) -> MithrilResult<T>
+    where
+        MithrilError: From<E>,
+    {
         let pb = progress_bar.add(ProgressBar::new_spinner());
         let spinner = async move {
             loop {
@@ -36,7 +39,7 @@ impl CardanoDbUtils {
 
         tokio::select! {
             _ = spinner => Err(anyhow!("timeout")),
-            res = future => res,
+            res = future => res.map_err(Into::into),
         }
     }