github_actions/
input.rs

1use std::env;
2
3#[derive(Debug)]
4pub enum InputResult {
5    VarError(env::VarError),
6}
7
8pub fn get_input(name: &str) -> Result<String, InputResult> {
9    env::var(format!("INPUT_{}", name.to_uppercase())).map_err(InputResult::VarError)
10}
11
12pub fn get_multiline_input(name: &str) -> Result<Vec<String>, InputResult> {
13    get_input(name).map(|input| {
14        input
15            .split('\n')
16            .filter(|v| !v.is_empty())
17            .map(|v| v.to_string())
18            .collect()
19    })
20}
21
22#[derive(Debug)]
23pub enum BoolInputResult {
24    VarError(env::VarError),
25    TypeError,
26}
27
28pub fn get_bool_input(name: &str) -> Result<bool, BoolInputResult> {
29    const TRUE_VALUE: [&str; 3] = ["true", "True", "TRUE"];
30    const FALSE_VALUE: [&str; 3] = ["false", "False", "FALSE"];
31    let val =
32        get_input(name).map_err(|InputResult::VarError(err)| BoolInputResult::VarError(err))?;
33    if TRUE_VALUE.contains(&val.as_str()) {
34        return Ok(true);
35    }
36    if FALSE_VALUE.contains(&val.as_str()) {
37        return Ok(false);
38    }
39    Err(BoolInputResult::TypeError)
40}
41
42#[cfg(test)]
43mod tests {
44    use super::*;
45
46    #[test]
47    fn test_get_input() {
48        env::set_var("INPUT_NAME", "value");
49        assert_eq!("value".to_string(), get_input("name").unwrap());
50    }
51
52    #[test]
53    fn test_get_input_none() {
54        assert!(get_input("unknown").is_err())
55    }
56
57    #[test]
58    fn test_get_bool_input() {
59        env::set_var("INPUT_CI", "true");
60        assert!(get_bool_input("ci").unwrap());
61    }
62
63    #[test]
64    fn test_get_multiline_input() {
65        env::set_var("INPUT_LINES", "true\nfalse\n");
66        assert_eq!(vec!["true", "false"], get_multiline_input("lines").unwrap());
67    }
68}
github_actions/input.rs

github_actions/
input.rs
