diff --git a/applications/tari_console_wallet/src/automation/command_parser.rs b/applications/tari_console_wallet/src/automation/command_parser.rs index bb9c0ad2c1..362133a99a 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 { 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 3c88928e2b..43f420e303 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)?; + }, } }