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

#include <csignal>
#include <cstdint>
#include <cstdlib>
#include <cstring>
#include <iterator>
#include <optional>
#include <string>
#include <type_traits>
#include <vector>

#include "llvm/ADT/STLExtras.h"
#include "llvm/Support/DynamicLibrary.h"
#include "llvm/Support/Error.h"
#include "llvm/Support/MemoryBuffer.h"
#include "llvm/Support/raw_ostream.h"

#include "revng/Pipeline/AllRegistries.h"
#include "revng/Pipeline/Container.h"
#include "revng/Pipeline/Runner.h"
#include "revng/Pipeline/Target.h"
#include "revng/PipelineC/PipelineC.h"
#include "revng/Pipes/ModelGlobal.h"
#include "revng/Pipes/PipelineManager.h"
#include "revng/Support/Assert.h"
#include "revng/Support/InitRevng.h"
#include "revng/TupleTree/TupleTreeDiff.h"

using namespace pipeline;
using namespace ::revng::pipes;

template<typename T>
concept default_constructible = std::is_default_constructible<T>::value;

template<default_constructible T>
class ExistingOrNew {
private:
  T *Pointer;
  std::optional<T> Default;

public:
  ExistingOrNew(T *Pointer) {
    if (Pointer != nullptr) {
      this->Pointer = Pointer;
    } else {
      Default.emplace();
      this->Pointer = &*Default;
    }
  }

  ExistingOrNew(ExistingOrNew &&) = delete;
  ExistingOrNew(const ExistingOrNew &) = delete;
  ExistingOrNew &operator=(ExistingOrNew &&) = delete;
  ExistingOrNew &operator=(const ExistingOrNew &) = delete;
  ~ExistingOrNew() = default;

  T &operator*() { return *Pointer; }
  T *operator->() { return Pointer; }
};

/// Used when we want to return a stack allocated string. Copies the string onto
/// the heap and gives ownership of to the caller
static char *copyString(llvm::StringRef str) {
  char *ToReturn = (char *) malloc(sizeof(char) * (str.size() + 1));
  strncpy(ToReturn, str.data(), str.size());
  ToReturn[str.size()] = 0;
  return ToReturn;
}

static bool Initialized = false;

static bool loadLibraryPermanently(const char *LibraryPath) {
  revng_check(not Initialized);
  revng_check(LibraryPath != nullptr);

  std::string Msg;
  return llvm::sys::DynamicLibrary::LoadLibraryPermanently(LibraryPath, &Msg);
}

static std::optional<revng::InitRevng> InitRevngInstance = std::nullopt;
typedef void (*sighandler_t)(int);

bool rp_initialize(int argc,
                   char *argv[],
                   int libraries_count,
                   const char *libraries_path[],
                   int signals_to_preserve_count,
                   int signals_to_preserve[]) {
  if (argc != 0)
    revng_check(argv != nullptr);
  if (libraries_count != 0)
    revng_check(libraries_path != nullptr);

  if (Initialized)
    return false;

  std::map<int, sighandler_t> Signals;
  for (int I = 0; I < signals_to_preserve_count; I++) {
    // For each signal number we are asked to preserve we need to extract the
    // function pointer to the signal and save it. The constructor for
    // revng::InitRevng will call LLVM's RegisterHandlers which overrides most
    // signal handlers and chains the previous one afterwards, we want instead
    // to keep the already existing handler and remove the LLVM's one
    int SigNumber = signals_to_preserve[I];
    sighandler_t Handler = signal(SigNumber, SIG_DFL);
    if (Handler != SIG_ERR && Handler != NULL) {
      // We save the signal handler for restoration after we initialize LLVM's
      // machinery, as said we do not restore the signal to avoid LLVM chaining
      // it after its own
      Signals[SigNumber] = Handler;
    }
  }

  revng_check(not InitRevngInstance.has_value());
  InitRevngInstance.emplace(argc, argv);

  llvm::cl::ParseCommandLineOptions(argc, argv);
  for (int I = 0; I < libraries_count; I++)
    if (loadLibraryPermanently(libraries_path[I]))
      return false;

  Initialized = true;

  Registry::runAllInitializationRoutines();

  for (const auto &[SigNumber, Handler] : Signals) {
    // All of LLVM's initialization is complete, restore the original signals to
    // the respective signal number
    signal(SigNumber, Handler);
  }

  return true;
}

