//
// This file is distributed under the MIT License. See LICENSE.md for details.
//

#include "revng/ABI/DefaultFunctionPrototype.h"
#include "revng/ABI/Definition.h"
#include "revng/ABI/FunctionType/Support.h"

static model::UpcastableType defaultPrototype(model::Binary &Binary,
                                              model::ABI::Values ABI) {
  auto &&[Definition, Type] = Binary.makeRawFunctionDefinition();

  revng_assert(ABI != model::ABI::Invalid);
  Definition.Architecture() = model::ABI::getArchitecture(ABI);

  const abi::Definition &Defined = abi::Definition::get(ABI);
  for (const auto &Register : Defined.GeneralPurposeArgumentRegisters())
    Definition.addArgument(Register, model::PrimitiveType::make(Register));

  for (const auto &Register : Defined.GeneralPurposeReturnValueRegisters())
    Definition.addReturnValue(Register, model::PrimitiveType::make(Register));

  for (const auto &Register : Defined.CalleeSavedRegisters())
    Definition.PreservedRegisters().insert(Register);

  Definition.FinalStackOffset() = getCallPushSize(Binary.Architecture());

  return Type;
}

using OptionalABI = std::optional<model::ABI::Values>;
model::UpcastableType
abi::registerDefaultFunctionPrototype(model::Binary &Binary,
                                      OptionalABI MaybeABI) {
  if (!MaybeABI.has_value())
    MaybeABI = Binary.DefaultABI();
  revng_assert(*MaybeABI != model::ABI::Invalid);
  return defaultPrototype(Binary, MaybeABI.value());
}