feat: support list roles with --list-roles

1 files changed, 12 insertions, 0 deletions

diff --git a/src/main.rs b/src/main.rs
index de61a03..8ee2e96 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -49,6 +49,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')
                 .long("role")
@@ -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 {