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        unsafe {
49            env::set_var("INPUT_NAME", "value");
50        }
51        assert_eq!("value".to_string(), get_input("name").unwrap());
52    }
53
54    #[test]
55    fn test_get_input_none() {
56        assert!(get_input("unknown").is_err())
57    }
58
59    #[test]
60    fn test_get_bool_input() {
61        unsafe {
62            env::set_var("INPUT_CI", "true");
63        }
64        assert!(get_bool_input("ci").unwrap());
65    }
66
67    #[test]
68    fn test_get_multiline_input() {
69        unsafe {
70            env::set_var("INPUT_LINES", "true\nfalse\n");
71        }
72        assert_eq!(vec!["true", "false"], get_multiline_input("lines").unwrap());
73    }
74}
github_actions/input.rs

github_actions/
input.rs
