#include <QLibrary>

#include <QtQml>

#include <QDebug>

#include <QPainter>

#include <QUuid>

#include "hwengine.h"

#include "previewimageprovider.h"

#include "themeiconprovider.h"

extern "C" {

    RunEngine_t *flibRunEngine;

    registerUIMessagesCallback_t *flibRegisterUIMessagesCallback;

    setSeed_t *flibSetSeed;

    getSeed_t *flibGetSeed;

    setTheme_t *flibSetTheme;

    setScript_t *flibSetScript;

    setScheme_t *flibSetScheme;

    setAmmo_t *flibSetAmmo;

    getPreview_t *flibGetPreview;

    runQuickGame_t *flibRunQuickGame;

    runLocalGame_t *flibRunLocalGame;

    flibInit_t *flibInit;

    flibFree_t *flibFree;

    resetGameConfig_t * flibResetGameConfig;

    getThemesList_t *flibGetThemesList;

    freeThemesList_t *flibFreeThemesList;

    getThemeIcon_t *flibGetThemeIcon;

    getScriptsList_t *flibGetScriptsList;

    getSchemesList_t *flibGetSchemesList;

    getAmmosList_t *flibGetAmmosList;

    getTeamsList_t *flibGetTeamsList;

    tryAddTeam_t * flibTryAddTeam;

    tryRemoveTeam_t * flibTryRemoveTeam;

    changeTeamColor_t * flibChangeTeamColor;

    connectOfficialServer_t * flibConnectOfficialServer;

    passNetData_t * flibPassNetData;

    passToNet_t * flibPassToNet;

    passFlibEvent_t * flibPassFlibEvent;

    sendChatLine_t * flibSendChatLine;

    joinRoom_t * flibJoinRoom;

    partRoom_t * flibPartRoom;

}

Q_DECLARE_METATYPE(MessageType)

HWEngine::HWEngine(QQmlEngine *engine, QObject *parent) :

    QObject(parent),

    m_engine(engine)

{

    qRegisterMetaType<MessageType>("MessageType");

#ifdef Q_OS_WIN    

    QLibrary hwlib("./libhwengine.dll");

#else

    QLibrary hwlib("./libhwengine.so");

#endif

    if(!hwlib.load())

        qWarning() << "Engine library not found" << hwlib.errorString();

    flibRunEngine = (RunEngine_t*) hwlib.resolve("RunEngine");

    flibRegisterUIMessagesCallback = (registerUIMessagesCallback_t*) hwlib.resolve("registerUIMessagesCallback");

    flibGetSeed = (getSeed_t*) hwlib.resolve("getSeed");

    flibGetPreview = (getPreview_t*) hwlib.resolve("getPreview");

    flibRunQuickGame = (runQuickGame_t*) hwlib.resolve("runQuickGame");

    flibRunLocalGame = (runLocalGame_t*) hwlib.resolve("runLocalGame");

    flibInit = (flibInit_t*) hwlib.resolve("flibInit");

    flibFree = (flibFree_t*) hwlib.resolve("flibFree");

    flibSetSeed = (setSeed_t*) hwlib.resolve("setSeed");

    flibSetTheme = (setTheme_t*) hwlib.resolve("setTheme");

    flibSetScript = (setScript_t*) hwlib.resolve("setScript");

    flibSetScheme = (setScheme_t*) hwlib.resolve("setScheme");

    flibSetAmmo = (setAmmo_t*) hwlib.resolve("setAmmo");

    flibGetThemesList = (getThemesList_t*) hwlib.resolve("getThemesList");

    flibFreeThemesList = (freeThemesList_t*) hwlib.resolve("freeThemesList");

    flibGetThemeIcon = (getThemeIcon_t*) hwlib.resolve("getThemeIcon");

    flibGetScriptsList = (getScriptsList_t*) hwlib.resolve("getScriptsList");

    flibGetSchemesList = (getSchemesList_t*) hwlib.resolve("getSchemesList");

    flibGetAmmosList = (getAmmosList_t*) hwlib.resolve("getAmmosList");

    flibResetGameConfig = (resetGameConfig_t*) hwlib.resolve("resetGameConfig");

    flibGetTeamsList = (getTeamsList_t*) hwlib.resolve("getTeamsList");

    flibTryAddTeam = (tryAddTeam_t*) hwlib.resolve("tryAddTeam");

    flibTryRemoveTeam = (tryRemoveTeam_t*) hwlib.resolve("tryRemoveTeam");

    flibChangeTeamColor = (changeTeamColor_t*) hwlib.resolve("changeTeamColor");

    flibConnectOfficialServer = (connectOfficialServer_t*) hwlib.resolve("connectOfficialServer");

    flibPassNetData = (passNetData_t*) hwlib.resolve("passNetData");

    flibPassToNet = (passToNet_t*) hwlib.resolve("passToNet");

    flibPassFlibEvent = (passFlibEvent_t*) hwlib.resolve("passFlibEvent");

    flibSendChatLine = (sendChatLine_t*) hwlib.resolve("sendChatLine");

    flibJoinRoom = (joinRoom_t*) hwlib.resolve("joinRoom");

    flibPartRoom = (partRoom_t*) hwlib.resolve("partRoom");

    flibRegisterUIMessagesCallback(this, &guiMessagesCallback);

    ThemeIconProvider * themeIcon = (ThemeIconProvider *)m_engine->imageProvider(QLatin1String("theme"));

    themeIcon->setFileContentsFunction(flibGetThemeIcon);

    fillModels();

}

