carbon-lang/toolchain/check/handle_pattern_list.cpp

// 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/check/class.h"
#include "toolchain/check/context.h"
#include "toolchain/check/handle.h"
#include "toolchain/check/inst.h"
#include "toolchain/check/pattern.h"
#include "toolchain/check/type.h"
#include "toolchain/diagnostics/emitter.h"

namespace Carbon::Check {

// Handle the start of any kind of pattern list.
static auto HandlePatternListStart(Context& context, Parse::NodeId node_id)
    -> bool {
  context.node_stack().Push(node_id);
  context.param_and_arg_refs_stack().Push();
  BeginExprRegionForPattern(context);
  return true;
}

auto HandleParseNode(Context& context, Parse::ImplicitParamListStartId node_id)
    -> bool {
  context.full_pattern_stack().StartImplicitParamList();
  return HandlePatternListStart(context, node_id);
}

auto HandleParseNode(Context& context, Parse::TuplePatternStartId node_id)
    -> bool {
  // End the pending `ExprRegion`, so that we can start a new one in
  // `HandlePatternListStart`.
  EndEmptyExprRegionForPattern(context);
  return HandlePatternListStart(context, node_id);
}

auto HandleParseNode(Context& context, Parse::ExplicitParamListStartId node_id)
    -> bool {
  context.full_pattern_stack().StartExplicitParamList();
  return HandlePatternListStart(context, node_id);
}

// Handle the end of any kind of parameter list (tuple patterns have separate
// logic).
static auto HandleParamListEnd(Context& context, Parse::NodeId node_id,
                               Parse::NodeKind start_kind) -> bool {
  if (context.node_stack().PeekIs(start_kind)) {
    // End the pending region started by a trailing comma, or the opening
    // delimiter of an empty list.
    EndEmptyExprRegionForPattern(context);
  } else {
    // End the pending region for the last pattern in the list.
    EndExprRegionForPattern(context, context.node_stack());
  }
  // Note the Start node remains on the stack, where the param list handler can
  // make use of it.
  auto refs_id = context.param_and_arg_refs_stack().EndAndPop(start_kind);
  context.node_stack().Push(node_id, refs_id);
  return true;
}

auto HandleParseNode(Context& context, Parse::ImplicitParamListId node_id)
    -> bool {
  context.full_pattern_stack().EndImplicitParamList();
  return HandleParamListEnd(context, node_id,
                            Parse::NodeKind::ImplicitParamListStart);
}

auto HandleParseNode(Context& context, Parse::ExplicitParamListId node_id)
    -> bool {
  context.full_pattern_stack().EndExplicitParamList();
  return HandleParamListEnd(context, node_id,
                            Parse::NodeKind::ExplicitParamListStart);
}

auto HandleParseNode(Context& context, Parse::ParenPatternId node_id) -> bool {
  EndExprRegionForPattern(context, context.node_stack());
  auto pattern_id = context.node_stack().PopPattern();
  context.param_and_arg_refs_stack().PopAndDiscard();
  context.node_stack()
      .PopAndDiscardSoloNodeId<Parse::NodeKind::TuplePatternStart>();
  context.node_stack().Push(node_id, pattern_id);
  // Start a new pending `ExprRegion`, to maintain the invariant that one is
  // pending at the end of handling for a pattern.
  BeginExprRegionForPattern(context);
  return true;
}

auto HandleParseNode(Context& context, Parse::TuplePatternId node_id) -> bool {
  if (context.node_stack().PeekIs(Parse::NodeKind::TuplePatternStart)) {
    // End the pending region started by a trailing comma, or the opening
    // delimiter of an empty list.
    EndEmptyExprRegionForPattern(context);
  } else {
    // End the pending region for the last pattern in the list.
    EndExprRegionForPattern(context, context.node_stack());
  }
  auto refs_id = context.param_and_arg_refs_stack().EndAndPop(
      Parse::NodeKind::TuplePatternStart);
  context.node_stack()
      .PopAndDiscardSoloNodeId<Parse::NodeKind::TuplePatternStart>();

  const auto& inst_block = context.inst_blocks().Get(refs_id);
  llvm::SmallVector<SemIR::InstId> type_inst_ids;
  type_inst_ids.reserve(inst_block.size());
  for (auto inst : inst_block) {
    if (InNonStaticFieldDecl(context)) {
      CARBON_DIAGNOSTIC(FieldWithTuplePattern, Error,
                        "found tuple pattern in class `var` decl");
      context.emitter().Emit(LocIdForDiagnostics::TokenOnly(node_id),
                             FieldWithTuplePattern);

      return false;
    }

    auto type_id = ExtractScrutineeType(context.sem_ir(),
                                        context.insts().Get(inst).type_id());
    type_inst_ids.push_back(context.types().GetTypeInstId(type_id));
  }
  auto type_id = GetPatternType(context, GetTupleType(context, type_inst_ids));
  context.node_stack().Push(
      node_id,
      AddInst<SemIR::TuplePattern>(
          context, node_id, {.type_id = type_id, .elements_id = refs_id}));
  // Start a new pending `ExprRegion`, to maintain the invariant that one is
  // pending at the end of handling for a pattern.
  BeginExprRegionForPattern(context);
  return true;
}

auto HandleParseNode(Context& context, Parse::PatternListCommaId /*node_id*/)
    -> bool {
  EndExprRegionForPattern(context, context.node_stack());
  context.param_and_arg_refs_stack().ApplyComma();
  BeginExprRegionForPattern(context);
  return true;
}

}  // namespace Carbon::Check