provekit_nargo_cli 1.0.0-beta.20

Noir's package manager
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

use clap::{Args, CommandFactory};
use clap_complete::Shell;

use crate::errors::CliError;

use super::NargoCli;

/// Generates a shell completion script for your favorite shell
#[derive(Debug, Clone, Args)]
pub(crate) struct GenerateCompletionScriptCommand {
    /// The shell to generate completions for. One of: bash, elvish, fish, powershell, zsh
    pub(crate) shell: String,
}

pub(crate) fn run(command: GenerateCompletionScriptCommand) -> Result<(), CliError> {
    let shell = match command.shell.to_lowercase().as_str() {
        "bash" => Shell::Bash,
        "elvish" => Shell::Elvish,
        "fish" => Shell::Fish,
        "powershell" => Shell::PowerShell,
        "zsh" => Shell::Zsh,
        _ => {
            return Err(CliError::Generic(
                "Invalid shell. Supported shells are: bash, elvish, fish, powershell, zsh"
                    .to_string(),
            ));
        }
    };

    clap_complete::generate(shell, &mut NargoCli::command(), "nargo", &mut std::io::stdout());

    Ok(())
}