1 | /// `match` for parsers |
2 | /// |
3 | /// While `match` works by accepting a value and returning values: |
4 | /// ```rust,ignore |
5 | /// let result_value = match scrutinee_value { |
6 | /// ArmPattern => arm_value, |
7 | /// }; |
8 | /// ``` |
9 | /// `dispatch!` composes parsers: |
10 | /// ```rust,ignore |
11 | /// let result_parser = dispatch!{scrutinee_parser; |
12 | /// ArmPattern => arm_parser, |
13 | /// }; |
14 | /// ``` |
15 | /// |
16 | /// This is useful when parsers have unique prefixes to test for. |
17 | /// This offers better performance over |
18 | /// [`alt`][crate::combinator::alt] though it might be at the cost of duplicating parts of your grammar |
19 | /// if you needed to [`peek(input_parser)`][crate::combinator::peek] the scrutinee. |
20 | /// |
21 | /// For tight control over the error in a catch-all case, use [`fail`][crate::combinator::fail]. |
22 | /// |
23 | /// # Example |
24 | /// |
25 | /// ```rust |
26 | /// use winnow::prelude::*; |
27 | /// use winnow::combinator::dispatch; |
28 | /// # use winnow::token::take; |
29 | /// # use winnow::token::take_while; |
30 | /// # use winnow::combinator::fail; |
31 | /// |
32 | /// fn integer(input: &mut &str) -> ModalResult<u64> { |
33 | /// dispatch! {take(2usize); |
34 | /// "0b" => take_while(1.., '0' ..='1' ).try_map(|s| u64::from_str_radix(s, 2)), |
35 | /// "0o" => take_while(1.., '0' ..='7' ).try_map(|s| u64::from_str_radix(s, 8)), |
36 | /// "0d" => take_while(1.., '0' ..='9' ).try_map(|s| u64::from_str_radix(s, 10)), |
37 | /// "0x" => take_while(1.., ('0' ..='9' , 'a' ..='f' , 'A' ..='F' )).try_map(|s| u64::from_str_radix(s, 16)), |
38 | /// _ => fail::<_, u64, _>, |
39 | /// } |
40 | /// .parse_next(input) |
41 | /// } |
42 | /// |
43 | /// assert_eq!(integer.parse_peek("0x100 Hello" ), Ok((" Hello" , 0x100))); |
44 | /// ``` |
45 | /// |
46 | /// ```rust |
47 | /// use winnow::prelude::*; |
48 | /// use winnow::combinator::dispatch; |
49 | /// # use winnow::token::any; |
50 | /// # use winnow::combinator::preceded; |
51 | /// # use winnow::combinator::empty; |
52 | /// # use winnow::combinator::fail; |
53 | /// |
54 | /// fn escaped(input: &mut &str) -> ModalResult<char> { |
55 | /// preceded(' \\' , escape_seq_char).parse_next(input) |
56 | /// } |
57 | /// |
58 | /// fn escape_seq_char(input: &mut &str) -> ModalResult<char> { |
59 | /// dispatch! {any; |
60 | /// 'b' => empty.value(' \u{8}' ), |
61 | /// 'f' => empty.value(' \u{c}' ), |
62 | /// 'n' => empty.value(' \n' ), |
63 | /// 'r' => empty.value(' \r' ), |
64 | /// 't' => empty.value(' \t' ), |
65 | /// ' \\' => empty.value(' \\' ), |
66 | /// '"' => empty.value('"' ), |
67 | /// _ => fail::<_, char, _>, |
68 | /// } |
69 | /// .parse_next(input) |
70 | /// } |
71 | /// |
72 | /// assert_eq!(escaped.parse_peek(" \\nHello" ), Ok(("Hello" , ' \n' ))); |
73 | /// ``` |
74 | #[macro_export ] |
75 | #[doc (hidden)] // forced to be visible in intended location |
76 | macro_rules! dispatch { |
77 | ( |
78 | $scrutinee_parser:expr; |
79 | $( $arm_pat:pat $(if $arm_pred:expr)? => $arm_parser: expr ),+ $(,)? |
80 | ) => { |
81 | $crate::combinator::trace("dispatch" , move |i: &mut _| |
82 | { |
83 | use $crate::Parser; |
84 | let initial = $scrutinee_parser.parse_next(i)?; |
85 | match initial { |
86 | $( |
87 | $arm_pat $(if $arm_pred)? => $arm_parser.parse_next(i), |
88 | )* |
89 | } |
90 | }) |
91 | } |
92 | } |
93 | |