proto_cli 0.58.1

A multi-language version manager, a unified toolchain.
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

use crate::app::App;
use crate::session::{ProtoSession, SessionResult};
use clap::{Args, CommandFactory};
use clap_complete::{Shell, generate};
use clap_complete_nushell::Nushell;
use starbase_console::ui::*;
use starbase_shell::ShellType;
use tracing::instrument;

#[derive(Args, Clone, Debug)]
pub struct CompletionsArgs {
    #[arg(long, help = "Shell to generate for")]
    shell: Option<ShellType>,
}

#[instrument(skip(session))]
pub async fn completions(session: ProtoSession, args: CompletionsArgs) -> SessionResult {
    let shell = match args.shell {
        Some(value) => value,
        None => ShellType::try_detect()?,
    };

    let mut app = App::command();
    let mut stdio = std::io::stdout();

    let clap_shell = match shell {
        ShellType::Bash => Shell::Bash,
        ShellType::Elvish => Shell::Elvish,
        ShellType::Fish => Shell::Fish,
        ShellType::Pwsh => Shell::PowerShell,
        ShellType::Zsh => Shell::Zsh,
        ShellType::Nu => {
            generate(Nushell, &mut app, "proto", &mut stdio);

            return Ok(None);
        }
        unsupported => {
            session.console.notice(
                Variant::Caution,
                format!("<id>{unsupported}</id> does not currently support completions"),
            )?;

            return Ok(Some(1));
        }
    };

    generate(clap_shell, &mut app, "proto", &mut stdio);

    Ok(None)
}