bool rp_shutdown() {
  if (InitRevngInstance.has_value()) {
    InitRevngInstance.reset();
    return true;
  }
  return false;
}

static rp_manager *rp_manager_create_impl(uint64_t pipelines_count,
                                          const char *pipelines_path[],
                                          uint64_t pipeline_flags_count,
                                          const char *pipeline_flags[],
                                          const char *execution_directory,
                                          bool is_path = true) {
  revng_check(Initialized);
  revng_check(pipelines_path != nullptr);
  if (pipeline_flags == nullptr)
    revng_check(pipeline_flags_count == 0);
  revng_check(execution_directory != nullptr);

  std::vector<std::string> FlagsVector;
  for (size_t I = 0; I < pipeline_flags_count; I++)
    FlagsVector.push_back(pipeline_flags[I]);

  std::vector<std::string> Pipelines;
  for (size_t I = 0; I < pipelines_count; I++)
    Pipelines.push_back(pipelines_path[I]);
  auto Pipeline = is_path ?
                    PipelineManager::create(Pipelines,
                                            FlagsVector,
                                            execution_directory) :
                    PipelineManager::createFromMemory(Pipelines,
                                                      FlagsVector,
                                                      execution_directory);
  if (not Pipeline) {
    auto Error = Pipeline.takeError();
    llvm::errs() << Error;
    llvm::consumeError(std::move(Error));
    return nullptr;
  }

  auto manager = new PipelineManager(std::move(*Pipeline));
  return manager;
}

rp_manager *rp_manager_create_from_string(uint64_t pipelines_count,
                                          const char *pipelines[],
                                          uint64_t pipeline_flags_count,
                                          const char *pipeline_flags[],
                                          const char *execution_directory) {
  return rp_manager_create_impl(pipelines_count,
                                pipelines,
                                pipeline_flags_count,
                                pipeline_flags,
                                execution_directory,
                                false);
}

rp_manager *rp_manager_create(uint64_t pipelines_count,
                              const char *pipelines_path[],
                              uint64_t pipeline_flags_count,
                              const char *pipeline_flags[],
                              const char *execution_directory) {
  return rp_manager_create_impl(pipelines_count,
                                pipelines_path,
                                pipeline_flags_count,
                                pipeline_flags,
                                execution_directory);
}

bool rp_manager_save(rp_manager *manager, const char *path) {
  revng_check(manager != nullptr);

  llvm::StringRef DirPath;
  if (path != nullptr)
    DirPath = llvm::StringRef(path);

  auto Error = manager->storeToDisk(DirPath);
  if (not Error)
    return true;

  llvm::consumeError(std::move(Error));
  return false;
}

bool rp_step_save(rp_step *step, const char *path) {
  revng_check(step != nullptr);
  revng_check(path != nullptr);

  auto Error = step->storeToDisk(path);
  if (not Error)
    return true;

  llvm::consumeError(std::move(Error));
  return false;
}

bool rp_manager_save_context(rp_manager *manager, const char *path) {
  revng_check(manager != nullptr);
  revng_check(path != nullptr);

  auto Error = manager->context().storeToDisk(path);
  if (not Error)
    return true;

  llvm::consumeError(std::move(Error));
  return false;
}

void rp_manager_destroy(rp_manager *manager) {
  revng_check(manager != nullptr);
  delete manager;
}

uint64_t rp_manager_steps_count(rp_manager *manager) {
  revng_check(manager != nullptr);
  return manager->getRunner().size();
}

uint64_t rp_manager_step_name_to_index(rp_manager *manager, const char *name) {
  revng_check(manager != nullptr);
  revng_check(name != nullptr);
  size_t I = 0;
  for (const auto &Step : manager->getRunner()) {
    if (Step.getName() == name)
      return I;
    I++;
  }

  return RP_STEP_NOT_FOUND;
}

rp_step *rp_manager_get_step(rp_manager *manager, uint64_t index) {
  revng_check(manager != nullptr);
  revng_check(index < manager->getRunner().size());
  return &(*(std::next(manager->getRunner().begin(), index)));
}

