From 97af9322e76ff5ec9ca89fc938835ec622731528 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 2 Mar 2023 22:25:55 +0800
Subject: feat: support list roles with --list-roles

---
 src/main.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index de61a03..8ee2e96 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -48,6 +48,13 @@ fn start() -> Result<()> {
             " - ",
             env!("CARGO_PKG_REPOSITORY")
         ))
+        .arg(
+            Arg::new("list-roles")
+                .short('L')
+                .long("list-roles")
+                .action(ArgAction::SetTrue)
+                .help("List all roles"),
+        )
         .arg(
             Arg::new("role")
                 .short('r')
@@ -82,6 +89,11 @@ fn start() -> Result<()> {
         text = Some(role.generate(text_));
     };
 
+    if matches.get_flag("list-roles") {
+        config.roles.iter().for_each(|v| println!("{}", v.name));
+        exit(1);
+    }
+
     let client = init_client(&config)?;
     let runtime = init_runtime()?;
     match text {
-- 
cgit v1.2.3