1#[macro_use]
2use crate::parser::*;
3use crate::*;
4use crate::label;
5use crate::labelr;
6use nom::{
7 multi::separated_list0,
8 sequence::tuple as nom_tuple,
9};
10use crate::nodes::Kind;
11
12macro_rules! leaf {
17 ($name:ident, $byte:expr, $token:expr) => (
18 pub fn $name(input: ParseString) -> ParseResult<Token> {
19 if input.is_empty() {
20 return Err(nom::Err::Error(ParseError::new(input, "Unexpected eof")))
21 }
22 let start = input.loc();
23 let byte = input.graphemes[input.cursor];
24 let (input, _) = tag($byte)(input)?;
25 let end = input.loc();
26 let src_range = SourceRange { start, end };
27 Ok((input, Token{kind: $token, chars: $byte.chars().collect::<Vec<char>>(), src_range}))
28 }
29 )
30}
31
32macro_rules! ws0_leaf {
33 ($name:ident, $byte:expr, $token:expr) => (
34 pub fn $name(input: ParseString) -> ParseResult<Token> {
35 if input.is_empty() {
36 return Err(nom::Err::Error(ParseError::new(input, "Unexpected eof")))
37 }
38 let start = input.loc();
39 let byte = input.graphemes[input.cursor];
40 let (input, _) = whitespace0(input)?;
41 let (input, _) = tag($byte)(input)?;
42 let (input, _) = whitespace0(input)?;
43 let end = input.loc();
44 let src_range = SourceRange { start, end };
45 Ok((input, Token{kind: $token, chars: $byte.chars().collect::<Vec<char>>(), src_range}))
46 }
47 )
48}
49
50macro_rules! ws1_leaf {
51 ($name:ident, $byte:expr, $token:expr) => (
52 pub fn $name(input: ParseString) -> ParseResult<Token> {
53 if input.is_empty() {
54 return Err(nom::Err::Error(ParseError::new(input, "Unexpected eof")))
55 }
56 let (input, _) = whitespace1(input)?;
57 let start = input.loc();
58 let byte = input.graphemes[input.cursor];
59 let (input, _) = tag($byte)(input)?;
60 let end = input.loc();
61 let (input, _) = whitespace1(input)?;
62 let src_range = SourceRange { start, end };
63 Ok((input, Token{kind: $token, chars: $byte.chars().collect::<Vec<char>>(), src_range}))
64 }
65 )
66}
67
68leaf!{ampersand, "&", TokenKind::Ampersand}
73leaf!{apostrophe, "'", TokenKind::Apostrophe}
74leaf!{asterisk, "*", TokenKind::Asterisk}
75leaf!{at, "@", TokenKind::At}
76leaf!{bar, "|", TokenKind::Bar}
77leaf!{backslash, "\\", TokenKind::Backslash}
78leaf!{caret, "^", TokenKind::Caret}
79leaf!{colon, ":", TokenKind::Colon}
80leaf!{comma, ",", TokenKind::Comma}
81leaf!{dash, "-", TokenKind::Dash}
82leaf!{dollar, "$", TokenKind::Dollar}
83leaf!{equal, "=", TokenKind::Equal}
84leaf!{exclamation, "!", TokenKind::Exclamation}
85leaf!{grave, "`", TokenKind::Grave}
86leaf!{hashtag, "#", TokenKind::HashTag}
87leaf!{negate, "¬", TokenKind::Not}
88leaf!{percent, "%", TokenKind::Percent}
89leaf!{period, ".", TokenKind::Period}
90leaf!{plus, "+", TokenKind::Plus}
91leaf!{question, "?", TokenKind::Question}
92leaf!{quote, "\"", TokenKind::Quote}
93leaf!{semicolon, ";", TokenKind::Semicolon}
94leaf!{slash, "/", TokenKind::Slash}
95leaf!{tilde, "~", TokenKind::Tilde}
96leaf!{underscore, "_", TokenKind::Underscore}
97
98leaf!{check_mark, "✓", TokenKind::True}
99leaf!{cross, "✗", TokenKind::False}
100leaf!{english_true_literal, "true", TokenKind::True}
101leaf!{english_false_literal, "false", TokenKind::False}
102
103leaf!{space, " ", TokenKind::Space}
104leaf!{new_line_char, "\n", TokenKind::Newline}
105leaf!{carriage_return, "\r", TokenKind::CarriageReturn}
106leaf!{carriage_return_new_line, "\r\n", TokenKind::CarriageReturn}
107leaf!{tab, "\t", TokenKind::Tab}
108
109leaf!{left_bracket, "[", TokenKind::LeftBracket}
110leaf!{left_parenthesis, "(", TokenKind::LeftParenthesis}
111leaf!{left_brace, "{", TokenKind::LeftBrace}
112leaf!{left_angle, "<", TokenKind::LeftAngle}
113
114leaf!{right_bracket, "]", TokenKind::RightBracket}
115leaf!{right_parenthesis, ")", TokenKind::RightParenthesis}
116leaf!{right_brace, "}", TokenKind::RightBrace}
117leaf!{right_angle, ">", TokenKind::RightAngle}
118
119leaf!{box_tl_round, "╭", TokenKind::BoxDrawing}
120leaf!{box_tr_round, "╮", TokenKind::BoxDrawing}
121leaf!{box_bl_round, "╰", TokenKind::BoxDrawing}
122leaf!{box_br_round, "╯", TokenKind::BoxDrawing}
123
124leaf!{box_tl_bold, "┏", TokenKind::BoxDrawing}
125leaf!{box_tr_bold, "┓", TokenKind::BoxDrawing}
126leaf!{box_bl_bold, "┗", TokenKind::BoxDrawing}
127leaf!{box_br_bold, "┛", TokenKind::BoxDrawing}
128
129leaf!{box_tl, "┌", TokenKind::BoxDrawing}
130leaf!{box_tr, "┐", TokenKind::BoxDrawing}
131leaf!{box_bl, "└", TokenKind::BoxDrawing}
132leaf!{box_br, "┘", TokenKind::BoxDrawing}
133
134leaf!{box_cross, "┼", TokenKind::BoxDrawing}
135leaf!{box_horz, "─", TokenKind::BoxDrawing}
136leaf!{box_t_left, "├", TokenKind::BoxDrawing}
137leaf!{box_t_right, "┤", TokenKind::BoxDrawing}
138leaf!{box_t_top, "┬", TokenKind::BoxDrawing}
139leaf!{box_t_bottom, "┴", TokenKind::BoxDrawing}
140leaf!{box_vert, "│", TokenKind::BoxDrawing}
141leaf!{box_vert_bold, "┃", TokenKind::BoxDrawing}
142
143leaf!(abstract_sigil, "%%", TokenKind::AbstractSigil);
144leaf!(emphasis_sigil, "*", TokenKind::EmphasisSigil);
145leaf!(equation_sigil, "$$", TokenKind::EquationSigil);
146leaf!(footnote_prefix, "[^", TokenKind::FootnotePrefix);
147leaf!(float_left, "<<", TokenKind::FloatLeft);
148leaf!(float_right, ">>", TokenKind::FloatRight);
149leaf!(http_prefix, "http", TokenKind::HttpPrefix);
150leaf!(highlight_sigil, "!!", TokenKind::HighlightSigil);
151leaf!(img_prefix, "![", TokenKind::ImgPrefix);
152leaf!(query_sigil, "??", TokenKind::QuerySigil);
153leaf!(quote_sigil, ">", TokenKind::QuoteSigil);
154leaf!(strike_sigil, "~~", TokenKind::StrikeSigil);
155leaf!(strong_sigil, "**", TokenKind::StrongSigil);
156leaf!(underline_sigil, "__", TokenKind::UnderlineSigil);
157
158ws0_leaf!(assign_operator, "=", TokenKind::AssignOperator);
159ws0_leaf!(async_transition_operator, "~>", TokenKind::AsyncTransitionOperator);
160ws0_leaf!(define_operator, ":=", TokenKind::DefineOperator);
161ws0_leaf!(output_operator, "=>", TokenKind::OutputOperator);
162ws0_leaf!(transition_operator, "->", TokenKind::TransitionOperator);
163
164
165pub fn emoji_grapheme(mut input: ParseString) -> ParseResult<String> {
167 if let Some(matched) = input.consume_emoji() {
168 Ok((input, matched))
169 } else {
170 Err(nom::Err::Error(ParseError::new(input, "Unexpected character")))
171 }
172}
173
174pub fn alpha(mut input: ParseString) -> ParseResult<String> {
176 if let Some(matched) = input.consume_alpha() {
177 Ok((input, matched))
178 } else {
179 Err(nom::Err::Error(ParseError::new(input, "Unexpected character")))
180 }
181}
182
183pub fn digit(mut input: ParseString) -> ParseResult<String> {
185 if let Some(matched) = input.consume_digit() {
186 Ok((input, matched))
187 } else {
188 Err(nom::Err::Error(ParseError::new(input, "Unexpected character")))
189 }
190}
191
192pub fn any(mut input: ParseString) -> ParseResult<String> {
194 if let Some(matched) = input.consume_one() {
195 Ok((input, matched))
196 } else {
197 Err(nom::Err::Error(ParseError::new(input, "Unexpected eof")))
198 }
199}
200
201pub fn any_token(mut input: ParseString) -> ParseResult<Token> {
202 if input.is_empty() {
203 return Err(nom::Err::Error(ParseError::new(input, "Unexpected eof")))
204 }
205 let start = input.loc();
206 let byte = input.graphemes[input.cursor];
207 if let Some(matched) = input.consume_one() {
208 let end = input.loc();
209 let src_range = SourceRange { start, end };
210 Ok((input, Token{kind: TokenKind::Any, chars: byte.chars().collect::<Vec<char>>(), src_range}))
211 } else {
212 Err(nom::Err::Error(ParseError::new(input, "Unexpected eof")))
213 }
214}
215
216pub fn forbidden_emoji(input: ParseString) -> ParseResult<Token> {
218 alt((box_tl, box_br, box_bl, box_tr, box_tr_bold, box_tl_bold, box_br_bold, box_bl_bold, box_t_left,box_tl_round,box_br_round, box_tr_round, box_bl_round, box_vert, box_cross, box_horz, box_t_right, box_t_top, box_t_bottom))(input)
219}
220
221pub fn emoji(input: ParseString) -> ParseResult<Token> {
223 let msg1 = "Cannot be a box-drawing emoji";
224 let start = input.loc();
225 let (input, _) = is_not(forbidden_emoji)(input)?;
226 let (input, g) = emoji_grapheme(input)?;
227 let end = input.loc();
228 let src_range = SourceRange { start, end };
229 Ok((input, Token{kind: TokenKind::Emoji, chars: g.chars().collect::<Vec<char>>(), src_range}))
230}
231
232pub fn alpha_token(input: ParseString) -> ParseResult<Token> {
234 let (input, (g, src_range)) = range(alpha)(input)?;
235 Ok((input, Token{kind: TokenKind::Alpha, chars: g.chars().collect::<Vec<char>>(), src_range}))
236}
237
238pub fn digit_token(input: ParseString) -> ParseResult<Token> {
240 let (input, (g, src_range)) = range(digit)(input)?;
241 Ok((input, Token{kind: TokenKind::Digit, chars: g.chars().collect::<Vec<char>>(), src_range}))
242}
243
244pub fn alphanumeric(input: ParseString) -> ParseResult<Token> {
246 let (input, token) = alt((alpha_token, digit_token))(input)?;
247 Ok((input, token))
248}
249
250pub fn underscore_digit(input: ParseString) -> ParseResult<Token> {
252 let (input, _) = underscore(input)?;
253 let (input, digit) = digit_token(input)?;
254 Ok((input,digit))
255}
256
257pub fn digit_sequence(input: ParseString) -> ParseResult<Vec<Token>> {
259 let (input, mut start) = digit_token(input)?;
260 let (input, mut tokens) = many0(alt((underscore_digit,digit_token)))(input)?;
261 let mut all = vec![start];
262 all.append(&mut tokens);
263 Ok((input,all))
264}
265
266pub fn grouping_symbol(input: ParseString) -> ParseResult<Token> {
268 let (input, grouping) = alt((left_parenthesis, right_parenthesis, left_angle, right_angle, left_brace, right_brace, left_bracket, right_bracket))(input)?;
269 Ok((input, grouping))
270}
271
272pub fn punctuation(input: ParseString) -> ParseResult<Token> {
274 let (input, punctuation) = alt((period, exclamation, question, comma, colon, semicolon, quote, apostrophe))(input)?;
275 Ok((input, punctuation))
276}
277
278pub fn escaped_char(input: ParseString) -> ParseResult<Token> {
280 let (input, _) = backslash(input)?;
281 let (input, symbol) = alt((alpha_token, symbol, punctuation))(input)?;
282 Ok((input, symbol))
283}
284
285pub fn symbol(input: ParseString) -> ParseResult<Token> {
287 let (input, symbol) = alt((ampersand, dollar, bar, percent, at, slash, hashtag, equal, backslash, tilde, plus, dash, asterisk, caret, underscore))(input)?;
288 Ok((input, symbol))
289}
290
291pub fn text(input: ParseString) -> ParseResult<Token> {
293 let (input, text) = alt((alpha_token, digit_token, emoji, forbidden_emoji, space, tab, escaped_char, punctuation, grouping_symbol, symbol))(input)?;
294 Ok((input, text))
295}
296
297pub fn new_line(input: ParseString) -> ParseResult<Token> {
303 let (input, result) = alt((carriage_return_new_line,new_line_char,carriage_return))(input)?;
304 Ok((input, result))
305}
306
307pub fn whitespace(input: ParseString) -> ParseResult<Token> {
309 let (input, space) = alt((space,tab,new_line))(input)?;
310 Ok((input, space))
311}
312
313pub fn whitespace0(input: ParseString) -> ParseResult<()> {
315 let (input, _) = many0(whitespace)(input)?;
316 Ok((input, ()))
317}
318
319pub fn whitespace1(input: ParseString) -> ParseResult<()> {
321 let (input, _) = many1(whitespace)(input)?;
322 Ok((input, ()))
323}
324
325pub fn newline_indent(input: ParseString) -> ParseResult<()> {
327 let (input, _) = new_line(input)?;
328 let (input, _) = many0(space_tab)(input)?;
329 Ok((input, ()))
330}
331
332pub fn ws1e(input: ParseString) -> ParseResult<()> {
334 let (input, _) = many1(space_tab)(input)?;
335 let (input, _) = opt(newline_indent)(input)?;
336 Ok((input, ()))
337}
338
339pub fn space_tab(input: ParseString) -> ParseResult<Token> {
341 let (input, space) = alt((space,tab))(input)?;
342 Ok((input, space))
343}
344
345pub fn list_separator(input: ParseString) -> ParseResult<()> {
347 let (input,_) = nom_tuple((whitespace0,tag(","),whitespace0))(input)?;
348 Ok((input, ()))
349}
350
351pub fn enum_separator(input: ParseString) -> ParseResult<()> {
353 let (input,_) = nom_tuple((whitespace0,tag("|"),whitespace0))(input)?;
354 Ok((input, ()))
355}
356
357pub fn identifier(input: ParseString) -> ParseResult<Identifier> {
363 let (input, (first, mut rest)) = nom_tuple((alt((alpha_token, emoji)), many0(alt((alpha_token, digit_token, symbol, emoji)))))(input)?;
364 let mut tokens = vec![first];
365 tokens.append(&mut rest);
366 let mut merged = Token::merge_tokens(&mut tokens).unwrap();
367 merged.kind = TokenKind::Identifier;
368 Ok((input, Identifier{name: merged}))
369}