const char *rp_step_get_name(rp_step *step) {
  revng_check(step != nullptr);
  return step->getName().data();
}

int rp_step_get_analyses_count(rp_step *step) {
  return step->getAnalysesSize();
}

rp_analysis *rp_step_get_analysis(rp_step *step, int index) {
  revng_check(step != nullptr);
  if (index >= rp_step_get_analyses_count(step))
    return nullptr;

  return &*(std::next(step->analysesBegin(), index));
}

rp_container *
rp_step_get_container(rp_step *step, rp_container_identifier *container) {
  revng_check(step != nullptr);
  revng_check(container != nullptr);
  if (step->containers().isContainerRegistered(container->first())) {
    step->containers()[container->first()];
    return &*step->containers().find(container->first());
  } else {
    return nullptr;
  }
}

rp_kind *rp_step_get_artifacts_kind(rp_step *step) {
  revng_check(step != nullptr);
  return step->getArtifactsKind();
}

rp_container *rp_step_get_artifacts_container(rp_step *step) {
  revng_check(step != nullptr);
  return step->getArtifactsContainer();
}

const char *rp_step_get_artifacts_single_target_filename(rp_step *step) {
  revng_check(step != nullptr);
  return copyString(step->getArtifactsSingleTargetFilename());
}

uint64_t rp_targets_list_targets_count(rp_targets_list *targets_list) {
  revng_check(targets_list != nullptr);
  return targets_list->size();
}

rp_kind *
rp_manager_get_kind_from_name(rp_manager *manager, const char *kind_name) {
  revng_check(manager != nullptr);
  revng_check(kind_name != nullptr);

  return manager->getKind(kind_name);
}

const char *rp_kind_get_name(rp_kind *kind) {
  revng_check(kind != nullptr);
  return kind->name().data();
}

rp_kind *rp_kind_get_parent(rp_kind *kind) {
  revng_check(kind != nullptr);
  return kind->parent();
}

rp_diff_map *rp_manager_run_analysis(rp_manager *manager,
                                     const char *step_name,
                                     const char *analysis_name,
                                     rp_container_targets_map *target_map,
                                     rp_invalidations *invalidations,
                                     const rp_string_map *options) {
  revng_check(manager != nullptr);
  revng_check(step_name != nullptr);
  revng_check(analysis_name != nullptr);
  revng_check(target_map != nullptr);

  ExistingOrNew<rp_invalidations> Invalidations(invalidations);
  ExistingOrNew<const rp_string_map> Options(options);

  auto MaybeDiffs = manager->runAnalysis(analysis_name,
                                         step_name,
                                         *target_map,
                                         *Invalidations,
                                         *Options);
  if (!MaybeDiffs) {
    llvm::consumeError(MaybeDiffs.takeError());
    return nullptr;
  }

  return new rp_diff_map(std::move(*MaybeDiffs));
}

void rp_diff_map_destroy(rp_diff_map *to_free) {
  delete to_free;
}

const char *rp_diff_map_get_diff(rp_diff_map *map, const char *global_name) {
  auto It = map->find(global_name);
  if (It == map->end())
    return nullptr;

  std::string S;
  llvm::raw_string_ostream OS(S);
  It->second.serialize(OS);
  OS.flush();
  return copyString(S);
}

const rp_buffer *rp_manager_produce_targets(rp_manager *manager,
                                            uint64_t targets_count,
                                            rp_target *targets[],
                                            rp_step *step,
                                            rp_container *container) {
  revng_check(manager != nullptr);
  revng_check(targets_count != 0);
  revng_check(targets != nullptr);
  revng_check(step != nullptr);
  revng_check(container != nullptr);

  ContainerToTargetsMap Targets;
  for (size_t I = 0; I < targets_count; I++)
    Targets[container->second->name()].push_back(*targets[I]);

  auto Error = manager->getRunner().run(step->getName(), Targets);
  if (Error) {
    llvm::consumeError(std::move(Error));
    return nullptr;
  }

  rp_buffer *Out = new rp_buffer();
  llvm::raw_svector_ostream Serialized(*Out);
  const auto &ToFilter = Targets[container->second->name()];
  const auto &Cloned = container->second->cloneFiltered(ToFilter);
  llvm::cantFail(Cloned->serialize(Serialized));

  return Out;
}

