#include "FinishedPduCreator.h"

FinishPduCreator::FinishPduCreator(PduConfig &conf, FinishedInfo &finishInfo)
    : FileDirectiveCreator(conf, cfdp::FileDirective::FINISH, 0), finishInfo(finishInfo) {
  updateDirectiveFieldLen();
}

size_t FinishPduCreator::getSerializedSize() const { return FinishPduCreator::getWholePduSize(); }

void FinishPduCreator::updateDirectiveFieldLen() {
  setDirectiveDataFieldLen(finishInfo.getSerializedSize());
}

ReturnValue_t FinishPduCreator::serialize(uint8_t **buffer, size_t *size, size_t maxSize,
                                          Endianness streamEndianness) const {
  ReturnValue_t result = FileDirectiveCreator::serialize(buffer, size, maxSize, streamEndianness);
  if (result != returnvalue::OK) {
    return result;
  }
  if (*size + 1 > maxSize) {
    return SerializeIF::BUFFER_TOO_SHORT;
  }
  **buffer = finishInfo.getConditionCode() << 4 | finishInfo.getDeliveryCode() << 2 |
             finishInfo.getFileStatus();
  *size += 1;
  *buffer += 1;

  if (finishInfo.hasFsResponses()) {
    FilestoreResponseTlv **fsResponsesArray = nullptr;
    size_t fsResponsesArrayLen = 0;
    finishInfo.getFilestoreResonses(&fsResponsesArray, &fsResponsesArrayLen, nullptr);
    for (size_t idx = 0; idx < fsResponsesArrayLen; idx++) {
      result = fsResponsesArray[idx]->serialize(buffer, size, maxSize, streamEndianness);
      if (result != returnvalue::OK) {
        return result;
      }
    }
  }
  EntityIdTlv *entityId = nullptr;
  if (finishInfo.getFaultLocation(&entityId) == returnvalue::OK) {
    result = entityId->serialize(buffer, size, maxSize, streamEndianness);
  }
  return result;
}