src/client/clienttransfer.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-2020 by the Quassel Project                        *
   3  *   devel@quassel-irc.org                                                 *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) version 3.                                           *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  19  ***************************************************************************/
  20
  21 #include "clienttransfer.h"
  22
  23 #include <QFile>
  24
  25 ClientTransfer::ClientTransfer(const QUuid& uuid, QObject* parent)
  26     : Transfer(uuid, parent)
  27     , _file(nullptr)
  28 {
  29     connect(this, &Transfer::statusChanged, this, &ClientTransfer::onStatusChanged);
  30 }
  31
  32 quint64 ClientTransfer::transferred() const
  33 {
  34     if (status() == Status::Completed)
  35         return fileSize();
  36
  37     return _file ? _file->size() : 0;
  38 }
  39
  40 void ClientTransfer::cleanUp()
  41 {
  42     if (_file) {
  43         _file->close();
  44         _file->deleteLater();
  45         _file = nullptr;
  46     }
  47 }
  48
  49 QString ClientTransfer::savePath() const
  50 {
  51     return _savePath;
  52 }
  53
  54 void ClientTransfer::accept(const QString& savePath) const
  55 {
  56     _savePath = savePath;
  57     PeerPtr ptr = nullptr;
  58     REQUEST_OTHER(requestAccepted, ARG(ptr));
  59     emit accepted();
  60 }
  61
  62 void ClientTransfer::reject() const
  63 {
  64     PeerPtr ptr = nullptr;
  65     REQUEST_OTHER(requestRejected, ARG(ptr));
  66     emit rejected();
  67 }
  68
  69 void ClientTransfer::dataReceived(PeerPtr, const QByteArray& data)
  70 {
  71     // TODO: proper error handling (relay to core)
  72     if (!_file) {
  73         _file = new QFile(_savePath, this);
  74         if (!_file->open(QFile::WriteOnly | QFile::Truncate)) {
  75             qWarning() << Q_FUNC_INFO << "Could not open file:" << _file->errorString();
  76             return;
  77         }
  78     }
  79
  80     if (!_file->isOpen())
  81         return;
  82
  83     if (_file->write(data) < 0) {
  84         qWarning() << Q_FUNC_INFO << "Could not write to file:" << _file->errorString();
  85         return;
  86     }
  87
  88     emit transferredChanged(transferred());
  89 }
  90
  91 void ClientTransfer::onStatusChanged(Transfer::Status status)
  92 {
  93     switch (status) {
  94     case Status::Completed:
  95         if (_file)
  96             _file->close();
  97         break;
  98     case Status::Failed:
  99         if (_file)
 100             _file->remove();
 101         break;
 102     default:
 103         ;
 104     }
 105 }