aboutsummaryrefslogtreecommitdiffhomepage
path: root/crates/mozart/src
diff options
context:
space:
mode:
Diffstat (limited to 'crates/mozart/src')
-rw-r--r--crates/mozart/src/commands.rs5
-rw-r--r--crates/mozart/src/commands/completion.rs20
2 files changed, 25 insertions, 0 deletions
diff --git a/crates/mozart/src/commands.rs b/crates/mozart/src/commands.rs
index 694b4e1..285f65d 100644
--- a/crates/mozart/src/commands.rs
+++ b/crates/mozart/src/commands.rs
@@ -5,6 +5,7 @@ pub mod browse;
pub mod bump;
pub mod check_platform_reqs;
pub mod clear_cache;
+pub mod completion;
pub mod config;
pub mod create_project;
pub mod dependency;
@@ -105,6 +106,9 @@ pub enum Commands {
#[command(name = "clear-cache", alias = "clearcache", alias = "cc")]
ClearCache(clear_cache::ClearCacheArgs),
+ /// Generate shell completion scripts
+ Completion(completion::CompletionArgs),
+
/// Sets config options
Config(config::ConfigArgs),
@@ -203,6 +207,7 @@ pub fn execute(cli: &Cli) -> anyhow::Result<()> {
Commands::Bump(args) => bump::execute(args, cli, &console),
Commands::CheckPlatformReqs(args) => check_platform_reqs::execute(args, cli, &console),
Commands::ClearCache(args) => clear_cache::execute(args, cli, &console),
+ Commands::Completion(args) => completion::execute(args, cli, &console),
Commands::Config(args) => config::execute(args, cli, &console),
Commands::CreateProject(args) => create_project::execute(args, cli, &console),
Commands::Depends(args) => depends::execute(args, cli, &console),
diff --git a/crates/mozart/src/commands/completion.rs b/crates/mozart/src/commands/completion.rs
new file mode 100644
index 0000000..406749c
--- /dev/null
+++ b/crates/mozart/src/commands/completion.rs
@@ -0,0 +1,20 @@
+use clap::Args;
+use clap::CommandFactory;
+use clap_complete::aot::Shell;
+
+#[derive(Args)]
+pub struct CompletionArgs {
+ /// The shell to generate completions for
+ #[arg(value_enum)]
+ pub shell: Shell,
+}
+
+pub fn execute(
+ args: &CompletionArgs,
+ _cli: &super::Cli,
+ _console: &crate::console::Console,
+) -> anyhow::Result<()> {
+ let mut cmd = super::Cli::command();
+ clap_complete::aot::generate(args.shell, &mut cmd, "mozart", &mut std::io::stdout());
+ Ok(())
+}