feat/cli: generate shell completions

Close #22

Cargo.toml

@@ -68,6 +68,7 @@ unicode-segmentation = "1.11.0"
 unicode-width = "0.1.12"
 serial_test = { version = "3.1.1", features = ["file_locks"]}
 arcstr = "1.2.0"
+clap_complete = "4.5.2"
 # tui-prompts = { version = "0.3.11", path = "../../contrib/tui-prompts" }
 # tui-popup = { version = "0.3.0", path = "../../contrib/tui-popup" }
 

src/cli.rs

@@ -1,6 +1,6 @@
-use std::{num::ParseFloatError, path::PathBuf};
+use std::{io::stdout, num::ParseFloatError, path::PathBuf};
 
-use clap::{Parser, Subcommand};
+use clap::{CommandFactory, Parser, Subcommand};
 
 use crate::tui::app::AppLayout;
 
@@ -106,6 +106,11 @@ pub enum CliCommand {
     )]
     frame_rate: f64,
   },
+  #[clap(about = "Generate shell completions for tracexec")]
+  GenerateCompletions {
+    #[arg(required = true, help = "The shell to generate completions for")]
+    shell: clap_complete::Shell,
+  },
 }
 
 #[derive(thiserror::Error, Debug)]
@@ -134,3 +139,10 @@ fn frame_rate_parser(s: &str) -> Result<f64, ParseFrameRateError> {
     Ok(v)
   }
 }
+
+impl Cli {
+  pub fn generate_completions(shell: clap_complete::Shell) {
+    let mut cmd = Cli::command();
+    clap_complete::generate(shell, &mut cmd, env!("CARGO_CRATE_NAME"), &mut stdout())
+  }
+}

src/main.rs

@@ -1,5 +1,6 @@
 mod action;
 mod arch;
+mod cache;
 mod cli;
 mod cmdbuilder;
 mod event;
@@ -11,7 +12,6 @@ mod pty;
 mod seccomp;
 mod tracer;
 mod tui;
-mod cache;
 
 use std::{
   io::{stderr, stdout, BufWriter},
@@ -202,6 +202,9 @@ async fn main() -> color_eyre::Result<()> {
       tui::restore_tui()?;
       tracer_thread.join().unwrap()?;
     }
+    CliCommand::GenerateCompletions { shell } => {
+      Cli::generate_completions(shell);
+    }
   }
   Ok(())
 }