From ba45cc56cf74ba8ef70889b3a3c7db8dd115572f Mon Sep 17 00:00:00 2001
From: Tan Chee Keong <tanck@sigmaprime.io>
Date: Tue, 17 Dec 2024 15:58:43 +0800
Subject: [PATCH] Add delete all validators

---
 validator_manager/src/delete_validators.rs | 24 ++++++++++++++++------
 validator_manager/src/exit_validators.rs   |  5 ++++-
 2 files changed, 22 insertions(+), 7 deletions(-)

diff --git a/validator_manager/src/delete_validators.rs b/validator_manager/src/delete_validators.rs
index a2d6c062fa2..98765f27ea0 100644
--- a/validator_manager/src/delete_validators.rs
+++ b/validator_manager/src/delete_validators.rs
@@ -45,7 +45,10 @@ pub fn cli_app() -> Command {
             Arg::new(VALIDATOR_FLAG)
                 .long(VALIDATOR_FLAG)
                 .value_name("STRING")
-                .help("Comma-separated list of validators (pubkey) that will be deleted.")
+                .help(
+                    "Comma-separated list of validators (pubkey) that will be deleted. \
+                 To delete all validators, use the keyword \"all\".",
+                )
                 .action(ArgAction::Set)
                 .required(true)
                 .display_order(0),
@@ -64,10 +67,14 @@ impl DeleteConfig {
         let validators_to_delete_str =
             clap_utils::parse_required::<String>(matches, VALIDATOR_FLAG)?;
 
-        let validators_to_delete = validators_to_delete_str
-            .split(',')
-            .map(|s| s.trim().parse())
-            .collect::<Result<Vec<PublicKeyBytes>, _>>()?;
+        let validators_to_delete = if validators_to_delete_str.trim() == "all" {
+            Vec::new()
+        } else {
+            validators_to_delete_str
+                .split(',')
+                .map(|s| s.trim().parse())
+                .collect::<Result<Vec<PublicKeyBytes>, _>>()?
+        };
 
         Ok(Self {
             vc_token_path: clap_utils::parse_required(matches, VC_TOKEN_FLAG)?,
@@ -90,11 +97,16 @@ async fn run<'a>(config: DeleteConfig) -> Result<(), String> {
     let DeleteConfig {
         vc_url,
         vc_token_path,
-        validators_to_delete,
+        mut validators_to_delete,
     } = config;
 
     let (http_client, validators) = vc_http_client(vc_url.clone(), &vc_token_path).await?;
 
+    // Delete all validators on the VC
+    if validators_to_delete.is_empty() {
+        validators_to_delete = validators.iter().map(|v| v.validating_pubkey).collect();
+    }
+
     for validator_to_delete in &validators_to_delete {
         if !validators
             .iter()
diff --git a/validator_manager/src/exit_validators.rs b/validator_manager/src/exit_validators.rs
index c86a75ee41e..099db676a9f 100644
--- a/validator_manager/src/exit_validators.rs
+++ b/validator_manager/src/exit_validators.rs
@@ -53,7 +53,10 @@ pub fn cli_app() -> Command {
             Arg::new(VALIDATOR_FLAG)
                 .long(VALIDATOR_FLAG)
                 .value_name("STRING")
-                .help("List of validators (pubkey) to exit.")
+                .help(
+                    "Comma-separated list of validators (pubkey) to exit. \
+                 To exit all validators, use the keyword \"all\".",
+                )
                 .action(ArgAction::Set)
                 .required(true)
                 .display_order(0),