#include "UioMapper.h"

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

#include <filesystem>
#include <fstream>
#include <sstream>

#include "fsfw/serviceinterface.h"

const char UioMapper::UIO_PATH_PREFIX[] = "/sys/class/uio/";
const char UioMapper::MAP_SUBSTR[] = "/maps/map";
const char UioMapper::SIZE_FILE_PATH[] = "/size";

UioMapper::UioMapper(std::string uioFile, int mapNum) : uioFile(uioFile), mapNum(mapNum) {}

UioMapper::~UioMapper() {}

ReturnValue_t UioMapper::getMappedAdress(uint32_t** address, Permissions permissions) {
  ReturnValue_t result = HasReturnvaluesIF::RETURN_OK;
  int fd = open(uioFile.c_str(), O_RDWR);
  if (fd < 1) {
#if FSFW_CPP_OSTREAM_ENABLED == 1
    sif::warning << "PtmeAxiConfig::initialize: Invalid UIO device file" << std::endl;
#endif
    return HasReturnvaluesIF::RETURN_FAILED;
  }
  size_t size = 0;
  result = getMapSize(&size);
  if (result != HasReturnvaluesIF::RETURN_OK) {
    return result;
  }
  *address = static_cast<uint32_t*>(
      mmap(NULL, size, static_cast<int>(permissions), MAP_SHARED, fd, mapNum * getpagesize()));

  if (*address == MAP_FAILED) {
#if FSFW_CPP_OSTREAM_ENABLED == 1
    sif::warning << "UioMapper::getMappedAdress: Failed to map physical address of uio device "
                 << uioFile.c_str() << " and map" << static_cast<int>(mapNum) << std::endl;
#endif
    return HasReturnvaluesIF::RETURN_FAILED;
  }
  return HasReturnvaluesIF::RETURN_OK;
}

ReturnValue_t UioMapper::getMapSize(size_t* size) {
  std::stringstream namestream;
  namestream << UIO_PATH_PREFIX << uioFile.substr(5, std::string::npos) << MAP_SUBSTR << mapNum
             << SIZE_FILE_PATH;
  FILE* fp;
  fp = fopen(namestream.str().c_str(), "r");
  if (fp == nullptr) {
#if FSFW_CPP_OSTREAM_ENABLED == 1
    sif::warning << "UioMapper::getMapSize: Failed to open file " << namestream.str() << std::endl;
#endif
    return HasReturnvaluesIF::RETURN_FAILED;
  }
  char hexstring[SIZE_HEX_STRING] = "";
  int items = fscanf(fp, "%s", hexstring);
  if (items != 1) {
#if FSFW_CPP_OSTREAM_ENABLED == 1
    sif::warning << "UioMapper::getMapSize: Failed with error code " << errno
                 << " to read size "
                    "string from file "
                 << namestream.str() << std::endl;
#endif
    return HasReturnvaluesIF::RETURN_FAILED;
  }
  items = sscanf(hexstring, "%lx", size);
  if (items != 1) {
#if FSFW_CPP_OSTREAM_ENABLED == 1
    sif::warning << "UioMapper::getMapSize: Failed with error code " << errno << "to convert "
                 << "size of map" << mapNum << " to integer" << std::endl;
#endif
    return HasReturnvaluesIF::RETURN_FAILED;
  }
  fclose(fp);
  return HasReturnvaluesIF::RETURN_OK;
}