himitsu 1.0.0

An encrypted CLI tool for managing secrets on your machine
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

//! Contains the execution flow for adding a new secret.

use inquire::{self, validator::StringValidator, Password, PasswordDisplayMode, Text};

use crate::errors::HimitsuError;

use super::config::{self, ConfigType};

/// Run the add secret prompts. Ask for a label (if one is not provided), then ask for the secret
/// itself, and finally asks to set any categories or tags.
pub fn run_add_secret(
    category: &Option<String>,
    label: &Option<String>,
    tags: &Option<Vec<String>>,
) -> Result<(String, String, String, Vec<String>), HimitsuError> {
    let render_config = config::get_inquire_config(ConfigType::Standard, true);

    let label_validator: StringValidator = &|input| {
        if input.is_empty() {
            Err("A label is required!".to_string())
        } else {
            Ok(())
        }
    };

    let label_input = if label.is_some() {
        label.to_owned()
    } else {
        Text::new("Enter a label for this secret:")
            .with_render_config(render_config)
            .with_validator(label_validator)
            .prompt_skippable()?
    };
    if label_input.is_none() {
        return Err(HimitsuError::UserCancelled);
    }

    let secret_input = Password::new("Enter your secret:")
        .with_display_mode(PasswordDisplayMode::Hidden)
        .with_display_toggle_enabled()
        .with_render_config(render_config)
        .with_help_message("Press \"<CTRL> + r\" to reveal input.")
        .prompt_skippable()?;
    if secret_input.is_none() {
        return Err(HimitsuError::UserCancelled);
    }

    let category = match category {
        Some(category_name) => category_name.to_lowercase(),
        None => {
            let input = Text::new("Set a category for this secret:")
                .with_default("unclassified")
                .with_help_message("(OPTIONAL) Defaults to \"unclassified\"")
                .with_render_config(render_config)
                .prompt_skippable()?;

            if input.is_none() {
                return Err(HimitsuError::UserCancelled);
            }

            if input.as_ref().unwrap().is_empty() {
                "unclassified".to_string()
            } else {
                input.unwrap().to_lowercase()
            }
        }
    };
    let tags = match tags {
        Some(tag_values) => tag_values
            .to_owned()
            .iter()
            .map(|tag| tag.to_lowercase())
            .collect(),
        None => {
            let tags_input = Text::new("Set tags for this secret:")
                .with_help_message("(OPTIONAL) Enter a list of space-delimited tags. No default tags are applied if none are specified")
                .with_render_config(render_config)
                .prompt_skippable()?;

            if tags_input.is_none() {
                return Err(HimitsuError::UserCancelled);
            }

            tags_input
                .unwrap()
                .split(' ')
                .map(|tag| tag.to_string().to_lowercase())
                .collect::<Vec<String>>()
        }
    };

    Ok((label_input.unwrap(), secret_input.unwrap(), category, tags))
}