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

#include "llvm/Support/MemoryBuffer.h"

#include "revng/Support/Assert.h"
#include "revng/Support/Debug.h"
#include "revng/Support/PathList.h"

static Logger<> Log("find-resources");

std::string getCurrentExecutableFullPath() {
  // TODO: make this optional
  llvm::SmallString<128> SelfPath("/proc/self/exe");
  llvm::SmallString<128> FullPath;
  std::error_code Err = llvm::sys::fs::real_path(SelfPath, FullPath);
  if (Err)
    revng_abort("Could not determine executable path");

  revng_assert(not FullPath.empty());

  return FullPath.str().str();
}

std::string getCurrentRoot() {
  using llvm::sys::path::parent_path;

  auto MaybeBuffer = llvm::MemoryBuffer::getFileAsStream("/proc/self/maps");
  if (!MaybeBuffer)
    revng_abort("Unable to open /proc/self/maps");

  auto Maps = (*MaybeBuffer)->getBuffer();
  llvm::SmallVector<llvm::StringRef, 128> MapsLines;
  Maps.split(MapsLines, "\n");

  for (const auto &Line : MapsLines) {
    llvm::StringRef File = std::get<1>(Line.rsplit(" "));

    if (File.endswith("librevngSupport.so")) {
      llvm::SmallString<128> FullPath;

      std::error_code Err = llvm::sys::fs::real_path(File, FullPath);
      if (Err)
        revng_abort("Could not find real path of librevngSupport.so");

      revng_assert(!FullPath.empty());

      return parent_path(parent_path(FullPath)).str();
    }
  }
  revng_abort("Could not determine root folder");
}

static std::optional<std::string>
findFileInPaths(const std::string &FileName,
                const std::vector<std::string> &SearchPaths) {

  std::optional<std::string> FoundFileName;

  for (const auto &Path : SearchPaths) {
    revng_log(Log, "Looking in path: " << Path);
    llvm::SmallString<64> FullFileName;
    llvm::sys::path::append(FullFileName, Path, FileName);

    LoggerIndent<> Indent(Log);

    if (not llvm::sys::fs::exists(FullFileName)) {
      revng_log(Log, "File not found: " << FullFileName.str().str());
      continue;
    }

    std::error_code
      Err = llvm::sys::fs::access(FullFileName,
                                  llvm::sys::fs::AccessMode::Exist);
    if (Err) {
      revng_log(Log, "Cannot access file: " << FullFileName.str().str());
    } else {
      revng_log(Log, "Found file: " << FullFileName.str().str());
      FoundFileName = FullFileName.str().str();
      break;
    }
  }

  return FoundFileName;
}

std::optional<std::string>
PathList::findFile(const std::string &FileName) const {
  return findFileInPaths(FileName, SearchPaths);
}