#include <QCoreApplication>
#include <QCommandLineParser>
#include <QCommandLineOption>
#include <QDateTime>
#include <unistd.h>
#include <stdio.h>

#include "core.h"
#include "zigbeecommander.h"
#include "loggingcategory.h"
#include "terminalcommander.h"

static const char *const normal = "\033[0m";
static const char *const warning = "\e[33m";
static const char *const error = "\e[31m";

static QHash<QString, bool> s_loggingFilters;

static void loggingCategoryFilter(QLoggingCategory *category)
{
    // If this is a known category
    if (s_loggingFilters.contains(category->categoryName())) {
        category->setEnabled(QtDebugMsg, s_loggingFilters.value(category->categoryName()));
        category->setEnabled(QtWarningMsg, true);
        category->setEnabled(QtCriticalMsg, true);
        category->setEnabled(QtFatalMsg, true);
    } else {
        //Disable default debug messages, print only >= warnings
        category->setEnabled(QtDebugMsg, false);
        category->setEnabled(QtWarningMsg, true);
        category->setEnabled(QtCriticalMsg, true);
        category->setEnabled(QtFatalMsg, true);
    }
}

static void consoleLogHandler(QtMsgType type, const QMessageLogContext& context, const QString& message)
{
    switch (type) {
    case QtInfoMsg:
        TerminalCommander::instance()->printToTerminal(QString("%1: %2\n").arg(context.category).arg(message.toUtf8().data()));
        break;
    case QtDebugMsg:
        TerminalCommander::instance()->printToTerminal(QString("%1: %2\n").arg(context.category).arg(message.toUtf8().data()));
        break;
    case QtWarningMsg:
        TerminalCommander::instance()->printToTerminal(QString("%1%2: %3%4\n").arg(terminalColorYellow).arg(context.category).arg(message.toUtf8().data()).arg(terminalColorNormal));
        break;
    case QtCriticalMsg:
        TerminalCommander::instance()->printToTerminal(QString("%1%2: %3%4\n").arg(terminalColorRed).arg(context.category).arg(message.toUtf8().data()).arg(terminalColorNormal));
        break;
    case QtFatalMsg:
        TerminalCommander::instance()->printToTerminal(QString("%1%2: %3%4\n").arg(terminalColorRed).arg(context.category).arg(message.toUtf8().data()).arg(terminalColorNormal));
        break;
    }
}

int main(int argc, char *argv[])
{
    qInstallMessageHandler(consoleLogHandler);

    QCoreApplication application(argc, argv);
    application.setOrganizationName("guh");
    application.setApplicationName("qt-zigbee");

    // Command line parser
    QCommandLineParser parser;
    parser.addHelpOption();
    parser.addVersionOption();
    parser.setApplicationDescription(QString("\nDaemon for the zigbee NXP uart bridge.\n\nCopyright %1 2016 Simon Stürz <simon.stuerz@guh.io>\nAll rights reserved.").arg(QChar(0xA9)));

    // Debug level
    QCommandLineOption debugLevelOption(QStringList() << "d" << "debug-level", "Set debug level [1-4].");
    debugLevelOption.setDefaultValue("1");
    debugLevelOption.setValueName("level");
    parser.addOption(debugLevelOption);

    // Channel
    QCommandLineOption channelOption(QStringList() << "c" << "channel", "Set channel for the zigbee network. Channel between [11-26] are allowed. If not specified, the quitest channel will be choosen automatically.");
    channelOption.setDefaultValue(0);
    channelOption.setValueName("channel");
    parser.addOption(channelOption);

    parser.process(application);

    // Check debug level
    bool debugLevelValueOk = false;
    int debugLevel = parser.value(debugLevelOption).toInt(&debugLevelValueOk);
    if (debugLevel < 1 || debugLevel > 4 || !debugLevelValueOk) {
        qCritical() << "Invalid debug level passed:" << parser.value(debugLevelOption) << "Reset to default debug level 1.";
        debugLevel = 1;
    }

    s_loggingFilters.insert("Zigbee", true);
    s_loggingFilters.insert("ZigbeeController", (debugLevel > 1));
    s_loggingFilters.insert("ZigbeeInterface", (debugLevel > 2));
    s_loggingFilters.insert("ZigbeeInterfaceTraffic", (debugLevel > 3));

    QLoggingCategory::installFilter(loggingCategoryFilter);

    // Check channel
    bool channelValueOk = false;
    int channel = parser.value(channelOption).toInt(&channelValueOk);
    if (channel != 0) {
        if (channel < 11 || channel > 26 || !channelValueOk) {
            qCritical() << "Invalid channel value passed:" << parser.value(channelOption) << "Selecting automatically quitest channel.";
            channel = 0;
        }
    }

    Core core(channel);

    return application.exec();
}