theshell/statuscenter/BluetoothPane/btobexagent.cpp

#include "btobexagent.h"

BTObexAgent::BTObexAgent(BluezQt::ObexManager *mgr, BluezQt::Manager* btMgr, QObject *parent) : BluezQt::ObexAgent(parent) {
    this->mgr = mgr;
    this->btMgr = btMgr;

    QDBusConnection dbus = QDBusConnection::sessionBus();
    dbus.connect("org.freedesktop.Notifications", "/org/freedesktop/Notifications", "org.freedesktop.Notifications", "ActionInvoked", this, SLOT(ActionInvoked(uint,QString)));

    BluezQt::PendingCall* serviceStarter = BluezQt::ObexManager::startService();
    serviceStarter->waitForFinished();

    BluezQt::InitObexManagerJob* job = mgr->init();
    connect(job, &BluezQt::InitObexManagerJob::result, [=] {
        if (job->error() == BluezQt::InitObexManagerJob::NoError) {
            mgr->registerAgent(this);
        }
    });
    job->start();
}

QDBusObjectPath BTObexAgent::objectPath() const {
    return QDBusObjectPath("/org/thesuite/tsbt/obex");
}

void BTObexAgent::authorizePush(BluezQt::ObexTransferPtr transfer, BluezQt::ObexSessionPtr session, const BluezQt::Request<QString> &request) {
    this->currentRequest = request;
    this->currentTransfer = transfer;

    BluezQt::DevicePtr dev = btMgr->deviceForAddress(session.data()->destination());

    QDBusInterface interface("org.freedesktop.Notifications", "/org/freedesktop/Notifications", "org.freedesktop.Notifications");
    QVariantMap hints;
    hints.insert("transient", true);

    QStringList actions;
    actions.append("true");
    actions.append("Accept File");
    actions.append("false");
    actions.append("Cancel");

    QList<QVariant> args;
    args << tr("Bluetooth") << notificationNumber << "preferences-system-bluetooth" << tr("Incoming File over Bluetooth") <<
                            tr("%1 wants to send the file <b>%2</b>.").arg(dev.data()->name(), transfer.data()->name()) <<
                            actions << hints << (int) 0;

    QDBusReply<uint> reply = interface.callWithArgumentList(QDBus::Block, "Notify", args);
    this->notificationNumber = reply.value();
}

void BTObexAgent::ActionInvoked(uint id, QString action) {
    if (id == this->notificationNumber) {
        if (action == "true") {
            emit newTransfer(currentTransfer);
            QString transferName = currentTransfer.data()->name();

            QString cachePath = QFile::decodeName(qgetenv("XDG_CACHE_HOME"));
            if (cachePath == "") cachePath = QDir::homePath() + "/.cache";
            cachePath += "/obexd/";

            QString temporaryPath = cachePath + transferName;
            int number = 0;
            while (QFile::exists(temporaryPath)) {
                temporaryPath = cachePath + transferName + "_" + QString::number(number);
                number++;
            }

            connect(currentTransfer.data(), &BluezQt::ObexTransfer::statusChanged, [=](BluezQt::ObexTransfer::Status status) {
                if (status == BluezQt::ObexTransfer::Complete) {
                    //Move the file to a good place

                    QString permanantPath = QDir::homePath() + "/Downloads/Bluetooth/" + transferName;
                    int number = 0;
                    while (QFile::exists(permanantPath)) {
                        permanantPath = QDir::homePath() + "/Downloads/Bluetooth/" + transferName + " (" + QString::number(number) + ")";
                        number++;
                    }

                    QFile::rename(temporaryPath, permanantPath);
                }
            });

            currentRequest.accept(temporaryPath);
        } else {
            currentRequest.reject();
        }
    }
}

void BTObexAgent::cancel() {
    QDBusInterface interface("org.freedesktop.Notifications", "/org/freedesktop/Notifications", "org.freedesktop.Notifications");
    interface.call("CloseNotification", this->notificationNumber);
}