Make identifier a token

This commit is contained in:
Greg Shuflin 2021-11-18 22:21:52 -08:00
parent 44cebec818
commit ec51659452
1 changed files with 24 additions and 19 deletions

View File

@ -154,7 +154,7 @@ fn decl_block(input: Span) -> ParseResult<Vec<Declaration>> {
} }
fn interface(input: Span) -> ParseResult<Declaration> { fn interface(input: Span) -> ParseResult<Declaration> {
map(preceded(kw("interface"), pair(tok(identifier), signature_block)), |(name, signatures)| { map(preceded(kw("interface"), pair(identifier, signature_block)), |(name, signatures)| {
Declaration::Interface { name: rc_string(name.fragment()), signatures } Declaration::Interface { name: rc_string(name.fragment()), signatures }
})(input) })(input)
} }
@ -190,7 +190,7 @@ fn func_decl(input: Span) -> ParseResult<Declaration> {
//TODO handle operators //TODO handle operators
fn func_signature(input: Span) -> ParseResult<Signature> { fn func_signature(input: Span) -> ParseResult<Signature> {
map(tuple((kw("fn"), tok(identifier), formal_params, opt(type_anno))), |(_, name, params, type_anno)| { map(tuple((kw("fn"), identifier, formal_params, opt(type_anno))), |(_, name, params, type_anno)| {
Signature { name: rc_string(name.fragment()), operator: false, params, type_anno } Signature { name: rc_string(name.fragment()), operator: false, params, type_anno }
})(input) })(input)
} }
@ -202,7 +202,7 @@ fn formal_params(input: Span) -> ParseResult<Vec<FormalParam>> {
//TODO support 256-limit //TODO support 256-limit
fn formal_param(input: Span) -> ParseResult<FormalParam> { fn formal_param(input: Span) -> ParseResult<FormalParam> {
map( map(
tuple((tok(identifier), opt(type_anno), opt(preceded(tok(char('=')), expression)))), tuple((identifier, opt(type_anno), opt(preceded(tok(char('=')), expression)))),
|(name, anno, default)| FormalParam { name: rc_string(name.fragment()), anno, default }, |(name, anno, default)| FormalParam { name: rc_string(name.fragment()), anno, default },
)(input) )(input)
} }
@ -210,7 +210,7 @@ fn formal_param(input: Span) -> ParseResult<FormalParam> {
fn type_decl(input: Span) -> ParseResult<Declaration> { fn type_decl(input: Span) -> ParseResult<Declaration> {
alt(( alt((
map( map(
tuple((kw("type"), kw("alias"), tok(identifier), tok(char('=')), tok(identifier))), tuple((kw("type"), kw("alias"), identifier, tok(char('=')), identifier)),
|(_, _, alias, _, name)| Declaration::TypeAlias { |(_, _, alias, _, name)| Declaration::TypeAlias {
alias: rc_string(alias.fragment()), alias: rc_string(alias.fragment()),
original: rc_string(name.fragment()), original: rc_string(name.fragment()),
@ -251,23 +251,22 @@ fn variant_spec(input: Span) -> ParseResult<Variant> {
let id = fresh_id(&input); let id = fresh_id(&input);
let (rest, (name, kind)) = alt(( let (rest, (name, kind)) = alt((
pair(tok(identifier), record_variant), pair(identifier, record_variant),
pair(tok(identifier), tuple_variant), pair(identifier, tuple_variant),
map(tok(identifier), |ident| (ident, VariantKind::UnitStruct)), map(identifier, |ident| (ident, VariantKind::UnitStruct)),
))(input)?; ))(input)?;
Ok((rest, Variant { id, name: rc_string(name.fragment()), kind })) Ok((rest, Variant { id, name: rc_string(name.fragment()), kind }))
} }
fn record_variant_item(input: Span) -> ParseResult<(Rc<String>, TypeIdentifier)> { fn record_variant_item(input: Span) -> ParseResult<(Rc<String>, TypeIdentifier)> {
map(tuple((tok(identifier), tok(char(':')), type_identifier)), |(name, _, ty)| { map(tuple((identifier, tok(char(':')), type_identifier)), |(name, _, ty)| {
(rc_string(name.fragment()), ty) (rc_string(name.fragment()), ty)
})(input) })(input)
} }
fn binding(input: Span) -> ParseResult<Declaration> { fn binding(input: Span) -> ParseResult<Declaration> {
let parser = let parser = tuple((kw("let"), opt(kw("mut")), identifier, opt(type_anno), tok(char('=')), expression));
tuple((kw("let"), opt(kw("mut")), tok(identifier), opt(type_anno), tok(char('=')), expression));
map(parser, |(_, maybe_mut, ident, type_anno, _, expr)| Declaration::Binding { map(parser, |(_, maybe_mut, ident, type_anno, _, expr)| Declaration::Binding {
name: rc_string(ident.fragment()), name: rc_string(ident.fragment()),
constant: maybe_mut.is_none(), constant: maybe_mut.is_none(),
@ -277,7 +276,7 @@ fn binding(input: Span) -> ParseResult<Declaration> {
} }
fn module(input: Span) -> ParseResult<Declaration> { fn module(input: Span) -> ParseResult<Declaration> {
map(tuple((kw("module"), tok(identifier), block)), |(_, name, items)| Declaration::Module { map(tuple((kw("module"), identifier, block)), |(_, name, items)| Declaration::Module {
name: rc_string(name.fragment()), name: rc_string(name.fragment()),
items, items,
})(input) })(input)
@ -316,7 +315,7 @@ fn type_identifier(input: Span) -> ParseResult<TypeIdentifier> {
} }
fn type_singleton_name(input: Span) -> ParseResult<TypeSingletonName> { fn type_singleton_name(input: Span) -> ParseResult<TypeSingletonName> {
map(pair(tok(identifier), opt(type_params)), |(name, params)| TypeSingletonName { map(pair(identifier, opt(type_params)), |(name, params)| TypeSingletonName {
name: rc_string(name.fragment()), name: rc_string(name.fragment()),
params: if let Some(params) = params { params } else { vec![] }, params: if let Some(params) = params { params } else { vec![] },
})(input) })(input)
@ -426,8 +425,9 @@ fn extended_expr_part(input: Span) -> ParseResult<ExtendedPart> {
fn invocation_argument(input: Span) -> ParseResult<InvocationArgument> { fn invocation_argument(input: Span) -> ParseResult<InvocationArgument> {
alt(( alt((
map(tok(char('_')), |_| InvocationArgument::Ignored), map(tok(char('_')), |_| InvocationArgument::Ignored),
map(tuple((tok(identifier), tok(char('=')), expression)), |(name, _, expr)| { map(tuple((identifier, tok(char('=')), expression)), |(name, _, expr)| InvocationArgument::Keyword {
InvocationArgument::Keyword { name: rc_string(name.fragment()), expr } name: rc_string(name.fragment()),
expr,
}), }),
map(expression, InvocationArgument::Positional), map(expression, InvocationArgument::Positional),
))(input) ))(input)
@ -470,7 +470,7 @@ fn named_struct(input: Span) -> ParseResult<ExpressionKind> {
//TODO support anonymous structs and Elm-style update syntax for structs //TODO support anonymous structs and Elm-style update syntax for structs
fn record_block(input: Span) -> ParseResult<Vec<(Rc<String>, Expression)>> { fn record_block(input: Span) -> ParseResult<Vec<(Rc<String>, Expression)>> {
let record_entry = let record_entry =
separated_pair(map(tok(identifier), |span| rc_string(span.fragment())), tok(char(':')), expression); separated_pair(map(identifier, |span| rc_string(span.fragment())), tok(char(':')), expression);
delimited(tok(char('{')), separated_list0(tok(char(',')), record_entry), tok(char('}')))(input) delimited(tok(char('{')), separated_list0(tok(char(',')), record_entry), tok(char('}')))(input)
} }
@ -495,12 +495,13 @@ fn for_expr(input: Span) -> ParseResult<ExpressionKind> {
//TODO add guards, etc. //TODO add guards, etc.
fn enumerator(input: Span) -> ParseResult<Enumerator> { fn enumerator(input: Span) -> ParseResult<Enumerator> {
alt(( alt((
map(separated_pair(tok(identifier), kw("<-"), expression_no_struct), |(ident, generator)| { map(separated_pair(identifier, kw("<-"), expression_no_struct), |(ident, generator)| {
Enumerator { id: rc_string(ident.fragment()), generator } Enumerator { id: rc_string(ident.fragment()), generator }
}), }),
//TODO distinguish these two cases in AST //TODO distinguish these two cases in AST
map(separated_pair(tok(identifier), kw("="), expression_no_struct), |(ident, generator)| { map(separated_pair(identifier, kw("="), expression_no_struct), |(ident, generator)| Enumerator {
Enumerator { id: rc_string(ident.fragment()), generator } id: rc_string(ident.fragment()),
generator,
}), }),
))(input) ))(input)
} }
@ -563,12 +564,16 @@ fn identifier_expr(input: Span) -> ParseResult<ExpressionKind> {
fn qualified_identifier(input: Span) -> ParseResult<QualifiedName> { fn qualified_identifier(input: Span) -> ParseResult<QualifiedName> {
let id = fresh_id(&input); let id = fresh_id(&input);
tok(map(separated_list1(tag("::"), map(identifier, |x| rc_string(x.fragment()))), move |items| { tok(map(separated_list1(tag("::"), map(identifier_span, |x| rc_string(x.fragment()))), move |items| {
QualifiedName { id, components: items } QualifiedName { id, components: items }
}))(input) }))(input)
} }
fn identifier(input: Span) -> ParseResult<Span> { fn identifier(input: Span) -> ParseResult<Span> {
tok(identifier_span)(input)
}
fn identifier_span(input: Span) -> ParseResult<Span> {
recognize(pair(alt((tag("_"), alpha1)), take_while(|ch: char| is_alphanumeric(ch as u8) || ch == '_')))( recognize(pair(alt((tag("_"), alpha1)), take_while(|ch: char| is_alphanumeric(ch as u8) || ch == '_')))(
input, input,
) )