rp_target *rp_target_create(rp_kind *kind,
                            uint64_t path_components_count,
                            const char *path_components[]) {
  revng_check(kind != nullptr);
  revng_check(path_components != nullptr);
  revng_check(kind->rank().depth() == path_components_count);
  std::vector<std::string> List;
  for (size_t I = 0; I < path_components_count; I++) {
    llvm::StringRef Rank(path_components[I]);
    List.push_back(Rank.str());
  }

  return new Target(std::move(List), *kind);
}

void rp_target_destroy(rp_target *target) {
  revng_check(target != nullptr);
  delete target;
}

uint64_t rp_manager_kinds_count(rp_manager *manager) {
  revng_check(manager != nullptr);
  return manager->getRunner().getKindsRegistry().size();
}

rp_kind *rp_manager_get_kind(rp_manager *manager, uint64_t index) {
  revng_check(manager != nullptr);
  revng_check(index < manager->getRunner().getKindsRegistry().size());
  return &*std::next(manager->getRunner().getKindsRegistry().begin(), index);
}

bool rp_container_store(rp_container *container, const char *path) {
  revng_check(container != nullptr);
  revng_check(path != nullptr);
  auto Error = container->second->storeToDisk(path);
  if (not Error)
    return true;

  llvm::consumeError(std::move(Error));
  return false;
}

bool rp_manager_container_deserialize(rp_manager *manager,
                                      rp_step *step,
                                      const char *container_name,
                                      const char *content,
                                      uint64_t size) {
  revng_check(manager != nullptr);
  revng_check(step != nullptr);
  revng_check(container_name != nullptr);
  revng_check(content != nullptr);

  auto Buffer = llvm::MemoryBuffer::getMemBuffer(llvm::StringRef(content, size),
                                                 "",
                                                 false);
  auto Error = manager->deserializeContainer(*step, container_name, *Buffer);
  if (!!Error) {
    llvm::consumeError(std::move(Error));
    return false;
  }
  return true;
}

rp_kind *rp_target_get_kind(rp_target *target) {
  revng_check(target != nullptr);
  return &target->getKind();
}

uint64_t rp_target_path_components_count(rp_target *target) {
  revng_check(target != nullptr);
  return target->getPathComponents().size();
}
const char *rp_target_get_path_component(rp_target *target, uint64_t index) {
  revng_check(target != nullptr);
  auto &PathComponents = target->getPathComponents();
  revng_check(index < PathComponents.size());

  return PathComponents[index].c_str();
}

char *rp_manager_create_container_path(rp_manager *manager,
                                       const char *step_name,
                                       const char *container_name) {
  revng_check(manager != nullptr);
  revng_check(step_name != nullptr);
  revng_check(container_name != nullptr);

  if (manager->executionDirectory().empty())
    return nullptr;

  llvm::SmallString<128> Path;
  llvm::sys::path::append(Path,
                          manager->executionDirectory(),
                          step_name,
                          container_name);
  return copyString(Path);
}

rp_targets_list *
rp_manager_get_container_targets_list(rp_manager *manager,
                                      rp_container *container) {
  revng_check(manager != nullptr);
  revng_check(container != nullptr);

  return manager->getTargetsAvailableFor(*container);
}

rp_target *
rp_targets_list_get_target(rp_targets_list *targets_list, uint64_t index) {
  revng_check(targets_list != nullptr);
  revng_check(index < targets_list->size());
  return &(*targets_list)[index];
}

void rp_string_destroy(char *string) {
  revng_check(string != nullptr);
  free(string);
}

uint64_t rp_manager_containers_count(rp_manager *manager) {
  revng_check(manager != nullptr);
  return manager->getRunner().registeredContainersCount();
}

rp_container_identifier *
rp_manager_get_container_identifier(rp_manager *manager, uint64_t index) {
  revng_check(manager != nullptr);
  const auto &ContainerRegistry = manager->getRunner().getContainerFactorySet();
  revng_check(index < ContainerRegistry.size());
  return &*std::next(ContainerRegistry.begin(), index);
}

