Merge #3888

3888: feat: revalidate ui command r=delta1 a=SWvheerden Description --- Add a UI command to call the revalidate all outputs and transactions from a wallet. Currently, this is only supported by GRPC command. Co-authored-by: SW van Heerden <[email protected]> Co-authored-by: Philip Robinson <[email protected]> Co-authored-by: Byron Hambly <[email protected]>
tari-project · Mar 3, 2022 · a099f14 · a099f14
2 parents fc516a4 + b755e78
commit a099f14
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git 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
@@ -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)?;
+            },
         }
     }