mirror of
https://github.com/cmclark00/retro-imager.git
synced 2025-05-18 07:55:21 +01:00
111 lines
3.1 KiB
C++
111 lines
3.1 KiB
C++
#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;
|
|
|
|
/* Function to return version */
|
|
Q_INVOKABLE QString constantVersion() 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
|