qTox/widget/filetransfertwidget.cpp

/*
    Copyright (C) 2014 by Project Tox <https://tox.im>

    This file is part of qTox, a Qt-based graphical interface for Tox.

    This program is libre 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.
    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 COPYING file for more details.
*/

#include "filetransfertwidget.h"
#include "widget.h"
#include "core.h"
#include "math.h"
#include "style.h"
#include <QFileDialog>
#include <QPixmap>
#include <QPainter>
#include <QMessageBox>

FileTransfertWidget::FileTransfertWidget(ToxFile File)
    : lastUpdate{QDateTime::currentDateTime()}, lastBytesSent{0},
      fileNum{File.fileNum}, friendId{File.friendId}, direction{File.direction}
{
    pic=new QLabel(), filename=new QLabel(), size=new QLabel(), speed=new QLabel(), eta=new QLabel();
    topright = new QPushButton(), bottomright = new QPushButton();
    progress = new QProgressBar();
    mainLayout = new QHBoxLayout(), textLayout = new QHBoxLayout();
    infoLayout = new QVBoxLayout(), buttonLayout = new QVBoxLayout();
    buttonWidget = new QWidget();
    QFont prettysmall;
    prettysmall.setPixelSize(10);
    this->setObjectName("default");
    this->setStyleSheet(Style::get(":/ui/fileTransferWidget/fileTransferWidget.css"));
    QPalette greybg;
    greybg.setColor(QPalette::Window, QColor(209,209,209));
    greybg.setColor(QPalette::Base, QColor(150,150,150));
    setPalette(greybg);
    setAutoFillBackground(true);

    setMinimumSize(250,58);
    setMaximumHeight(58);
    setLayout(mainLayout);
    mainLayout->setMargin(0);

    pic->setMaximumHeight(40);
    pic->setContentsMargins(5,0,0,0);
    filename->setText(File.fileName);
    filename->setFont(prettysmall);
    size->setText(getHumanReadableSize(File.filesize));
    size->setFont(prettysmall);
    speed->setText("0B/s");
    speed->setFont(prettysmall);
    eta->setText("00:00");
    eta->setFont(prettysmall);
    progress->setValue(0);
    progress->setMinimumHeight(11);
    progress->setFont(prettysmall);
    progress->setTextVisible(false);
    QPalette whitebg;
    whitebg.setColor(QPalette::Window, QColor(255,255,255));
    buttonWidget->setPalette(whitebg);
    buttonWidget->setAutoFillBackground(true);
    buttonWidget->setLayout(buttonLayout);

    stopFileButtonStylesheet = Style::get(":/ui/stopFileButton/style.css");
    pauseFileButtonStylesheet = Style::get(":/ui/pauseFileButton/style.css");
    acceptFileButtonStylesheet = Style::get(":/ui/acceptFileButton/style.css");

    topright->setStyleSheet(stopFileButtonStylesheet);
    if (File.direction == ToxFile::SENDING)
    {
        bottomright->setStyleSheet(pauseFileButtonStylesheet);
        connect(topright, SIGNAL(clicked()), this, SLOT(cancelTransfer()));
        connect(bottomright, SIGNAL(clicked()), this, SLOT(pauseResumeSend()));

        QPixmap preview;
        File.file->seek(0);
        if (preview.loadFromData(File.file->readAll()))
        {
            preview = preview.scaledToHeight(40);
            pic->setPixmap(preview);
        }
        File.file->seek(0);
    }
    else
    {
        bottomright->setStyleSheet(acceptFileButtonStylesheet);
        connect(topright, SIGNAL(clicked()), this, SLOT(rejectRecvRequest()));
        connect(bottomright, SIGNAL(clicked()), this, SLOT(acceptRecvRequest()));
    }

    QPalette toxgreen;
    toxgreen.setColor(QPalette::Button, QColor(107,194,96)); // Tox Green
    topright->setIconSize(QSize(10,10));
    topright->setMinimumSize(25,28);
    topright->setFlat(true);
    topright->setAutoFillBackground(true);
    topright->setPalette(toxgreen);
    bottomright->setIconSize(QSize(10,10));
    bottomright->setMinimumSize(25,28);
    bottomright->setFlat(true);
    bottomright->setAutoFillBackground(true);
    bottomright->setPalette(toxgreen);

    mainLayout->addStretch();
    mainLayout->addWidget(pic);
    mainLayout->addLayout(infoLayout,3);
    mainLayout->addStretch();
    mainLayout->addWidget(buttonWidget);
    mainLayout->setMargin(0);
    mainLayout->setSpacing(0);

    infoLayout->addWidget(filename);
    infoLayout->addLayout(textLayout);
    infoLayout->addWidget(progress);
    infoLayout->setMargin(4);
    infoLayout->setSpacing(4);

    textLayout->addWidget(size);
    textLayout->addStretch(0);
    textLayout->addWidget(speed);
    textLayout->addStretch(0);
    textLayout->addWidget(eta);
    textLayout->setMargin(2);
    textLayout->setSpacing(5);

    buttonLayout->addWidget(topright);
    buttonLayout->addSpacing(2);
    buttonLayout->addWidget(bottomright);
    buttonLayout->setContentsMargins(2,0,0,0);
    buttonLayout->setSpacing(0);
}

