mago_syntax/parser/internal/function_like/
arrow_function.rs

1use crate::T;
2use crate::ast::ast::*;
3use crate::ast::sequence::Sequence;
4use crate::error::ParseError;
5use crate::parser::internal::expression::parse_expression;
6use crate::parser::internal::function_like::parameter::parse_function_like_parameter_list;
7use crate::parser::internal::function_like::r#return::parse_optional_function_like_return_type_hint;
8use crate::parser::internal::token_stream::TokenStream;
9use crate::parser::internal::utils;
10
11pub fn parse_arrow_function_with_attributes(
12    stream: &mut TokenStream<'_, '_>,
13    attributes: Sequence<AttributeList>,
14) -> Result<ArrowFunction, ParseError> {
15    Ok(ArrowFunction {
16        attribute_lists: attributes,
17        r#static: utils::maybe_expect_keyword(stream, T!["static"])?,
18        r#fn: utils::expect_keyword(stream, T!["fn"])?,
19        ampersand: utils::maybe_expect(stream, T!["&"])?.map(|t| t.span),
20        parameter_list: parse_function_like_parameter_list(stream)?,
21        return_type_hint: parse_optional_function_like_return_type_hint(stream)?,
22        arrow: utils::expect_span(stream, T!["=>"])?,
23        expression: Box::new(parse_expression(stream)?),
24    })
25}