HWEngine::~HWEngine()

{

    flibFree();

}

void HWEngine::getPreview()

{

    flibSetSeed(QUuid::createUuid().toString().toLatin1());

    flibGetPreview();

}

void HWEngine::runQuickGame()

{

    flibSetSeed(QUuid::createUuid().toString().toLatin1());

    flibRunQuickGame();

}

void HWEngine::runLocalGame()

{

    flibRunLocalGame();

}

static QObject *hwengine_singletontype_provider(QQmlEngine *engine, QJSEngine *scriptEngine)

{

    Q_UNUSED(scriptEngine)

    HWEngine *hwengine = new HWEngine(engine);

    return hwengine;

}

void HWEngine::exposeToQML()

{

    qDebug("HWEngine::exposeToQML");

    qmlRegisterSingletonType<HWEngine>("Hedgewars.Engine", 1, 0, "HWEngine", hwengine_singletontype_provider);

}

void HWEngine::guiMessagesCallback(void *context, MessageType mt, const char * msg, uint32_t len)

{

    HWEngine * obj = (HWEngine *)context;

    QByteArray b = QByteArray(msg, len);

    //qDebug() << "FLIPC in" << mt << " size = " << b.size();

    QMetaObject::invokeMethod(obj, "engineMessageHandler", Qt::QueuedConnection, Q_ARG(MessageType, mt), Q_ARG(QByteArray, b));

}

void HWEngine::engineMessageHandler(MessageType mt, const QByteArray &msg)

