#include "jsonrpcserver.h"

#include "tcpserver.h"

#include "hivecore.h"
#include "devicemanager.h"
#include "deviceplugin.h"
#include "deviceclass.h"
#include "device.h"
#include "rule.h"
#include "ruleengine.h"

#include <QJsonDocument>
#include <QStringList>

JsonRPCServer::JsonRPCServer(QObject *parent):
    QObject(parent),
    m_tcpServer(new TcpServer(this))
{
    connect(m_tcpServer, &TcpServer::jsonDataAvailable, this, &JsonRPCServer::processData);
    m_tcpServer->startServer();
}

void JsonRPCServer::processData(int clientId, const QByteArray &jsonData)
{
    QJsonParseError error;
    QJsonDocument jsonDoc = QJsonDocument::fromJson(jsonData, &error);

    if(error.error != QJsonParseError::NoError) {
        qDebug() << "failed to parse data" << jsonData << ":" << error.errorString();
    }
    qDebug() << "-------------------------\n" << jsonDoc.toJson();

    QVariantMap message = jsonDoc.toVariant().toMap();

    bool success;
    int commandId = message.value("id").toInt(&success);
    if (!success) {
        qWarning() << "Error parsing command. Missing \"id\":" << jsonData;
        return;
    }

    QStringList commandList = message.value("method").toString().split('.');
    if (commandList.count() != 2) {
        qWarning() << "Error parsing method.\nGot:" << message.value("method").toString() << "\nExpected: \"Namespace.method\"";
        return;
    }

    QString targetNamespace = commandList.first();
    QString method = commandList.last();
    QVariantMap params = message.value("params").toMap();

    qDebug() << "got:" << targetNamespace << method << params;
    emit commandReceived(targetNamespace, method, params);

    if (targetNamespace == "Devices") {
        handleDevicesMessage(clientId, commandId, method, params);
    } else if (targetNamespace == "Rules") {
        handleRulesMessage(clientId, commandId, method, params);
    } else if (targetNamespace == "Actions") {
        handleActionMessage(clientId, commandId, method, params);
    } else {
        qDebug() << "got unknown namespace" << targetNamespace;
    }
}

void JsonRPCServer::handleDevicesMessage(int clientId, int commandId, const QString &method, const QVariantMap &params)
{
    if (method == "GetSupportedDevices") {
        QVariantMap params;
        QVariantList supportedDeviceList;
        foreach (const DeviceClass &deviceClass, HiveCore::instance()->deviceManager()->supportedDevices()) {
            supportedDeviceList.append(packDeviceClass(deviceClass));
        }
        params.insert("deviceClasses", supportedDeviceList);
        sendResponse(clientId, commandId, params);
    } else if (method == "GetPlugins") {
        QVariantMap params;
        QVariantList plugins;
        foreach (DevicePlugin *plugin, HiveCore::instance()->deviceManager()->plugins()) {
            QVariantMap pluginMap;
            pluginMap.insert("id", plugin->pluginId());
            pluginMap.insert("name", plugin->pluginName());
            pluginMap.insert("config", plugin->configuration());
            plugins.append(pluginMap);
        }
        params.insert("plugins", plugins);
        sendResponse(clientId, commandId, params);
    } else if (method == "SetPluginConfig") {
        QUuid pluginId = params.value("pluginId").toUuid();
        QVariantMap pluginParams = params.value("params").toMap();
        HiveCore::instance()->deviceManager()->plugin(pluginId)->setConfiguration(pluginParams);
        sendResponse(clientId, commandId);
    } else if (method == "AddConfiguredDevice") {
        QUuid deviceClass = params.value("deviceClass").toUuid();
        QVariantMap deviceParams = params.value("params").toMap();
        DeviceManager::DeviceError status = HiveCore::instance()->deviceManager()->addConfiguredDevice(deviceClass, deviceParams);
        switch(status) {
        case DeviceManager::DeviceErrorNoError:
            sendResponse(clientId, commandId);
            break;
        case DeviceManager::DeviceErrorDeviceClassNotFound:
            sendErrorResponse(clientId, commandId, "Error creating device. Device class not found.");
            break;
        case DeviceManager::DeviceErrorMissingParameter:
            sendErrorResponse(clientId, commandId, "Error creating device. Missing parameter.");
            break;
        default:
            sendErrorResponse(clientId, commandId, "Unknown error.");
        }
    } else if (method == "GetConfiguredDevices") {
        QVariantMap rspParams;
        QVariantList configuredDeviceList;
        foreach (Device *device, HiveCore::instance()->deviceManager()->configuredDevices()) {
            configuredDeviceList.append(packDevice(device));
        }
        rspParams.insert("devices", configuredDeviceList);
        sendResponse(clientId, commandId, rspParams);
    } else {
        sendErrorResponse(clientId, commandId, "No such method");
    }
}