const char *rp_container_get_name(rp_container *container) {
  revng_check(container != nullptr);
  return container->second->name().data();
}

const char *
rp_container_identifier_get_name(rp_container_identifier *identifier) {
  revng_check(identifier != nullptr);
  return identifier->first().data();
}

char *rp_target_create_serialized_string(rp_target *target) {
  revng_check(target != nullptr);
  return copyString(target->serialize());
}

rp_target *
rp_target_create_from_string(rp_manager *manager, const char *string) {
  revng_check(manager != nullptr);
  revng_check(string != nullptr);
  TargetsList Targets;
  auto Error = parseTarget(manager->context(),
                           string,
                           manager->context().getKindsRegistry(),
                           Targets);
  revng_check(Targets.size() == 1);
  if (not Error)
    return new Target(std::move(Targets.front()));

  llvm::consumeError(std::move(Error));
  return nullptr;
}

bool rp_target_is_ready(rp_target *target, rp_container *container) {
  revng_assert(target);
  revng_assert(container);
  return container->second->enumerate().contains(*target);
}

/// TODO Remove the redundant copy by writing a custom string stream that writes
/// direclty to a buffer to return.
const char *
rp_manager_create_global_copy(rp_manager *manager, const char *global_name) {
  std::string Out;
  llvm::raw_string_ostream Serialized(Out);
  auto &GlobalsMap = manager->context().getGlobals();
  if (auto Error = GlobalsMap.serialize(global_name, Serialized); Error) {
    llvm::consumeError(std::move(Error));
    return nullptr;
  }
  Serialized.flush();
  return copyString(Out);
}

int rp_manager_get_globals_count(rp_manager *manager) {
  return manager->context().getGlobals().size();
}

const char *rp_manager_get_global_name(rp_manager *manager, int index) {
  if (index < rp_manager_get_globals_count(manager))
    return copyString(manager->context().getGlobals().getName(index).str());
  return nullptr;
}

static bool llvmErrorToRpError(llvm::Error Error, rp_error *out) {
  if (out == nullptr)
    return !!Error;

  bool Success = true;
  // clang-format off
  llvm::handleAllErrors(std::move(Error),
    [&](const revng::DocumentErrorBase &Error) {
      rp_document_error ErrorBody(Error.getTypeName(),
                                  Error.getLocationTypeName());

      for (size_t I = 0; I < Error.size(); I++) {
        rp_error_reason reason(Error.getMessage(I), Error.getLocation(I));
        ErrorBody.Reasons.emplace_back(reason);
      }

      **out = std::move(ErrorBody);
      Success = false;
    },
    [&](const llvm::ErrorInfoBase &OtherErrors) {
      std::string Reason;
      llvm::raw_string_ostream OS(Reason);
      OtherErrors.log(OS);
      OS.flush();
      **out = rp_simple_error(Reason, "");
      Success = false;
    });
  // clang-format on

  return Success;
}

template<bool commit>
inline bool rp_manager_set_global_impl(rp_manager *manager,
                                       const char *serialized,
                                       const char *global_name,
                                       rp_invalidations *invalidations,
                                       rp_error *error) {
  revng_check(manager != nullptr);
  revng_check(serialized != nullptr);
  revng_check(global_name != nullptr);

  auto &GlobalsMap = manager->context().getGlobals();
  auto Buffer = llvm::MemoryBuffer::getMemBuffer(serialized);

  auto MaybeGlobal = GlobalsMap.get(global_name);
  if (!MaybeGlobal) {
    llvmErrorToRpError(MaybeGlobal.takeError(), error);
    return false;
  }

  auto MaybeNewGlobal = GlobalsMap.createNew(global_name, *Buffer);
  if (!MaybeNewGlobal) {
    llvmErrorToRpError(MaybeNewGlobal.takeError(), error);
    return false;
  }

  if (auto Error = MaybeNewGlobal->get()->verify(); Error) {
    if (error)

      **error = rp_simple_error(std::string("could not verify ") + global_name,
                                "");
    return false;
  }

  if constexpr (commit) {
    auto &NewGlobal = MaybeNewGlobal.get();
    auto Diff = MaybeGlobal.get()->diff(*NewGlobal);
    *MaybeGlobal.get() = *NewGlobal;

    auto MaybeInvalidations = manager->invalidateFromDiff(global_name, Diff);
    if (!MaybeInvalidations) {
      llvmErrorToRpError(MaybeInvalidations.takeError(), error);
      return false;
    }

    ExistingOrNew<rp_invalidations> Invalidations(invalidations);
    *Invalidations = MaybeInvalidations.get();
  }

  return true;
}

