Qt/QML edition

imagewriter.h

@@ -0,0 +1,108 @@
+#ifndef IMAGEWRITER_H
+#define IMAGEWRITER_H
+
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright (C) 2020 Raspberry Pi (Trading) Limited
+ */
+
+#include <QObject>
+#include <QTimer>
+#include <QUrl>
+#include <QSettings>
+#include <QVariant>
+#include "config.h"
+#include "powersaveblocker.h"
+#include "drivelistmodel.h"
+
+class QQmlApplicationEngine;
+class DownloadThread;
+
+class ImageWriter : public QObject
+{
+    Q_OBJECT
+public:
+    explicit ImageWriter(QObject *parent = nullptr);
+    virtual ~ImageWriter();
+    void setEngine(QQmlApplicationEngine *engine);
+
+    /* Set URL to download from, and if known download length and uncompressed length */
+    Q_INVOKABLE void setSrc(const QUrl &url, quint64 downloadLen = 0, quint64 extrLen = 0, QByteArray expectedHash = "", bool multifilesinzip = false);
+
+    /* Set device to write to */
+    Q_INVOKABLE void setDst(const QString &device, quint64 deviceSize = 0);
+
+    /* Enable/disable verification */
+    Q_INVOKABLE void setVerifyEnabled(bool verify);
+
+    /* Returns true if src and dst are set */
+    Q_INVOKABLE bool readyToWrite();
+
+    /* Start writing */
+    Q_INVOKABLE void startWrite();
+
+    /* Cancel write */
+    Q_INVOKABLE void cancelWrite();
+
+    /* Return true if url is in our local disk cache */
+    Q_INVOKABLE bool isCached(const QUrl &url, const QByteArray &sha256);
+
+    /* Refresh the list of available drives */
+    Q_INVOKABLE void refreshDriveList();
+
+    /* Return list of available drives. Call refreshDriveList() first */
+    DriveListModel *getDriveList();
+
+    /* Utility function to return filename part from URL */
+    Q_INVOKABLE QString fileNameFromUrl(const QUrl &url);
+
+    /* Function to return OS list URL */
+    Q_INVOKABLE QUrl constantOsListUrl() const;
+
+    /* Set custom repository */
+    Q_INVOKABLE void setCustomOsListUrl(const QUrl &url);
+
+    /* Utility function to open OS file dialog */
+    Q_INVOKABLE void openFileDialog();
+
+    /* Return filename part of URL set */
+    Q_INVOKABLE QString srcFileName();
+
+signals:
+    /* We are emiting signals with QVariant as parameters because QML likes it that way */
+
+    void downloadProgress(QVariant dlnow, QVariant dltotal);
+    void verifyProgress(QVariant now, QVariant total);
+    void error(QVariant msg);
+    void success();
+    void fileSelected(QVariant filename);
+    void cancelled();
+    void finalizing();
+
+protected slots:
+
+    void pollProgress();
+    void onSuccess();
+    void onError(QString msg);
+    void onFileSelected(QString filename);
+    void onCancelled();
+    void onCacheFileUpdated(QByteArray sha256);
+    void onFinalizing();
+
+protected:
+    QUrl _src, _repo;
+    QString _dst, _cacheFileName;
+    QByteArray _expectedHash, _cachedFileHash;
+    quint64 _downloadLen, _extrLen, _devLen, _dlnow, _verifynow;
+    DriveListModel _drivelist;
+    QQmlApplicationEngine *_engine;
+    QTimer _polltimer;
+    PowerSaveBlocker _powersave;
+    DownloadThread *_thread;
+    bool _verifyEnabled, _multipleFilesInZip, _cachingEnabled;
+    QSettings _settings;
+
+    void _parseCompressedFile();
+};
+
+#endif // IMAGEWRITER_H