{

    switch(mt)

    {

    case MSG_RENDERINGPREVIEW: {

        emit previewIsRendering();

        break;

    }

    case MSG_PREVIEW: {

        PreviewImageProvider * preview = (PreviewImageProvider *)m_engine->imageProvider(QLatin1String("preview"));

        preview->setPixmap(msg);

        emit previewImageChanged();

        break;

    }

    case MSG_PREVIEWHOGCOUNT: {

        emit previewHogCountChanged((quint8)msg.data()[0]);

        break;

    }

    case MSG_ADDPLAYINGTEAM: {

        QStringList l = QString::fromUtf8(msg).split('\n');

        emit playingTeamAdded(l[1], l[0].toInt(), true);

        break;

    }

    case MSG_REMOVEPLAYINGTEAM: {

        emit playingTeamRemoved(QString::fromUtf8(msg));

        break;

    }

    case MSG_ADDTEAM: {

        emit localTeamAdded(QString::fromUtf8(msg), 0);

        break;

    }

    case MSG_REMOVETEAM: {

        emit localTeamRemoved(QString::fromUtf8(msg));

        break;

    }

    case MSG_TEAMCOLOR: {

        QStringList l = QString::fromUtf8(msg).split('\n');

        emit teamColorChanged(l[0], QColor::fromRgba(l[1].toInt()).name());

        break;

    }

    case MSG_HEDGEHOGSNUMBER: {

        QStringList l = QString::fromUtf8(msg).split('\n');

        emit hedgehogsNumberChanged(l[0], l[1].toInt());

        break;

    }

    case MSG_NETDATA: {

        flibPassNetData(msg.constData());

        break;

    }

    case MSG_TONET: {

        flibPassToNet(msg.constData(), msg.size());

        break;

    }

    case MSG_FLIBEVENT: {

        flibPassFlibEvent(msg.constData());

        break;

    }

    case MSG_CONNECTED: {

        emit netConnected();

        break;

    }

    case MSG_DISCONNECTED: {

        emit netDisconnected(QString::fromUtf8(msg));

        break;

    }

    case MSG_ADDLOBBYCLIENT: {

        emit lobbyClientAdded(QString::fromUtf8(msg));

        break;

    }

    case MSG_REMOVELOBBYCLIENT: {

        QStringList l = QString::fromUtf8(msg).split('\n');

        if(l.size() < 2)

            l.append("");

        emit lobbyClientRemoved(l[0], l[1]);

        break;

    }

    case MSG_LOBBYCHATLINE: {

        QStringList l = QString::fromUtf8(msg).split('\n');

        emit lobbyChatLine(l[0], l[1]);

        break;

    }

    case MSG_ADDROOMCLIENT: {

        emit roomClientAdded(QString::fromUtf8(msg));

        break;

    }

    case MSG_REMOVEROOMCLIENT: {

        QStringList l = QString::fromUtf8(msg).split('\n');

        if(l.size() < 2)

            l.append("");

        emit roomClientRemoved(l[0], l[1]);

        break;

    }

    case MSG_ROOMCHATLINE: {

        QStringList l = QString::fromUtf8(msg).split('\n');

        emit roomChatLine(l[0], l[1]);

        break;

    }

    case MSG_ADDROOM: {

        QStringList l = QString::fromUtf8(msg).split('\n');

        emit roomAdded(0, l[1], l[2].toInt(), l[3].toInt(), l[4], l[5], l[6], l[7], l[8]);

        break;

    }

    case MSG_UPDATEROOM: {

        QStringList l = QString::fromUtf8(msg).split('\n');

        emit roomUpdated(l[0], 0, l[2], l[3].toInt(), l[4].toInt(), l[5], l[6], l[7], l[8], l[9]);

        break;

    }

    case MSG_REMOVEROOM: {

        emit roomRemoved(QString::fromUtf8(msg));

        break;

    }

    case MSG_ERROR: {

        emit errorMessage(QString::fromUtf8(msg));

        break;

    }

    case MSG_WARNING: {

        emit warningMessage(QString::fromUtf8(msg));

        break;

    }

    case MSG_MOVETOLOBBY: {

        emit movedToLobby();

        break;

    }

    case MSG_MOVETOROOM: {

        emit movedToRoom();

        break;

    }

    case MSG_NICKNAME: {

        m_myNickname = QString::fromUtf8(msg);

        break;

    }

    case MSG_SEED: {

        emit seedChanged(QString::fromUtf8(msg));

        break;

    }

    case MSG_THEME: {

        emit themeChanged(QString::fromUtf8(msg));

        break;

    }

    case MSG_SCRIPT: {

        emit scriptChanged(QString::fromUtf8(msg));

        break;

    }

    case MSG_FEATURESIZE: {

        emit featureSizeChanged(msg.toInt());

        break;

    }

    case MSG_MAPGEN: {

        emit mapGenChanged(msg.toInt());

        break;

    }

    case MSG_MAP: {

        emit mapChanged(QString::fromUtf8(msg));

        break;

    }

    case MSG_MAZESIZE: {

        emit mazeSizeChanged(msg.toInt());

        break;

    }

    case MSG_TEMPLATE: {

        emit templateChanged(msg.toInt());

        break;

    }

    case MSG_AMMO: {

        emit ammoChanged(QString::fromUtf8(msg));

        break;

    }

    case MSG_SCHEME: {

        emit schemeChanged(QString::fromUtf8(msg));

        break;

    }

    }

}

