toml-rs/src/de.rs

2056 lines
66 KiB
Rust
Raw Normal View History

//! Deserializing TOML into Rust structures.
//!
//! This module contains all the Serde support for deserializing TOML documents
//! into Rust structures. Note that some top-level functions here are also
//! provided at the top of the crate.
use std::borrow::Cow;
use std::error;
use std::f64;
use std::fmt;
use std::mem::discriminant;
use std::str;
use std::vec;
use serde::de;
use serde::de::value::BorrowedStrDeserializer;
2018-12-17 19:45:35 -06:00
use serde::de::IntoDeserializer;
2019-05-08 14:12:14 -05:00
use crate::datetime;
use crate::spanned;
use crate::tokens::{Error as TokenError, Span, Token, Tokenizer};
/// Type Alias for a TOML Table pair
type TablePair<'a> = (Cow<'a, str>, Value<'a>);
/// Deserializes a byte slice into a type.
///
/// This function will attempt to interpret `bytes` as UTF-8 data and then
/// deserialize `T` from the TOML document provided.
2017-04-20 12:16:00 -05:00
pub fn from_slice<'de, T>(bytes: &'de [u8]) -> Result<T, Error>
2018-12-17 19:45:35 -06:00
where
T: de::Deserialize<'de>,
{
match str::from_utf8(bytes) {
Ok(s) => from_str(s),
2019-07-30 12:20:18 -05:00
Err(e) => Err(Error::custom(None, e.to_string())),
}
}
/// Deserializes a string into a type.
///
/// This function will attempt to interpret `s` as a TOML document and
/// deserialize `T` from the document.
2017-05-10 09:49:35 -05:00
///
/// # Examples
///
2017-05-10 10:17:58 -05:00
/// ```
2019-05-08 19:37:38 -05:00
/// use serde_derive::Deserialize;
2017-05-10 09:49:35 -05:00
///
/// #[derive(Deserialize)]
/// struct Config {
/// title: String,
/// owner: Owner,
/// }
///
/// #[derive(Deserialize)]
/// struct Owner {
/// name: String,
/// }
///
/// fn main() {
/// let config: Config = toml::from_str(r#"
/// title = 'TOML Example'
///
/// [owner]
/// name = 'Lisa'
/// "#).unwrap();
///
/// assert_eq!(config.title, "TOML Example");
/// assert_eq!(config.owner.name, "Lisa");
/// }
/// ```
2017-04-20 12:16:00 -05:00
pub fn from_str<'de, T>(s: &'de str) -> Result<T, Error>
2018-12-17 19:45:35 -06:00
where
T: de::Deserialize<'de>,
{
let mut d = Deserializer::new(s);
let ret = T::deserialize(&mut d)?;
d.end()?;
2017-03-30 06:38:48 -05:00
Ok(ret)
}
/// Errors that can occur when deserializing a type.
#[derive(Debug, PartialEq, Clone)]
pub struct Error {
inner: Box<ErrorInner>,
}
#[derive(Debug, PartialEq, Clone)]
struct ErrorInner {
kind: ErrorKind,
line: Option<usize>,
col: usize,
2019-07-30 12:20:18 -05:00
at: Option<usize>,
message: String,
key: Vec<String>,
}
/// Errors that can occur when deserializing a type.
#[derive(Debug, PartialEq, Clone)]
enum ErrorKind {
/// EOF was reached when looking for a value
UnexpectedEof,
/// An invalid character not allowed in a string was found
InvalidCharInString(char),
/// An invalid character was found as an escape
InvalidEscape(char),
/// An invalid character was found in a hex escape
InvalidHexEscape(char),
/// An invalid escape value was specified in a hex escape in a string.
///
/// Valid values are in the plane of unicode codepoints.
InvalidEscapeValue(u32),
/// A newline in a string was encountered when one was not allowed.
NewlineInString,
/// An unexpected character was encountered, typically when looking for a
/// value.
Unexpected(char),
/// An unterminated string was found where EOF was found before the ending
/// EOF mark.
UnterminatedString,
/// A newline was found in a table key.
NewlineInTableKey,
/// A number failed to parse
NumberInvalid,
/// A date or datetime was invalid
DateInvalid,
/// Wanted one sort of token, but found another.
Wanted {
/// Expected token type
expected: &'static str,
/// Actually found token type
found: &'static str,
},
/// An array was decoded but the types inside of it were mixed, which is
/// disallowed by TOML.
MixedArrayType,
/// A duplicate table definition was found.
DuplicateTable(String),
/// A previously defined table was redefined as an array.
RedefineAsArray,
/// An empty table key was found.
EmptyTableKey,
/// Multiline strings are not allowed for key
MultilineStringKey,
/// A custom error which could be generated when deserializing a particular
/// type.
Custom,
/// A tuple with a certain number of elements was expected but something
/// else was found.
ExpectedTuple(usize),
/// Expected table keys to be in increasing tuple index order, but something
/// else was found.
ExpectedTupleIndex {
/// Expected index.
expected: usize,
/// Key that was specified.
found: String,
},
/// An empty table was expected but entries were found
ExpectedEmptyTable,
2018-07-11 01:29:47 -05:00
/// Dotted key attempted to extend something that is not a table.
DottedKeyInvalidType,
/// An unexpected key was encountered.
///
/// Used when deserializing a struct with a limited set of fields.
UnexpectedKeys {
/// The unexpected keys.
keys: Vec<String>,
/// Keys that may be specified.
available: &'static [&'static str],
},
#[doc(hidden)]
__Nonexhaustive,
}
/// Deserialization implementation for TOML.
pub struct Deserializer<'a> {
require_newline_after_table: bool,
allow_duplciate_after_longer_table: bool,
input: &'a str,
tokens: Tokenizer<'a>,
}
2017-04-20 12:16:00 -05:00
impl<'de, 'b> de::Deserializer<'de> for &'b mut Deserializer<'de> {
type Error = Error;
2017-04-20 12:16:00 -05:00
fn deserialize_any<V>(self, visitor: V) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::Visitor<'de>,
{
let mut tables = self.tables()?;
2019-07-30 12:20:18 -05:00
let res = visitor.visit_map(MapVisitor {
values: Vec::new().into_iter(),
next_value: None,
depth: 0,
cur: 0,
cur_parent: 0,
max: tables.len(),
tables: &mut tables,
array: false,
de: self,
2019-07-30 12:20:18 -05:00
});
res.map_err(|mut err| {
// Errors originating from this library (toml), have an offset
// attached to them already. Other errors, like those originating
// from serde (like "missing field") or from a custom deserializer,
// do not have offsets on them. Here, we do a best guess at their
// location, by attributing them to the "current table" (the last
// item in `tables`).
err.fix_offset(|| tables.last().map(|table| table.at));
err.fix_linecol(|at| self.to_linecol(at));
err
})
}
// Called when the type to deserialize is an enum, as opposed to a field in the type.
fn deserialize_enum<V>(
self,
_name: &'static str,
_variants: &'static [&'static str],
2018-12-17 19:45:35 -06:00
visitor: V,
) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::Visitor<'de>,
{
let (value, name) = self.string_or_table()?;
match value.e {
E::String(val) => visitor.visit_enum(val.into_deserializer()),
E::InlineTable(values) => {
if values.len() != 1 {
2019-07-30 12:20:18 -05:00
Err(Error::from_kind(
Some(value.start),
ErrorKind::Wanted {
expected: "exactly 1 element",
found: if values.is_empty() {
"zero elements"
} else {
"more than 1 element"
},
},
2019-07-30 12:20:18 -05:00
))
} else {
visitor.visit_enum(InlineTableDeserializer {
values: values.into_iter(),
next_value: None,
})
}
}
E::DottedTable(_) => visitor.visit_enum(DottedTableDeserializer {
name: name.expect("Expected table header to be passed."),
value,
}),
e => Err(Error::from_kind(
2019-07-30 12:20:18 -05:00
Some(value.start),
ErrorKind::Wanted {
expected: "string or table",
found: e.type_name(),
},
)),
}
}
2019-05-08 19:37:38 -05:00
serde::forward_to_deserialize_any! {
bool u8 u16 u32 u64 i8 i16 i32 i64 f32 f64 char str string seq
bytes byte_buf map struct unit newtype_struct
2017-04-20 12:16:00 -05:00
ignored_any unit_struct tuple_struct tuple option identifier
}
}
struct Table<'a> {
at: usize,
header: Vec<Cow<'a, str>>,
values: Option<Vec<TablePair<'a>>>,
array: bool,
}
#[doc(hidden)]
pub struct MapVisitor<'de, 'b> {
values: vec::IntoIter<TablePair<'de>>,
next_value: Option<TablePair<'de>>,
depth: usize,
cur: usize,
cur_parent: usize,
max: usize,
2017-04-20 12:16:00 -05:00
tables: &'b mut [Table<'de>],
array: bool,
2017-04-20 12:16:00 -05:00
de: &'b mut Deserializer<'de>,
}
2017-04-20 12:16:00 -05:00
impl<'de, 'b> de::MapAccess<'de> for MapVisitor<'de, 'b> {
type Error = Error;
2017-04-20 12:16:00 -05:00
fn next_key_seed<K>(&mut self, seed: K) -> Result<Option<K::Value>, Error>
2018-12-17 19:45:35 -06:00
where
K: de::DeserializeSeed<'de>,
{
if self.cur_parent == self.max || self.cur == self.max {
2018-12-17 19:45:35 -06:00
return Ok(None);
}
loop {
assert!(self.next_value.is_none());
if let Some((key, value)) = self.values.next() {
let ret = seed.deserialize(StrDeserializer::new(key.clone()))?;
self.next_value = Some((key, value));
2018-12-17 19:45:35 -06:00
return Ok(Some(ret));
}
let next_table = {
let prefix = &self.tables[self.cur_parent].header[..self.depth];
2018-12-17 19:45:35 -06:00
self.tables[self.cur..self.max]
.iter()
.enumerate()
.find(|&(_, t)| {
if t.values.is_none() {
return false;
}
match t.header.get(..self.depth) {
Some(header) => header == prefix,
None => false,
}
})
.map(|(i, _)| i + self.cur)
};
let pos = match next_table {
Some(pos) => pos,
None => return Ok(None),
};
self.cur = pos;
// Test to see if we're duplicating our parent's table, and if so
// then this is an error in the toml format
if self.cur_parent != pos {
if self.tables[self.cur_parent].header == self.tables[pos].header {
let at = self.tables[pos].at;
let name = self.tables[pos].header.join(".");
return Err(self.de.error(at, ErrorKind::DuplicateTable(name)));
}
// If we're here we know we should share the same prefix, and if
// the longer table was defined first then we want to narrow
// down our parent's length if possible to ensure that we catch
// duplicate tables defined afterwards.
if !self.de.allow_duplciate_after_longer_table {
let parent_len = self.tables[self.cur_parent].header.len();
let cur_len = self.tables[pos].header.len();
if cur_len < parent_len {
self.cur_parent = pos;
}
}
}
let table = &mut self.tables[pos];
// If we're not yet at the appropriate depth for this table then we
2017-04-20 12:16:00 -05:00
// just next the next portion of its header and then continue
// decoding.
if self.depth != table.header.len() {
let key = &table.header[self.depth];
2017-04-20 12:16:00 -05:00
let key = seed.deserialize(StrDeserializer::new(key.clone()))?;
2018-12-17 19:45:35 -06:00
return Ok(Some(key));
}
// Rule out cases like:
//
// [[foo.bar]]
// [[foo]]
2018-12-17 19:45:35 -06:00
if table.array {
let kind = ErrorKind::RedefineAsArray;
2018-12-17 19:45:35 -06:00
return Err(self.de.error(table.at, kind));
}
2018-12-17 19:45:35 -06:00
self.values = table
.values
.take()
.expect("Unable to read table values")
.into_iter();
}
}
2017-04-20 12:16:00 -05:00
fn next_value_seed<V>(&mut self, seed: V) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::DeserializeSeed<'de>,
{
if let Some((k, v)) = self.next_value.take() {
match seed.deserialize(ValueDeserializer::new(v)) {
Ok(v) => return Ok(v),
Err(mut e) => {
e.add_key_context(&k);
2018-12-17 19:45:35 -06:00
return Err(e);
}
}
}
2018-12-17 19:45:35 -06:00
let array =
self.tables[self.cur].array && self.depth == self.tables[self.cur].header.len() - 1;
self.cur += 1;
let res = seed.deserialize(MapVisitor {
values: Vec::new().into_iter(),
next_value: None,
2018-12-17 19:45:35 -06:00
depth: self.depth + if array { 0 } else { 1 },
cur_parent: self.cur - 1,
cur: 0,
max: self.max,
array,
tables: &mut *self.tables,
de: &mut *self.de,
});
res.map_err(|mut e| {
e.add_key_context(&self.tables[self.cur - 1].header[self.depth]);
e
})
}
}
2017-04-20 12:16:00 -05:00
impl<'de, 'b> de::SeqAccess<'de> for MapVisitor<'de, 'b> {
type Error = Error;
2017-04-20 12:16:00 -05:00
fn next_element_seed<K>(&mut self, seed: K) -> Result<Option<K::Value>, Error>
2018-12-17 19:45:35 -06:00
where
K: de::DeserializeSeed<'de>,
{
assert!(self.next_value.is_none());
assert!(self.values.next().is_none());
if self.cur_parent == self.max {
2018-12-17 19:45:35 -06:00
return Ok(None);
}
let next = self.tables[..self.max]
.iter()
.enumerate()
.skip(self.cur_parent + 1)
2018-12-17 19:45:35 -06:00
.find(|&(_, table)| table.array && table.header == self.tables[self.cur_parent].header)
.map(|p| p.0)
.unwrap_or(self.max);
let ret = seed.deserialize(MapVisitor {
2018-12-17 19:45:35 -06:00
values: self.tables[self.cur_parent]
.values
.take()
.expect("Unable to read table values")
.into_iter(),
next_value: None,
depth: self.depth + 1,
cur_parent: self.cur_parent,
max: next,
cur: 0,
array: false,
tables: &mut self.tables,
de: &mut self.de,
})?;
self.cur_parent = next;
2017-03-30 06:38:48 -05:00
Ok(Some(ret))
}
}
2017-04-20 12:16:00 -05:00
impl<'de, 'b> de::Deserializer<'de> for MapVisitor<'de, 'b> {
type Error = Error;
2017-04-20 12:16:00 -05:00
fn deserialize_any<V>(self, visitor: V) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::Visitor<'de>,
{
2018-12-17 19:45:35 -06:00
if self.array {
visitor.visit_seq(self)
} else {
visitor.visit_map(self)
}
}
// `None` is interpreted as a missing field so be sure to implement `Some`
// as a present field.
fn deserialize_option<V>(self, visitor: V) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::Visitor<'de>,
{
visitor.visit_some(self)
}
fn deserialize_newtype_struct<V>(
self,
_name: &'static str,
2018-12-17 19:45:35 -06:00
visitor: V,
) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::Visitor<'de>,
{
visitor.visit_newtype_struct(self)
}
2019-05-08 19:37:38 -05:00
serde::forward_to_deserialize_any! {
bool u8 u16 u32 u64 i8 i16 i32 i64 f32 f64 char str string seq
bytes byte_buf map struct unit identifier
2017-04-20 12:16:00 -05:00
ignored_any unit_struct tuple_struct tuple enum
}
}
struct StrDeserializer<'a> {
key: Cow<'a, str>,
}
impl<'a> StrDeserializer<'a> {
fn new(key: Cow<'a, str>) -> StrDeserializer<'a> {
StrDeserializer { key }
}
}
2017-04-20 12:16:00 -05:00
impl<'de> de::Deserializer<'de> for StrDeserializer<'de> {
type Error = Error;
2017-04-20 12:16:00 -05:00
fn deserialize_any<V>(self, visitor: V) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::Visitor<'de>,
{
match self.key {
2017-04-20 12:16:00 -05:00
Cow::Borrowed(s) => visitor.visit_borrowed_str(s),
Cow::Owned(s) => visitor.visit_string(s),
}
}
2019-05-08 19:37:38 -05:00
serde::forward_to_deserialize_any! {
bool u8 u16 u32 u64 i8 i16 i32 i64 f32 f64 char str string seq
2017-04-20 12:16:00 -05:00
bytes byte_buf map struct option unit newtype_struct
ignored_any unit_struct tuple_struct tuple enum identifier
}
}
struct ValueDeserializer<'a> {
value: Value<'a>,
validate_struct_keys: bool,
}
impl<'a> ValueDeserializer<'a> {
fn new(value: Value<'a>) -> ValueDeserializer<'a> {
ValueDeserializer {
value,
validate_struct_keys: false,
}
}
fn with_struct_key_validation(mut self) -> Self {
self.validate_struct_keys = true;
self
}
}
2017-04-20 12:16:00 -05:00
impl<'de> de::Deserializer<'de> for ValueDeserializer<'de> {
type Error = Error;
2017-04-20 12:16:00 -05:00
fn deserialize_any<V>(self, visitor: V) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::Visitor<'de>,
{
2019-07-30 12:20:18 -05:00
let start = self.value.start;
let res = match self.value.e {
E::Integer(i) => visitor.visit_i64(i),
E::Boolean(b) => visitor.visit_bool(b),
E::Float(f) => visitor.visit_f64(f),
E::String(Cow::Borrowed(s)) => visitor.visit_borrowed_str(s),
E::String(Cow::Owned(s)) => visitor.visit_string(s),
E::Datetime(s) => visitor.visit_map(DatetimeDeserializer {
date: s,
visited: false,
}),
E::Array(values) => {
let mut s = de::value::SeqDeserializer::new(values.into_iter());
let ret = visitor.visit_seq(&mut s)?;
s.end()?;
Ok(ret)
}
E::InlineTable(values) | E::DottedTable(values) => {
visitor.visit_map(InlineTableDeserializer {
values: values.into_iter(),
next_value: None,
})
}
2019-07-30 12:20:18 -05:00
};
res.map_err(|mut err| {
// Attribute the error to whatever value returned the error.
err.fix_offset(|| Some(start));
err
})
}
2018-12-17 19:45:35 -06:00
fn deserialize_struct<V>(
self,
name: &'static str,
fields: &'static [&'static str],
visitor: V,
) -> Result<V::Value, Error>
where
V: de::Visitor<'de>,
{
if name == datetime::NAME && fields == [datetime::FIELD] {
if let E::Datetime(s) = self.value.e {
return visitor.visit_map(DatetimeDeserializer {
date: s,
visited: false,
2018-12-17 19:45:35 -06:00
});
}
}
if self.validate_struct_keys {
match self.value.e {
E::InlineTable(ref values) | E::DottedTable(ref values) => {
2018-12-17 19:45:35 -06:00
let extra_fields = values
.iter()
.filter_map(|key_value| {
let (ref key, ref _val) = *key_value;
if !fields.contains(&&(**key)) {
Some(key.clone())
} else {
None
}
})
.collect::<Vec<Cow<'de, str>>>();
if !extra_fields.is_empty() {
2019-07-30 12:20:18 -05:00
return Err(Error::from_kind(
Some(self.value.start),
ErrorKind::UnexpectedKeys {
keys: extra_fields
.iter()
.map(|k| k.to_string())
.collect::<Vec<_>>(),
available: fields,
},
));
}
}
_ => {}
}
}
if name == spanned::NAME && fields == [spanned::START, spanned::END, spanned::VALUE] {
let start = self.value.start;
let end = self.value.end;
return visitor.visit_map(SpannedDeserializer {
start: Some(start),
value: Some(self.value),
end: Some(end),
});
}
2017-04-20 12:16:00 -05:00
self.deserialize_any(visitor)
}
// `None` is interpreted as a missing field so be sure to implement `Some`
// as a present field.
fn deserialize_option<V>(self, visitor: V) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::Visitor<'de>,
{
visitor.visit_some(self)
}
2017-04-27 23:00:37 -05:00
fn deserialize_enum<V>(
self,
_name: &'static str,
_variants: &'static [&'static str],
2018-12-17 19:45:35 -06:00
visitor: V,
2017-04-27 23:00:37 -05:00
) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::Visitor<'de>,
2017-04-27 23:00:37 -05:00
{
match self.value.e {
E::String(val) => visitor.visit_enum(val.into_deserializer()),
E::InlineTable(values) => {
if values.len() != 1 {
2019-07-30 12:20:18 -05:00
Err(Error::from_kind(
Some(self.value.start),
ErrorKind::Wanted {
expected: "exactly 1 element",
found: if values.is_empty() {
"zero elements"
} else {
"more than 1 element"
},
},
2019-07-30 12:20:18 -05:00
))
} else {
visitor.visit_enum(InlineTableDeserializer {
values: values.into_iter(),
next_value: None,
})
}
}
e => Err(Error::from_kind(
2019-07-30 12:20:18 -05:00
Some(self.value.start),
ErrorKind::Wanted {
expected: "string or inline table",
found: e.type_name(),
},
)),
2017-04-27 23:00:37 -05:00
}
}
fn deserialize_newtype_struct<V>(
self,
_name: &'static str,
2018-12-17 19:45:35 -06:00
visitor: V,
) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::Visitor<'de>,
{
visitor.visit_newtype_struct(self)
}
2019-05-08 19:37:38 -05:00
serde::forward_to_deserialize_any! {
bool u8 u16 u32 u64 i8 i16 i32 i64 f32 f64 char str string seq
bytes byte_buf map unit identifier
2017-04-27 23:00:37 -05:00
ignored_any unit_struct tuple_struct tuple
}
}
2017-04-20 12:16:00 -05:00
impl<'de> de::IntoDeserializer<'de, Error> for Value<'de> {
type Deserializer = ValueDeserializer<'de>;
fn into_deserializer(self) -> Self::Deserializer {
ValueDeserializer::new(self)
}
}
struct SpannedDeserializer<'a> {
start: Option<usize>,
end: Option<usize>,
value: Option<Value<'a>>,
}
impl<'de> de::MapAccess<'de> for SpannedDeserializer<'de> {
type Error = Error;
fn next_key_seed<K>(&mut self, seed: K) -> Result<Option<K::Value>, Error>
where
K: de::DeserializeSeed<'de>,
{
if self.start.is_some() {
2018-12-17 19:45:35 -06:00
seed.deserialize(BorrowedStrDeserializer::new(spanned::START))
.map(Some)
} else if self.end.is_some() {
2018-12-17 19:45:35 -06:00
seed.deserialize(BorrowedStrDeserializer::new(spanned::END))
.map(Some)
} else if self.value.is_some() {
2018-12-17 19:45:35 -06:00
seed.deserialize(BorrowedStrDeserializer::new(spanned::VALUE))
.map(Some)
} else {
Ok(None)
}
}
fn next_value_seed<V>(&mut self, seed: V) -> Result<V::Value, Error>
where
V: de::DeserializeSeed<'de>,
{
if let Some(start) = self.start.take() {
seed.deserialize(start.into_deserializer())
2018-12-17 19:45:35 -06:00
} else if let Some(end) = self.end.take() {
seed.deserialize(end.into_deserializer())
} else if let Some(value) = self.value.take() {
seed.deserialize(value.into_deserializer())
} else {
panic!("next_value_seed called before next_key_seed")
}
}
}
struct DatetimeDeserializer<'a> {
visited: bool,
date: &'a str,
}
2017-04-20 12:16:00 -05:00
impl<'de> de::MapAccess<'de> for DatetimeDeserializer<'de> {
type Error = Error;
2017-04-20 12:16:00 -05:00
fn next_key_seed<K>(&mut self, seed: K) -> Result<Option<K::Value>, Error>
2018-12-17 19:45:35 -06:00
where
K: de::DeserializeSeed<'de>,
{
if self.visited {
2018-12-17 19:45:35 -06:00
return Ok(None);
}
self.visited = true;
seed.deserialize(DatetimeFieldDeserializer).map(Some)
}
2017-04-20 12:16:00 -05:00
fn next_value_seed<V>(&mut self, seed: V) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::DeserializeSeed<'de>,
{
seed.deserialize(StrDeserializer::new(self.date.into()))
}
}
struct DatetimeFieldDeserializer;
2017-04-20 12:16:00 -05:00
impl<'de> de::Deserializer<'de> for DatetimeFieldDeserializer {
type Error = Error;
2017-04-20 12:16:00 -05:00
fn deserialize_any<V>(self, visitor: V) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::Visitor<'de>,
{
visitor.visit_borrowed_str(datetime::FIELD)
}
2019-05-08 19:37:38 -05:00
serde::forward_to_deserialize_any! {
bool u8 u16 u32 u64 i8 i16 i32 i64 f32 f64 char str string seq
2017-04-20 12:16:00 -05:00
bytes byte_buf map struct option unit newtype_struct
ignored_any unit_struct tuple_struct tuple enum identifier
}
}
struct DottedTableDeserializer<'a> {
name: Cow<'a, str>,
value: Value<'a>,
}
impl<'de> de::EnumAccess<'de> for DottedTableDeserializer<'de> {
type Error = Error;
type Variant = TableEnumDeserializer<'de>;
fn variant_seed<V>(self, seed: V) -> Result<(V::Value, Self::Variant), Self::Error>
where
V: de::DeserializeSeed<'de>,
{
let (name, value) = (self.name, self.value);
seed.deserialize(StrDeserializer::new(name))
.map(|val| (val, TableEnumDeserializer { value }))
}
}
struct InlineTableDeserializer<'a> {
values: vec::IntoIter<TablePair<'a>>,
next_value: Option<Value<'a>>,
}
2017-04-20 12:16:00 -05:00
impl<'de> de::MapAccess<'de> for InlineTableDeserializer<'de> {
type Error = Error;
2017-04-20 12:16:00 -05:00
fn next_key_seed<K>(&mut self, seed: K) -> Result<Option<K::Value>, Error>
2018-12-17 19:45:35 -06:00
where
K: de::DeserializeSeed<'de>,
{
let (key, value) = match self.values.next() {
Some(pair) => pair,
None => return Ok(None),
};
self.next_value = Some(value);
seed.deserialize(StrDeserializer::new(key)).map(Some)
}
2017-04-20 12:16:00 -05:00
fn next_value_seed<V>(&mut self, seed: V) -> Result<V::Value, Error>
2018-12-17 19:45:35 -06:00
where
V: de::DeserializeSeed<'de>,
{
let value = self.next_value.take().expect("Unable to read table values");
seed.deserialize(ValueDeserializer::new(value))
}
}
impl<'de> de::EnumAccess<'de> for InlineTableDeserializer<'de> {
type Error = Error;
type Variant = TableEnumDeserializer<'de>;
fn variant_seed<V>(mut self, seed: V) -> Result<(V::Value, Self::Variant), Self::Error>
where
V: de::DeserializeSeed<'de>,
{
let (key, value) = match self.values.next() {
Some(pair) => pair,
None => {
2019-07-30 12:20:18 -05:00
return Err(Error::from_kind(
None, // FIXME: How do we get an offset here?
ErrorKind::Wanted {
expected: "table with exactly 1 entry",
found: "empty table",
},
));
}
};
seed.deserialize(StrDeserializer::new(key))
.map(|val| (val, TableEnumDeserializer { value }))
}
}
/// Deserializes table values into enum variants.
struct TableEnumDeserializer<'a> {
value: Value<'a>,
}
impl<'de> de::VariantAccess<'de> for TableEnumDeserializer<'de> {
type Error = Error;
fn unit_variant(self) -> Result<(), Self::Error> {
match self.value.e {
E::InlineTable(values) | E::DottedTable(values) => {
if values.is_empty() {
Ok(())
} else {
2019-07-30 12:20:18 -05:00
Err(Error::from_kind(
Some(self.value.start),
ErrorKind::ExpectedEmptyTable,
))
}
}
e => Err(Error::from_kind(
2019-07-30 12:20:18 -05:00
Some(self.value.start),
ErrorKind::Wanted {
expected: "table",
found: e.type_name(),
},
)),
}
}
fn newtype_variant_seed<T>(self, seed: T) -> Result<T::Value, Self::Error>
where
T: de::DeserializeSeed<'de>,
{
seed.deserialize(ValueDeserializer::new(self.value))
}
fn tuple_variant<V>(self, len: usize, visitor: V) -> Result<V::Value, Self::Error>
where
V: de::Visitor<'de>,
{
match self.value.e {
E::InlineTable(values) | E::DottedTable(values) => {
let tuple_values = values
.into_iter()
.enumerate()
.map(|(index, (key, value))| match key.parse::<usize>() {
Ok(key_index) if key_index == index => Ok(value),
2019-07-30 12:20:18 -05:00
Ok(_) | Err(_) => Err(Error::from_kind(
Some(value.start),
ErrorKind::ExpectedTupleIndex {
expected: index,
found: key.to_string(),
},
)),
})
// Fold all values into a `Vec`, or return the first error.
.fold(Ok(Vec::with_capacity(len)), |result, value_result| {
result.and_then(move |mut tuple_values| match value_result {
Ok(value) => {
tuple_values.push(value);
Ok(tuple_values)
}
// `Result<de::Value, Self::Error>` to `Result<Vec<_>, Self::Error>`
Err(e) => Err(e),
})
})?;
if tuple_values.len() == len {
de::Deserializer::deserialize_seq(
ValueDeserializer::new(Value {
e: E::Array(tuple_values),
start: self.value.start,
end: self.value.end,
}),
visitor,
)
} else {
2019-07-30 12:20:18 -05:00
Err(Error::from_kind(
Some(self.value.start),
ErrorKind::ExpectedTuple(len),
))
}
}
e => Err(Error::from_kind(
2019-07-30 12:20:18 -05:00
Some(self.value.start),
ErrorKind::Wanted {
expected: "table",
found: e.type_name(),
},
)),
}
}
fn struct_variant<V>(
self,
fields: &'static [&'static str],
visitor: V,
) -> Result<V::Value, Self::Error>
where
V: de::Visitor<'de>,
{
de::Deserializer::deserialize_struct(
ValueDeserializer::new(self.value).with_struct_key_validation(),
"", // TODO: this should be the variant name
fields,
visitor,
)
}
}
2017-04-24 23:57:35 -05:00
impl<'a> Deserializer<'a> {
/// Creates a new deserializer which will be deserializing the string
/// provided.
pub fn new(input: &'a str) -> Deserializer<'a> {
Deserializer {
tokens: Tokenizer::new(input),
input,
require_newline_after_table: true,
allow_duplciate_after_longer_table: false,
}
}
/// The `Deserializer::end` method should be called after a value has been
/// fully deserialized. This allows the `Deserializer` to validate that the
/// input stream is at the end or that it only has trailing
/// whitespace/comments.
pub fn end(&mut self) -> Result<(), Error> {
Ok(())
}
/// Historical versions of toml-rs accidentally allowed a newline after a
/// table definition, but the TOML spec requires a newline after a table
/// definition header.
///
/// This option can be set to `false` (the default is `true`) to emulate
/// this behavior for backwards compatibility with older toml-rs versions.
pub fn set_require_newline_after_table(&mut self, require: bool) {
self.require_newline_after_table = require;
}
/// Historical versions of toml-rs accidentally allowed a duplicate table
/// header after a longer table header was previously defined. This is
/// invalid according to the TOML spec, however.
///
/// This option can be set to `true` (the default is `false`) to emulate
/// this behavior for backwards compatibility with older toml-rs versions.
pub fn set_allow_duplicate_after_longer_table(&mut self, allow: bool) {
self.allow_duplciate_after_longer_table = allow;
}
fn tables(&mut self) -> Result<Vec<Table<'a>>, Error> {
let mut tables = Vec::new();
let mut cur_table = Table {
at: 0,
header: Vec::new(),
values: None,
array: false,
};
while let Some(line) = self.line()? {
match line {
Line::Table {
at,
mut header,
array,
} => {
if !cur_table.header.is_empty() || cur_table.values.is_some() {
tables.push(cur_table);
}
cur_table = Table {
at,
header: Vec::new(),
values: Some(Vec::new()),
array,
};
loop {
let part = header.next().map_err(|e| self.token_error(e));
match part? {
Some(part) => cur_table.header.push(part),
None => break,
}
}
}
Line::KeyValue(key, value) => {
if cur_table.values.is_none() {
cur_table.values = Some(Vec::new());
}
self.add_dotted_key(key, value, cur_table.values.as_mut().unwrap())?;
}
}
}
if !cur_table.header.is_empty() || cur_table.values.is_some() {
tables.push(cur_table);
}
Ok(tables)
}
fn line(&mut self) -> Result<Option<Line<'a>>, Error> {
loop {
self.eat_whitespace()?;
if self.eat_comment()? {
2018-12-17 19:45:35 -06:00
continue;
}
if self.eat(Token::Newline)? {
2018-12-17 19:45:35 -06:00
continue;
}
2018-12-17 19:45:35 -06:00
break;
}
match self.peek()? {
Some((_, Token::LeftBracket)) => self.table_header().map(Some),
Some(_) => self.key_value().map(Some),
None => Ok(None),
}
}
fn table_header(&mut self) -> Result<Line<'a>, Error> {
let start = self.tokens.current();
self.expect(Token::LeftBracket)?;
let array = self.eat(Token::LeftBracket)?;
2018-12-17 19:45:35 -06:00
let ret = Header::new(self.tokens.clone(), array, self.require_newline_after_table);
if self.require_newline_after_table {
self.tokens.skip_to_newline();
} else {
loop {
match self.next()? {
Some((_, Token::RightBracket)) => {
2017-03-31 20:45:00 -05:00
if array {
self.eat(Token::RightBracket)?;
}
2018-12-17 19:45:35 -06:00
break;
2017-03-31 20:45:00 -05:00
}
2018-12-17 19:45:35 -06:00
Some((_, Token::Newline)) | None => break,
_ => {}
}
}
self.eat_whitespace()?;
}
2018-12-17 19:45:35 -06:00
Ok(Line::Table {
at: start,
header: ret,
array,
2018-12-17 19:45:35 -06:00
})
}
fn key_value(&mut self) -> Result<Line<'a>, Error> {
2018-07-11 01:29:47 -05:00
let key = self.dotted_key()?;
self.eat_whitespace()?;
self.expect(Token::Equals)?;
self.eat_whitespace()?;
let value = self.value()?;
self.eat_whitespace()?;
if !self.eat_comment()? {
self.eat_newline_or_eof()?;
}
Ok(Line::KeyValue(key, value))
}
fn value(&mut self) -> Result<Value<'a>, Error> {
let at = self.tokens.current();
let value = match self.next()? {
2018-12-17 19:45:35 -06:00
Some((Span { start, end }, Token::String { val, .. })) => Value {
e: E::String(val),
start,
end,
2018-12-17 19:45:35 -06:00
},
Some((Span { start, end }, Token::Keylike("true"))) => Value {
e: E::Boolean(true),
start,
end,
2018-12-17 19:45:35 -06:00
},
Some((Span { start, end }, Token::Keylike("false"))) => Value {
e: E::Boolean(false),
start,
end,
2018-12-17 19:45:35 -06:00
},
Some((span, Token::Keylike(key))) => self.number_or_date(span, key)?,
2018-05-06 21:05:05 -05:00
Some((span, Token::Plus)) => self.number_leading_plus(span)?,
Some((Span { start, .. }, Token::LeftBrace)) => {
self.inline_table().map(|(Span { end, .. }, table)| Value {
e: E::InlineTable(table),
start,
end,
})?
}
2018-05-06 21:05:05 -05:00
Some((Span { start, .. }, Token::LeftBracket)) => {
self.array().map(|(Span { end, .. }, array)| Value {
e: E::Array(array),
start,
end,
2018-05-06 21:05:05 -05:00
})?
}
Some(token) => {
2018-12-17 19:45:35 -06:00
return Err(self.error(
at,
ErrorKind::Wanted {
expected: "a value",
found: token.1.describe(),
},
))
}
None => return Err(self.eof()),
};
Ok(value)
}
2018-12-17 19:45:35 -06:00
fn number_or_date(&mut self, span: Span, s: &'a str) -> Result<Value<'a>, Error> {
2019-02-19 12:59:42 -06:00
if s.contains('T')
|| s.contains('t')
|| (s.len() > 1 && s[1..].contains('-') && !s.contains("e-") && !s.contains("E-"))
{
2018-12-17 19:45:35 -06:00
self.datetime(span, s, false)
.map(|(Span { start, end }, d)| Value {
e: E::Datetime(d),
start,
end,
2018-12-17 19:45:35 -06:00
})
} else if self.eat(Token::Colon)? {
2018-12-17 19:45:35 -06:00
self.datetime(span, s, true)
.map(|(Span { start, end }, d)| Value {
e: E::Datetime(d),
start,
end,
2018-12-17 19:45:35 -06:00
})
} else {
self.number(span, s)
}
}
/// Returns a string or table value type.
///
/// Used to deserialize enums. Unit enums may be represented as a string or a table, all other
/// structures (tuple, newtype, struct) must be represented as a table.
fn string_or_table(&mut self) -> Result<(Value<'a>, Option<Cow<'a, str>>), Error> {
match self.peek()? {
2019-07-30 12:20:18 -05:00
Some((span, Token::LeftBracket)) => {
let tables = self.tables()?;
if tables.len() != 1 {
2019-07-30 12:20:18 -05:00
return Err(Error::from_kind(
Some(span.start),
ErrorKind::Wanted {
expected: "exactly 1 table",
found: if tables.is_empty() {
"zero tables"
} else {
"more than 1 table"
},
},
2019-07-30 12:20:18 -05:00
));
}
let table = tables
.into_iter()
.next()
.expect("Expected exactly one table");
let header = table
.header
.last()
.expect("Expected at least one header value for table.");
let start = table.at;
let end = table
.values
.as_ref()
.and_then(|values| values.last())
.map(|&(_, ref val)| val.end)
.unwrap_or_else(|| header.len());
Ok((
Value {
e: E::DottedTable(table.values.unwrap_or_else(Vec::new)),
start,
end,
},
Some(header.clone()),
))
}
Some(_) => self.value().map(|val| (val, None)),
None => Err(self.eof()),
}
}
fn number(&mut self, Span { start, end }: Span, s: &'a str) -> Result<Value<'a>, Error> {
2018-12-17 19:45:35 -06:00
let to_integer = |f| Value {
e: E::Integer(f),
start,
end,
2018-12-17 19:45:35 -06:00
};
if s.starts_with("0x") {
self.integer(&s[2..], 16).map(to_integer)
} else if s.starts_with("0o") {
self.integer(&s[2..], 8).map(to_integer)
} else if s.starts_with("0b") {
self.integer(&s[2..], 2).map(to_integer)
} else if s.contains('e') || s.contains('E') {
2018-12-17 19:45:35 -06:00
self.float(s, None).map(|f| Value {
e: E::Float(f),
start,
end,
2018-12-17 19:45:35 -06:00
})
} else if self.eat(Token::Period)? {
let at = self.tokens.current();
match self.next()? {
Some((Span { start, end }, Token::Keylike(after))) => {
self.float(s, Some(after)).map(|f| Value {
2018-12-17 19:45:35 -06:00
e: E::Float(f),
start,
end,
})
}
_ => Err(self.error(at, ErrorKind::NumberInvalid)),
}
} else if s == "inf" {
2018-12-17 19:45:35 -06:00
Ok(Value {
e: E::Float(f64::INFINITY),
start,
end,
2018-12-17 19:45:35 -06:00
})
} else if s == "-inf" {
2018-12-17 19:45:35 -06:00
Ok(Value {
e: E::Float(f64::NEG_INFINITY),
start,
end,
2018-12-17 19:45:35 -06:00
})
} else if s == "nan" {
2018-12-17 19:45:35 -06:00
Ok(Value {
e: E::Float(f64::NAN),
start,
end,
2018-12-17 19:45:35 -06:00
})
} else if s == "-nan" {
2018-12-17 19:45:35 -06:00
Ok(Value {
e: E::Float(-f64::NAN),
start,
end,
2018-12-17 19:45:35 -06:00
})
} else {
self.integer(s, 10).map(to_integer)
}
}
2018-05-06 21:05:05 -05:00
fn number_leading_plus(&mut self, Span { start, .. }: Span) -> Result<Value<'a>, Error> {
let start_token = self.tokens.current();
match self.next()? {
Some((Span { end, .. }, Token::Keylike(s))) => self.number(Span { start, end }, s),
2018-05-06 21:05:05 -05:00
_ => Err(self.error(start_token, ErrorKind::NumberInvalid)),
}
}
fn integer(&self, s: &'a str, radix: u32) -> Result<i64, Error> {
let allow_sign = radix == 10;
let allow_leading_zeros = radix != 10;
let (prefix, suffix) = self.parse_integer(s, allow_sign, allow_leading_zeros, radix)?;
let start = self.tokens.substr_offset(s);
if suffix != "" {
2018-12-17 19:45:35 -06:00
return Err(self.error(start, ErrorKind::NumberInvalid));
}
i64::from_str_radix(&prefix.replace("_", "").trim_start_matches('+'), radix)
.map_err(|_e| self.error(start, ErrorKind::NumberInvalid))
}
fn parse_integer(
&self,
s: &'a str,
allow_sign: bool,
allow_leading_zeros: bool,
radix: u32,
) -> Result<(&'a str, &'a str), Error> {
let start = self.tokens.substr_offset(s);
let mut first = true;
let mut first_zero = false;
let mut underscore = false;
let mut end = s.len();
for (i, c) in s.char_indices() {
let at = i + start;
if i == 0 && (c == '+' || c == '-') && allow_sign {
2018-12-17 19:45:35 -06:00
continue;
}
if c == '0' && first {
first_zero = true;
} else if c.to_digit(radix).is_some() {
if !first && first_zero && !allow_leading_zeros {
return Err(self.error(at, ErrorKind::NumberInvalid));
}
underscore = false;
} else if c == '_' && first {
return Err(self.error(at, ErrorKind::NumberInvalid));
} else if c == '_' && !underscore {
underscore = true;
} else {
end = i;
break;
}
first = false;
}
if first || underscore {
2018-12-17 19:45:35 -06:00
return Err(self.error(start, ErrorKind::NumberInvalid));
}
Ok((&s[..end], &s[end..]))
}
2018-12-17 19:45:35 -06:00
fn float(&mut self, s: &'a str, after_decimal: Option<&'a str>) -> Result<f64, Error> {
let (integral, mut suffix) = self.parse_integer(s, true, false, 10)?;
let start = self.tokens.substr_offset(integral);
let mut fraction = None;
if let Some(after) = after_decimal {
if suffix != "" {
2018-12-17 19:45:35 -06:00
return Err(self.error(start, ErrorKind::NumberInvalid));
}
let (a, b) = self.parse_integer(after, false, true, 10)?;
fraction = Some(a);
suffix = b;
}
let mut exponent = None;
2017-03-30 06:40:27 -05:00
if suffix.starts_with('e') || suffix.starts_with('E') {
let (a, b) = if suffix.len() == 1 {
self.eat(Token::Plus)?;
match self.next()? {
Some((_, Token::Keylike(s))) => self.parse_integer(s, false, true, 10)?,
_ => return Err(self.error(start, ErrorKind::NumberInvalid)),
}
} else {
self.parse_integer(&suffix[1..], true, true, 10)?
};
if b != "" {
2018-12-17 19:45:35 -06:00
return Err(self.error(start, ErrorKind::NumberInvalid));
}
exponent = Some(a);
2019-04-05 15:51:28 -05:00
} else if !suffix.is_empty() {
return Err(self.error(start, ErrorKind::NumberInvalid));
}
2018-12-17 19:45:35 -06:00
let mut number = integral
.trim_start_matches('+')
2018-12-17 19:45:35 -06:00
.chars()
.filter(|c| *c != '_')
.collect::<String>();
if let Some(fraction) = fraction {
number.push_str(".");
number.extend(fraction.chars().filter(|c| *c != '_'));
}
if let Some(exponent) = exponent {
number.push_str("E");
number.extend(exponent.chars().filter(|c| *c != '_'));
}
2018-12-17 19:45:35 -06:00
number
.parse()
.map_err(|_e| self.error(start, ErrorKind::NumberInvalid))
.and_then(|n: f64| {
if n.is_finite() {
Ok(n)
} else {
Err(self.error(start, ErrorKind::NumberInvalid))
}
})
}
2018-12-17 19:45:35 -06:00
fn datetime(
&mut self,
mut span: Span,
date: &'a str,
colon_eaten: bool,
) -> Result<(Span, &'a str), Error> {
let start = self.tokens.substr_offset(date);
// Check for space separated date and time.
let mut lookahead = self.tokens.clone();
if let Ok(Some((_, Token::Whitespace(" ")))) = lookahead.next() {
// Check if hour follows.
if let Ok(Some((_, Token::Keylike(_)))) = lookahead.next() {
2018-12-17 19:45:35 -06:00
self.next()?; // skip space
self.next()?; // skip keylike hour
}
}
if colon_eaten || self.eat(Token::Colon)? {
// minutes
match self.next()? {
Some((_, Token::Keylike(_))) => {}
_ => return Err(self.error(start, ErrorKind::DateInvalid)),
}
// Seconds
self.expect(Token::Colon)?;
match self.next()? {
Some((Span { end, .. }, Token::Keylike(_))) => {
span.end = end;
2018-12-17 19:45:35 -06:00
}
_ => return Err(self.error(start, ErrorKind::DateInvalid)),
}
// Fractional seconds
if self.eat(Token::Period)? {
match self.next()? {
Some((Span { end, .. }, Token::Keylike(_))) => {
span.end = end;
2018-12-17 19:45:35 -06:00
}
_ => return Err(self.error(start, ErrorKind::DateInvalid)),
}
}
// offset
if self.eat(Token::Plus)? {
match self.next()? {
Some((Span { end, .. }, Token::Keylike(_))) => {
span.end = end;
2018-12-17 19:45:35 -06:00
}
_ => return Err(self.error(start, ErrorKind::DateInvalid)),
}
}
if self.eat(Token::Colon)? {
match self.next()? {
Some((Span { end, .. }, Token::Keylike(_))) => {
span.end = end;
2018-12-17 19:45:35 -06:00
}
_ => return Err(self.error(start, ErrorKind::DateInvalid)),
}
}
}
let end = self.tokens.current();
Ok((span, &self.tokens.input()[start..end]))
}
// TODO(#140): shouldn't buffer up this entire table in memory, it'd be
// great to defer parsing everything until later.
fn inline_table(&mut self) -> Result<(Span, Vec<TablePair<'a>>), Error> {
let mut ret = Vec::new();
self.eat_whitespace()?;
2018-05-06 21:05:05 -05:00
if let Some(span) = self.eat_spanned(Token::RightBrace)? {
2018-12-17 19:45:35 -06:00
return Ok((span, ret));
}
loop {
2018-07-11 01:29:47 -05:00
let key = self.dotted_key()?;
self.eat_whitespace()?;
self.expect(Token::Equals)?;
self.eat_whitespace()?;
2018-07-11 01:29:47 -05:00
let value = self.value()?;
self.add_dotted_key(key, value, &mut ret)?;
self.eat_whitespace()?;
2018-05-06 21:05:05 -05:00
if let Some(span) = self.eat_spanned(Token::RightBrace)? {
2018-12-17 19:45:35 -06:00
return Ok((span, ret));
}
self.expect(Token::Comma)?;
self.eat_whitespace()?;
}
}
// TODO(#140): shouldn't buffer up this entire array in memory, it'd be
// great to defer parsing everything until later.
2018-05-06 21:05:05 -05:00
fn array(&mut self) -> Result<(Span, Vec<Value<'a>>), Error> {
let mut ret = Vec::new();
2019-05-08 19:37:38 -05:00
let intermediate = |me: &mut Deserializer<'_>| {
loop {
me.eat_whitespace()?;
if !me.eat(Token::Newline)? && !me.eat_comment()? {
2018-12-17 19:45:35 -06:00
break;
}
}
Ok(())
};
loop {
intermediate(self)?;
2018-05-06 21:05:05 -05:00
if let Some(span) = self.eat_spanned(Token::RightBracket)? {
2018-12-17 19:45:35 -06:00
return Ok((span, ret));
}
let at = self.tokens.current();
let value = self.value()?;
if let Some(last) = ret.last() {
if !value.same_type(last) {
2018-12-17 19:45:35 -06:00
return Err(self.error(at, ErrorKind::MixedArrayType));
}
}
ret.push(value);
intermediate(self)?;
if !self.eat(Token::Comma)? {
2018-12-17 19:45:35 -06:00
break;
}
}
intermediate(self)?;
2018-05-06 21:05:05 -05:00
let span = self.expect_spanned(Token::RightBracket)?;
Ok((span, ret))
}
fn table_key(&mut self) -> Result<Cow<'a, str>, Error> {
2018-12-17 19:45:35 -06:00
self.tokens
.table_key()
.map(|t| t.1)
.map_err(|e| self.token_error(e))
}
2018-07-11 01:29:47 -05:00
fn dotted_key(&mut self) -> Result<Vec<Cow<'a, str>>, Error> {
let mut result = Vec::new();
result.push(self.table_key()?);
self.eat_whitespace()?;
while self.eat(Token::Period)? {
self.eat_whitespace()?;
result.push(self.table_key()?);
self.eat_whitespace()?;
}
Ok(result)
}
/// Stores a value in the appropriate hierachical structure positioned based on the dotted key.
///
/// Given the following definition: `multi.part.key = "value"`, `multi` and `part` are
/// intermediate parts which are mapped to the relevant fields in the deserialized type's data
/// hierarchy.
///
/// # Parameters
///
/// * `key_parts`: Each segment of the dotted key, e.g. `part.one` maps to
/// `vec![Cow::Borrowed("part"), Cow::Borrowed("one")].`
/// * `value`: The parsed value.
/// * `values`: The `Vec` to store the value in.
2018-07-11 01:29:47 -05:00
fn add_dotted_key(
&self,
mut key_parts: Vec<Cow<'a, str>>,
value: Value<'a>,
values: &mut Vec<TablePair<'a>>,
2018-07-11 01:29:47 -05:00
) -> Result<(), Error> {
let key = key_parts.remove(0);
if key_parts.is_empty() {
values.push((key, value));
return Ok(());
}
2018-07-11 02:13:47 -05:00
match values.iter_mut().find(|&&mut (ref k, _)| *k == key) {
2018-12-17 19:45:35 -06:00
Some(&mut (
_,
Value {
e: E::DottedTable(ref mut v),
..
},
)) => {
2018-07-11 01:29:47 -05:00
return self.add_dotted_key(key_parts, value, v);
}
2018-07-11 02:13:47 -05:00
Some(&mut (_, Value { start, .. })) => {
return Err(self.error(start, ErrorKind::DottedKeyInvalidType));
2018-07-11 01:29:47 -05:00
}
None => {}
}
// The start/end value is somewhat misleading here.
let table_values = Value {
e: E::DottedTable(Vec::new()),
2018-07-11 01:29:47 -05:00
start: value.start,
end: value.end,
};
values.push((key, table_values));
2018-07-11 01:29:47 -05:00
let last_i = values.len() - 1;
2018-12-17 19:45:35 -06:00
if let (
_,
Value {
e: E::DottedTable(ref mut v),
..
},
) = values[last_i]
{
2018-07-11 01:29:47 -05:00
self.add_dotted_key(key_parts, value, v)?;
}
Ok(())
}
fn eat_whitespace(&mut self) -> Result<(), Error> {
2018-12-17 19:45:35 -06:00
self.tokens
.eat_whitespace()
.map_err(|e| self.token_error(e))
}
fn eat_comment(&mut self) -> Result<bool, Error> {
self.tokens.eat_comment().map_err(|e| self.token_error(e))
}
fn eat_newline_or_eof(&mut self) -> Result<(), Error> {
2018-12-17 19:45:35 -06:00
self.tokens
.eat_newline_or_eof()
.map_err(|e| self.token_error(e))
}
fn eat(&mut self, expected: Token<'a>) -> Result<bool, Error> {
self.tokens.eat(expected).map_err(|e| self.token_error(e))
}
2018-05-06 21:05:05 -05:00
fn eat_spanned(&mut self, expected: Token<'a>) -> Result<Option<Span>, Error> {
2018-12-17 19:45:35 -06:00
self.tokens
.eat_spanned(expected)
.map_err(|e| self.token_error(e))
2018-05-06 21:05:05 -05:00
}
fn expect(&mut self, expected: Token<'a>) -> Result<(), Error> {
2018-12-17 19:45:35 -06:00
self.tokens
.expect(expected)
.map_err(|e| self.token_error(e))
}
2018-05-06 21:05:05 -05:00
fn expect_spanned(&mut self, expected: Token<'a>) -> Result<Span, Error> {
2018-12-17 19:45:35 -06:00
self.tokens
.expect_spanned(expected)
.map_err(|e| self.token_error(e))
2018-05-06 21:05:05 -05:00
}
fn next(&mut self) -> Result<Option<(Span, Token<'a>)>, Error> {
self.tokens.next().map_err(|e| self.token_error(e))
}
fn peek(&mut self) -> Result<Option<(Span, Token<'a>)>, Error> {
self.tokens.peek().map_err(|e| self.token_error(e))
}
fn eof(&self) -> Error {
self.error(self.input.len(), ErrorKind::UnexpectedEof)
}
fn token_error(&self, error: TokenError) -> Error {
match error {
TokenError::InvalidCharInString(at, ch) => {
self.error(at, ErrorKind::InvalidCharInString(ch))
}
2018-12-17 19:45:35 -06:00
TokenError::InvalidEscape(at, ch) => self.error(at, ErrorKind::InvalidEscape(ch)),
TokenError::InvalidEscapeValue(at, v) => {
self.error(at, ErrorKind::InvalidEscapeValue(v))
}
2018-12-17 19:45:35 -06:00
TokenError::InvalidHexEscape(at, ch) => self.error(at, ErrorKind::InvalidHexEscape(ch)),
TokenError::NewlineInString(at) => self.error(at, ErrorKind::NewlineInString),
TokenError::Unexpected(at, ch) => self.error(at, ErrorKind::Unexpected(ch)),
TokenError::UnterminatedString(at) => self.error(at, ErrorKind::UnterminatedString),
TokenError::NewlineInTableKey(at) => self.error(at, ErrorKind::NewlineInTableKey),
TokenError::Wanted {
at,
expected,
found,
} => self.error(at, ErrorKind::Wanted { expected, found }),
2018-12-17 19:45:35 -06:00
TokenError::EmptyTableKey(at) => self.error(at, ErrorKind::EmptyTableKey),
TokenError::MultilineStringKey(at) => self.error(at, ErrorKind::MultilineStringKey),
}
}
fn error(&self, at: usize, kind: ErrorKind) -> Error {
2019-07-30 12:20:18 -05:00
let mut err = Error::from_kind(Some(at), kind);
err.fix_linecol(|at| self.to_linecol(at));
2017-03-30 06:38:48 -05:00
err
}
/// Converts a byte offset from an error message to a (line, column) pair
///
/// All indexes are 0-based.
fn to_linecol(&self, offset: usize) -> (usize, usize) {
let mut cur = 0;
// Use split_terminator instead of lines so that if there is a `\r`,
// it is included in the offset calculation. The `+1` values below
// account for the `\n`.
for (i, line) in self.input.split_terminator('\n').enumerate() {
if cur + line.len() + 1 > offset {
2018-12-17 19:45:35 -06:00
return (i, offset - cur);
}
cur += line.len() + 1;
}
(self.input.lines().count(), 0)
}
}
impl Error {
/// Produces a (line, column) pair of the position of the error if available
///
/// All indexes are 0-based.
pub fn line_col(&self) -> Option<(usize, usize)> {
self.inner.line.map(|line| (line, self.inner.col))
}
2019-07-30 12:20:18 -05:00
fn from_kind(at: Option<usize>, kind: ErrorKind) -> Error {
Error {
inner: Box::new(ErrorInner {
kind,
line: None,
col: 0,
2019-07-30 12:20:18 -05:00
at,
message: String::new(),
key: Vec::new(),
}),
}
}
2019-07-30 12:20:18 -05:00
fn custom(at: Option<usize>, s: String) -> Error {
Error {
inner: Box::new(ErrorInner {
kind: ErrorKind::Custom,
line: None,
col: 0,
2019-07-30 12:20:18 -05:00
at,
message: s,
key: Vec::new(),
}),
}
}
/// Do not call this method, it may be removed at any time, it's just an
/// internal implementation detail.
#[doc(hidden)]
pub fn add_key_context(&mut self, key: &str) {
self.inner.key.insert(0, key.to_string());
}
2019-07-30 12:20:18 -05:00
fn fix_offset<F>(&mut self, f: F)
2019-07-30 12:20:18 -05:00
where
F: FnOnce() -> Option<usize>,
{
// An existing offset is always better positioned than anything we
// might want to add later.
if self.inner.at.is_none() {
self.inner.at = f();
}
}
fn fix_linecol<F>(&mut self, f: F)
2019-07-30 12:20:18 -05:00
where
F: FnOnce(usize) -> (usize, usize),
{
if let Some(at) = self.inner.at {
let (line, col) = f(at);
self.inner.line = Some(line);
self.inner.col = col;
}
}
}
2019-08-12 15:50:44 -05:00
impl std::convert::From<Error> for std::io::Error {
fn from(e: Error) -> Self {
std::io::Error::new(std::io::ErrorKind::InvalidData, e.to_string())
2019-08-12 15:50:44 -05:00
}
}
impl fmt::Display for Error {
2019-05-08 19:37:38 -05:00
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self.inner.kind {
ErrorKind::UnexpectedEof => "unexpected eof encountered".fmt(f)?,
2018-12-17 19:45:35 -06:00
ErrorKind::InvalidCharInString(c) => write!(
f,
"invalid character in string: `{}`",
c.escape_default().collect::<String>()
)?,
ErrorKind::InvalidEscape(c) => write!(
f,
"invalid escape character in string: `{}`",
c.escape_default().collect::<String>()
)?,
ErrorKind::InvalidHexEscape(c) => write!(
f,
"invalid hex escape character in string: `{}`",
c.escape_default().collect::<String>()
)?,
ErrorKind::InvalidEscapeValue(c) => write!(f, "invalid escape value: `{}`", c)?,
ErrorKind::NewlineInString => "newline in string found".fmt(f)?,
2018-12-17 19:45:35 -06:00
ErrorKind::Unexpected(ch) => write!(
f,
"unexpected character found: `{}`",
ch.escape_default().collect::<String>()
)?,
ErrorKind::UnterminatedString => "unterminated string".fmt(f)?,
ErrorKind::NewlineInTableKey => "found newline in table key".fmt(f)?,
ErrorKind::Wanted { expected, found } => {
write!(f, "expected {}, found {}", expected, found)?
}
ErrorKind::NumberInvalid => "invalid number".fmt(f)?,
ErrorKind::DateInvalid => "invalid date".fmt(f)?,
ErrorKind::MixedArrayType => "mixed types in an array".fmt(f)?,
ErrorKind::DuplicateTable(ref s) => {
write!(f, "redefinition of table `{}`", s)?;
}
ErrorKind::RedefineAsArray => "table redefined as array".fmt(f)?,
ErrorKind::EmptyTableKey => "empty table key found".fmt(f)?,
ErrorKind::MultilineStringKey => "multiline strings are not allowed for key".fmt(f)?,
ErrorKind::Custom => self.inner.message.fmt(f)?,
ErrorKind::ExpectedTuple(l) => write!(f, "expected table with length {}", l)?,
ErrorKind::ExpectedTupleIndex {
expected,
ref found,
} => write!(f, "expected table key `{}`, but was `{}`", expected, found)?,
ErrorKind::ExpectedEmptyTable => "expected empty table".fmt(f)?,
ErrorKind::DottedKeyInvalidType => {
"dotted key attempted to extend non-table type".fmt(f)?
}
2018-12-17 19:45:35 -06:00
ErrorKind::UnexpectedKeys {
ref keys,
available,
} => write!(
f,
"unexpected keys in table: `{:?}`, available keys: `{:?}`",
keys, available
)?,
ErrorKind::__Nonexhaustive => panic!(),
}
2017-03-30 06:39:46 -05:00
if !self.inner.key.is_empty() {
write!(f, " for key `")?;
for (i, k) in self.inner.key.iter().enumerate() {
if i > 0 {
write!(f, ".")?;
}
write!(f, "{}", k)?;
}
write!(f, "`")?;
}
if let Some(line) = self.inner.line {
2019-07-28 12:40:49 -05:00
write!(f, " at line {} column {}", line + 1, self.inner.col + 1)?;
}
Ok(())
}
}
impl error::Error for Error {
fn description(&self) -> &str {
match self.inner.kind {
ErrorKind::UnexpectedEof => "unexpected eof encountered",
ErrorKind::InvalidCharInString(_) => "invalid char in string",
ErrorKind::InvalidEscape(_) => "invalid escape in string",
ErrorKind::InvalidHexEscape(_) => "invalid hex escape in string",
ErrorKind::InvalidEscapeValue(_) => "invalid escape value in string",
ErrorKind::NewlineInString => "newline in string found",
ErrorKind::Unexpected(_) => "unexpected or invalid character",
ErrorKind::UnterminatedString => "unterminated string",
ErrorKind::NewlineInTableKey => "found newline in table key",
ErrorKind::Wanted { .. } => "expected a token but found another",
ErrorKind::NumberInvalid => "invalid number",
ErrorKind::DateInvalid => "invalid date",
ErrorKind::MixedArrayType => "mixed types in an array",
ErrorKind::DuplicateTable(_) => "duplicate table",
ErrorKind::RedefineAsArray => "table redefined as array",
ErrorKind::EmptyTableKey => "empty table key found",
ErrorKind::MultilineStringKey => "invalid multiline string for key",
ErrorKind::Custom => "a custom error",
ErrorKind::ExpectedTuple(_) => "expected table length",
ErrorKind::ExpectedTupleIndex { .. } => "expected table key",
ErrorKind::ExpectedEmptyTable => "expected empty table",
2018-07-11 01:29:47 -05:00
ErrorKind::DottedKeyInvalidType => "dotted key invalid type",
ErrorKind::UnexpectedKeys { .. } => "unexpected keys in table",
ErrorKind::__Nonexhaustive => panic!(),
}
}
}
impl de::Error for Error {
fn custom<T: fmt::Display>(msg: T) -> Error {
2019-07-30 12:20:18 -05:00
Error::custom(None, msg.to_string())
}
}
enum Line<'a> {
2018-12-17 19:45:35 -06:00
Table {
at: usize,
header: Header<'a>,
array: bool,
},
2018-07-11 01:29:47 -05:00
KeyValue(Vec<Cow<'a, str>>, Value<'a>),
}
struct Header<'a> {
first: bool,
array: bool,
require_newline_after_table: bool,
tokens: Tokenizer<'a>,
}
impl<'a> Header<'a> {
2018-12-17 19:45:35 -06:00
fn new(tokens: Tokenizer<'a>, array: bool, require_newline_after_table: bool) -> Header<'a> {
Header {
first: true,
array,
tokens,
require_newline_after_table,
}
}
fn next(&mut self) -> Result<Option<Cow<'a, str>>, TokenError> {
self.tokens.eat_whitespace()?;
if self.first || self.tokens.eat(Token::Period)? {
self.first = false;
self.tokens.eat_whitespace()?;
2018-05-01 06:43:02 -05:00
self.tokens.table_key().map(|t| t.1).map(Some)
} else {
self.tokens.expect(Token::RightBracket)?;
if self.array {
self.tokens.expect(Token::RightBracket)?;
}
self.tokens.eat_whitespace()?;
if self.require_newline_after_table && !self.tokens.eat_comment()? {
self.tokens.eat_newline_or_eof()?;
}
Ok(None)
}
}
}
2017-04-24 23:57:35 -05:00
#[derive(Debug)]
struct Value<'a> {
e: E<'a>,
start: usize,
end: usize,
}
#[derive(Debug)]
enum E<'a> {
Integer(i64),
Float(f64),
Boolean(bool),
String(Cow<'a, str>),
Datetime(&'a str),
Array(Vec<Value<'a>>),
InlineTable(Vec<TablePair<'a>>),
DottedTable(Vec<TablePair<'a>>),
}
impl<'a> E<'a> {
fn type_name(&self) -> &'static str {
match *self {
E::String(..) => "string",
E::Integer(..) => "integer",
E::Float(..) => "float",
E::Boolean(..) => "boolean",
E::Datetime(..) => "datetime",
E::Array(..) => "array",
E::InlineTable(..) => "inline table",
E::DottedTable(..) => "dotted table",
}
}
}
impl<'a> Value<'a> {
fn same_type(&self, other: &Value<'a>) -> bool {
discriminant(&self.e) == discriminant(&other.e)
}
}