#include "HousekeepingMessage.h"

#include "../objectmanager/ObjectManager.h"
#include <cstring>

HousekeepingMessage::~HousekeepingMessage() {}

void HousekeepingMessage::setHkReportReply(CommandMessage* message, sid_t sid,
		store_address_t storeId) {
	message->setCommand(HK_REPORT);
	message->setMessageSize(HK_MESSAGE_SIZE);
	setSid(message, sid);
	message->setParameter3(storeId.raw);
}

void HousekeepingMessage::setHkDiagnosticsReply(CommandMessage* message,
		sid_t sid, store_address_t storeId) {
	message->setCommand(DIAGNOSTICS_REPORT);
	message->setMessageSize(HK_MESSAGE_SIZE);
	setSid(message, sid);
	message->setParameter3(storeId.raw);
}

sid_t HousekeepingMessage::getHkDataReply(const CommandMessage *message,
		store_address_t *storeIdToSet) {
	if(storeIdToSet != nullptr) {
	    *storeIdToSet = message->getParameter3();
	}
	return getSid(message);
}

void HousekeepingMessage::setToggleReportingCommand(CommandMessage *message,
		sid_t sid, bool enableReporting, bool isDiagnostics) {
	if(isDiagnostics) {
		if(enableReporting) {
			message->setCommand(ENABLE_PERIODIC_DIAGNOSTICS_GENERATION);
		}
		else {
			message->setCommand(DISABLE_PERIODIC_DIAGNOSTICS_GENERATION);
		}
	}
	else {
		if(enableReporting) {
			message->setCommand(ENABLE_PERIODIC_HK_REPORT_GENERATION);
		}
		else {
			message->setCommand(DISABLE_PERIODIC_HK_REPORT_GENERATION);
		}
	}

	setSid(message, sid);
}

void HousekeepingMessage::setStructureReportingCommand(CommandMessage *command,
		sid_t sid, bool isDiagnostics) {
	if(isDiagnostics) {
		command->setCommand(REPORT_DIAGNOSTICS_REPORT_STRUCTURES);
	}
	else {
		command->setCommand(REPORT_HK_REPORT_STRUCTURES);
	}

	setSid(command, sid);
}

void HousekeepingMessage::setOneShotReportCommand(CommandMessage *command,
		sid_t sid, bool isDiagnostics) {
	if(isDiagnostics) {
		command->setCommand(GENERATE_ONE_DIAGNOSTICS_REPORT);
	}
	else {
		command->setCommand(GENERATE_ONE_PARAMETER_REPORT);
	}

	setSid(command, sid);
}

void HousekeepingMessage::setCollectionIntervalModificationCommand(
		CommandMessage *command, sid_t sid, float collectionInterval,
		bool isDiagnostics) {
	if(isDiagnostics) {
		command->setCommand(MODIFY_DIAGNOSTICS_REPORT_COLLECTION_INTERVAL);
	}
	else {
		command->setCommand(MODIFY_PARAMETER_REPORT_COLLECTION_INTERVAL);
	}

	/* Raw storage of the float in the message. Do not use setParameter3, does
	implicit conversion to integer type! */
	std::memcpy(command->getData() + 2 * sizeof(uint32_t), &collectionInterval,
	        sizeof(collectionInterval));

	setSid(command, sid);
}

sid_t HousekeepingMessage::getCollectionIntervalModificationCommand(
		const CommandMessage* command, float* newCollectionInterval) {

	if(newCollectionInterval != nullptr) {
	    std::memcpy(newCollectionInterval, command->getData() + 2 * sizeof(uint32_t),
	            sizeof(*newCollectionInterval));
	}

	return getSid(command);
}

void HousekeepingMessage::setHkRequestSuccessReply(CommandMessage *reply,
		sid_t sid) {
	setSid(reply, sid);
	reply->setCommand(HK_REQUEST_SUCCESS);
}

void HousekeepingMessage::setHkRequestFailureReply(CommandMessage *reply,
		sid_t sid, ReturnValue_t error) {
	setSid(reply, sid);
	reply->setCommand(HK_REQUEST_FAILURE);
	reply->setParameter3(error);
}

