#include <FEXCore/fextl/fmt.h>

#include "Common/JitSymbols.h"

#include <fcntl.h>
#include <unistd.h>

namespace FEXCore {
  JITSymbols::JITSymbols() {
  }

  JITSymbols::~JITSymbols() {
    if (fd != -1) {
      close(fd);
    }
  }

  void JITSymbols::InitFile() {
    // We can't use FILE here since we must be robust against forking processes closing our FD from under us.
#ifdef __ANDROID__
    // Android simpleperf looks in /data/local/tmp instead of /tmp
    const auto PerfMap = fextl::fmt::format("/data/local/tmp/perf-{}.map", getpid());
#else
    const auto PerfMap = fextl::fmt::format("/tmp/perf-{}.map", getpid());
#endif
    fd = open(PerfMap.c_str(), O_CREAT | O_TRUNC | O_WRONLY | O_APPEND, 0644);
  }

  void JITSymbols::Register(const void *HostAddr, uint64_t GuestAddr, uint32_t CodeSize) {
    if (fd == -1) return;

    // Linux perf format is very straightforward
    // `<HostPtr> <Size> <Name>\n`
    const auto Buffer = fextl::fmt::format("{} {:x} JIT_0x{:x}_{}\n", HostAddr, CodeSize, GuestAddr, HostAddr);
    auto Result = write(fd, Buffer.c_str(), Buffer.size());
    if (Result == -1 && errno == EBADF) {
      fd = -1;
    }
  }

  void JITSymbols::Register(const void *HostAddr, uint32_t CodeSize, std::string_view Name) {
    if (fd == -1) return;

    // Linux perf format is very straightforward
    // `<HostPtr> <Size> <Name>\n`
    const auto Buffer = fextl::fmt::format("{} {:x} {}_{}\n", HostAddr, CodeSize, Name, HostAddr);
    auto Result = write(fd, Buffer.c_str(), Buffer.size());
    if (Result == -1 && errno == EBADF) {
      fd = -1;
    }
  }

  void JITSymbols::Register(const void *HostAddr, uint32_t CodeSize, std::string_view Name, uintptr_t Offset) {
    if (fd == -1) return;

    // Linux perf format is very straightforward
    // `<HostPtr> <Size> <Name>\n`
    const auto Buffer = fextl::fmt::format("{} {:x} {}+0x{:x} ({})\n", HostAddr, CodeSize, Name, Offset, HostAddr);
    auto Result = write(fd, Buffer.c_str(), Buffer.size());
    if (Result == -1 && errno == EBADF) {
      fd = -1;
    }
  }

  void JITSymbols::RegisterNamedRegion(const void *HostAddr, uint32_t CodeSize, std::string_view Name) {
    if (fd == -1) return;

    // Linux perf format is very straightforward
    // `<HostPtr> <Size> <Name>\n`
    const auto Buffer = fextl::fmt::format("{} {:x} {}\n", HostAddr, CodeSize, Name);
    auto Result = write(fd, Buffer.c_str(), Buffer.size());
    if (Result == -1 && errno == EBADF) {
      fd = -1;
    }
  }

  void JITSymbols::RegisterJITSpace(const void *HostAddr, uint32_t CodeSize) {
    if (fd == -1) return;

    // Linux perf format is very straightforward
    // `<HostPtr> <Size> <Name>\n`
    const auto Buffer = fextl::fmt::format("{} {:x} FEXJIT\n", HostAddr, CodeSize);
    auto Result = write(fd, Buffer.c_str(), Buffer.size());
    if (Result == -1 && errno == EBADF) {
      fd = -1;
    }
  }

} // namespace FEXCore