mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 22:02:23 +01:00
Jon Ross-Perkins
19f6cd2023
Just a small factoring thing. The `if` logic is sizeable, and there's no need for it to be in handle_expr (we do also have handle_brace_expr, handle_index_expr, etc)
76 lines
2.6 KiB
C++
76 lines
2.6 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/lex/token_kind.h"
|
|
#include "toolchain/parse/context.h"
|
|
#include "toolchain/parse/handle.h"
|
|
|
|
namespace Carbon::Parse {
|
|
|
|
auto HandleIfExprFinishCondition(Context& context) -> void {
|
|
auto state = context.PopState();
|
|
|
|
context.AddNode(NodeKind::IfExprIf, state.token, state.has_error);
|
|
|
|
if (context.PositionIs(Lex::TokenKind::Then)) {
|
|
context.PushState(State::IfExprFinishThen);
|
|
context.ConsumeChecked(Lex::TokenKind::Then);
|
|
context.PushStateForExpr(*PrecedenceGroup::ForLeading(Lex::TokenKind::If));
|
|
} else {
|
|
// TODO: Include the location of the `if` token.
|
|
CARBON_DIAGNOSTIC(ExpectedThenAfterIf, Error,
|
|
"Expected `then` after `if` condition.");
|
|
if (!state.has_error) {
|
|
context.emitter().Emit(*context.position(), ExpectedThenAfterIf);
|
|
}
|
|
// Add placeholders for `IfExprThen` and final `Expr`.
|
|
context.AddLeafNode(NodeKind::InvalidParse, *context.position(),
|
|
/*has_error=*/true);
|
|
context.AddLeafNode(NodeKind::InvalidParse, *context.position(),
|
|
/*has_error=*/true);
|
|
context.ReturnErrorOnState();
|
|
}
|
|
}
|
|
|
|
auto HandleIfExprFinishThen(Context& context) -> void {
|
|
auto state = context.PopState();
|
|
|
|
context.AddNode(NodeKind::IfExprThen, state.token, state.has_error);
|
|
|
|
if (context.PositionIs(Lex::TokenKind::Else)) {
|
|
context.PushState(State::IfExprFinishElse);
|
|
context.ConsumeChecked(Lex::TokenKind::Else);
|
|
context.PushStateForExpr(*PrecedenceGroup::ForLeading(Lex::TokenKind::If));
|
|
} else {
|
|
// TODO: Include the location of the `if` token.
|
|
CARBON_DIAGNOSTIC(ExpectedElseAfterIf, Error,
|
|
"Expected `else` after `if ... then ...`.");
|
|
if (!state.has_error) {
|
|
context.emitter().Emit(*context.position(), ExpectedElseAfterIf);
|
|
}
|
|
// Add placeholder for the final `Expr`.
|
|
context.AddLeafNode(NodeKind::InvalidParse, *context.position(),
|
|
/*has_error=*/true);
|
|
context.ReturnErrorOnState();
|
|
}
|
|
}
|
|
|
|
auto HandleIfExprFinishElse(Context& context) -> void {
|
|
auto else_state = context.PopState();
|
|
|
|
// Propagate the location of `else`.
|
|
auto if_state = context.PopState();
|
|
if_state.token = else_state.token;
|
|
if_state.has_error |= else_state.has_error;
|
|
context.PushState(if_state);
|
|
}
|
|
|
|
auto HandleIfExprFinish(Context& context) -> void {
|
|
auto state = context.PopState();
|
|
|
|
context.AddNode(NodeKind::IfExprElse, state.token, state.has_error);
|
|
}
|
|
|
|
} // namespace Carbon::Parse
|