enhancement(cli): Shell autocompletion for vector cli

Cargo.toml

@@ -145,6 +145,7 @@ cfg-if = { version = "1.0.3", default-features = false }
 chrono = { version = "0.4.41", default-features = false, features = ["clock", "serde"] }
 chrono-tz = { version = "0.10.4", default-features = false, features = ["serde"] }
 clap = { version = "4.5.48", default-features = false, features = ["derive", "error-context", "env", "help", "std", "string", "usage", "wrap_help"] }
+clap_complete = "4.5.58"
 colored = { version = "3.0.0", default-features = false }
 crossbeam-utils = { version = "0.8.21", default-features = false }
 darling = { version = "0.20.11", default-features = false, features = ["suggestions"] }
@@ -208,6 +209,7 @@ serial_test = { version = "3.2" }
 [dependencies]
 cfg-if.workspace = true
 clap.workspace = true
+clap_complete.workspace = true
 indoc.workspace = true
 paste.workspace = true
 pin-project.workspace = true

changelog.d/24414_shell_autocompletion.enhancement.md

@@ -0,0 +1,3 @@
+Autocompletion scripts for the vector cli can now be generated (via `vector completion`).
+
+authors: weriomat

src/cli.rs

@@ -12,8 +12,8 @@ use crate::tap;
 use crate::top;
 
 use crate::{
-    config, convert_config, generate, generate_schema, get_version, graph, list, signal, unit_test,
-    validate,
+    completion, config, convert_config, generate, generate_schema, get_version, graph, list,
+    signal, unit_test, validate,
 };
 
 #[derive(Parser, Debug)]
@@ -318,6 +318,10 @@ pub enum SubCommand {
     /// By default all output is writen to stdout. The `output_path` option can be used to redirect to a file.
     GenerateSchema(generate_schema::Opts),
 
+    /// Generate shell completion, then exit.
+    #[command(hide = true)]
+    Completion(completion::Opts),
+
     /// Output a provided Vector configuration file/dir as a single JSON object, useful for checking in to version control.
     #[command(hide = true)]
     Config(config::Opts),
@@ -355,6 +359,7 @@ impl SubCommand {
         color: bool,
     ) -> exitcode::ExitCode {
         match self {
+            Self::Completion(s) => completion::cmd(s),
             Self::Config(c) => config::cmd(c),
             Self::ConvertConfig(opts) => convert_config::cmd(opts),
             Self::Generate(g) => generate::cmd(g),

src/completion.rs

@@ -0,0 +1,23 @@
+#![allow(missing_docs)]
+use clap::{CommandFactory, Parser};
+use clap_complete::{Shell, generate};
+use std::io;
+
+use crate::cli::Opts as RootCli;
+
+#[derive(Parser, Debug, Clone)]
+#[command(rename_all = "kebab-case")]
+pub struct Opts {
+    /// Shell to generate completion for
+    #[clap(value_enum)]
+    shell: Shell,
+}
+
+pub fn cmd(opts: &Opts) -> exitcode::ExitCode {
+    let mut cmd = RootCli::command();
+    let bin_name = cmd.get_name().to_string();
+
+    generate(opts.shell, &mut cmd, bin_name, &mut io::stdout());
+
+    exitcode::OK
+}

src/lib.rs

@@ -75,6 +75,7 @@ pub mod aws;
 #[allow(unreachable_pub)]
 pub mod codecs;
 pub mod common;
+pub mod completion;
 mod convert_config;
 pub mod encoding_transcode;
 pub mod enrichment_tables;

vdev/Cargo.toml

@@ -21,7 +21,7 @@ anyhow.workspace = true
 chrono.workspace = true
 clap.workspace = true
 clap-verbosity-flag = "3.0.4"
-clap_complete = "4.5.58"
+clap_complete.workspace = true
 directories = "6.0.0"
 glob.workspace = true
 hex = "0.4.3"