bool rp_manager_set_global(rp_manager *manager,
                           const char *serialized,
                           const char *global_name,
                           rp_invalidations *invalidations,
                           rp_error *error) {
  return rp_manager_set_global_impl<true>(manager,
                                          serialized,
                                          global_name,
                                          invalidations,
                                          error);
}

bool rp_manager_verify_global(rp_manager *manager,
                              const char *serialized,
                              const char *global_name,
                              rp_error *error) {
  return rp_manager_set_global_impl<false>(manager,
                                           serialized,
                                           global_name,
                                           nullptr,
                                           error);
}

template<bool commit>
inline bool rp_manager_apply_diff_impl(rp_manager *manager,
                                       const char *diff,
                                       const char *global_name,
                                       rp_invalidations *invalidations,
                                       rp_error *error) {
  revng_check(manager != nullptr);
  revng_check(diff != nullptr);
  revng_check(global_name != nullptr);

  auto &GlobalsMap = manager->context().getGlobals();
  auto Buffer = llvm::MemoryBuffer::getMemBuffer(diff);

  auto GlobalOrError = GlobalsMap.get(global_name);
  if (!GlobalOrError) {
    llvmErrorToRpError(GlobalOrError.takeError(), error);
    return false;
  }

  auto &Global = GlobalOrError.get();
  auto MaybeDiff = Global->deserializeDiff(*Buffer);
  if (!MaybeDiff) {
    llvmErrorToRpError(MaybeDiff.takeError(), error);
    return false;
  }

  auto &Diff = MaybeDiff.get();
  auto GlobalClone = Global->clone();
  auto Error = GlobalClone->applyDiff(Diff);
  if (Error) {
    llvmErrorToRpError(std::move(Error), error);
    return false;
  }

  if (auto Error = GlobalClone->verify(); Error) {
    **error = rp_simple_error(std::string("could not verify ") + global_name,
                              "");
    return false;
  }

  if constexpr (commit) {
    *Global = *GlobalClone;
    auto MaybeInvalidations = manager->invalidateFromDiff(global_name, Diff);
    if (!MaybeInvalidations) {
      llvmErrorToRpError(MaybeInvalidations.takeError(), error);
      return false;
    }

    ExistingOrNew<rp_invalidations> Invalidations(invalidations);
    *Invalidations = MaybeInvalidations.get();
  }

  return true;
}

bool rp_manager_apply_diff(rp_manager *manager,
                           const char *diff,
                           const char *global_name,
                           rp_invalidations *invalidations,
                           rp_error *error) {
  return rp_manager_apply_diff_impl<true>(manager,
                                          diff,
                                          global_name,
                                          invalidations,
                                          error);
}

bool rp_manager_verify_diff(rp_manager *manager,
                            const char *diff,
                            const char *global_name,
                            rp_error *error) {
  return rp_manager_apply_diff_impl<false>(manager,
                                           diff,
                                           global_name,
                                           nullptr,
                                           error);
}

uint64_t rp_ranks_count() {
  return Rank::getAll().size();
}

rp_rank *rp_rank_get(uint64_t index) {
  revng_check(index < Rank::getAll().size());
  return Rank::getAll()[index];
}

rp_rank *rp_rank_get_from_name(const char *rank_name) {
  revng_check(rank_name != nullptr);
  for (auto rank : Rank::getAll()) {
    if (rank->name() == rank_name) {
      return rank;
    }
  }
  return nullptr;
}

const char *rp_rank_get_name(rp_rank *rank) {
  revng_check(rank != nullptr);
  return rank->name().data();
}

uint64_t rp_rank_get_depth(rp_rank *rank) {
  revng_check(rank != nullptr);
  return rank->depth();
}

