Implement GUI for receiving incoming multicast transfers

1 files changed, 82 insertions, 1 deletions

diff --git a/src/gui/clientFileReceiveDialog.cpp b/src/gui/clientFileReceiveDialog.cpp
index 669ca81..ff3226a 100644
--- a/src/gui/clientFileReceiveDialog.cpp
+++ b/src/gui/clientFileReceiveDialog.cpp
@@ -16,6 +16,9 @@
  */
 
 #include "clientFileReceiveDialog.h"
+#include <pvsinterface.h>
+#include <QFileInfo>
+#include <limits.h>
 
 ClientFileReceiveDialog::ClientFileReceiveDialog(QTcpSocket *socket, QWidget *parent) :
         QDialog(parent)
@@ -37,9 +40,44 @@ ClientFileReceiveDialog::ClientFileReceiveDialog(QTcpSocket *socket, QWidget *pa
     connect(this, SIGNAL(finished(int)), this, SLOT(deleteLater()));
 }
 
+ClientFileReceiveDialog::ClientFileReceiveDialog(QString const& sender, qulonglong transferID,
+        QString const& filename, qulonglong size, OrgOpenslxPvsInterface* ifaceDBus, QWidget* parent)
+{
+    setupUi(this);
+
+    _transferID = transferID;
+    _filename = filename;
+    _ifaceDBus = ifaceDBus;
+    _bytesToRead = size;
+    _socket = 0;
+    _file = 0;
+    div = 1;
+    while((size / div) > INT_MAX)
+    {
+        div <<= 1;
+    }
+
+    connect(ifaceDBus, SIGNAL(incomingMulticastTransferStarted(qulonglong)), SLOT(mcastTransferStarted(qulonglong)));
+    connect(ifaceDBus, SIGNAL(incomingMulticastTransferProgress(qulonglong, qulonglong, qulonglong)),
+            SLOT(mcastTransferProgress(qulonglong, qulonglong, qulonglong)));
+    connect(ifaceDBus, SIGNAL(incomingMulticastTransferFinished(qulonglong)), SLOT(mcastTransferFinished(qulonglong)));
+    connect(ifaceDBus, SIGNAL(incomingMulticastTransferFailed(qulonglong, QString)), SLOT(mcastTransferFailed(qulonglong, QString)));
+    connect(cancelButton, SIGNAL(clicked()), SLOT(cancelTransfer()));
+
+    qDebug("[%s] New multicast incoming transfer: %s from %s", metaObject()->className(),
+            filename.toLocal8Bit().constData(), sender.toLocal8Bit().constData());
+
+
+    filenameLabel->setText(QFileInfo(filename).baseName());
+    labelNick->setText(sender);
+    progressBar->setRange(0, 0);
+}
+
+
 ClientFileReceiveDialog::~ClientFileReceiveDialog()
 {
-    _socket->deleteLater();
+    if(_socket)
+        _socket->deleteLater();
     qDebug("[%s] Deleted!", metaObject()->className());
 }
 
@@ -168,6 +206,49 @@ void ClientFileReceiveDialog::error(QAbstractSocket::SocketError error)
     close();
 }
 
+void ClientFileReceiveDialog::mcastTransferStarted(qulonglong transferID)
+{
+    if(transferID != _transferID)
+        return;
+}
+
+void ClientFileReceiveDialog::mcastTransferProgress(qulonglong transferID, qulonglong bytes, qulonglong of)
+{
+    if(transferID != _transferID)
+        return;
+
+    progressBar->setRange(0, of);
+    progressBar->setValue(bytes);
+}
+
+void ClientFileReceiveDialog::mcastTransferFinished(qulonglong transferID)
+{
+    if(transferID != _transferID)
+        return;
+
+    QString filename = QFileDialog::getSaveFileName(this, tr("Where should I save %1?").arg(_filename), _filename);
+    QFile* file = new QFile(_filename);
+    if(!file->rename(filename))
+    {
+        QMessageBox::warning(this, tr("Could not rename file"), tr("Failed to rename %1 to %2").arg(_filename).arg(filename));
+    }
+    accept();
+}
+
+void ClientFileReceiveDialog::mcastTransferFailed(qulonglong transferID, QString reason)
+{
+    if(transferID != _transferID)
+        return;
+
+    QMessageBox::warning(this, tr("File transfer failed"), tr("File transfer failed for the following reason:\n%1").arg(reason));
+    reject();
+}
+
+void ClientFileReceiveDialog::cancelTransfer()
+{
+    _ifaceDBus->cancelIncomingMulticastTransfer(_transferID);
+}
+
 QString ClientFileReceiveDialog::formatSize(qint64 size)
 {
     if (size >= 1000000000) // GB