carbon-lang/toolchain/check/handle_literal.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/context.h"

namespace Carbon::Check {

auto HandleLiteral(Context& context, Parse::Node parse_node) -> bool {
  auto token = context.parse_tree().node_token(parse_node);
  switch (auto token_kind = context.tokens().GetKind(token)) {
    case Lex::TokenKind::False:
    case Lex::TokenKind::True: {
      context.AddNodeAndPush(
          parse_node,
          SemIR::BoolLiteral{
              parse_node, context.GetBuiltinType(SemIR::BuiltinKind::BoolType),
              token_kind == Lex::TokenKind::True ? SemIR::BoolValue::True
                                                 : SemIR::BoolValue::False});
      break;
    }
    case Lex::TokenKind::IntegerLiteral: {
      context.AddNodeAndPush(
          parse_node,
          SemIR::IntegerLiteral{
              parse_node,
              context.GetBuiltinType(SemIR::BuiltinKind::IntegerType),
              context.tokens().GetIntegerLiteral(token)});
      break;
    }
    case Lex::TokenKind::RealLiteral: {
      context.AddNodeAndPush(
          parse_node,
          SemIR::RealLiteral{
              parse_node,
              context.GetBuiltinType(SemIR::BuiltinKind::FloatingPointType),
              context.tokens().GetRealLiteral(token)});
      break;
    }
    case Lex::TokenKind::StringLiteral: {
      auto id = context.tokens().GetStringLiteral(token);
      context.AddNodeAndPush(
          parse_node,
          SemIR::StringLiteral{
              parse_node,
              context.GetBuiltinType(SemIR::BuiltinKind::StringType), id});
      break;
    }
    case Lex::TokenKind::Type: {
      context.node_stack().Push(parse_node, SemIR::NodeId::BuiltinTypeType);
      break;
    }
    case Lex::TokenKind::Bool: {
      context.node_stack().Push(parse_node, SemIR::NodeId::BuiltinBoolType);
      break;
    }
    case Lex::TokenKind::IntegerTypeLiteral: {
      auto text = context.tokens().GetTokenText(token);
      if (text != "i32") {
        return context.TODO(parse_node, "Currently only i32 is allowed");
      }
      context.node_stack().Push(parse_node, SemIR::NodeId::BuiltinIntegerType);
      break;
    }
    case Lex::TokenKind::FloatingPointTypeLiteral: {
      auto text = context.tokens().GetTokenText(token);
      if (text != "f64") {
        return context.TODO(parse_node, "Currently only f64 is allowed");
      }
      context.node_stack().Push(parse_node,
                                SemIR::NodeId::BuiltinFloatingPointType);
      break;
    }
    case Lex::TokenKind::StringTypeLiteral: {
      context.node_stack().Push(parse_node, SemIR::NodeId::BuiltinStringType);
      break;
    }
    default: {
      return context.TODO(parse_node, llvm::formatv("Handle {0}", token_kind));
    }
  }

  return true;
}

}  // namespace Carbon::Check