mirror of
https://github.com/cmclark00/retro-imager.git
synced 2025-05-18 07:55:21 +01:00
282 lines
9 KiB
C++
282 lines
9 KiB
C++
/*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
* Copyright (C) 2020 Raspberry Pi (Trading) Limited
|
|
*/
|
|
|
|
#include <QFileInfo>
|
|
#include <QGuiApplication>
|
|
#include <QQmlApplicationEngine>
|
|
#include <QQmlContext>
|
|
#include <QIcon>
|
|
#include <QDebug>
|
|
#include <QTextStream>
|
|
#include "imagewriter.h"
|
|
#include "drivelistmodel.h"
|
|
#include "networkaccessmanagerfactory.h"
|
|
#include "cli.h"
|
|
#include <QMessageLogContext>
|
|
#include <QQuickWindow>
|
|
#include <QTranslator>
|
|
#include <QLocale>
|
|
#include <QScreen>
|
|
#include <QSettings>
|
|
#ifndef QT_NO_WIDGETS
|
|
#include <QtWidgets/QApplication>
|
|
#endif
|
|
#ifdef Q_OS_DARWIN
|
|
#include <CoreFoundation/CoreFoundation.h>
|
|
#endif
|
|
|
|
static QTextStream cerr(stderr);
|
|
|
|
#ifdef Q_OS_WIN
|
|
static void consoleMsgHandler(QtMsgType, const QMessageLogContext &, const QString &str) {
|
|
cerr << str << endl;
|
|
}
|
|
#endif
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
for (int i = 1; i < argc; i++)
|
|
{
|
|
if (strcmp(argv[i], "--cli") == 0)
|
|
{
|
|
/* CLI mode */
|
|
Cli cli(argc, argv);
|
|
return cli.main();
|
|
}
|
|
}
|
|
|
|
QCoreApplication::setAttribute(Qt::AA_EnableHighDpiScaling);
|
|
#ifdef Q_OS_WIN
|
|
// prefer ANGLE (DirectX) over desktop OpenGL
|
|
QCoreApplication::setAttribute(Qt::AA_UseOpenGLES);
|
|
#endif
|
|
#ifdef QT_NO_WIDGETS
|
|
{
|
|
QGuiApplication tmp(argc, argv);
|
|
int h = QGuiApplication::primaryScreen()->geometry().height();
|
|
if (h > 720)
|
|
{
|
|
qputenv("QT_SCALE_FACTOR", QByteArray::number(h / 720.0, 'f', 2));
|
|
}
|
|
}
|
|
|
|
QGuiApplication app(argc, argv);
|
|
#else
|
|
QApplication app(argc, argv);
|
|
#endif
|
|
app.setOrganizationName("Raspberry Pi");
|
|
app.setOrganizationDomain("raspberrypi.org");
|
|
app.setApplicationName("Imager");
|
|
app.setWindowIcon(QIcon(":/icons/rpi-imager.ico"));
|
|
ImageWriter imageWriter;
|
|
NetworkAccessManagerFactory namf;
|
|
QQmlApplicationEngine engine;
|
|
QTranslator translator;
|
|
QString customQm;
|
|
QSettings settings;
|
|
|
|
/* Parse commandline arguments (if any) */
|
|
QString customRepo;
|
|
QUrl url;
|
|
QStringList args = app.arguments();
|
|
for (int i=1; i < args.size(); i++)
|
|
{
|
|
if (!args[i].startsWith("-") && url.isEmpty())
|
|
{
|
|
if (args[i].startsWith("http:", Qt::CaseInsensitive) || args[i].startsWith("https:", Qt::CaseInsensitive))
|
|
{
|
|
url = args[i];
|
|
}
|
|
else
|
|
{
|
|
QFileInfo fi(args[i]);
|
|
|
|
if (fi.isFile())
|
|
{
|
|
url = QUrl::fromLocalFile(args[i]);
|
|
}
|
|
else
|
|
{
|
|
cerr << "Argument ignored because it is not a regular file: " << args[i] << endl;;
|
|
}
|
|
}
|
|
}
|
|
else if (args[i] == "--repo")
|
|
{
|
|
if (args.size()-i < 2 || args[i+1].startsWith("-"))
|
|
{
|
|
cerr << "Missing URL after --repo" << endl;
|
|
return 1;
|
|
}
|
|
|
|
customRepo = args[++i];
|
|
if (customRepo.startsWith("http://") || customRepo.startsWith("https://"))
|
|
{
|
|
imageWriter.setCustomOsListUrl(customRepo);
|
|
}
|
|
else
|
|
{
|
|
QFileInfo fi(customRepo);
|
|
if (!fi.isFile())
|
|
{
|
|
cerr << "Custom repository file does not exist or is not a regular file: " << customRepo << endl;
|
|
return 1;
|
|
}
|
|
|
|
imageWriter.setCustomOsListUrl(QUrl::fromLocalFile(customRepo));
|
|
}
|
|
}
|
|
else if (args[i] == "--qm")
|
|
{
|
|
if (args.size()-i < 2 || args[i+1].startsWith("-"))
|
|
{
|
|
cerr << "Missing QM file after --qm" << endl;
|
|
return 1;
|
|
}
|
|
customQm = args[++i];
|
|
|
|
QFileInfo fi(customQm);
|
|
if (!fi.isFile())
|
|
{
|
|
cerr << "Custom QM file does not exist or is not a regular file: " << customQm << endl;
|
|
return 1;
|
|
}
|
|
}
|
|
else if (args[i] == "--debug")
|
|
{
|
|
#ifdef Q_OS_WIN
|
|
/* Allocate console for debug messages on Windows */
|
|
if (::AttachConsole(ATTACH_PARENT_PROCESS) || ::AllocConsole())
|
|
{
|
|
freopen("CONOUT$", "w", stdout);
|
|
freopen("CONOUT$", "w", stderr);
|
|
std::ios::sync_with_stdio();
|
|
qInstallMessageHandler(consoleMsgHandler);
|
|
}
|
|
#endif
|
|
}
|
|
else if (args[i] == "--help")
|
|
{
|
|
cerr << args[0] << " [--debug] [--version] [--repo <repository URL>] [--qm <custom qm translation file>] [--disable-telemetry] [<image file to write>]" << endl;
|
|
cerr << "-OR- " << args[0] << " --cli [--disable-verify] [--sha256 <expected hash>] [--debug] [--quiet] <image file to write> <destination drive device>" << endl;
|
|
return 0;
|
|
}
|
|
else if (args[i] == "--version")
|
|
{
|
|
cerr << args[0] << " version " << imageWriter.constantVersion() << endl;
|
|
cerr << "Repository: " << imageWriter.constantOsListUrl().toString() << endl;
|
|
return 0;
|
|
}
|
|
else if (args[i] == "--disable-telemetry")
|
|
{
|
|
cerr << "Disabled telemetry" << endl;
|
|
settings.setValue("telemetry", false);
|
|
settings.sync();
|
|
}
|
|
else if (args[i] == "--enable-telemetry")
|
|
{
|
|
cerr << "Using default telemetry setting" << endl;
|
|
settings.remove("telemetry");
|
|
settings.sync();
|
|
}
|
|
else
|
|
{
|
|
cerr << "Ignoring unknown argument: " << args[i] << endl;
|
|
}
|
|
}
|
|
|
|
if (customQm.isEmpty())
|
|
{
|
|
#ifdef Q_OS_DARWIN
|
|
QString langcode = "en_GB";
|
|
CFArrayRef prefLangs = CFLocaleCopyPreferredLanguages();
|
|
if (CFArrayGetCount(prefLangs))
|
|
{
|
|
char buf[32] = {0};
|
|
CFStringRef strRef = (CFStringRef) CFArrayGetValueAtIndex(prefLangs, 0);
|
|
CFStringGetCString(strRef, buf, sizeof(buf), kCFStringEncodingUTF8);
|
|
langcode = buf;
|
|
langcode.replace('-', '_');
|
|
qDebug() << "OSX most preferred language:" << langcode;
|
|
}
|
|
|
|
CFRelease(prefLangs);
|
|
QLocale::setDefault(QLocale(langcode));
|
|
#endif
|
|
|
|
if (translator.load(QLocale(), "rpi-imager", "_", QLatin1String(":/i18n")))
|
|
QCoreApplication::installTranslator(&translator);
|
|
}
|
|
else
|
|
{
|
|
if (translator.load(customQm))
|
|
QCoreApplication::installTranslator(&translator);
|
|
}
|
|
|
|
if (!url.isEmpty())
|
|
imageWriter.setSrc(url);
|
|
imageWriter.setEngine(&engine);
|
|
engine.setNetworkAccessManagerFactory(&namf);
|
|
engine.rootContext()->setContextProperty("imageWriter", &imageWriter);
|
|
engine.rootContext()->setContextProperty("driveListModel", imageWriter.getDriveList());
|
|
engine.load(QUrl(QStringLiteral("qrc:/main.qml")));
|
|
|
|
if (engine.rootObjects().isEmpty())
|
|
return -1;
|
|
|
|
QObject *qmlwindow = engine.rootObjects().value(0);
|
|
qmlwindow->connect(&imageWriter, SIGNAL(downloadProgress(QVariant,QVariant)), qmlwindow, SLOT(onDownloadProgress(QVariant,QVariant)));
|
|
qmlwindow->connect(&imageWriter, SIGNAL(verifyProgress(QVariant,QVariant)), qmlwindow, SLOT(onVerifyProgress(QVariant,QVariant)));
|
|
qmlwindow->connect(&imageWriter, SIGNAL(preparationStatusUpdate(QVariant)), qmlwindow, SLOT(onPreparationStatusUpdate(QVariant)));
|
|
qmlwindow->connect(&imageWriter, SIGNAL(error(QVariant)), qmlwindow, SLOT(onError(QVariant)));
|
|
qmlwindow->connect(&imageWriter, SIGNAL(success()), qmlwindow, SLOT(onSuccess()));
|
|
qmlwindow->connect(&imageWriter, SIGNAL(fileSelected(QVariant)), qmlwindow, SLOT(onFileSelected(QVariant)));
|
|
qmlwindow->connect(&imageWriter, SIGNAL(cancelled()), qmlwindow, SLOT(onCancelled()));
|
|
qmlwindow->connect(&imageWriter, SIGNAL(finalizing()), qmlwindow, SLOT(onFinalizing()));
|
|
qmlwindow->connect(&imageWriter, SIGNAL(networkOnline()), qmlwindow, SLOT(fetchOSlist()));
|
|
|
|
#ifndef QT_NO_WIDGETS
|
|
/* Set window position */
|
|
auto screensize = app.primaryScreen()->geometry();
|
|
int x = settings.value("x", -1).toInt();
|
|
int y = settings.value("y", -1).toInt();
|
|
int w = qmlwindow->property("width").toInt();
|
|
int h = qmlwindow->property("height").toInt();
|
|
|
|
if (x != -1 && y != -1)
|
|
{
|
|
if ( !app.screenAt(QPoint(x,y)) || !app.screenAt(QPoint(x+w,y+h)) )
|
|
{
|
|
qDebug() << "Not restoring saved window position as it falls outside any currently attached screen";
|
|
x = y = -1;
|
|
}
|
|
}
|
|
|
|
if (x == -1 || y == -1)
|
|
{
|
|
x = qMax(1, (screensize.width()-w)/2);
|
|
y = qMax(1, (screensize.height()-h)/2);
|
|
}
|
|
|
|
qmlwindow->setProperty("x", x);
|
|
qmlwindow->setProperty("y", y);
|
|
#endif
|
|
|
|
int rc = app.exec();
|
|
|
|
#ifndef QT_NO_WIDGETS
|
|
int newX = qmlwindow->property("x").toInt();
|
|
int newY = qmlwindow->property("y").toInt();
|
|
if (x != newX || y != newY)
|
|
{
|
|
settings.setValue("x", newX);
|
|
settings.setValue("y", newY);
|
|
settings.sync();
|
|
}
|
|
#endif
|
|
|
|
return rc;
|
|
}
|
|
|