QString FileTransfertWidget::getHumanReadableSize(int size)
{
    static const char* suffix[] = {"B","kiB","MiB","GiB","TiB"};
    int exp = 0;
    if (size)
        exp = std::min( (int) (log(size) / log(1024)), (int) (sizeof(suffix) / sizeof(suffix[0]) - 1));
    return QString().setNum(size / pow(1024, exp),'g',3).append(suffix[exp]);
}

void FileTransfertWidget::onFileTransferInfo(int FriendId, int FileNum, int64_t Filesize, int64_t BytesSent, ToxFile::FileDirection Direction)
{
    if (FileNum != fileNum || FriendId != friendId || Direction != direction)
            return;
    QDateTime newtime = QDateTime::currentDateTime();
    int timediff = lastUpdate.secsTo(newtime);
    if (timediff <= 0)
        return;
    qint64 diff = BytesSent - lastBytesSent;
    if (diff < 0)
    {
        qWarning() << "FileTransfertWidget::onFileTransferInfo: Negative transfer speed !";
        diff = 0;
    }
    int rawspeed = diff / timediff;
    speed->setText(getHumanReadableSize(rawspeed)+"/s");
    size->setText(getHumanReadableSize(Filesize));
    if (!rawspeed)
        return;
    int etaSecs = (Filesize - BytesSent) / rawspeed;
    QTime etaTime(0,0);
    etaTime = etaTime.addSecs(etaSecs);
    eta->setText(etaTime.toString("mm:ss"));
    if (!Filesize)
    {
        progress->setValue(0);
        qDebug() << QString("FT: received %1 bytes of an empty file, stop sending sequential devices, zetok!").arg(BytesSent);
    }
    else
    {
        progress->setValue(BytesSent*100/Filesize);
        qDebug() << QString("FT: received %1/%2 bytes, progress is %3%").arg(BytesSent).arg(Filesize).arg(BytesSent*100/Filesize);
    }
    lastUpdate = newtime;
    lastBytesSent = BytesSent;
}

void FileTransfertWidget::onFileTransferCancelled(int FriendId, int FileNum, ToxFile::FileDirection Direction)
{
    if (FileNum != fileNum || FriendId != friendId || Direction != direction)
            return;
    buttonLayout->setContentsMargins(0,0,0,0);
    disconnect(topright);
    disconnect(Widget::getInstance()->getCore(),0,this,0);
    progress->hide();
    speed->hide();
    eta->hide();
    topright->hide();
    bottomright->hide();
    QPalette whiteText;
    whiteText.setColor(QPalette::WindowText, Qt::white);
    filename->setPalette(whiteText);
    size->setPalette(whiteText);
    this->setObjectName("error");
    this->style()->polish(this);

    //Toggle window visibility to fix draw order bug
    this->hide();
    this->show();
}

