#include "PDU1Handler.h"

#include <fsfw/datapool/PoolReadGuard.h>
#include <mission/devices/devicedefinitions/GomSpacePackets.h>

#include "devices/powerSwitcherList.h"

PDU1Handler::PDU1Handler(object_id_t objectId, object_id_t comIF, CookieIF *comCookie,
                         FailureIsolationBase *customFdir)
    : GomspaceDeviceHandler(objectId, comIF, comCookie, customFdir, PDU::MAX_CONFIGTABLE_ADDRESS,
                            PDU::MAX_HKTABLE_ADDRESS, PDU::HK_TABLE_REPLY_SIZE),
      coreHk(this),
      auxHk(this) {}

PDU1Handler::~PDU1Handler() {}

ReturnValue_t PDU1Handler::buildNormalDeviceCommand(DeviceCommandId_t *id) {
  *id = GomspaceCommands::REQUEST_HK_TABLE;
  return requestHkTableAction.handle();
}

void PDU1Handler::letChildHandleHkReply(DeviceCommandId_t id, const uint8_t *packet) {
  parseHkTableReply(packet);
  handleDeviceTM(&coreHk, id, true);
}

void PDU1Handler::assignChannelHookFunction(GOMSPACE::ChannelSwitchHook hook, void *args) {
  this->channelSwitchHook = hook;
  this->hookArgs = args;
}

ReturnValue_t PDU1Handler::setParamCallback(uint16_t address, uint32_t value, uint8_t valueBytes, bool afterExecution) {
  using namespace PDU1;
  GOMSPACE::Pdu pdu = GOMSPACE::Pdu::PDU1;
  if (not afterExecution) {
    return HasReturnvaluesIF::RETURN_OK;
  }
  if (channelSwitchHook != nullptr and valueBytes == 1) {
    switch (address) {
      case (CONFIG_ADDRESS_OUT_EN_TCS_BOARD_3V3): {
        channelSwitchHook(pdu, 0, value, hookArgs);
        break;
      }
      case (CONFIG_ADDRESS_OUT_EN_SYRLINKS): {
        channelSwitchHook(pdu, 1, value, hookArgs);
        break;
      }
      case (CONFIG_ADDRESS_OUT_EN_STAR_TRACKER): {
        channelSwitchHook(pdu, 2, value, hookArgs);
        break;
      }
      case (CONFIG_ADDRESS_OUT_EN_MGT): {
        channelSwitchHook(pdu, 3, value, hookArgs);
        break;
      }
      case (CONFIG_ADDRESS_OUT_EN_SUS_NOMINAL): {
        channelSwitchHook(pdu, 4, value, hookArgs);
        break;
      }
      case (CONFIG_ADDRESS_OUT_EN_SOLAR_CELL_EXP): {
        channelSwitchHook(pdu, 5, value, hookArgs);
        break;
      }
      case (CONFIG_ADDRESS_OUT_EN_PLOC): {
        channelSwitchHook(pdu, 6, value, hookArgs);
        break;
      }
      case (CONFIG_ADDRESS_OUT_EN_ACS_BOARD_SIDE_A): {
        channelSwitchHook(pdu, 7, value, hookArgs);
        break;
      }
      case (CONFIG_ADDRESS_OUT_EN_CHANNEL8): {
        channelSwitchHook(pdu, 8, value, hookArgs);
        break;
      }
    }
  }
  return HasReturnvaluesIF::RETURN_OK;
}

void PDU1Handler::parseHkTableReply(const uint8_t *packet) {
  GomspaceDeviceHandler::parsePduHkTable(coreHk, auxHk, packet);
}

ReturnValue_t PDU1Handler::initializeLocalDataPool(localpool::DataPool &localDataPoolMap,
                                                   LocalDataPoolManager &poolManager) {
  initializePduPool(localDataPoolMap, poolManager, pcdu::INIT_SWITCHES_PDU1);
  poolManager.subscribeForPeriodicPacket(coreHk.getSid(), false, 10.0, true);
  poolManager.subscribeForPeriodicPacket(auxHk.getSid(), false, 30.0, false);
  return HasReturnvaluesIF::RETURN_OK;
}