sid_t HousekeepingMessage::getHkRequestFailureReply(const CommandMessage *reply,
		ReturnValue_t *error) {
	if(error != nullptr) {
		*error = reply->getParameter3();
	}
	return getSid(reply);
}

sid_t HousekeepingMessage::getSid(const CommandMessage* message) {
	sid_t sid;
	std::memcpy(&sid.raw, message->getData(), sizeof(sid.raw));
	return sid;
}

gp_id_t HousekeepingMessage::getGpid(const CommandMessage* message) {
    gp_id_t globalPoolId;
    std::memcpy(&globalPoolId.raw, message->getData(), sizeof(globalPoolId.raw));
    return globalPoolId;
}

void HousekeepingMessage::setHkStuctureReportReply(CommandMessage *reply,
        sid_t sid, store_address_t storeId) {
    reply->setCommand(HK_DEFINITIONS_REPORT);
    setSid(reply, sid);
    reply->setParameter3(storeId.raw);
}

void HousekeepingMessage::setDiagnosticsStuctureReportReply(
        CommandMessage *reply, sid_t sid, store_address_t storeId) {
    reply->setCommand(DIAGNOSTICS_DEFINITION_REPORT);
    setSid(reply, sid);
    reply->setParameter3(storeId.raw);
}

void HousekeepingMessage::clear(CommandMessage* message) {
    switch(message->getCommand()) {
    case(HK_REPORT):
    case(DIAGNOSTICS_REPORT):
    case(HK_DEFINITIONS_REPORT):
    case(DIAGNOSTICS_DEFINITION_REPORT):
    case(UPDATE_SNAPSHOT_SET):
    case(UPDATE_SNAPSHOT_VARIABLE): {
        store_address_t storeId;
        getHkDataReply(message, &storeId);
        StorageManagerIF *ipcStore = ObjectManager::instance()->get<StorageManagerIF>(
                objects::IPC_STORE);
        if (ipcStore != nullptr) {
            ipcStore->deleteData(storeId);
        }
    }
    }
    message->setCommand(CommandMessage::CMD_NONE);
}

void HousekeepingMessage::setUpdateNotificationSetCommand(
        CommandMessage *command, sid_t sid) {
    command->setCommand(UPDATE_NOTIFICATION_SET);
    setSid(command, sid);
}

void HousekeepingMessage::setUpdateNotificationVariableCommand(
        CommandMessage *command, gp_id_t globalPoolId) {
    command->setCommand(UPDATE_NOTIFICATION_VARIABLE);
    setGpid(command, globalPoolId);
}

void HousekeepingMessage::setUpdateSnapshotSetCommand(CommandMessage *command,
        sid_t sid, store_address_t storeId) {
    command->setCommand(UPDATE_SNAPSHOT_SET);
    setSid(command, sid);
    command->setParameter3(storeId.raw);
}

void HousekeepingMessage::setUpdateSnapshotVariableCommand(
        CommandMessage *command, gp_id_t globalPoolId, store_address_t storeId) {
    command->setCommand(UPDATE_SNAPSHOT_VARIABLE);
    setGpid(command, globalPoolId);
    command->setParameter3(storeId.raw);
}

sid_t HousekeepingMessage::getUpdateNotificationSetCommand(
        const CommandMessage *command) {
    return getSid(command);
}

gp_id_t HousekeepingMessage::getUpdateNotificationVariableCommand(
        const CommandMessage *command) {
    return getGpid(command);
}

sid_t HousekeepingMessage::getUpdateSnapshotSetCommand(
        const CommandMessage *command, store_address_t *storeId) {
    if(storeId != nullptr) {
        *storeId = command->getParameter3();
    }
    return getSid(command);
}

gp_id_t HousekeepingMessage::getUpdateSnapshotVariableCommand(
        const CommandMessage *command, store_address_t *storeId) {
    if(storeId != nullptr) {
        *storeId = command->getParameter3();
    }
    return getGpid(command);
}

void HousekeepingMessage::setSid(CommandMessage *message, sid_t sid) {
    std::memcpy(message->getData(), &sid.raw, sizeof(sid.raw));
}

void HousekeepingMessage::setGpid(CommandMessage *message, gp_id_t globalPoolId) {
    std::memcpy(message->getData(), &globalPoolId.raw, sizeof(globalPoolId.raw));
}