void FileTransfertWidget::onFileTransferFinished(ToxFile File)
{
    if (File.fileNum != fileNum || File.friendId != friendId || File.direction != direction)
            return;
    topright->disconnect();
    disconnect(Widget::getInstance()->getCore(),0,this,0);
    progress->hide();
    speed->hide();
    eta->hide();
    topright->hide();
    bottomright->hide();
    buttonLayout->setContentsMargins(0,0,0,0);
    QPalette whiteText;
    whiteText.setColor(QPalette::WindowText, Qt::white);
    filename->setPalette(whiteText);
    size->setPalette(whiteText);
    this->setObjectName("success");
    this->style()->polish(this);

    //Toggle window visibility to fix draw order bug
    this->hide();
    this->show();

    if (File.direction == ToxFile::RECEIVING)
    {
        QPixmap preview;
        QFile previewFile(File.filePath);
        if (previewFile.open(QIODevice::ReadOnly) && previewFile.size() <= 1024*1024*25) // Don't preview big (>25MiB) images
        {
            if (preview.loadFromData(previewFile.readAll()))
            {
                preview = preview.scaledToHeight(40);
                pic->setPixmap(preview);
            }
            previewFile.close();
        }
    }
}

void FileTransfertWidget::cancelTransfer()
{
    Widget::getInstance()->getCore()->cancelFileSend(friendId, fileNum);
}

void FileTransfertWidget::rejectRecvRequest()
{
    Widget::getInstance()->getCore()->rejectFileRecvRequest(friendId, fileNum);
    onFileTransferCancelled(friendId, fileNum, direction);
}

// for whatever the fuck reason, QFileInfo::isWritable() always fails for files that don't exist
// which makes it useless for our case
// since QDir doesn't have an isWritable(), the only option I can think of is to make/delete the file
// surely this is a common problem that has a qt-implemented solution?
bool isWritable(QString& path)
{
    QFile file(path);
    bool exists = file.exists();
    bool out = file.open(QIODevice::WriteOnly);
    file.close();
    if (!exists)
        file.remove();
    return out;
}

void FileTransfertWidget::acceptRecvRequest()
{
    QString path;
    while (true)
    {
        path = QFileDialog::getSaveFileName(this,tr("Save a file","Title of the file saving dialog"),QDir::currentPath()+'/'+filename->text());
        if (path.isEmpty())
            return;
        else
        {
            //bool savable = QFileInfo(path).isWritable();
            //qDebug() << path << " is writable: " << savable;
            //qDebug() << "/home/bill/bliss.pdf writable: " << QFileInfo("/home/bill/bliss.pdf").isWritable();
            if (isWritable(path))
                break;
            else
                QMessageBox::warning(this, tr("Location not writable","Title of permissions popup"), tr("You do not have permission to write that location. Choose another, or cancel the save dialog.", "text of permissions popup"));
        }
    }

    savePath = path;

    bottomright->setStyleSheet(pauseFileButtonStylesheet);
    bottomright->disconnect();
    connect(bottomright, SIGNAL(clicked()), this, SLOT(pauseResumeRecv()));
    Widget::getInstance()->getCore()->acceptFileRecvRequest(friendId, fileNum, path);
}

void FileTransfertWidget::pauseResumeRecv()
{
    Widget::getInstance()->getCore()->pauseResumeFileRecv(friendId, fileNum);
}

void FileTransfertWidget::pauseResumeSend()
{
    Widget::getInstance()->getCore()->pauseResumeFileSend(friendId, fileNum);
}

void FileTransfertWidget::paintEvent(QPaintEvent *)
{
    QStyleOption opt;
    opt.init(this);
    QPainter p(this);
    style()->drawPrimitive(QStyle::PE_Widget, &opt, &p, this);
}