#include "MessageQueue.h"
#include "QueueMapManager.h"
#include "../../objectmanager/ObjectManager.h"
#include "../../serviceinterface/ServiceInterface.h"

MessageQueue::MessageQueue(size_t messageDepth, size_t maxMessageSize):
        maxMessageSize(maxMessageSize) {
    handle = xQueueCreate(messageDepth, maxMessageSize);
    if (handle == nullptr) {
#if FSFW_CPP_OSTREAM_ENABLED == 1
        sif::error << "MessageQueue::MessageQueue: Creation failed" << std::endl;
        sif::error << "Specified Message Depth: " << messageDepth << std::endl;
        sif::error << "Specified Maximum Message Size: " << maxMessageSize << std::endl;
#else
        sif::printError("MessageQueue::MessageQueue: Creation failed\n");
        sif::printError("Specified Message Depth: %d\n", messageDepth);
        sif::printError("Specified MAximum Message Size: %d\n", maxMessageSize);
#endif
    }
    QueueMapManager::instance()->addMessageQueue(handle, &queueId);
}

MessageQueue::~MessageQueue() {
    if (handle != nullptr) {
        vQueueDelete(handle);
    }
}

void MessageQueue::switchSystemContext(CallContext callContext) {
    this->callContext = callContext;
}

ReturnValue_t MessageQueue::sendMessage(MessageQueueId_t sendTo,
        MessageQueueMessageIF* message, bool ignoreFault) {
    return sendMessageFrom(sendTo, message, this->getId(), ignoreFault);
}

ReturnValue_t MessageQueue::sendToDefault(MessageQueueMessageIF* message) {
    return sendToDefaultFrom(message, this->getId());
}

ReturnValue_t MessageQueue::sendToDefaultFrom(MessageQueueMessageIF* message,
        MessageQueueId_t sentFrom, bool ignoreFault) {
    return sendMessageFrom(defaultDestination,message,sentFrom,ignoreFault);
}

ReturnValue_t MessageQueue::reply(MessageQueueMessageIF* message) {
    if (this->lastPartner != MessageQueueIF::NO_QUEUE) {
        return sendMessageFrom(this->lastPartner, message, this->getId());
    } else {
        return NO_REPLY_PARTNER;
    }
}

ReturnValue_t MessageQueue::sendMessageFrom(MessageQueueId_t sendTo,
        MessageQueueMessageIF* message, MessageQueueId_t sentFrom,
        bool ignoreFault) {
    return sendMessageFromMessageQueue(sendTo, message, sentFrom, ignoreFault,
            callContext);
}

QueueHandle_t MessageQueue::getNativeQueueHandle() {
    return handle;
}

ReturnValue_t MessageQueue::handleSendResult(BaseType_t result, bool ignoreFault) {
	if (result != pdPASS) {
		if (not ignoreFault) {
			InternalErrorReporterIF* internalErrorReporter = ObjectManager::instance()->
			        get<InternalErrorReporterIF>(objects::INTERNAL_ERROR_REPORTER);
			if (internalErrorReporter != nullptr) {
				internalErrorReporter->queueMessageNotSent();
			}
		}
		return MessageQueueIF::FULL;
	}
	return HasReturnvaluesIF::RETURN_OK;
}

ReturnValue_t MessageQueue::receiveMessage(MessageQueueMessageIF* message,
        MessageQueueId_t* receivedFrom) {
    ReturnValue_t status = this->receiveMessage(message);
    if(status == HasReturnvaluesIF::RETURN_OK) {
        *receivedFrom = this->lastPartner;
    }
    return status;
}

ReturnValue_t MessageQueue::receiveMessage(MessageQueueMessageIF* message) {
    BaseType_t result = xQueueReceive(handle,reinterpret_cast<void*>(
            message->getBuffer()), 0);
    if (result == pdPASS){
        this->lastPartner = message->getSender();
        return HasReturnvaluesIF::RETURN_OK;
    } else {
        return MessageQueueIF::EMPTY;
    }
}

MessageQueueId_t MessageQueue::getLastPartner() const {
    return lastPartner;
}

ReturnValue_t MessageQueue::flush(uint32_t* count) {
    //TODO FreeRTOS does not support flushing partially
    //Is always successful
    xQueueReset(handle);
    return HasReturnvaluesIF::RETURN_OK;
}

MessageQueueId_t MessageQueue::getId() const {
    return queueId;
}

void MessageQueue::setDefaultDestination(MessageQueueId_t defaultDestination) {
    defaultDestinationSet = true;
    this->defaultDestination = defaultDestination;
}

MessageQueueId_t MessageQueue::getDefaultDestination() const {
    return defaultDestination;
}

bool MessageQueue::isDefaultDestinationSet() const {
    return defaultDestinationSet;
}


// static core function to send messages.
ReturnValue_t MessageQueue::sendMessageFromMessageQueue(MessageQueueId_t sendTo,
        MessageQueueMessageIF* message, MessageQueueId_t sentFrom,
        bool ignoreFault, CallContext callContext) {
    BaseType_t result = pdFALSE;
    if(sendTo == MessageQueueIF::NO_QUEUE) {
        return MessageQueueIF::DESTINATION_INVALID;
    }

    QueueHandle_t destination = QueueMapManager::instance()->getMessageQueue(sendTo);
    if(destination == nullptr) {
        return MessageQueueIF::DESTINATION_INVALID;
    }

    message->setSender(sentFrom);
    if(callContext == CallContext::TASK) {
        result = xQueueSendToBack(destination, static_cast<const void*>(message->getBuffer()), 0);
    }
    else {
        /* If the call context is from an interrupt, request a context switch if a higher priority
        task was blocked by the interrupt. */
        BaseType_t xHigherPriorityTaskWoken = pdFALSE;
        result = xQueueSendFromISR(destination, static_cast<const void*>(message->getBuffer()),
                &xHigherPriorityTaskWoken);
        if(xHigherPriorityTaskWoken == pdTRUE) {
            TaskManagement::requestContextSwitch(callContext);
        }
    }
    return handleSendResult(result, ignoreFault);
}