aboutsummaryrefslogtreecommitdiff
path: root/src/downloadwidget.cpp
blob: d228074748daf7e4385dd3596012da3c03db3cd5 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
#include "downloadwidget.h"

#include <QFileInfo>
#include <QUrl>
#include <QWebEngineDownloadItem>

DownloadWidget::DownloadWidget(QWebEngineDownloadItem *download,
                               QWidget *parent)
    : QFrame(parent), m_download(download) {
  setupUi(this);
  m_dstName->setText(QFileInfo(m_download->downloadDirectory()).fileName());
  m_srcUrl->setText(m_download->url().toDisplayString());

  connect(m_cancelButton, &QPushButton::clicked, m_download, [this](bool) {
    if (m_download->state() == QWebEngineDownloadItem::DownloadInProgress)
      m_download->cancel();
    else
      emit removeClicked(this);
  });

  connect(m_download, &QWebEngineDownloadItem::downloadProgress, this,
          &DownloadWidget::updateWidget);

  connect(m_download, &QWebEngineDownloadItem::stateChanged, this,
          &DownloadWidget::updateWidget);

  updateWidget();
}

inline QString DownloadWidget::withUnit(qreal bytes) {
  if (bytes < (1 << 10))
    return tr("%L1 B").arg(bytes);
  else if (bytes < (1 << 20))
    return tr("%L1 KiB").arg(bytes / (1 << 10), 0, 'f', 2);
  else if (bytes < (1 << 30))
    return tr("%L1 MiB").arg(bytes / (1 << 20), 0, 'f', 2);
  else
    return tr("%L1 GiB").arg(bytes / (1 << 30), 0, 'f', 2);
}

void DownloadWidget::updateWidget() {
  qreal totalBytes = m_download->totalBytes();
  qreal receivedBytes = m_download->receivedBytes();
  qreal bytesPerSecond = receivedBytes / m_timeAdded.elapsed() * 1000;

  auto state = m_download->state();
  switch (state) {
  case QWebEngineDownloadItem::DownloadRequested:
    Q_UNREACHABLE();
    break;
  case QWebEngineDownloadItem::DownloadInProgress:
    if (totalBytes >= 0) {
      m_progressBar->setValue(qRound(100 * receivedBytes / totalBytes));
      m_progressBar->setDisabled(false);
      m_progressBar->setFormat(tr("%p% - %1 of %2 downloaded - %3/s")
                                   .arg(withUnit(receivedBytes),
                                        withUnit(totalBytes),
                                        withUnit(bytesPerSecond)));
    } else {
      m_progressBar->setValue(0);
      m_progressBar->setDisabled(false);
      m_progressBar->setFormat(
          tr("unknown size - %1 downloaded - %2/s")
              .arg(withUnit(receivedBytes), withUnit(bytesPerSecond)));
    }
    break;
  case QWebEngineDownloadItem::DownloadCompleted:
    m_progressBar->setValue(100);
    m_progressBar->setDisabled(true);
    m_progressBar->setFormat(
        tr("completed - %1 downloaded - %2/s")
            .arg(withUnit(receivedBytes), withUnit(bytesPerSecond)));
    break;
  case QWebEngineDownloadItem::DownloadCancelled:
    m_progressBar->setValue(0);
    m_progressBar->setDisabled(true);
    m_progressBar->setFormat(
        tr("cancelled - %1 downloaded - %2/s")
            .arg(withUnit(receivedBytes), withUnit(bytesPerSecond)));
    break;
  case QWebEngineDownloadItem::DownloadInterrupted:
    m_progressBar->setValue(0);
    m_progressBar->setDisabled(true);
    m_progressBar->setFormat(
        tr("interrupted: %1").arg(m_download->interruptReasonString()));
    break;
  }

  if (state == QWebEngineDownloadItem::DownloadInProgress) {
    static QIcon cancelIcon(":/icons/stop-line.png");
    m_cancelButton->setIcon(cancelIcon);
    m_cancelButton->setToolTip(tr("Stop downloading"));
  } else {
    static QIcon removeIcon(":/icons/close-fill.png");
    m_cancelButton->setIcon(removeIcon);
    m_cancelButton->setToolTip(tr("Remove from list"));
  }
}