carbon-lang/toolchain/check/handle_name.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 "llvm/ADT/STLExtras.h"
#include "toolchain/check/context.h"
#include "toolchain/check/convert.h"
#include "toolchain/lex/token_kind.h"
#include "toolchain/sem_ir/node.h"

namespace Carbon::Check {

// Returns the name scope corresponding to base_id, or nullopt if not a scope.
// On invalid scopes, prints a diagnostic and still returns the scope.
static auto GetAsNameScope(Context& context, SemIR::NodeId base_id)
    -> std::optional<SemIR::NameScopeId> {
  auto base = context.nodes().Get(context.FollowNameReferences(base_id));
  if (auto base_as_namespace = base.TryAs<SemIR::Namespace>()) {
    return base_as_namespace->name_scope_id;
  }
  if (auto base_as_class = base.TryAs<SemIR::ClassType>()) {
    auto& class_info = context.classes().Get(base_as_class->class_id);
    if (!class_info.is_defined()) {
      CARBON_DIAGNOSTIC(QualifiedExpressionInIncompleteClassScope, Error,
                        "Member access into incomplete class `{0}`.",
                        std::string);
      auto builder = context.emitter().Build(
          context.nodes().Get(base_id).parse_node(),
          QualifiedExpressionInIncompleteClassScope,
          context.sem_ir().StringifyTypeExpression(base_id, true));
      context.NoteIncompleteClass(base_as_class->class_id, builder);
      builder.Emit();
    }
    return class_info.scope_id;
  }
  return std::nullopt;
}

auto HandleMemberAccessExpression(Context& context, Parse::Node parse_node)
    -> bool {
  StringId name_id = context.node_stack().Pop<Parse::NodeKind::Name>();
  auto base_id = context.node_stack().PopExpression();

  // If the base is a name scope, such as a class or namespace, perform lookup
  // into that scope.
  if (auto name_scope_id = GetAsNameScope(context, base_id)) {
    auto node_id = name_scope_id->is_valid()
                       ? context.LookupName(parse_node, name_id, *name_scope_id,
                                            /*print_diagnostics=*/true)
                       : SemIR::NodeId::BuiltinError;
    auto node = context.nodes().Get(node_id);
    // TODO: Track that this node was named within `base_id`.
    context.AddNodeAndPush(
        parse_node,
        SemIR::NameReference{parse_node, node.type_id(), name_id, node_id});
    return true;
  }

  // If the base isn't a scope, it must have a complete type.
  auto base_type_id = context.nodes().Get(base_id).type_id();
  if (!context.TryToCompleteType(base_type_id, [&] {
        CARBON_DIAGNOSTIC(IncompleteTypeInMemberAccess, Error,
                          "Member access into object of incomplete type `{0}`.",
                          std::string);
        return context.emitter().Build(
            context.nodes().Get(base_id).parse_node(),
            IncompleteTypeInMemberAccess,
            context.sem_ir().StringifyType(base_type_id, true));
      })) {
    context.node_stack().Push(parse_node, SemIR::NodeId::BuiltinError);
    return true;
  }

  // Materialize a temporary for the base expression if necessary.
  base_id = ConvertToValueOrReferenceExpression(context, base_id);
  base_type_id = context.nodes().Get(base_id).type_id();

  auto base_type = context.nodes().Get(
      context.sem_ir().GetTypeAllowBuiltinTypes(base_type_id));

  switch (base_type.kind()) {
    case SemIR::ClassType::Kind: {
      // Perform lookup for the name in the class scope.
      auto class_scope_id = context.classes()
                                .Get(base_type.As<SemIR::ClassType>().class_id)
                                .scope_id;
      auto member_id = context.LookupName(parse_node, name_id, class_scope_id,
                                          /*print_diagnostics=*/true);
      if (!member_id.is_valid()) {
        break;
      }

      // Perform instance binding if we found an instance member.
      auto member_type_id = context.nodes().Get(member_id).type_id();
      auto member_type_node = context.nodes().Get(
          context.sem_ir().GetTypeAllowBuiltinTypes(member_type_id));
      if (auto unbound_field_type =
              member_type_node.TryAs<SemIR::UnboundFieldType>()) {
        // TODO: Check that the unbound field type describes a member of this
        // class. Perform a conversion of the base if necessary.

        // Find the named field and build a field access expression.
        auto field_id = context.GetConstantValue(member_id);
        CARBON_CHECK(field_id.is_valid())
            << "Non-constant value " << context.nodes().Get(member_id)
            << " of unbound field type";
        auto field = context.nodes().Get(field_id).TryAs<SemIR::Field>();
        CARBON_CHECK(field)
            << "Unexpected value " << context.nodes().Get(field_id)
            << " for field name expression";
        context.AddNodeAndPush(
            parse_node, SemIR::ClassFieldAccess{
                            parse_node, unbound_field_type->field_type_id,
                            base_id, field->index});
        return true;
      }
      if (member_type_id ==
          context.GetBuiltinType(SemIR::BuiltinKind::FunctionType)) {
        // Find the named function and check whether it's an instance method.
        auto function_name_id = context.GetConstantValue(member_id);
        CARBON_CHECK(function_name_id.is_valid())
            << "Non-constant value " << context.nodes().Get(member_id)
            << " of function type";
        auto function_decl = context.nodes()
                                 .Get(function_name_id)
                                 .TryAs<SemIR::FunctionDeclaration>();
        CARBON_CHECK(function_decl)
            << "Unexpected value " << context.nodes().Get(function_name_id)
            << " of function type";
        auto& function = context.functions().Get(function_decl->function_id);
        for (auto param_id :
             context.node_blocks().Get(function.implicit_param_refs_id)) {
          if (context.nodes().Get(param_id).Is<SemIR::SelfParameter>()) {
            context.AddNodeAndPush(
                parse_node,
                SemIR::BoundMethod{
                    parse_node,
                    context.GetBuiltinType(SemIR::BuiltinKind::BoundMethodType),
                    base_id, member_id});
            return true;
          }
        }
      }

      // For a non-instance member, the result is that member.
      // TODO: Track that this was named within `base_id`.
      context.AddNodeAndPush(
          parse_node,
          SemIR::NameReference{parse_node, member_type_id, name_id, member_id});
      return true;
    }
    case SemIR::StructType::Kind: {
      auto refs = context.node_blocks().Get(
          base_type.As<SemIR::StructType>().fields_id);
      // TODO: Do we need to optimize this with a lookup table for O(1)?
      for (auto [i, ref_id] : llvm::enumerate(refs)) {
        auto field = context.nodes().GetAs<SemIR::StructTypeField>(ref_id);
        if (name_id == field.name_id) {
          context.AddNodeAndPush(
              parse_node, SemIR::StructAccess{parse_node, field.field_type_id,
                                              base_id, SemIR::MemberIndex(i)});
          return true;
        }
      }
      CARBON_DIAGNOSTIC(QualifiedExpressionNameNotFound, Error,
                        "Type `{0}` does not have a member `{1}`.", std::string,
                        llvm::StringRef);
      context.emitter().Emit(parse_node, QualifiedExpressionNameNotFound,
                             context.sem_ir().StringifyType(base_type_id),
                             context.strings().Get(name_id));
      break;
    }
    // TODO: `ConstType` should support member access just like the
    // corresponding non-const type, except that the result should have `const`
    // type if it creates a reference expression performing field access.
    default: {
      if (base_type_id != SemIR::TypeId::Error) {
        CARBON_DIAGNOSTIC(QualifiedExpressionUnsupported, Error,
                          "Type `{0}` does not support qualified expressions.",
                          std::string);
        context.emitter().Emit(parse_node, QualifiedExpressionUnsupported,
                               context.sem_ir().StringifyType(base_type_id));
      }
      break;
    }
  }

  // Should only be reached on error.
  context.node_stack().Push(parse_node, SemIR::NodeId::BuiltinError);
  return true;
}

auto HandlePointerMemberAccessExpression(Context& context,
                                         Parse::Node parse_node) -> bool {
  return context.TODO(parse_node, "HandlePointerMemberAccessExpression");
}

auto HandleName(Context& context, Parse::Node parse_node) -> bool {
  auto name_id = context.tokens().GetIdentifier(
      context.parse_tree().node_token(parse_node));
  // The parent is responsible for binding the name.
  context.node_stack().Push(parse_node, name_id);
  return true;
}

auto HandleNameExpression(Context& context, Parse::Node parse_node) -> bool {
  auto name_id = context.tokens().GetIdentifier(
      context.parse_tree().node_token(parse_node));
  auto value_id =
      context.LookupName(parse_node, name_id, SemIR::NameScopeId::Invalid,
                         /*print_diagnostics=*/true);
  auto value = context.nodes().Get(value_id);

  // If lookup finds a class declaration, the value is its `Self` type.
  if (auto class_decl = value.TryAs<SemIR::ClassDeclaration>()) {
    value_id = context.sem_ir().GetTypeAllowBuiltinTypes(
        context.classes().Get(class_decl->class_id).self_type_id);
    value = context.nodes().Get(value_id);
  }

  CARBON_CHECK(value.kind().value_kind() == SemIR::NodeValueKind::Typed);
  context.AddNodeAndPush(
      parse_node,
      SemIR::NameReference{parse_node, value.type_id(), name_id, value_id});
  return true;
}

auto HandleQualifiedDeclaration(Context& context, Parse::Node parse_node)
    -> bool {
  auto [parse_node2, name_id2] =
      context.node_stack().PopWithParseNode<Parse::NodeKind::Name>();

  Parse::Node parse_node1 = context.node_stack().PeekParseNode();
  switch (context.parse_tree().node_kind(parse_node1)) {
    case Parse::NodeKind::QualifiedDeclaration:
      // This is the second or subsequent QualifiedDeclaration in a chain.
      // Nothing to do: the first QualifiedDeclaration remains as a
      // bracketing node for later QualifiedDeclarations.
      break;

    case Parse::NodeKind::Name: {
      // This is the first QualifiedDeclaration in a chain, and starts with a
      // name.
      auto name_id = context.node_stack().Pop<Parse::NodeKind::Name>();
      context.declaration_name_stack().ApplyNameQualifier(parse_node1, name_id);
      // Add the QualifiedDeclaration so that it can be used for bracketing.
      context.node_stack().Push(parse_node);
      break;
    }

    default:
      CARBON_FATAL() << "Unexpected node kind on left side of qualified "
                        "declaration name";
  }

  context.declaration_name_stack().ApplyNameQualifier(parse_node2, name_id2);
  return true;
}

auto HandleSelfTypeNameExpression(Context& context, Parse::Node parse_node)
    -> bool {
  // TODO: This will find a local variable declared with name `r#Self`, but
  // should not. See #2984 and the corresponding code in
  // HandleClassDefinitionStart.
  auto name_id = context.strings().Add(
      Lex::TokenKind::SelfTypeIdentifier.fixed_spelling());
  auto value_id =
      context.LookupName(parse_node, name_id, SemIR::NameScopeId::Invalid,
                         /*print_diagnostics=*/true);
  auto value = context.nodes().Get(value_id);
  context.AddNodeAndPush(
      parse_node,
      SemIR::NameReference{parse_node, value.type_id(), name_id, value_id});
  return true;
}

auto HandleSelfValueName(Context& context, Parse::Node parse_node) -> bool {
  context.node_stack().Push(parse_node);
  return true;
}

auto HandleSelfValueNameExpression(Context& context, Parse::Node parse_node)
    -> bool {
  // TODO: This will find a local variable declared with name `r#self`, but
  // should not. See #2984 and the corresponding code in
  // HandleFunctionDefinitionStart.
  auto name_id = context.strings().Add(SemIR::SelfParameter::Name);
  auto value_id =
      context.LookupName(parse_node, name_id, SemIR::NameScopeId::Invalid,
                         /*print_diagnostics=*/true);
  auto value = context.nodes().Get(value_id);
  context.AddNodeAndPush(
      parse_node,
      SemIR::NameReference{parse_node, value.type_id(), name_id, value_id});
  return true;
}

}  // namespace Carbon::Check