#include <fsfw/tasks/TaskFactory.h>
#include <fsfw/timemanager/Stopwatch.h>
#include <mission/tmtc/PersistentSingleTmStoreTask.h>
#include <unistd.h>

PersistentSingleTmStoreTask::PersistentSingleTmStoreTask(
    object_id_t objectId, StorageManagerIF& ipcStore, PersistentTmStoreWithTmQueue& tmStore,
    VirtualChannel& channel, Event eventIfDumpDone, SdCardMountedIF& sdcMan)
    : TmStoreTaskBase(objectId, ipcStore, channel, sdcMan),
      storeWithQueue(tmStore),
      dumpContext(eventIfDumpDone) {}

ReturnValue_t PersistentSingleTmStoreTask::performOperation(uint8_t opCode) {
  ReturnValue_t result = returnvalue::OK;
  while (true) {
    // Delay done by the check
    if (not cyclicStoreCheck()) {
      continue;
    }
    bool busy = handleOneStore(storeWithQueue, dumpContext);
    if (not busy) {
      result = TaskFactory::delayTask(100);
    } else if (dumpContext.vcBusyDuringDump) {
      result = TaskFactory::delayTask(10);
    } else if (dumpContext.packetWasDumped and
               dumpContext.dumpedBytes - dumpContext.bytesDumpedAtLastDelay >= 2048) {
      dumpContext.bytesDumpedAtLastDelay = dumpContext.dumpedBytes;
      result = TaskFactory::delayTask(10);
    } else {
      // TODO: Lower this as much as possible...
      result = TaskFactory::delayTask(10);
    }
    if(result != returnvalue::OK) {
      sif::warning << "TmStoreTask: Delay failed" << std::endl;
    }
  }
}

bool PersistentSingleTmStoreTask::initStoresIfPossible() {
  if (sdcMan.isSdCardUsable(std::nullopt)) {
    storeWithQueue.initializeTmStore();
    return true;
  }
  return false;
}