winnow/macros/dispatch.rs
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
76macro_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}