// C++ includes
#include <iostream> // for cerr, cout
using std::cerr;
using std::cout;
using std::endl;

// Qt headers
#include <QHostAddress>
#include <QUdpSocket>
#include <QString>
#include <QFile>

// libmyth* includes
#include "compat.h"
#include "exitcodes.h"
#include "mythlogging.h"

// Local includes
#include "messageutils.h"

const QString kMessage =
"<mythmessage version=\"1\">\n"
"  <text>%message_text%</text>\n"
"  <timeout>%timeout%</timeout>\n"
"</mythmessage>";

const QString kNotification =
"<mythnotification version=\"1\">\n"
"  <text>%message_text%</text>\n"
"  <origin>%origin%</origin>\n"
"  <description>%description%</description>\n"
"  <error>%error%</error>\n"
"  <timeout>%timeout%</timeout>\n"
"  <image>%image%</image>\n"
"  <extra>%extra%</extra>\n"
"  <progress_text>%progress_text%</progress_text>\n"
"  <progress>%progress%</progress>\n"
"  <fullscreen>%fullscreen%</fullscreen>\n"
"  <visibility>%visibility%</visibility>\n"
"  <type>%type%</type>\n"
"</mythnotification>";

static int PrintMTemplate(const MythUtilCommandLineParser &/*cmdline*/)
{
    cerr << kMessage.toLocal8Bit().constData() << endl;
    return GENERIC_EXIT_OK;
}

static int PrintNTemplate(const MythUtilCommandLineParser &/*cmdline*/)
{
    cerr << kNotification.toLocal8Bit().constData() << endl;
    return GENERIC_EXIT_OK;
}

static int SendMessage(const MythUtilCommandLineParser &cmdline)
{
    QHostAddress address = QHostAddress::Broadcast;
    unsigned short port = 6948;
    QString name = cmdline.GetPassthrough();
    bool notification = cmdline.toBool("notification");
    QString text /* = "message" */ ;
    QString timeout = "0";
    QString image = "";
    QString origin = "MythUtils";
    QString description = "";
    QString extra = "";
    QString progress_text = "";
    QString progress = "-1";
    QString fullscreen = "false";
    QString error = "false";
    QString visibility = "0";
    QString type = "normal";

    QString message = notification ? kNotification : kMessage;

    if (cmdline.toBool("udpport"))
        port = (unsigned short)cmdline.toUInt("udpport");
    if (cmdline.toBool("bcastaddr"))
        address.setAddress(cmdline.toString("bcastaddr"));
    if (cmdline.toBool("message_text"))
        text = cmdline.toString("message_text");
    message.replace("%message_text%", text);
    if (cmdline.toBool("timeout"))
        timeout = cmdline.toString("timeout");
    message.replace("%timeout%", timeout);
    if (notification)
    {
        if (cmdline.toBool("image"))
            image = cmdline.toString("image");
        message.replace("%image%", image);
        if (cmdline.toBool("origin"))
            origin = cmdline.toString("origin");
        message.replace("%origin%", origin);
        if (cmdline.toBool("description"))
            description = cmdline.toString("description");
        message.replace("%description%", description);
        if (cmdline.toBool("extra"))
            extra = cmdline.toString("extra");
        message.replace("%extra%", extra);
        if (cmdline.toBool("progress_text"))
            progress_text = cmdline.toString("progress_text");
        message.replace("%progress_text%", progress_text);
        if (cmdline.toBool("progress"))
            progress = cmdline.toString("progress");
        message.replace("%progress%", progress);
        if (cmdline.toBool("fullscreen"))
            fullscreen = cmdline.toString("fullscreen");
        message.replace("%fullscreen%", fullscreen);
        if (cmdline.toBool("error"))
            error = cmdline.toString("error");
        message.replace("%error%", error);
        if (cmdline.toBool("visibility"))
            visibility = cmdline.toString("visibility");
        message.replace("%visibility%", visibility);
        if (cmdline.toBool("type"))
            type = cmdline.toString("type");
        message.replace("%type%", type);
    }

    // extra optional argument
    // in effeect not use as the above code provides default for all possible
    // cases
    QMap<QString,QString>::const_iterator i;
    QMap<QString,QString> extras = cmdline.GetExtra();
    for (i = extras.begin(); i != extras.end(); ++i)
    {
        QString kv_name = i.key();
        QString kv_value = i.value();

        kv_name.replace("--", "");
        cerr << "name: " << kv_name.toLocal8Bit().constData()
             << " -- value: " << kv_value.toLocal8Bit().constData() << endl;

        kv_name.append("%");
        kv_name.prepend("%");
        message.replace(kv_name, kv_value);
    }

    cout << "output:\n" << message.toLocal8Bit().constData() << endl;

    QUdpSocket *sock = new QUdpSocket();
    QByteArray utf8 = message.toUtf8();

    int result = GENERIC_EXIT_OK;
    if (sock->writeDatagram(utf8, address, port) < 0)
    {
        cout << "Failed to send UDP/XML packet" << endl;
        result = GENERIC_EXIT_NOT_OK;
    }
    else
    {
        cout << "Sent UDP/XML packet to IP "
             << address.toString().toLocal8Bit().constData()
             << " and port: " << port << endl;
    }

    sock->deleteLater();

    return result;
}

void registerMessageUtils(UtilMap &utilMap)
{
    utilMap["message"]                 = &SendMessage;
    utilMap["printmtemplate"]          = &PrintMTemplate;
    utilMap["notification"]            = &SendMessage;
    utilMap["printntemplate"]          = &PrintNTemplate;
}

/* vim: set expandtab tabstop=4 shiftwidth=4: */
