add basic completion

Author: emouty

Cargo.lock

@@ -25,8 +25,8 @@ pkg-fmt = "bin"
 actix-web = { version = "4", features = ["openssl"] }
 anyhow = "1"
 biscuit = "0.7"
-chrono = "0.4"
 clap = { version = "4", features = ["derive", "env"] }
+clap_complete = "4"
 colored_json = "5"
 comfy-table = "7"
 directories = "5"

Cargo.toml

@@ -4,7 +4,15 @@ mod inspect;
 mod list;
 mod token;
 
+use crate::Cli;
+use clap::CommandFactory;
+use clap_complete::{
+    generate,
+    shells::{Bash, Fish, Zsh},
+};
+use std::path::Path;
 use std::process::ExitCode;
+use std::{env, io};
 
 #[derive(Debug, clap::Subcommand)]
 pub enum Command {
@@ -13,6 +21,7 @@ pub enum Command {
     Token(token::GetToken),
     List(list::List),
     Inspect(inspect::Inspect),
+    Completion { shell: String },
 }
 
 impl Command {
@@ -23,6 +32,25 @@ impl Command {
             Self::Token(cmd) => cmd.run().await,
             Self::List(cmd) => cmd.run().await,
             Self::Inspect(cmd) => cmd.run().await,
+            Self::Completion { shell } => {
+                let mut cmd = Cli::command();
+                let bin_name = env::args()
+                    .next()
+                    .and_then(|path| {
+                        Path::new(&path)
+                            .file_stem()
+                            .map(|name| name.to_string_lossy().into_owned())
+                    })
+                    .unwrap();
+
+                match shell.as_str() {
+                    "bash" => generate(Bash, &mut cmd, &bin_name, &mut io::stdout()),
+                    "zsh" => generate(Zsh, &mut cmd, &bin_name, &mut io::stdout()),
+                    "fish" => generate(Fish, &mut cmd, &bin_name, &mut io::stdout()),
+                    _ => eprintln!("Unsupported shell: {}", shell),
+                }
+                Ok(())
+            }
         }
         .map(|()| ExitCode::SUCCESS)
     }