Files
carbon-lang/toolchain/parse/handle_function.cpp
T
Richard SmithandJon Ross-Perkins f9ce0b194d Defer parsing of method bodies until the end of a suitable enclosing scope. (#3832)
In parse, form a list of methods that are defined inline, tracking where
they start, where they end, and which other inline methods are nested
within them.

In check, when we reach an inline method body, skip it and add it to a
worklist to be processed later. We also track when we reach the start
and end of a context in which inline method bodies are deferred, so that
we know when to replay the bodies.

When suspending a function definition to be processed later, the
`DeclNameStack` entry is moved to separate storage, including popping
the corresponding scopes from the scope stack and removing the
corresponding lexical names from lexical lookup. Later, when we return
to the function and parse its definition, the `DeclNameStack` entry is
restored. The same is done when we reach the end of a nested context
that can have inline methods, so that we can reenter the nested scope
before processing its members.

---------

Co-authored-by: Jon Ross-Perkins <jperkins@google.com>
2024-04-01 18:25:27 +00:00

101 lines
3.7 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/parse/context.h"
namespace Carbon::Parse {
auto HandleFunctionIntroducer(Context& context) -> void {
auto state = context.PopState();
context.PushState(state, State::FunctionAfterParams);
context.PushState(State::DeclNameAndParamsAsRequired, state.token);
}
auto HandleFunctionAfterParams(Context& context) -> void {
auto state = context.PopState();
// Regardless of whether there's a return type, we'll finish the signature.
context.PushState(state, State::FunctionSignatureFinish);
// If there is a return type, parse the expression before adding the return
// type node.
if (context.PositionIs(Lex::TokenKind::MinusGreater)) {
context.PushState(State::FunctionReturnTypeFinish);
context.ConsumeAndDiscard();
context.PushStateForExpr(PrecedenceGroup::ForType());
}
}
auto HandleFunctionReturnTypeFinish(Context& context) -> void {
auto state = context.PopState();
context.AddNode(NodeKind::ReturnType, state.token, state.subtree_start,
state.has_error);
}
auto HandleFunctionSignatureFinish(Context& context) -> void {
auto state = context.PopState();
switch (context.PositionKind()) {
case Lex::TokenKind::Semi: {
context.AddNode(NodeKind::FunctionDecl, context.Consume(),
state.subtree_start, state.has_error);
break;
}
case Lex::TokenKind::OpenCurlyBrace: {
context.AddFunctionDefinitionStart(context.Consume(), state.subtree_start,
state.has_error);
// Any error is recorded on the FunctionDefinitionStart.
state.has_error = false;
context.PushState(state, State::FunctionDefinitionFinish);
context.PushState(State::StatementScopeLoop);
break;
}
case Lex::TokenKind::Equal: {
context.AddNode(NodeKind::BuiltinFunctionDefinitionStart,
context.Consume(), state.subtree_start, state.has_error);
if (!context.ConsumeAndAddLeafNodeIf(Lex::TokenKind::StringLiteral,
NodeKind::BuiltinName)) {
CARBON_DIAGNOSTIC(ExpectedBuiltinName, Error,
"Expected builtin function name after `=`.");
context.emitter().Emit(*context.position(), ExpectedBuiltinName);
state.has_error = true;
}
auto semi = context.ConsumeIf(Lex::TokenKind::Semi);
if (!semi && !state.has_error) {
context.EmitExpectedDeclSemi(context.tokens().GetKind(state.token));
state.has_error = true;
}
if (state.has_error) {
context.RecoverFromDeclError(state, NodeKind::BuiltinFunctionDefinition,
/*skip_past_likely_end=*/true);
} else {
context.AddNode(NodeKind::BuiltinFunctionDefinition, *semi,
state.subtree_start, state.has_error);
}
break;
}
default: {
if (!state.has_error) {
context.EmitExpectedDeclSemiOrDefinition(Lex::TokenKind::Fn);
}
// Only need to skip if we've not already found a new line.
bool skip_past_likely_end =
context.tokens().GetLine(*context.position()) ==
context.tokens().GetLine(state.token);
context.RecoverFromDeclError(state, NodeKind::FunctionDecl,
skip_past_likely_end);
break;
}
}
}
auto HandleFunctionDefinitionFinish(Context& context) -> void {
auto state = context.PopState();
context.AddFunctionDefinition(context.Consume(), state.subtree_start,
state.has_error);
}
} // namespace Carbon::Parse