Files
carbon-lang/toolchain/parse/handle_let.cpp
T
Richard Smith 181a592b8c Support for parsing expression patterns (#6977)
When parsing a pattern, if we encounter something that isn't pattern
syntax, try parsing as an expression instead. We only need one-token
lookahead to distinguish pattern syntax from expression syntax.

Track a precedence group through pattern parsing so that we can allow
different kinds of expressions in a top-level pattern (such as the
operand of `let`) and in a nested pattern (such as a subpattern of a
tuple pattern or within grouping parens). For example, we do not allow
`case if ...`, and for now I've chosen to also not allow logical or
relational operators at the top level of a pattern, so `case 1 + 1` is
OK, but `case 1 == 1` and `case true and false` require parentheses.
This decision should be ratified or revisited by a design proposal.

Very basic check support is also provided, only sufficient to form an
`ExprPattern` instruction and nothing beyond that. For now, all pattern
matching against an `ExprPattern` fails with a TODO error. To support
that, I've switched from calling `BeginSubpattern` in the parent handler
of a pattern and `EndSubpatternAs*` in the pattern handler itself to
calling both functions in parent handlers, with `EndSubpattern`
converting an expression into an expression pattern where needed.

Depends on #6976.

Assisted-by: Gemini via Google Antigravity
2026-03-28 00:06:06 +00:00

115 lines
4.0 KiB
C++

// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
// Exceptions. See /LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
#include "toolchain/diagnostics/diagnostic.h"
#include "toolchain/lex/token_kind.h"
#include "toolchain/parse/context.h"
#include "toolchain/parse/handle.h"
#include "toolchain/parse/node_kind.h"
namespace Carbon::Parse {
auto HandleLet(Context& context) -> void {
auto state = context.PopState();
// These will start at the `let`.
context.PushState(state, StateKind::LetFinishAsRegular);
context.PushState(state, StateKind::LetAfterPatternAsRegular);
context.PushStateForPattern(StateKind::Pattern, /*in_var_pattern=*/false,
/*in_unused_pattern=*/false,
PrecedenceGroup::ForTopLevelPattern());
}
auto HandleAssociatedConstant(Context& context) -> void {
auto state = context.PopState();
// Parse the associated constant pattern: identifier :! type
auto identifier = context.ConsumeIf(Lex::TokenKind::Identifier);
if (!identifier) {
CARBON_DIAGNOSTIC(ExpectedAssociatedConstantIdentifier, Error,
"expected identifier in associated constant declaration");
context.emitter().Emit(*context.position(),
ExpectedAssociatedConstantIdentifier);
state.has_error = true;
}
auto colon_exclaim = context.ConsumeIf(Lex::TokenKind::ColonExclaim);
if (identifier && !colon_exclaim) {
CARBON_DIAGNOSTIC(ExpectedAssociatedConstantColonExclaim, Error,
"found runtime binding pattern in associated constant "
"declaration; expected a `:!` binding");
context.emitter().Emit(*context.position(),
ExpectedAssociatedConstantColonExclaim);
state.has_error = true;
}
if (!identifier || !colon_exclaim) {
auto end_token = context.SkipPastLikelyEnd(*(context.position() - 1));
context.AddNode(NodeKind::AssociatedConstantDecl, end_token,
/*has_error=*/true);
state.has_error = true;
return;
}
context.AddLeafNode(NodeKind::IdentifierNameNotBeforeSignature, *identifier);
state.token = *colon_exclaim;
context.PushState(state, StateKind::LetFinishAsAssociatedConstant);
context.PushState(state, StateKind::LetAfterPatternAsAssociatedConstant);
context.PushState(StateKind::Expr);
}
static auto HandleLetAfterPattern(Context& context, NodeKind init_kind)
-> void {
auto state = context.PopState();
if (state.has_error) {
if (auto after_pattern =
context.FindNextOf({Lex::TokenKind::Equal, Lex::TokenKind::Semi})) {
context.SkipTo(*after_pattern);
}
}
if (auto equals = context.ConsumeIf(Lex::TokenKind::Equal)) {
context.AddLeafNode(init_kind, *equals);
context.PushState(StateKind::Expr);
}
}
auto HandleLetAfterPatternAsRegular(Context& context) -> void {
HandleLetAfterPattern(context, NodeKind::LetInitializer);
}
auto HandleLetAfterPatternAsAssociatedConstant(Context& context) -> void {
auto state = context.PopState();
context.AddNode(NodeKind::AssociatedConstantNameAndType, state.token,
state.has_error);
context.PushState(state);
HandleLetAfterPattern(context, NodeKind::AssociatedConstantInitializer);
}
static auto HandleLetFinish(Context& context, NodeKind node_kind) -> void {
auto state = context.PopState();
auto end_token = state.token;
if (context.PositionIs(Lex::TokenKind::Semi)) {
end_token = context.Consume();
} else {
context.DiagnoseExpectedDeclSemi(Lex::TokenKind::Let);
state.has_error = true;
end_token = context.SkipPastLikelyEnd(state.token);
}
context.AddNode(node_kind, end_token, state.has_error);
}
auto HandleLetFinishAsRegular(Context& context) -> void {
HandleLetFinish(context, NodeKind::LetDecl);
}
auto HandleLetFinishAsAssociatedConstant(Context& context) -> void {
HandleLetFinish(context, NodeKind::AssociatedConstantDecl);
}
} // namespace Carbon::Parse