Add subcommand to generate completions

Signed-off-by: Matthias Beyer <mail@beyermatthias.de>

2 files changed, 20 insertions, 0 deletions

diff --git a/src/cli.rs b/src/cli.rs
index 008fb28c..b6b0ec3c 100644
--- a/src/cli.rs
+++ b/src/cli.rs
@@ -51,4 +51,10 @@ pub enum ConfigCli {
         /// Disables loading of configuration file at default location
         clean: bool,
     },
+
+    #[structopt(name = "generate-completion")]
+    GenerateCompletion {
+        shell: String,
+    },
 }
+
diff --git a/src/main.rs b/src/main.rs
index ec249e8a..377ef4f6 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -45,6 +45,20 @@ pub fn main() {
         let mut stream = UnixStream::connect(ZELLIJ_IPC_PIPE).unwrap();
         let api_command = bincode::serialize(&ApiCommand::OpenFile(file_to_open)).unwrap();
         stream.write_all(&api_command).unwrap();
+    } else if let Some(crate::cli::ConfigCli::GenerateCompletion { shell }) = opts.option {
+        let shell = match shell.as_ref() {
+            "bash"       => structopt::clap::Shell::Bash,
+            "fish"       => structopt::clap::Shell::Fish,
+            "zsh"        => structopt::clap::Shell::Zsh,
+            "powerShell" => structopt::clap::Shell::PowerShell,
+            "elvish"     => structopt::clap::Shell::Elvish,
+            other        => {
+                eprintln!("Unsupported shell: {}", other);
+                std::process::exit(1);
+            },
+        };
+        let mut out = std::io::stdout();
+        CliArgs::clap().gen_completions_to("zellij", shell, &mut out);
     } else {
         let os_input = get_os_input();
         atomic_create_dir(ZELLIJ_TMP_DIR).unwrap();