/**
 * Copyright 2013 Albert Vaca <albertvaka@gmail.com>
 *
 * This program 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 2 of
 * the License or (at your option) version 3 or any later version
 * accepted by the membership of KDE e.V. (or its successor approved
 * by the membership of KDE e.V.), which shall act as a proxy
 * defined in Section 14 of version 3 of the license.
 *
 * This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "daemon.h"

#include "networkpackage.h"

#include "packageinterfaces/pingpackageinterface.h"
#include "packageinterfaces/notificationpackageinterface.h"
#include "packageinterfaces/pausemusicpackageinterface.h"
#include "packageinterfaces/clipboardpackageinterface.h"
#include "packageinterfaces/batterypackageinterface.h"
#include "packageinterfaces/mpriscontrolpackageinterface.h"

#include "linkproviders/avahitcplinkprovider.h"
#include "linkproviders/loopbacklinkprovider.h"

#include <QtNetwork/QUdpSocket>
#include <QFile>
#include <quuid.h>
#include <QDBusConnection>

#include <KIcon>
#include <KConfigGroup>

#include <sstream>
#include <iomanip>
#include <iostream>

K_PLUGIN_FACTORY(KdeConnectFactory, registerPlugin<Daemon>();)
K_EXPORT_PLUGIN(KdeConnectFactory("kdeconnect", "kdeconnect"))

Daemon::Daemon(QObject *parent, const QList<QVariant>&)
    : KDEDModule(parent)
{
    KSharedConfigPtr config = KSharedConfig::openConfig("kdeconnectrc");

    if (!config->group("myself").hasKey("id")) {
        config->group("myself").writeEntry("id",QUuid::createUuid().toString());
    }

    //Debugging
    qDebug() << "Starting KdeConnect daemon";

    //TODO: Do not hardcode the load of the package interfaces
    //use: https://techbase.kde.org/Development/Tutorials/Services/Plugins
    mPackageInterfaces.push_back(new PingPackageInterface());
    mPackageInterfaces.push_back(new NotificationPackageInterface());
    mPackageInterfaces.push_back(new PauseMusicPackageInterface());
    mPackageInterfaces.push_back(new ClipboardPackageInterface());
    mPackageInterfaces.push_back(new BatteryPackageInterface(this));
    mPackageInterfaces.push_back(new MprisControlPackageInterface());

    //TODO: Do not hardcode the load of the device locators
    //use: https://techbase.kde.org/Development/Tutorials/Services/Plugins
    mLinkProviders.insert(new AvahiTcpLinkProvider());
    //mLinkProviders.insert(new LoopbackLinkProvider());

    //Read remebered paired devices
    const KConfigGroup& known = config->group("devices").group("paired");
    const QStringList& list = known.groupList();
    const QString defaultName("unnamed");
    Q_FOREACH(const QString& id, list) {
        const KConfigGroup& data = known.group(id);
        const QString& name = data.readEntry<QString>("name",defaultName);
        Device* device = new Device(id,name);
        mDevices[id] = device;
        Q_FOREACH (PackageInterface* pr, mPackageInterfaces) {
            connect(device,SIGNAL(receivedPackage(const Device&, const NetworkPackage&)),
                    pr,SLOT(receivePackage(const Device&, const NetworkPackage&)));
            connect(pr,SIGNAL(sendPackage(const NetworkPackage&)),
                    device,SLOT(sendPackage(const NetworkPackage&)));
        }
    }

    //Listen to incomming connections
    Q_FOREACH (LinkProvider* a, mLinkProviders) {
        connect(a,SIGNAL(onNewDeviceLink(NetworkPackage,DeviceLink*)),
                this,SLOT(onNewDeviceLink(NetworkPackage,DeviceLink*)));
    }

    QDBusConnection::sessionBus().registerService("org.kde.kdeconnect");

    setDiscoveryEnabled(true);

}
void Daemon::setDiscoveryEnabled(bool b)
{
    //Listen to incomming connections
    Q_FOREACH (LinkProvider* a, mLinkProviders) {
        a->setDiscoverable(b);
    }

}

QStringList Daemon::devices()
{
    return mDevices.keys();
}


void Daemon::onNewDeviceLink(const NetworkPackage& identityPackage, DeviceLink* dl)
{
    const QString& id = identityPackage.get<QString>("deviceId");

    qDebug() << "Device discovered" << id << "via" << dl->provider()->name();

    if (mDevices.contains(id)) {
        qDebug() << "It is a known device";

        Device* device = mDevices[id];
        device->addLink(dl);

        KNotification* notification = new KNotification("pingReceived"); //KNotification::Persistent
        notification->setPixmap(KIcon("dialog-ok").pixmap(48, 48));
        notification->setComponentData(KComponentData("kdeconnect", "kdeconnect"));
        notification->setTitle(device->name());
        notification->setText("Succesfully connected");
        notification->sendEvent();

        emit deviceStatusChanged(id);
    } else {
        qDebug() << "It is a new device";

        const QString& name = identityPackage.get<QString>("deviceName");

        Device* device = new Device(id,name,dl);
        mDevices[id] = device;
        Q_FOREACH (PackageInterface* pr, mPackageInterfaces) {
            connect(device,SIGNAL(receivedPackage(const Device&, const NetworkPackage&)),
                    pr,SLOT(receivePackage(const Device&, const NetworkPackage&)));
            connect(pr,SIGNAL(sendPackage(const NetworkPackage&)),
                    device,SLOT(sendPackage(const NetworkPackage&)));
        }
        emit newDeviceAdded(id);
    }

}

Daemon::~Daemon()
{

}