/***************************************************************************
 *   This file is part of pytextmonitor                                    *
 *                                                                         *
 *   pytextmonitor is free software: you can redistribute it and/or        *
 *   modify it under the terms of the GNU General Public License as        *
 *   published by the Free Software Foundation, either version 3 of the    *
 *   License, or (at your option) any later version.                       *
 *                                                                         *
 *   pytextmonitor is distributed in the hope that it will be useful,      *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with pytextmonitor. If not, see http://www.gnu.org/licenses/    *
 ***************************************************************************/

#include "awesome-widget.h"

#include <KNotification>
#include <Plasma/ToolTipManager>
#include <QGraphicsLinearLayout>
#include <QGraphicsView>

#include "customlabel.h"
#include <pdebug/pdebug.h>


void AwesomeWidget::initValues()
{
    if (debug) qDebug() << PDEBUG;
}


void AwesomeWidget::reinit()
{
    if (debug) qDebug() << PDEBUG;

    mainLayout = new QGraphicsLinearLayout();
    mainLayout->setContentsMargins(1, 1, 1, 1);
    if (configuration[QString("layout")].toInt() == 0)
        mainLayout->setOrientation(Qt::Horizontal);
    else
        mainLayout->setOrientation(Qt::Vertical);
    setLayout(mainLayout);

    if (configuration[QString("background")].toInt() == 0)
        setBackgroundHints(NoBackground);
    else
        setBackgroundHints(DefaultBackground);
    if (configuration[QString("leftStretch")].toInt() == 2)
        mainLayout->addStretch(1);
    if (configuration[QString("popup")].toInt() == 0)
        textLabel->setPopupEnabled(false);
    else
        textLabel->setPopupEnabled(true);
    updateText(true);
    mainLayout->addItem(textLabel);
    if (configuration[QString("rightStretch")].toInt() == 2)
        mainLayout->addStretch(1);
    resize(10, 10);

    keys = getKeys();
    foundKeys = findKeys();
    initValues();
    updateNetworkDevice();
    connectToEngine();
}


void AwesomeWidget::sendNotification(const QString eventId, const QString message)
{
    if (debug) qDebug() << PDEBUG;
    if (debug) qDebug() << PDEBUG << ":" << "Event" << eventId;
    if (debug) qDebug() << PDEBUG << ":" << "Message" << message;

    KNotification *notification = new KNotification(eventId);
    notification->setComponentData(KComponentData("plasma_applet_awesome-widget"));
    notification->setTitle(QString("Awesome Widget ::: ") + eventId);
    notification->setText(message);
    notification->sendEvent();
    delete notification;
}


void AwesomeWidget::updateNetworkDevice()
{
    if (debug) qDebug() << PDEBUG;

    values[QString("netdev")] = getNetworkDevice();
}


void AwesomeWidget::updateText(bool clear)
{
    if (debug) qDebug() << PDEBUG;

    QString text = configuration[QString("text")];
    text.replace(QString("\n"), QString("<br>"));
    if (!clear)
        for (int i=0; i<foundKeys.count(); i++)
            text.replace(QString("$") + foundKeys[i] + QString("$"), values[foundKeys[i]]);
    textLabel->setText(formatLine[0] + text + formatLine[1]);
}


void AwesomeWidget::updateTooltip()
{
    if (debug) qDebug() << PDEBUG;

    toolTipView->resize(100.0 * counts[QString("tooltip")], 100.0);
    // remove nan

    // boundaries
    QMap<QString, float> boundaries;
    boundaries[QString("cpu")] = 100.0;
    boundaries[QString("cpucl")] = 4000.0;
    boundaries[QString("mem")] = 100.0;
    boundaries[QString("swap")] = 100.0;
    boundaries[QString("down")] = 1.0;
    if (configuration[QString("downTooltip")].toInt() == 2) {
        for (int i=0; i<tooltipValues[QString("down")].count(); i++)
            if (boundaries[QString("down")] < tooltipValues[QString("down")][i])
                boundaries[QString("down")] = tooltipValues[QString("down")][i];
        for (int i=0; i<tooltipValues[QString("up")].count(); i++)
            if (boundaries[QString("down")] < tooltipValues[QString("up")][i])
                boundaries[QString("down")] = tooltipValues[QString("up")][i];
        boundaries[QString("down")] *= 1.2;
    }
    boundaries[QString("up")] = boundaries[QString("down")];

    // create image
    toolTipScene->clear();
    QPen pen = QPen();
    if (configuration[QString("useTooltipBackground")].toInt() == 2)
        toolTipScene->setBackgroundBrush(QColor(configuration[QString("tooltipBackground")]));
    else
        toolTipScene->setBackgroundBrush(QBrush(Qt::NoBrush));
    bool down = false;
    QStringList trueKeys;
    if (tooltipValues.contains(QString("cpu"))) trueKeys.append(QString("cpu"));
    if (tooltipValues.contains(QString("cpucl"))) trueKeys.append(QString("cpucl"));
    if (tooltipValues.contains(QString("mem"))) trueKeys.append(QString("mem"));
    if (tooltipValues.contains(QString("swap"))) trueKeys.append(QString("swap"));
    if (tooltipValues.contains(QString("down"))) trueKeys.append(QString("down"));
    if (tooltipValues.contains(QString("up"))) trueKeys.append(QString("up"));
    for (int i=0; i<trueKeys.count(); i++) {
        float normX = 100.0 / (tooltipValues[trueKeys[i]].count() + 0.0);
        float normY = 100.0 / (1.5 * boundaries[trueKeys[i]]);
        pen.setColor(QColor(configuration[trueKeys[i] + QString("Color")]));
        float shift = i * 100.0;
        if (down)
            shift -= 100.0;
        for (int j=0; j<tooltipValues[trueKeys[i]].count()-1; j++) {
            float x1 = j * normX + shift;
            float y1 = - tooltipValues[trueKeys[i]][j] * normY;
            float x2 = (j + 1) * normX + shift;
            float y2 = - tooltipValues[trueKeys[i]][j+1] * normY;
            toolTipScene->addLine(x1, y1, x2, y2, pen);
        }
        if (trueKeys[i] == QString("down"))
            down = true;
    }

    toolTip.setImage(QPixmap::grabWidget(toolTipView));
    Plasma::ToolTipManager::self()->setContent(this, toolTip);
}