#[cfg(feature = "typed")]
use serde::Serialize;
use toml::Value;
use crate::error::{Error, Result};
use crate::tokenizer::tokenize_with_seperator;
use crate::tokenizer::Token;
pub trait TomlValueInsertExt {
fn insert_with_seperator(
&mut self,
query: &str,
sep: char,
value: Value,
) -> Result<Option<Value>>;
fn insert(&mut self, query: &str, value: Value) -> Result<Option<Value>> {
self.insert_with_seperator(query, '.', value)
}
#[cfg(feature = "typed")]
fn insert_serialized<S: Serialize>(&mut self, query: &str, value: S) -> Result<Option<Value>> {
let value = Value::try_from(value).map_err(Error::TomlSerialize)?;
self.insert(query, value)
}
}
impl TomlValueInsertExt for Value {
fn insert_with_seperator(
&mut self,
query: &str,
sep: char,
value: Value,
) -> Result<Option<Value>> {
use crate::resolver::mut_creating_resolver::resolve;
let mut tokens = tokenize_with_seperator(query, sep)?;
let (val, last) = match tokens.pop_last() {
None => (self, Box::new(tokens)),
Some(last) => (resolve(self, &tokens)?, last),
};
match *last {
Token::Identifier { ident, .. } => match val {
Value::Table(ref mut t) => Ok(t.insert(ident, value)),
_ => Err(Error::NoIdentifierInArray(ident)),
},
Token::Index { idx, .. } => match val {
Value::Array(ref mut a) => {
if a.len() > idx {
a.insert(idx, value);
Ok(None)
} else {
a.push(value);
Ok(None)
}
}
_ => Err(Error::NoIndexInTable(idx)),
},
}
}
}
#[cfg(test)]
mod test {
use super::*;
use toml::from_str as toml_from_str;
use toml::Value;
#[test]
fn test_insert_one_token() {
use toml::map::Map;
let mut toml = Value::Table(Map::new());
let res = toml.insert(&String::from("value"), Value::Integer(1));
println!("TOML: {:?}", toml);
assert!(res.is_ok());
let res = res.unwrap();
assert!(res.is_none());
assert!(is_match!(toml, Value::Table(_)));
match toml {
Value::Table(ref t) => {
assert!(!t.is_empty());
let val = t.get("value");
assert!(
val.is_some(),
"'value' from table {:?} should be Some(_), is None",
t
);
let val = val.unwrap();
assert!(is_match!(val, Value::Integer(1)), "Is not one: {:?}", val);
}
_ => panic!("What just happenend?"),
}
}
#[test]
fn test_insert_with_seperator_into_table() {
let mut toml: Value = toml_from_str(
r#"
[table]
"#,
)
.unwrap();
let res = toml.insert_with_seperator(&String::from("table.a"), '.', Value::Integer(1));
assert!(res.is_ok());
let res = res.unwrap();
assert!(res.is_none());
assert!(is_match!(toml, Value::Table(_)));
match toml {
Value::Table(ref t) => {
assert!(!t.is_empty());
let table = t.get("table");
assert!(table.is_some());
let table = table.unwrap();
assert!(is_match!(table, Value::Table(_)));
match table {
Value::Table(ref t) => {
assert!(!t.is_empty());
let a = t.get("a");
assert!(a.is_some());
let a = a.unwrap();
assert!(is_match!(a, Value::Integer(1)));
}
_ => panic!("What just happenend?"),
}
}
_ => panic!("What just happenend?"),
}
}
#[test]
fn test_insert_with_seperator_into_array() {
use std::ops::Index;
let mut toml: Value = toml_from_str(
r#"
array = []
"#,
)
.unwrap();
let res = toml.insert_with_seperator(&String::from("array.[0]"), '.', Value::Integer(1));
assert!(res.is_ok());
let res = res.unwrap();
assert!(res.is_none());
assert!(is_match!(toml, Value::Table(_)));
match toml {
Value::Table(ref t) => {
assert!(!t.is_empty());
let array = t.get("array");
assert!(array.is_some());
let array = array.unwrap();
assert!(is_match!(array, Value::Array(_)));
match array {
Value::Array(ref a) => {
assert!(!a.is_empty());
assert!(is_match!(a.index(0), Value::Integer(1)));
}
_ => panic!("What just happenend?"),
}
}
_ => panic!("What just happenend?"),
}
}
#[test]
fn test_insert_with_seperator_into_nested_table() {
let mut toml: Value = toml_from_str(
r#"
[a.b.c]
"#,
)
.unwrap();
let res = toml.insert_with_seperator(&String::from("a.b.c.d"), '.', Value::Integer(1));
assert!(res.is_ok());
let res = res.unwrap();
assert!(res.is_none());
assert!(is_match!(toml, Value::Table(_)));
match toml {
Value::Table(ref outer) => {
assert!(!outer.is_empty());
let a_tab = outer.get("a");
assert!(a_tab.is_some());
let a_tab = a_tab.unwrap();
assert!(is_match!(a_tab, Value::Table(_)));
match a_tab {
Value::Table(ref a) => {
assert!(!a.is_empty());
let b_tab = a.get("b");
assert!(b_tab.is_some());
let b_tab = b_tab.unwrap();
assert!(is_match!(b_tab, Value::Table(_)));
match b_tab {
Value::Table(ref b) => {
assert!(!b.is_empty());
let c_tab = b.get("c");
assert!(c_tab.is_some());
let c_tab = c_tab.unwrap();
assert!(is_match!(c_tab, Value::Table(_)));
match c_tab {
Value::Table(ref c) => {
assert!(!c.is_empty());
let d = c.get("d");
assert!(d.is_some());
let d = d.unwrap();
assert!(is_match!(d, Value::Integer(1)));
}
_ => panic!("What just happenend?"),
}
}
_ => panic!("What just happenend?"),
}
}
_ => panic!("What just happenend?"),
}
}
_ => panic!("What just happened?"),
}
}
#[test]
fn test_insert_with_seperator_into_table_where_array_is() {
let mut toml: Value = toml_from_str(
r#"
table = []
"#,
)
.unwrap();
let res = toml.insert_with_seperator(&String::from("table.a"), '.', Value::Integer(1));
assert!(res.is_err());
let err = res.unwrap_err();
assert!(is_match!(err, Error::NoIdentifierInArray(_)));
}
#[test]
fn test_insert_with_seperator_into_array_where_table_is() {
let mut toml: Value = toml_from_str(
r#"
[table]
"#,
)
.unwrap();
let res = toml.insert_with_seperator(&String::from("table.[0]"), '.', Value::Integer(1));
assert!(res.is_err());
let err = res.unwrap_err();
assert!(is_match!(err, Error::NoIndexInTable(_)));
}
#[test]
fn test_insert_with_seperator_into_array_between_values() {
use std::ops::Index;
let mut toml: Value = toml_from_str(
r#"
array = [1, 2, 3, 4, 5]
"#,
)
.unwrap();
let res = toml.insert_with_seperator(&String::from("array.[2]"), '.', Value::Integer(6));
assert!(res.is_ok());
let res = res.unwrap();
assert!(res.is_none());
assert!(is_match!(toml, Value::Table(_)));
match toml {
Value::Table(ref t) => {
assert!(!t.is_empty());
let array = t.get("array");
assert!(array.is_some());
let array = array.unwrap();
assert!(is_match!(array, Value::Array(_)));
match array {
Value::Array(ref a) => {
assert!(!a.is_empty());
assert!(is_match!(a.index(0), Value::Integer(1)));
assert!(is_match!(a.index(1), Value::Integer(2)));
assert!(is_match!(a.index(2), Value::Integer(6)));
assert!(is_match!(a.index(3), Value::Integer(3)));
assert!(is_match!(a.index(4), Value::Integer(4)));
assert!(is_match!(a.index(5), Value::Integer(5)));
}
_ => panic!("What just happenend?"),
}
}
_ => panic!("What just happenend?"),
}
}
#[test]
fn test_insert_with_seperator_into_table_with_nonexisting_keys() {
let mut toml: Value = toml_from_str(
r#"
"#,
)
.unwrap();
let res = toml.insert_with_seperator(&String::from("table.a"), '.', Value::Integer(1));
assert!(res.is_ok());
let res = res.unwrap();
assert!(res.is_none());
assert!(is_match!(toml, Value::Table(_)));
match toml {
Value::Table(ref t) => {
assert!(!t.is_empty());
let table = t.get("table");
assert!(table.is_some());
let table = table.unwrap();
assert!(is_match!(table, Value::Table(_)));
match table {
Value::Table(ref t) => {
assert!(!t.is_empty());
let a = t.get("a");
assert!(a.is_some());
let a = a.unwrap();
assert!(is_match!(a, Value::Integer(1)));
}
_ => panic!("What just happenend?"),
}
}
_ => panic!("What just happenend?"),
}
}
}