// 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/lower/function_context.h"
#include "toolchain/sem_ir/file.h"

namespace Carbon::Lower {

auto HandleBindValue(FunctionContext& context, SemIR::InstId inst_id,
                     SemIR::BindValue inst) -> void {
  switch (auto rep = SemIR::GetValueRepr(context.sem_ir(), inst.type_id);
          rep.kind) {
    case SemIR::ValueRepr::Unknown:
      CARBON_FATAL()
          << "Value binding for type with incomplete value representation";
    case SemIR::ValueRepr::None:
      // Nothing should use this value, but StubRef needs a value to
      // propagate.
      // TODO: Remove this now the StubRefs are gone.
      context.SetLocal(inst_id,
                       llvm::PoisonValue::get(context.GetType(inst.type_id)));
      break;
    case SemIR::ValueRepr::Copy:
      context.SetLocal(inst_id, context.builder().CreateLoad(
                                    context.GetType(inst.type_id),
                                    context.GetValue(inst.value_id)));
      break;
    case SemIR::ValueRepr::Pointer:
      context.SetLocal(inst_id, context.GetValue(inst.value_id));
      break;
    case SemIR::ValueRepr::Custom:
      CARBON_FATAL() << "TODO: Add support for BindValue with custom value rep";
  }
}

auto HandleTemporary(FunctionContext& context, SemIR::InstId inst_id,
                     SemIR::Temporary inst) -> void {
  context.FinishInit(inst.type_id, inst.storage_id, inst.init_id);
  context.SetLocal(inst_id, context.GetValue(inst.storage_id));
}

auto HandleTemporaryStorage(FunctionContext& context, SemIR::InstId inst_id,
                            SemIR::TemporaryStorage inst) -> void {
  context.SetLocal(inst_id,
                   context.builder().CreateAlloca(context.GetType(inst.type_id),
                                                  nullptr, "temp"));
}

auto HandleValueAsRef(FunctionContext& context, SemIR::InstId inst_id,
                      SemIR::ValueAsRef inst) -> void {
  CARBON_CHECK(SemIR::GetExprCategory(context.sem_ir(), inst.value_id) ==
               SemIR::ExprCategory::Value);
  CARBON_CHECK(SemIR::GetValueRepr(context.sem_ir(), inst.type_id).kind ==
               SemIR::ValueRepr::Pointer);
  context.SetLocal(inst_id, context.GetValue(inst.value_id));
}

auto HandleValueOfInitializer(FunctionContext& context, SemIR::InstId inst_id,
                              SemIR::ValueOfInitializer inst) -> void {
  CARBON_CHECK(SemIR::GetExprCategory(context.sem_ir(), inst.init_id) ==
               SemIR::ExprCategory::Initializing);
  CARBON_CHECK(SemIR::GetValueRepr(context.sem_ir(), inst.type_id).kind ==
               SemIR::ValueRepr::Copy);
  CARBON_CHECK(SemIR::GetInitRepr(context.sem_ir(), inst.type_id).kind ==
               SemIR::InitRepr::ByCopy);
  context.SetLocal(inst_id, context.GetValue(inst.init_id));
}

}  // namespace Carbon::Lower