From fd2f36c34c82e56a403d8b8f1474edaabf423cae Mon Sep 17 00:00:00 2001
From: SW van Heerden <swvheerden@gmail.com>
Date: Wed, 2 Mar 2022 10:22:26 +0200
Subject: [PATCH] add revalidate ui command

---
 .../src/automation/command_parser.rs                  |  2 ++
 .../tari_console_wallet/src/automation/commands.rs    | 11 +++++++++++
 2 files changed, 13 insertions(+)

diff --git a/applications/tari_console_wallet/src/automation/command_parser.rs b/applications/tari_console_wallet/src/automation/command_parser.rs
index bb9c0ad2c15..362133a99ac 100644
--- a/applications/tari_console_wallet/src/automation/command_parser.rs
+++ b/applications/tari_console_wallet/src/automation/command_parser.rs
@@ -66,6 +66,7 @@ impl Display for ParsedCommand {
             MintTokens => "mint-tokens",
             CreateInitialCheckpoint => "create-initial-checkpoint",
             CreateCommitteeDefinition => "create-committee-definition",
+            RevalidateWalletDb => "revalidate-wallet-db",
         };
 
         let args = self
@@ -143,6 +144,7 @@ pub fn parse_command(command: &str) -> Result<ParsedCommand, ParseError> {
         MintTokens => parser_builder(args).pub_key().text_array().build()?,
         CreateInitialCheckpoint => parser_builder(args).pub_key().text().build()?,
         CreateCommitteeDefinition => parser_builder(args).pub_key().pub_key_array().build()?,
+        RevalidateWalletDb => Vec::new(),
     };
 
     Ok(ParsedCommand { command, args })
diff --git a/applications/tari_console_wallet/src/automation/commands.rs b/applications/tari_console_wallet/src/automation/commands.rs
index 3c88928e2bf..43f420e3031 100644
--- a/applications/tari_console_wallet/src/automation/commands.rs
+++ b/applications/tari_console_wallet/src/automation/commands.rs
@@ -95,6 +95,7 @@ pub enum WalletCommand {
     MintTokens,
     CreateInitialCheckpoint,
     CreateCommitteeDefinition,
+    RevalidateWalletDb,
 }
 
 #[derive(Debug, EnumString, PartialEq, Clone)]
@@ -923,6 +924,16 @@ pub async fn command_runner(
                     .await?;
                 println!("Done!");
             },
+            RevalidateWalletDb => {
+                output_service
+                    .revalidate_all_outputs()
+                    .await
+                    .map_err(CommandError::OutputManagerError)?;
+                transaction_service
+                    .revalidate_all_transactions()
+                    .await
+                    .map_err(CommandError::TransactionServiceError)?;
+            },
         }
     }