adenosine_cli/
lib.rs

1use anyhow::anyhow;
2pub use anyhow::Result;
3use lazy_static::lazy_static;
4use regex::Regex;
5use serde_json::Value;
6use std::collections::HashMap;
7use std::str::FromStr;
8
9pub mod pretty;
10
11/// Represents fields/content specified on the command line.
12///
13/// Sort of like HTTPie. Query parameters are '==', body values (JSON) are '='. Only single-level
14/// body values are allowed currently, not JSON Pointer assignment.
15#[derive(Debug, PartialEq, Eq, Clone)]
16pub enum ArgField {
17    Query(String, serde_json::Value),
18    Body(String, serde_json::Value),
19}
20
21impl FromStr for ArgField {
22    type Err = anyhow::Error;
23
24    fn from_str(s: &str) -> Result<Self, Self::Err> {
25        lazy_static! {
26            static ref FIELD_RE: Regex = Regex::new(r"^([a-zA-Z_]+)=(=)?(.*)$").unwrap();
27        }
28        if let Some(captures) = FIELD_RE.captures(s) {
29            let key = captures[1].to_string();
30            let val =
31                Value::from_str(&captures[3]).unwrap_or(Value::String(captures[3].to_string()));
32            let val = match val {
33                Value::String(s) if s.is_empty() => Value::Null,
34                _ => val,
35            };
36            if captures.get(2).is_some() {
37                Ok(ArgField::Query(key, val))
38            } else {
39                Ok(ArgField::Body(key, val))
40            }
41        } else {
42            Err(anyhow!("could not parse as a field assignment: {}", s))
43        }
44    }
45}
46
47#[test]
48fn test_argfield() {
49    use serde_json::json;
50    assert_eq!(
51        ArgField::from_str("a=3").unwrap(),
52        ArgField::Body("a".to_string(), json!(3)),
53    );
54    assert_eq!(
55        ArgField::from_str("a==3").unwrap(),
56        ArgField::Query("a".to_string(), json!(3)),
57    );
58    assert_eq!(
59        ArgField::from_str("cream==\"something\"").unwrap(),
60        ArgField::Query("cream".to_string(), Value::String("something".to_string()))
61    );
62    assert_eq!(
63        ArgField::from_str("cream==something").unwrap(),
64        ArgField::Query("cream".to_string(), Value::String("something".to_string()))
65    );
66    assert_eq!(
67        ArgField::from_str("cream=").unwrap(),
68        ArgField::Body("cream".to_string(), Value::Null),
69    );
70
71    assert!(ArgField::from_str("a").is_err());
72    assert!(ArgField::from_str("").is_err());
73    assert!(ArgField::from_str("asdf.fee").is_err());
74
75    assert!(ArgField::from_str("text=\"other value\"").is_ok());
76}
77
78// TODO: what should type signature actually be here...
79pub fn update_params_from_fields(fields: &[ArgField], params: &mut HashMap<String, String>) {
80    for f in fields.iter() {
81        if let ArgField::Query(ref k, ref v) = f {
82            match v {
83                Value::String(s) => params.insert(k.to_string(), s.to_string()),
84                _ => params.insert(k.to_string(), v.to_string()),
85            };
86        }
87    }
88}
89
90pub fn update_value_from_fields(fields: Vec<ArgField>, value: &mut Value) {
91    if let Value::Object(map) = value {
92        for f in fields.into_iter() {
93            if let ArgField::Body(k, v) = f {
94                map.insert(k, v);
95            }
96        }
97    }
98}
99
100/// Consumes the entire Vec of fields passed in
101pub fn value_from_fields(fields: Vec<ArgField>) -> Value {
102    let mut map: HashMap<String, Value> = HashMap::new();
103    for f in fields.into_iter() {
104        if let ArgField::Body(k, v) = f {
105            map.insert(k, v);
106        }
107    }
108    Value::Object(serde_json::map::Map::from_iter(map.into_iter()))
109}
adenosine_cli/lib.rs

adenosine_cli/
lib.rs