QString HWEngine::currentSeed()

{

    return QString::fromLatin1(flibGetSeed());

}

void HWEngine::fillModels()

{

    QStringList resultModel;

    char ** themes = flibGetThemesList();

    for (char **i = themes; *i != NULL; i++)

        resultModel << QString::fromUtf8(*i);

    flibFreeThemesList(themes);

    m_engine->rootContext()->setContextProperty("themesModel", QVariant::fromValue(resultModel));

    // scripts model

    resultModel.clear();

    for (char **i = flibGetScriptsList(); *i != NULL; i++)

        resultModel << QString::fromUtf8(*i);

    m_engine->rootContext()->setContextProperty("scriptsModel", QVariant::fromValue(resultModel));

    // schemes model

    resultModel.clear();

    for (char **i = flibGetSchemesList(); *i != NULL; i++)

        resultModel << QString::fromUtf8(*i);

    m_engine->rootContext()->setContextProperty("schemesModel", QVariant::fromValue(resultModel));

    // ammos model

    resultModel.clear();

    for (char **i = flibGetAmmosList(); *i != NULL; i++)

        resultModel << QString::fromUtf8(*i);

    m_engine->rootContext()->setContextProperty("ammosModel", QVariant::fromValue(resultModel));

}

void HWEngine::getTeamsList()

{

    char ** teams = flibGetTeamsList();

    for (char **i = teams; *i != NULL; i++) {

        QString team = QString::fromUtf8(*i);

        emit localTeamAdded(team, 0);

    }

}

void HWEngine::tryAddTeam(const QString &teamName)

{

    flibTryAddTeam(teamName.toUtf8().constData());

}

void HWEngine::tryRemoveTeam(const QString &teamName)

{

    flibTryRemoveTeam(teamName.toUtf8().constData());

}

void HWEngine::resetGameConfig()

{

    flibResetGameConfig();

}

void HWEngine::changeTeamColor(const QString &teamName, int dir)

{

    flibChangeTeamColor(teamName.toUtf8().constData(), dir);

}

void HWEngine::connect(const QString &host, quint16 port)

{

    flibConnectOfficialServer();

}

void HWEngine::sendChatMessage(const QString &msg)

{

    flibSendChatLine(msg.toUtf8().constData());

}

void HWEngine::joinRoom(const QString &roomName)

{

    flibJoinRoom(roomName.toUtf8().constData());

}

void HWEngine::partRoom(const QString &message)

{

    flibPartRoom(message.toUtf8().constData());

}

QString HWEngine::myNickname()

{

    return m_myNickname;

}

void HWEngine::setTheme(const QString &theme)

{

    flibSetTheme(theme.toUtf8().constData());

}

void HWEngine::setScript(const QString &script)

{

    flibSetScript(script.toUtf8().constData());

}

void HWEngine::setScheme(const QString &scheme)

{

    flibSetScheme(scheme.toUtf8().constData());

}

void HWEngine::setAmmo(const QString &ammo)

{

    flibSetAmmo(ammo.toUtf8().constData());

}