rp_rank *rp_rank_get_parent(rp_rank *rank) {
  revng_check(rank != nullptr);
  return rank->parent();
}

rp_rank *rp_kind_get_rank(rp_kind *kind) {
  revng_check(kind != nullptr);
  return &kind->rank();
}

rp_step *rp_step_get_parent(rp_step *step) {
  revng_check(step != nullptr);
  if (step->hasPredecessor()) {
    return &step->getPredecessor();
  } else {
    return nullptr;
  }
}

const char *rp_container_get_mime(rp_container *container) {
  revng_check(container != nullptr);
  return container->getValue()->mimeType().data();
}

const rp_buffer *
rp_container_extract_one(rp_container *container, rp_target *target) {
  if (!container->second->enumerate().contains(*target)) {
    return nullptr;
  }

  rp_buffer *Out = new rp_buffer();
  llvm::raw_svector_ostream Serialized(*Out);
  llvm::cantFail(container->second->extractOne(Serialized, *target));

  return Out;
}

const char *rp_analysis_get_name(rp_analysis *analysis) {
  return analysis->second->getUserBoundName().c_str();
}

int rp_analysis_get_arguments_count(rp_analysis *analysis) {
  return analysis->second->getRunningContainersNames().size();
}

const char *rp_analysis_get_argument_name(rp_analysis *analysis, int index) {
  revng_check(index < rp_analysis_get_arguments_count(analysis));
  std::string Name = analysis->second->getRunningContainersNames()[index];

  return copyString(Name);
}

int rp_analysis_get_argument_acceptable_kinds_count(rp_analysis *analysis,
                                                    int argument_index) {
  return analysis->second->getAcceptedKinds(argument_index).size();
}

const rp_kind *rp_analysis_get_argument_acceptable_kind(rp_analysis *analysis,
                                                        int argument_index,
                                                        int kind_index) {
  const auto &Accepted = analysis->second->getAcceptedKinds(argument_index);
  if (static_cast<size_t>(kind_index) >= Accepted.size())
    return nullptr;
  return Accepted[kind_index];
}

rp_diff_map *rp_manager_run_all_analyses(rp_manager *manager,
                                         rp_invalidations *invalidations,
                                         const rp_string_map *options) {
  revng_check(manager != nullptr);

  ExistingOrNew<rp_invalidations> Invalidations(invalidations);
  ExistingOrNew<const rp_string_map> Options(options);

  auto MaybeDiffs = manager->runAllAnalyses(*Invalidations, *Options);
  if (!MaybeDiffs) {
    llvm::consumeError(MaybeDiffs.takeError());
    return nullptr;
  }

  return new rp_diff_map(std::move(*MaybeDiffs));
}

bool rp_diff_map_is_empty(rp_diff_map *map) {
  for (auto &Entry : *map) {
    if (!Entry.second.isEmpty()) {
      return false;
    }
  }
  return true;
}

rp_error *rp_error_create() {
  return new rp_error(nullptr);
}

bool rp_error_is_success(rp_error *error) {
  return error->get() == nullptr;
}

bool rp_error_is_document_error(rp_error *error) {
  return std::holds_alternative<rp_document_error>(**error);
}

rp_simple_error *rp_error_get_simple_error(rp_error *error) {
  if (error->get() == nullptr
      || not std::holds_alternative<rp_simple_error>(**error))
    return nullptr;

  return &std::get<rp_simple_error>(**error);
}

rp_document_error *rp_error_get_document_error(rp_error *error) {
  if (error->get() == nullptr
      || not std::holds_alternative<rp_document_error>(**error))
    return nullptr;

  return &std::get<rp_document_error>(**error);
}

void rp_error_destroy(rp_error *error_list) {
  revng_check(error_list != nullptr);
  delete error_list;
}

size_t rp_document_error_reasons_count(rp_document_error *error) {
  revng_check(error != nullptr);
  return error->Reasons.size();
}

const char *rp_document_error_get_error_type(rp_document_error *error) {
  revng_check(error != nullptr);
  return error->ErrorType.c_str();
}

const char *rp_document_error_get_location_type(rp_document_error *error_list) {
  revng_check(error_list != nullptr);
  return error_list->LocationType.c_str();
}

