#include "TcWinUdpPollingTask.h"
#include "../../globalfunctions/arrayprinter.h"
#include "../../serviceinterface/ServiceInterfaceStream.h"

#include <winsock2.h>

TcWinUdpPollingTask::TcWinUdpPollingTask(object_id_t objectId,
		object_id_t tmtcUnixUdpBridge, size_t frameSize,
		double timeoutSeconds): SystemObject(objectId),
		tmtcBridgeId(tmtcUnixUdpBridge) {
	if(frameSize > 0) {
		this->frameSize = frameSize;
	}
	else {
		this->frameSize = DEFAULT_MAX_FRAME_SIZE;
	}

	/* Set up reception buffer with specified frame size.
	For now, it is assumed that only one frame is held in the buffer! */
	receptionBuffer.reserve(this->frameSize);
	receptionBuffer.resize(this->frameSize);

	if(timeoutSeconds == -1) {
		receptionTimeout = DEFAULT_TIMEOUT;
	}
	else {
		receptionTimeout = timevalOperations::toTimeval(timeoutSeconds);
	}
}

TcWinUdpPollingTask::~TcWinUdpPollingTask() {}

ReturnValue_t TcWinUdpPollingTask::performOperation(uint8_t opCode) {
	/* Poll for new UDP datagrams in permanent loop. */
	while(true) {
		/* Sender Address is cached here. */
		struct sockaddr_in senderAddress;
		int senderAddressSize = sizeof(senderAddress);
		int bytesReceived = recvfrom(serverUdpSocket,
				reinterpret_cast<char*>(receptionBuffer.data()), frameSize,
				receptionFlags, reinterpret_cast<sockaddr*>(&senderAddress),
				&senderAddressSize);
		if(bytesReceived == SOCKET_ERROR) {
			/* Handle error */
#if FSFW_CPP_OSTREAM_ENABLED == 1
			sif::error << "TcWinUdpPollingTask::performOperation: Reception"
					" error." << std::endl;
#endif
			handleReadError();
			continue;
		}
#if FSFW_CPP_OSTREAM_ENABLED == 1 && FSFW_UDP_RCV_WIRETAPPING_ENABLED == 1
		sif::debug << "TcWinUdpPollingTask::performOperation: " << bytesReceived <<
		        " bytes received" << std::endl;
#endif

		ReturnValue_t result = handleSuccessfullTcRead(bytesReceived);
		if(result != HasReturnvaluesIF::RETURN_FAILED) {

		}
		tmtcBridge->checkAndSetClientAddress(senderAddress);
	}
	return HasReturnvaluesIF::RETURN_OK;
}


ReturnValue_t TcWinUdpPollingTask::handleSuccessfullTcRead(size_t bytesRead) {
	store_address_t storeId;
	ReturnValue_t result = tcStore->addData(&storeId,
			receptionBuffer.data(), bytesRead);
#if FSFW_UDP_RCV_WIRETAPPING_ENABLED == 1
	arrayprinter::print(receptionBuffer.data(), bytesRead);
#endif
	if (result != HasReturnvaluesIF::RETURN_OK) {
#if FSFW_CPP_OSTREAM_ENABLED == 1
		sif::warning<< "TcSerialPollingTask::transferPusToSoftwareBus: Data "
		        "storage failed" << std::endl;
		sif::warning << "Packet size: " << bytesRead << std::endl;
#endif
		return HasReturnvaluesIF::RETURN_FAILED;
	}

	TmTcMessage message(storeId);

	result  = MessageQueueSenderIF::sendMessage(targetTcDestination, &message);
	if (result != HasReturnvaluesIF::RETURN_OK) {
#if FSFW_CPP_OSTREAM_ENABLED == 1
		sif::warning << "Serial Polling: Sending message to queue failed" << std::endl;
#endif
		tcStore->deleteData(storeId);
	}
	return result;
}

ReturnValue_t TcWinUdpPollingTask::initialize() {
	tcStore = objectManager->get<StorageManagerIF>(objects::TC_STORE);
	if (tcStore == nullptr) {
#if FSFW_CPP_OSTREAM_ENABLED == 1
		sif::error << "TcSerialPollingTask::initialize: TC Store uninitialized!"
				<< std::endl;
#endif
		return ObjectManagerIF::CHILD_INIT_FAILED;
	}

	tmtcBridge = objectManager->get<TmTcWinUdpBridge>(tmtcBridgeId);
	if(tmtcBridge == nullptr) {
#if FSFW_CPP_OSTREAM_ENABLED == 1
		sif::error << "TcSocketPollingTask::TcSocketPollingTask: Invalid"
				" TMTC bridge object!" << std::endl;
#endif
		return ObjectManagerIF::CHILD_INIT_FAILED;
	}

	serverUdpSocket = tmtcBridge->serverSocket;
#if FSFW_CPP_OSTREAM_ENABLED == 1 && FSFW_UDP_RCV_WIRETAPPING_ENABLED == 1
	sif::info << "TcWinUdpPollingTask::initialize: Server UDP socket " << serverUdpSocket <<
	        std::endl;
#endif

	return HasReturnvaluesIF::RETURN_OK;
}

ReturnValue_t TcWinUdpPollingTask::initializeAfterTaskCreation() {
	/* Initialize the destination after task creation. This ensures
	that the destination has already been set in the TMTC bridge. */
	targetTcDestination = tmtcBridge->getRequestQueue();
	return HasReturnvaluesIF::RETURN_OK;
}

void TcWinUdpPollingTask::setTimeout(double timeoutSeconds) {
	DWORD timeoutMs = timeoutSeconds * 1000.0;
	int result = setsockopt(serverUdpSocket, SOL_SOCKET, SO_RCVTIMEO,
			reinterpret_cast<const char*>(&timeoutMs), sizeof(DWORD));
	if(result == -1) {
#if FSFW_CPP_OSTREAM_ENABLED == 1
		sif::error << "TcWinUdpPollingTask::TcSocketPollingTask: Setting "
				"receive timeout failed with " << strerror(errno) << std::endl;
#endif
	}
}

void TcWinUdpPollingTask::handleReadError() {
    int error = WSAGetLastError();
    switch(error) {
    case(WSANOTINITIALISED): {
#if FSFW_CPP_OSTREAM_ENABLED == 1
        sif::info << "TcWinUdpPollingTask::handleReadError: WSANOTINITIALISED: "
                << "WSAStartup(...) call " << "necessary" << std::endl;
#endif
        break;
    }
    case(WSAEFAULT): {
#if FSFW_CPP_OSTREAM_ENABLED == 1
        sif::info << "TcWinUdpPollingTask::handleReadError: WSADEFAULT: "
                << "Bad address " << std::endl;
#endif
        break;
    }
    case(WSAEINVAL): {
#if FSFW_CPP_OSTREAM_ENABLED == 1
        sif::info << "TcWinUdpPollingTask::handleReadError: WSAEINVAL: "
                << "Invalid input parameters. " << std::endl;
#endif
        break;
    }
    default: {
#if FSFW_CPP_OSTREAM_ENABLED == 1
        sif::info << "TcWinUdpPollingTask::handleReadError: Error code: "
                << error << std::endl;
#endif
        break;
    }
    }
    // to prevent spam.
    Sleep(1000);
}