#include "TmStore.h"

#include <mission/memory/SdCardMountedIF.h>

#include <algorithm>
#include <cinttypes>
#include <filesystem>

using namespace returnvalue;

TmStore::TmStore(object_id_t objectId, std::string baseName, RolloverInterval intervalUnit,
                 uint32_t intervalCount, PacketFilter filter, SdCardMountedIF& sdcMan)
    : SystemObject(objectId), filter(filter), baseName(std::move(baseName)), sdcMan(sdcMan) {
  calcDiffSeconds(intervalUnit, intervalCount);
}

ReturnValue_t TmStore::passPacket(PusTmReader& reader) {
  bool inApidList = false;
  if (filter.apid) {
    auto& apidFilter = filter.apid.value();
    if (std::find(apidFilter.begin(), apidFilter.end(), reader.getApid()) != apidFilter.end()) {
      if (not filter.serviceSubservices and not filter.services) {
        return storePacket(reader);
      }
      inApidList = true;
    }
  }
  std::pair<uint8_t, uint8_t> serviceSubservice;
  serviceSubservice.first = reader.getService();
  serviceSubservice.second = reader.getSubService();
  if (filter.services) {
    auto& serviceFilter = filter.services.value();
    if (std::find(serviceFilter.begin(), serviceFilter.end(), serviceSubservice.first) !=
        serviceFilter.end()) {
      if (filter.apid and inApidList) {
        return storePacket(reader);
      }
    }
  }
  if (filter.serviceSubservices) {
    auto& serviceSubserviceFilter = filter.serviceSubservices.value();
    if (std::find(serviceSubserviceFilter.begin(), serviceSubserviceFilter.end(),
                  serviceSubservice) != serviceSubserviceFilter.end()) {
      if (filter.apid and inApidList) {
        return storePacket(reader);
      }
    }
  }
  return returnvalue::OK;
}

ReturnValue_t TmStore::storePacket(PusTmReader& reader) {
  using namespace std::filesystem;
  if (baseDirUninitialized) {
    updateBaseDir();
  }
  // It is assumed here that the filesystem is usable.
  if (not mostRecentFile) {
    assignMostRecentFile();
  }
  // TODO: Need to find the file of the most recent file.
  // TODO: If file exists: Determine whether file rolls over: Maximum file size reached? Interval
  //       since last timestamp exceeds rollover interval?
  // TODO: If file does not exist or rollover criteria met: Create new file with current timestamp.
  return returnvalue::OK;
}

MessageQueueId_t TmStore::getCommandQueue() { return MessageQueueIF::NO_QUEUE; }

void TmStore::calcDiffSeconds(RolloverInterval intervalUnit, uint32_t intervalCount) {
  if (intervalUnit == RolloverInterval::HOURLY) {
    rolloverDiffSeconds = 60 * intervalCount;
  } else if (intervalUnit == RolloverInterval::DAILY) {
    rolloverDiffSeconds = 60 * 24 * intervalCount;
  }
}

void TmStore::updateBaseDir() {
  using namespace std::filesystem;
  std::string currentPrefix = sdcMan.getCurrentMountPrefix();
  baseDir = path(currentPrefix) / baseName;
  if (not exists(baseDir)) {
    create_directory(baseDir);
  }
  baseDirUninitialized = false;
}

void TmStore::assignMostRecentFile() {
  using namespace std::filesystem;
  for (auto const& file : directory_iterator(baseDir)) {
    if (file.is_directory()) {
      continue;
    }
    auto pathStr = file.path().string();
    Clock::TimeOfDay_t tod;
    if (pathStr.find(baseName) == std::string::npos) {
      continue;
    }
    unsigned int underscorePos = pathStr.find_last_of('_');
    std::string stampStr = pathStr.substr(underscorePos + 1);
    int count =
        sscanf(stampStr.c_str(),
               "%4" SCNu32 "-%2" SCNu32 "-%2" SCNu32 "T%2" SCNu32 ":%2" SCNu32 ":%2" SCNu32 "Z",
               &tod.year, &tod.month, &tod.day, &tod.hour, &tod.minute, &tod.second);
    if (count != 6) {
      continue;
    }
    timeval tv{};
    Clock::convertTimeOfDayToTimeval(&tod, &tv);
    if (not mostRecentTv || tv > mostRecentTv.value()) {
      mostRecentTv = tv;
      mostRecentFile = file.path();
    }
  }
}