const char *
rp_document_error_get_error_message(rp_document_error *error, uint64_t index) {
  revng_check(error != nullptr);
  return error->Reasons.at(index).Message.c_str();
}

const char *
rp_document_error_get_error_location(rp_document_error *error, uint64_t index) {
  revng_check(error != nullptr);
  return error->Reasons.at(index).Location.c_str();
}

const char *rp_simple_error_get_error_type(rp_simple_error *error) {
  revng_check(error != nullptr);
  return error->ErrorType.c_str();
}

const char *rp_simple_error_get_message(rp_simple_error *error) {
  revng_check(error != nullptr);
  return error->Message.c_str();
}

int rp_analysis_get_options_count(rp_analysis *analysis) {
  return analysis->second->getOptionsNames().size();
}

const char * /*owning*/
rp_analysis_get_option_name(rp_analysis *analysis, int extra_argument_index) {

  auto Names = analysis->second->getOptionsNames();
  if (extra_argument_index < 0
      or static_cast<size_t>(extra_argument_index) >= Names.size())
    return nullptr;
  return copyString(Names[extra_argument_index]);
}

const char * /*owning*/
rp_analysis_get_option_type(rp_analysis *analysis, int extra_argument_index) {

  auto Names = analysis->second->getOptionsTypes();
  if (extra_argument_index < 0
      or static_cast<size_t>(extra_argument_index) >= Names.size())
    return nullptr;
  return copyString(Names[extra_argument_index]);
}

rp_string_map *rp_string_map_create() {
  return new rp_string_map();
}

void rp_string_map_destroy(rp_string_map *map) {
  delete map;
}

void rp_string_map_insert(rp_string_map *map,
                          const char *key,
                          const char *value) {
  auto Result = map->insert_or_assign(key, value);
  revng_assert(Result.second);
}

rp_invalidations *rp_invalidations_create() {
  return new rp_invalidations();
}

void rp_invalidations_destroy(rp_invalidations *invalidations) {
  delete invalidations;
}

const char *rp_invalidations_serialize(const rp_invalidations *invalidations) {
  std::string Out;

  for (auto &StepPair : *invalidations) {
    for (auto &ContainerPair : StepPair.second) {
      for (auto Target : ContainerPair.second) {
        Out += llvm::join_items('/',
                                StepPair.first(),
                                ContainerPair.first(),
                                Target.serialize())
               + '\n';
      }
    }
  }

  return copyString(Out);
}

uint64_t rp_kind_get_defined_location_count(rp_kind *kind) {
  revng_check(kind != nullptr);
  return kind->definedLocations().size();
}

const rp_rank *rp_kind_get_defined_location(rp_kind *kind, uint64_t index) {
  revng_check(kind != nullptr);
  const llvm::ArrayRef<const Rank *> Locations = kind->definedLocations();
  if (index < Locations.size())
    return Locations[index];
  else
    return nullptr;
}

uint64_t rp_kind_get_preferred_kind_count(rp_kind *kind) {
  revng_check(kind != nullptr);
  return kind->preferredKinds().size();
}

const rp_kind *rp_kind_get_preferred_kind(rp_kind *kind, uint64_t index) {
  revng_check(kind != nullptr);
  const llvm::ArrayRef<const Kind *> PreferredKinds = kind->preferredKinds();
  if (index < PreferredKinds.size())
    return PreferredKinds[index];
  else
    return nullptr;
}

uint64_t rp_buffer_size(const rp_buffer *buffer) {
  return buffer->size();
}

const char *rp_buffer_data(const rp_buffer *buffer) {
  return buffer->data();
}

void rp_buffer_destroy(const rp_buffer *buffer) {
  delete buffer;
}

rp_container_targets_map *rp_container_targets_map_create() {
  return new ContainerToTargetsMap();
}

void rp_container_targets_map_destroy(rp_container_targets_map *map) {
  revng_check(map != nullptr);
  delete map;
}

void rp_container_targets_map_add(rp_container_targets_map *map,
                                  const rp_container *container,
                                  const rp_target *target) {
  revng_check(map != nullptr);
  revng_check(container != nullptr);
  revng_check(target != nullptr);
  (*map)[container->second->name()].push_back(*target);
}