void JsonRPCServer::handleRulesMessage(int clientId, int commandId, const QString &method, const QVariantMap &params)
{
    if (method == "GetRules") {
        QVariantList rulesList;
        foreach (const Rule &rule, HiveCore::instance()->ruleEngine()->rules()) {
            QVariantMap ruleMap;
            ruleMap.insert("id", rule.id());
            ruleMap.insert("trigger", packTrigger(rule.trigger()));
            ruleMap.insert("action", packAction(rule.action()));
            rulesList.append(ruleMap);
        }
        QVariantMap rspParams;
        rspParams.insert("rules", rulesList);
        sendResponse(clientId, commandId, rspParams);
    } else if (method == "AddRule") {
        QVariantMap triggerMap = params.value("trigger").toMap();

        QUuid triggerTypeId = triggerMap.value("triggerTypeId").toUuid();
        QUuid triggerDeviceId = triggerMap.value("deviceId").toUuid();
        QVariantMap triggerParams = triggerMap.value("params").toMap();
        Trigger trigger(triggerTypeId, triggerDeviceId, triggerParams);

        QVariantMap actionMap = params.value("action").toMap();
        Action action(actionMap.value("deviceId").toString());
        action.setName(actionMap.value("name").toString());
        action.setParams(actionMap.value("params").toMap());

        switch(HiveCore::instance()->ruleEngine()->addRule(trigger, action)) {
        case RuleEngine::RuleErrorNoError:
            sendResponse(clientId, commandId);
            break;
        case RuleEngine::RuleErrorDeviceNotFound:
            sendErrorResponse(clientId, commandId, "No such device.");
            break;
        case RuleEngine::RuleErrorTriggerTypeNotFound:
            sendErrorResponse(clientId, commandId, "Device does not have such a trigger type.");
            break;
        }
    }
}

void JsonRPCServer::handleActionMessage(int clientId, int commandId, const QString &method, const QVariantMap &params)
{
    if (method == "ExecuteAction") {
        QVariantMap actionMap = params.value("action").toMap();
        QUuid deviceId = actionMap.value("deviceId").toUuid();
        QVariantMap actionParams = actionMap.value("params").toMap();

        Action action(deviceId);
        action.setParams(actionParams);

        qDebug() << "actions params in json" << action.params();


        DeviceManager::DeviceError error = HiveCore::instance()->deviceManager()->executeAction(action);
        switch (error) {
        case DeviceManager::DeviceErrorNoError:
            sendResponse(clientId, commandId);
            break;
        case DeviceManager::DeviceErrorDeviceNotFound:
            sendErrorResponse(clientId, commandId, "No such device");
            break;
        default:
            sendErrorResponse(clientId, commandId, "Unknown error.");
        }
    }
}

QVariantMap JsonRPCServer::packDeviceClass(const DeviceClass &deviceClass)
{
    QVariantMap variant;
    variant.insert("name", deviceClass.name());
    variant.insert("id", deviceClass.id());
    QVariantList triggerTypes;
    foreach (const TriggerType &triggerType, deviceClass.triggers()) {
        QVariantMap triggerMap;
        triggerMap.insert("id", triggerType.id().toString());
        triggerMap.insert("name", triggerType.name());
        triggerMap.insert("params", triggerType.parameters());

        triggerTypes.append(triggerMap);
    }
    QVariantList actionTypes;
    qDebug() << "*******************" << deviceClass.actions().count();
    foreach (const ActionType &actionType, deviceClass.actions()) {
        QVariantMap actionMap;
        actionMap.insert("id", actionType.id().toString());
        actionMap.insert("name", actionType.name());
        actionMap.insert("params", actionType.parameters());

        actionTypes.append(actionMap);
    }
    variant.insert("params", deviceClass.params());
    variant.insert("triggers", triggerTypes);
    variant.insert("actions", actionTypes);
    return variant;
}

QVariantMap JsonRPCServer::packDevice(Device *device)
{
    QVariantMap variant;
    variant.insert("id", device->id());
    variant.insert("deviceClassId", device->deviceClassId());
    variant.insert("name", device->name());
    variant.insert("params", device->params());
    return variant;
}

QVariantMap JsonRPCServer::packTrigger(const Trigger &trigger)
{
    QVariantMap variant;
    variant.insert("id", trigger.triggerTypeId());
    variant.insert("deviceId", trigger.deviceId());
    variant.insert("params", trigger.params());
    return variant;
}

QVariantMap JsonRPCServer::packAction(const Action &action)
{
    QVariantMap variant;
    variant.insert("id", action.id());
    variant.insert("deviceId", action.deviceId());
    variant.insert("name", action.name());
    variant.insert("params", action.params());
    return variant;
}

void JsonRPCServer::sendResponse(int clientId, int commandId, const QVariantMap &params)
{
    QVariantMap rsp;
    rsp.insert("id", commandId);
    rsp.insert("status", "success");
    rsp.insert("params", params);

    QJsonDocument jsonDoc = QJsonDocument::fromVariant(rsp);
    m_tcpServer->sendResponse(clientId, jsonDoc.toJson());
}

void JsonRPCServer::sendErrorResponse(int clientId, int commandId, const QString &error)
{
    QVariantMap rsp;
    rsp.insert("id", commandId);
    rsp.insert("status", "error");
    rsp.insert("error", error);

    QJsonDocument jsonDoc = QJsonDocument::fromVariant(rsp);
    m_tcpServer->sendResponse(clientId, jsonDoc.toJson());
}