toml_edit/parser/
value.rs

Help
1use winnow::combinator::alt;
2use winnow::combinator::fail;
3use winnow::combinator::peek;
4use winnow::token::any;
5
6use crate::parser::array::array;
7use crate::parser::datetime::date_time;
8use crate::parser::inline_table::inline_table;
9use crate::parser::numbers::{float, integer};
10use crate::parser::prelude::*;
11use crate::parser::strings::string;
12use crate::repr::{Formatted, Repr};
13use crate::value as v;
14use crate::RawString;
15use crate::Value;
16
17// val = string / boolean / array / inline-table / date-time / float / integer
18pub(crate) fn value<'i>(check: RecursionCheck) -> impl Parser<Input<'i>, v::Value, ContextError> {
19    move |input: &mut Input<'i>| {
20        dispatch!{peek(any);
21            crate::parser::strings::QUOTATION_MARK |
22            crate::parser::strings::APOSTROPHE => string.map(|s| {
23                v::Value::String(Formatted::new(
24                    s.into_owned()
25                ))
26            }),
27            crate::parser::array::ARRAY_OPEN => array(check).map(v::Value::Array),
28            crate::parser::inline_table::INLINE_TABLE_OPEN => inline_table(check).map(v::Value::InlineTable),
29            // Date/number starts
30            b'+' | b'-' | b'0'..=b'9' => {
31                // Uncommon enough not to be worth optimizing at this time
32                alt((
33                    date_time
34                        .map(v::Value::from),
35                    float
36                        .map(v::Value::from),
37                    integer
38                        .map(v::Value::from),
39                ))
40            },
41            // Report as if they were numbers because its most likely a typo
42            b'_' => {
43                    integer
44                        .map(v::Value::from)
45                .context(StrContext::Expected(StrContextValue::Description("leading digit")))
46            },
47            // Report as if they were numbers because its most likely a typo
48            b'.' =>  {
49                    float
50                        .map(v::Value::from)
51                .context(StrContext::Expected(StrContextValue::Description("leading digit")))
52            },
53            b't' => {
54                crate::parser::numbers::true_.map(v::Value::from)
55                    .context(StrContext::Label("string"))
56                    .context(StrContext::Expected(StrContextValue::CharLiteral('"')))
57                    .context(StrContext::Expected(StrContextValue::CharLiteral('\'')))
58            },
59            b'f' => {
60                crate::parser::numbers::false_.map(v::Value::from)
61                    .context(StrContext::Label("string"))
62                    .context(StrContext::Expected(StrContextValue::CharLiteral('"')))
63                    .context(StrContext::Expected(StrContextValue::CharLiteral('\'')))
64            },
65            b'i' => {
66                crate::parser::numbers::inf.map(v::Value::from)
67                    .context(StrContext::Label("string"))
68                    .context(StrContext::Expected(StrContextValue::CharLiteral('"')))
69                    .context(StrContext::Expected(StrContextValue::CharLiteral('\'')))
70            },
71            b'n' => {
72                crate::parser::numbers::nan.map(v::Value::from)
73                    .context(StrContext::Label("string"))
74                    .context(StrContext::Expected(StrContextValue::CharLiteral('"')))
75                    .context(StrContext::Expected(StrContextValue::CharLiteral('\'')))
76            },
77            _ => {
78                fail
79                    .context(StrContext::Label("string"))
80                    .context(StrContext::Expected(StrContextValue::CharLiteral('"')))
81                    .context(StrContext::Expected(StrContextValue::CharLiteral('\'')))
82            },
83    }
84        .with_span()
85        .try_map(|(value, span)| apply_raw(value, span))
86        .parse_next(input)
87    }
88}
89
90fn apply_raw(mut val: Value, span: std::ops::Range<usize>) -> Result<Value, std::str::Utf8Error> {
91    match val {
92        Value::String(ref mut f) => {
93            let raw = RawString::with_span(span);
94            f.set_repr_unchecked(Repr::new_unchecked(raw));
95        }
96        Value::Integer(ref mut f) => {
97            let raw = RawString::with_span(span);
98            f.set_repr_unchecked(Repr::new_unchecked(raw));
99        }
100        Value::Float(ref mut f) => {
101            let raw = RawString::with_span(span);
102            f.set_repr_unchecked(Repr::new_unchecked(raw));
103        }
104        Value::Boolean(ref mut f) => {
105            let raw = RawString::with_span(span);
106            f.set_repr_unchecked(Repr::new_unchecked(raw));
107        }
108        Value::Datetime(ref mut f) => {
109            let raw = RawString::with_span(span);
110            f.set_repr_unchecked(Repr::new_unchecked(raw));
111        }
112        Value::Array(ref mut arr) => {
113            arr.span = Some(span);
114        }
115        Value::InlineTable(ref mut table) => {
116            table.span = Some(span);
117        }
118    };
119    val.decorate("", "");
120    Ok(val)
121}
122
123#[cfg(test)]
124mod test {
125    use super::*;
126
127    #[test]
128    fn values() {
129        let inputs = [
130            "1979-05-27T00:32:00.999999",
131            "-239",
132            "1e200",
133            "9_224_617.445_991_228_313",
134            r#"'''I [dw]on't need \d{2} apples'''"#,
135            r#"'''
136The first newline is
137trimmed in raw strings.
138   All other whitespace
139   is preserved.
140'''"#,
141            r#""Jos\u00E9\n""#,
142            r#""\\\"\b/\f\n\r\t\u00E9\U000A0000""#,
143            r#"{ hello = "world", a = 1}"#,
144            r#"[ { x = 1, a = "2" }, {a = "a",b = "b",     c =    "c"} ]"#,
145        ];
146        for input in inputs {
147            dbg!(input);
148            let mut parsed = value(Default::default()).parse(new_input(input));
149            if let Ok(parsed) = &mut parsed {
150                parsed.despan(input);
151            }
152            assert_eq!(parsed.map(|a| a.to_string()), Ok(input.to_owned()));
153        }
154    }
155}
toml_edit/parser/value.rs

toml_edit/parser/
value.rs
