Ventoy/LinuxGUI/Ventoy2Disk/QT/main.cpp
2021-09-22 14:44:50 +08:00

122 lines
2.9 KiB
C++

#include "ventoy2diskwindow.h"
#include <QApplication>
#include <QMessageBox>
#include <QFileInfo>
#include <QStyle>
#include <QDir>
#include <QDesktopWidget>
#include <QPixmap>
#include <unistd.h>
#include <sys/types.h>
extern "C" {
#include "ventoy_define.h"
#include "ventoy_util.h"
#include "ventoy_qt.h"
}
using namespace std;
char g_log_file[4096];
char g_ini_file[4096];
int main(int argc, char *argv[])
{
int ret;
long long size;
QApplication a(argc, argv);
Ventoy2DiskWindow w;
#ifdef QT_CHECK_EUID
if (geteuid() != 0)
{
QMessageBox::critical(NULL, "Error", "Permission denied!\nPlease run with root privileges.");
return 1;
}
#endif
if (!QFileInfo::exists("./boot/boot.img"))
{
QString curdir = a.applicationDirPath();
int index = curdir.indexOf("/tool/");
if (index >= 0)
{
QDir::setCurrent(curdir.left(index));
}
else
{
QDir::setCurrent(curdir);
}
}
if (!QFileInfo::exists("./boot/boot.img"))
{
QMessageBox::critical(NULL, "Error", "Please run under the correct directory.");
return 1;
}
ventoy_log_init();
snprintf(g_log_file, sizeof(g_log_file), "./log.txt");
snprintf(g_ini_file, sizeof(g_ini_file), "./Ventoy2Disk.ini");
for (int i = 0; i < argc; i++)
{
if (argv[i] && argv[i + 1] && strcmp(argv[i], "-l") == 0)
{
snprintf(g_log_file, sizeof(g_log_file), "%s", argv[i + 1]);
}
else if (argv[i] && argv[i + 1] && strcmp(argv[i], "-i") == 0)
{
snprintf(g_ini_file, sizeof(g_ini_file), "%s", argv[i + 1]);
}
}
QFileInfo Info(g_log_file);
size = (long long)Info.size();
if (size >= 4 * SIZE_1MB)
{
QFile::remove(g_log_file);
}
vlog("===================================================\n");
vlog("===== Ventoy2Disk %s powered by QT %s =====\n", ventoy_get_local_version(), qVersion());
vlog("===================================================\n");
vlog("log file is <%s> lastsize:%lld\n", g_log_file, (long long)size);
vlog("ini file is <%s>\n", g_ini_file);
ventoy_disk_init();
ventoy_http_init();
w.setGeometry(QStyle::alignedRect(Qt::LeftToRight,
Qt::AlignCenter,
w.size(),
a.desktop()->availableGeometry()));
QPixmap pix;
QIcon icon;
int len;
const uchar *data;
data = (const uchar *)get_window_icon_raw_data(&len);
pix.loadFromData(data, len);
icon.addPixmap(pix);
w.setWindowIcon(icon);
w.show();
w.setFixedSize(w.size());
ret = a.exec();
ventoy_disk_exit();
ventoy_http_exit();
vlog("######## Ventoy2Disk QT %s exit ########\n", ventoy_get_local_version());
/* log exit must at the end */
ventoy_log_exit();
return ret;
}