use std::collections::HashMap;
use prost_types::{ListValue, Struct};
use prost_types::value::Kind;
use semver::{Version, VersionReq};
use serde_json::{json, Value};
pub fn to_proto_struct(values: &HashMap<String, Value>) -> Struct {
Struct {
fields: values.iter().map(|(k, v)| (k.clone(), to_proto_value(v))).collect()
}
}
pub fn to_proto_value(val: &Value) -> prost_types::Value {
match val {
Value::Null => prost_types::Value { kind: Some(prost_types::value::Kind::NullValue(0)) },
Value::Bool(b) => prost_types::Value { kind: Some(prost_types::value::Kind::BoolValue(*b)) },
Value::Number(n) => if let Some(n) = n.as_u64() {
prost_types::Value { kind: Some(prost_types::value::Kind::NumberValue(n as f64)) }
} else if let Some(n) = n.as_i64() {
prost_types::Value { kind: Some(prost_types::value::Kind::NumberValue(n as f64)) }
} else {
prost_types::Value { kind: Some(prost_types::value::Kind::NumberValue(n.as_f64().unwrap_or_default())) }
}
Value::String(s) => prost_types::Value { kind: Some(prost_types::value::Kind::StringValue(s.clone())) },
Value::Array(a) => prost_types::Value { kind: Some(prost_types::value::Kind::ListValue(ListValue {
values: a.iter().map(|v| to_proto_value(v)).collect()
}))},
Value::Object(o) => prost_types::Value { kind: Some(prost_types::value::Kind::StructValue(Struct {
fields: o.iter().map(|(k, v)| (k.clone(), to_proto_value(v))).collect()
}))}
}
}
pub fn proto_struct_to_json(val: &prost_types::Struct) -> Value {
Value::Object(val.fields.iter()
.map(|(k, v)| (k.clone(), proto_value_to_json(v))).collect())
}
pub fn proto_value_to_json(val: &prost_types::Value) -> Value {
match &val.kind {
Some(kind) => match kind {
Kind::NullValue(_) => Value::Null,
Kind::NumberValue(n) => json!(n),
Kind::StringValue(s) => Value::String(s.clone()),
Kind::BoolValue(b) => Value::Bool(*b),
Kind::StructValue(s) => proto_struct_to_json(s),
Kind::ListValue(l) => Value::Array(l.values.iter()
.map(|v| proto_value_to_json(v)).collect())
}
None => Value::Null
}
}
pub fn proto_struct_to_map(val: &prost_types::Struct) -> HashMap<String, Value> {
val.fields.iter().map(|(k, v)| (k.clone(), proto_value_to_json(v))).collect()
}
pub fn proto_value_to_string(val: &prost_types::Value) -> Option<String> {
match &val.kind {
Some(kind) => match kind {
Kind::NullValue(_) => None,
Kind::NumberValue(n) => Some(n.to_string()),
Kind::StringValue(s) => Some(s.clone()),
Kind::BoolValue(b) => Some(b.to_string()),
Kind::StructValue(s) => Some(proto_struct_to_json(s).to_string()),
Kind::ListValue(l) => Some(Value::Array(l.values.iter()
.map(|v| proto_value_to_json(v)).collect()).to_string())
}
None => None
}
}
pub fn versions_compatible(version: &str, required: &Option<String>) -> bool {
match required {
None => true,
Some(required) => {
if required == version {
true
} else if let Ok(version) = Version::parse(version) {
if let Ok(req) = VersionReq::parse(format!(">={}", required).as_str()) {
req.matches(&version)
} else {
false
}
} else {
false
}
}
}
}
#[cfg(test)]
mod tests {
use expectest::prelude::*;
use super::versions_compatible;
#[test]
fn versions_compatible_test() {
expect!(versions_compatible("1.0.0", &None)).to(be_true());
expect!(versions_compatible("1.0.0", &Some("1.0.0".to_string()))).to(be_true());
expect!(versions_compatible("1.0.0", &Some("1.0.1".to_string()))).to(be_false());
expect!(versions_compatible("1.0.4", &Some("1.0.3".to_string()))).to(be_true());
expect!(versions_compatible("1.1.0", &Some("1.0.3".to_string()))).to(be_true());
expect!(versions_compatible("2.0.1", &Some("1.1.0".to_string()))).to(be_true());
expect!(versions_compatible("1.0.1", &Some("2.1.0".to_string()))).to(be_false());
expect!(versions_compatible("0.1.0", &Some("0.0.3".to_string()))).to(be_true());
}
}