LocalPoolDataSetBase *PDU1Handler::getDataSetHandle(sid_t sid) {
  if (sid == coreHk.getSid()) {
    return &coreHk;
  } else if (sid == auxHk.getSid()) {
    return &auxHk;
  }
  return nullptr;
}

ReturnValue_t PDU1Handler::printStatus(DeviceCommandId_t cmd) {
  ReturnValue_t result = RETURN_OK;
  switch (cmd) {
    case (GomspaceCommands::PRINT_SWITCH_V_I): {
      PoolReadGuard pg(&coreHk);
      result = pg.getReadResult();
      if (result != HasReturnvaluesIF::RETURN_OK) {
        break;
      }
      printHkTableSwitchVI();
      break;
    }
    case (GomspaceCommands::PRINT_LATCHUPS): {
      PoolReadGuard pg(&auxHk);
      result = pg.getReadResult();
      if (result != HasReturnvaluesIF::RETURN_OK) {
        break;
      }
      printHkTableLatchups();
      break;
    }
    default: {
      return DeviceHandlerIF::COMMAND_NOT_SUPPORTED;
    }
  }
  if (result != HasReturnvaluesIF::RETURN_OK) {
    sif::warning << "Reading PDU1 HK table failed!" << std::endl;
  }
  return result;
}

void PDU1Handler::printHkTableSwitchVI() {
  using namespace PDU1;
  sif::info << "PDU1 Info: " << std::endl;
  sif::info << "Boot Cause: " << auxHk.bootcause << " | Boot Count: " << std::setw(4) << std::right
            << coreHk.bootcount << std::endl;
  sif::info << "Reset Cause: " << auxHk.resetcause
            << " | Battery Mode: " << static_cast<int>(coreHk.battMode.value) << std::endl;
  sif::info << "SwitchState, Currents [mA], Voltages [mV]:" << std::endl;
  auto printerHelper = [&](std::string channelStr, Channels idx) {
    sif::info << std::setw(30) << std::left << channelStr << std::dec << "| "
              << unsigned(coreHk.outputEnables[idx]) << ", " << std::setw(4) << std::right
              << coreHk.currents[idx] << ", " << std::setw(4) << coreHk.voltages[idx] << std::endl;
  };

  printerHelper("TCS Board", Channels::TCS_BOARD_3V3);
  printerHelper("Syrlinks", Channels::SYRLINKS);
  printerHelper("Star Tracker", Channels::STR);
  printerHelper("MGT", Channels::MGT);
  printerHelper("SUS Nominal", Channels::SUS_NOMINAL);
  printerHelper("SCEX", Channels::SOL_CELL_EXPERIMENT);
  printerHelper("PLOC", Channels::PLOC);
  printerHelper("ACS Board A Side", Channels::ACS_A_SIDE);
  printerHelper("Channel 8", Channels::UNUSED);
  printerHelper("Syrlinks", Channels::SYRLINKS);
}

void PDU1Handler::printHkTableLatchups() {
  using namespace PDU1;
  sif::info << "PDU1 Latchup Information" << std::endl;
  auto printerHelper = [&](std::string channelStr, Channels idx) {
    sif::info << std::setw(MAX_CHANNEL_STR_WIDTH) << std::left << "TCS Board" << std::dec << "| "
              << std::setw(4) << std::right << auxHk.latchups[idx] << std::endl;
  };
  printerHelper("TCS Board", Channels::TCS_BOARD_3V3);
  printerHelper("Syrlinks", Channels::SYRLINKS);
  printerHelper("Star Tracker", Channels::STR);
  printerHelper("MGT", Channels::MGT);
  printerHelper("SUS Nominal", Channels::SUS_NOMINAL);
  printerHelper("SCEX", Channels::SOL_CELL_EXPERIMENT);
  printerHelper("PLOC", Channels::PLOC);
  printerHelper("ACS Board A Side", Channels::ACS_A_SIDE);
  printerHelper("Channel 8", Channels::UNUSED);
  printerHelper("Syrlinks", Channels::SYRLINKS);
}