rigg 1.4.3

Configuration-as-code CLI for Azure AI Search and Microsoft Foundry
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

//! Thin wrappers around `inquire` prompts: consistent styling, `--no-color`
//! support, and clean abort (Esc/Ctrl-C → error, nothing written).

use anyhow::{Result, anyhow};
use inquire::ui::RenderConfig;
use inquire::{Confirm, InquireError, MultiSelect, Select, Text};

fn config(plain: bool) -> RenderConfig<'static> {
    if plain {
        RenderConfig::empty()
    } else {
        RenderConfig::default_colored()
    }
}

fn map_err(e: InquireError) -> anyhow::Error {
    match e {
        InquireError::OperationCanceled | InquireError::OperationInterrupted => {
            anyhow!("aborted")
        }
        other => anyhow!(other),
    }
}

pub fn select(prompt: &str, options: Vec<String>, plain: bool) -> Result<String> {
    Select::new(prompt, options)
        .with_render_config(config(plain))
        .prompt()
        .map_err(map_err)
}

/// Returns the indices of the chosen options (order of the input list).
pub fn multi_select(prompt: &str, options: Vec<String>, plain: bool) -> Result<Vec<usize>> {
    let indexed: Vec<String> = options;
    let chosen = MultiSelect::new(prompt, indexed.clone())
        .with_render_config(config(plain))
        .prompt()
        .map_err(map_err)?;
    Ok(chosen
        .into_iter()
        .filter_map(|c| indexed.iter().position(|o| *o == c))
        .collect())
}

/// Multi-select with every option pre-checked when `checked` is true.
pub fn multi_select_checked(
    prompt: &str,
    options: Vec<String>,
    checked: bool,
    plain: bool,
) -> Result<Vec<usize>> {
    let indexed: Vec<String> = options;
    let mut ms = MultiSelect::new(prompt, indexed.clone()).with_render_config(config(plain));
    let all: Vec<usize> = (0..indexed.len()).collect();
    if checked {
        ms = ms.with_default(&all);
    }
    let chosen = ms.prompt().map_err(map_err)?;
    Ok(chosen
        .into_iter()
        .filter_map(|c| indexed.iter().position(|o| *o == c))
        .collect())
}

pub fn confirm_default_yes(prompt: &str, plain: bool) -> Result<bool> {
    Confirm::new(prompt)
        .with_default(true)
        .with_render_config(config(plain))
        .prompt()
        .map_err(map_err)
}

pub fn confirm_default_no(prompt: &str, plain: bool) -> Result<bool> {
    Confirm::new(prompt)
        .with_default(false)
        .with_render_config(config(plain))
        .prompt()
        .map_err(map_err)
}

/// Text prompt pre-filled with a default (Enter accepts it).
pub fn text_with_default(prompt: &str, default: &str, plain: bool) -> Result<String> {
    Text::new(prompt)
        .with_default(default)
        .with_render_config(config(plain))
        .prompt()
        .map_err(map_err)
}

pub fn text(prompt: &str, plain: bool) -> Result<String> {
    Text::new(prompt)
        .with_render_config(config(plain))
        .prompt()
        .map_err(map_err)
}