aichat 0.18.0

All-in-one AI CLI Tool
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

use inquire::{required, validator::Validation, Text};

const MSG_REQUIRED: &str = "This field is required";
const MSG_OPTIONAL: &str = "Optional field - Press ↵ to skip";

pub fn prompt_input_string(desc: &str, required: bool) -> anyhow::Result<String> {
    let mut text = Text::new(desc);
    if required {
        text = text.with_validator(required!(MSG_REQUIRED))
    } else {
        text = text.with_help_message(MSG_OPTIONAL)
    }
    let text = text.prompt()?;
    Ok(text)
}

pub fn prompt_input_integer(desc: &str, required: bool) -> anyhow::Result<String> {
    let mut text = Text::new(desc);
    if required {
        text = text.with_validator(|text: &str| {
            let out = if text.is_empty() {
                Validation::Invalid(MSG_REQUIRED.into())
            } else {
                validate_integer(text)
            };
            Ok(out)
        })
    } else {
        text = text
            .with_validator(|text: &str| {
                let out = if text.is_empty() {
                    Validation::Valid
                } else {
                    validate_integer(text)
                };
                Ok(out)
            })
            .with_help_message(MSG_OPTIONAL)
    }
    let text = text.prompt()?;
    Ok(text)
}

#[derive(Debug, Clone, Copy)]
pub enum PromptKind {
    String,
    Integer,
}

fn validate_integer(text: &str) -> Validation {
    if text.parse::<i32>().is_err() {
        Validation::Invalid("Must be a integer".into())
    } else {
        Validation::Valid
    }
}