#include "MainWindow.hpp" #include "ui_MainWindow.h" #include #include #include #include #include "EscapeSequenceParser.hpp" #include "FileDirDialog.hpp" #include "ExtractZip.hpp" #if _WIN32 #include #include #include static void KillProcessTree(QProcess& proc) { Q_PID pid = proc.pid(); if (!pid) return; DWORD myprocID = pid->dwProcessId; PROCESSENTRY32 pe = {}; pe.dwSize = sizeof(PROCESSENTRY32); HANDLE hSnap = ::CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0); if (::Process32First(hSnap, &pe)) { BOOL bContinue = TRUE; // kill child processes while (bContinue) { // only kill child processes if (pe.th32ParentProcessID == myprocID) { HANDLE hChildProc = ::OpenProcess(PROCESS_ALL_ACCESS, FALSE, pe.th32ProcessID); if (hChildProc) { ::TerminateProcess(hChildProc, 1); ::CloseHandle(hChildProc); } } bContinue = ::Process32Next(hSnap, &pe); } } proc.close(); proc.terminate(); } #else static void KillProcessTree(QProcess& proc) { proc.close(); proc.terminate(); } #endif const QStringList MainWindow::skUpdateTracks = {QStringLiteral("stable"), QStringLiteral("dev")}; MainWindow::MainWindow(QWidget* parent) : QMainWindow(parent) , m_ui(std::make_unique()) , m_fileMgr(_SYS_STR("urde")) , m_cvarManager(m_fileMgr) , m_cvarCommons(m_cvarManager) , m_heclProc(this) , m_dlManager(this) , m_launchMenu(m_cvarCommons, this) { if (m_settings.value(QStringLiteral("urde_arguments")).isNull()) { m_settings.setValue(QStringLiteral("urde_arguments"), QStringList{QStringLiteral("--no-shader-warmup")}); } if (m_settings.value(QStringLiteral("update_track")).isNull()) { m_settings.setValue(QStringLiteral("update_track"), QStringLiteral("stable")); } m_ui->setupUi(this); m_ui->heclTabs->setCurrentIndex(0); m_ui->aboutIcon->setPixmap(QApplication::windowIcon().pixmap(256, 256)); QFont mFont = QFontDatabase::systemFont(QFontDatabase::FixedFont); mFont.setPointSize(m_ui->currentBinaryLabel->font().pointSize()); m_ui->currentBinaryLabel->setFont(mFont); m_ui->recommendedBinaryLabel->setFont(mFont); mFont.setPointSize(10); m_ui->processOutput->setFont(mFont); m_cursor = QTextCursor(m_ui->processOutput->document()); m_updateURDEButton = new QPushButton(tr("Update URDE"), m_ui->centralwidget); m_ui->gridLayout->addWidget(m_updateURDEButton, 2, 3, 1, 1); m_updateURDEButton->hide(); QPalette pal = m_updateURDEButton->palette(); pal.setColor(QPalette::Button, QColor(53, 53, 72)); m_updateURDEButton->setPalette(pal); connect(m_updateURDEButton, &QPushButton::clicked, this, &MainWindow::onUpdateURDEPressed); qDebug() << "Stored track " << m_settings.value(QStringLiteral("update_track")); const int index = skUpdateTracks.indexOf(m_settings.value(QStringLiteral("update_track")).toString()); m_ui->devTrackWarning->setVisible(index == 1); m_ui->updateTrackComboBox->setCurrentIndex(index); connect(m_ui->updateTrackComboBox, qOverload(&QComboBox::currentIndexChanged), this, &MainWindow::onUpdateTrackChanged); m_dlManager.connectWidgets(m_ui->downloadProgressBar, m_ui->downloadErrorLabel, std::bind(&MainWindow::onIndexDownloaded, this, std::placeholders::_1), std::bind(&MainWindow::onBinaryDownloaded, this, std::placeholders::_1), std::bind(&MainWindow::onBinaryFailed, this)); initSlots(); m_dlManager.fetchIndex(); setPath(m_settings.value(QStringLiteral("working_dir")).toString()); } MainWindow::~MainWindow() { KillProcessTree(m_heclProc); } void MainWindow::onExtract() { if (m_path.isEmpty()) { return; } const QString imgPath = QFileDialog::getOpenFileName(this, tr("Extract Image"), m_path, tr("Images (*.iso *.wbfs *.gcm)")); if (imgPath.isEmpty()) { return; } m_ui->processOutput->clear(); KillProcessTree(m_heclProc); m_heclProc.setProcessChannelMode(QProcess::ProcessChannelMode::MergedChannels); m_heclProc.setWorkingDirectory(m_path); QProcessEnvironment env = QProcessEnvironment::systemEnvironment(); env.insert(QStringLiteral("TERM"), QStringLiteral("xterm-color")); env.insert(QStringLiteral("ConEmuANSI"), QStringLiteral("ON")); m_heclProc.setProcessEnvironment(env); disconnect(&m_heclProc, qOverload(&QProcess::finished), nullptr, nullptr); connect(&m_heclProc, qOverload(&QProcess::finished), this, &MainWindow::onExtractFinished); const QStringList heclProcArguments{ QStringLiteral("extract"), QStringLiteral("-y"), QStringLiteral("-g"), QStringLiteral("-o"), m_path, imgPath}; m_heclProc.start(m_heclPath, heclProcArguments, QIODevice::ReadOnly | QIODevice::Unbuffered); m_ui->heclTabs->setCurrentIndex(0); disableOperations(); m_ui->extractBtn->setText(tr("Cancel")); m_ui->extractBtn->setEnabled(true); disconnect(m_ui->extractBtn, &QPushButton::clicked, nullptr, nullptr); connect(m_ui->extractBtn, &QPushButton::clicked, this, &MainWindow::doHECLTerminate); } void MainWindow::onExtractFinished(int returnCode, QProcess::ExitStatus) { m_cursor.movePosition(QTextCursor::End); m_cursor.insertBlock(); disconnect(m_ui->extractBtn, &QPushButton::clicked, nullptr, nullptr); connect(m_ui->extractBtn, &QPushButton::clicked, this, &MainWindow::onExtract); checkDownloadedBinary(); } void MainWindow::onPackage() { if (m_path.isEmpty()) return; m_ui->processOutput->clear(); KillProcessTree(m_heclProc); m_heclProc.setProcessChannelMode(QProcess::ProcessChannelMode::MergedChannels); m_heclProc.setWorkingDirectory(m_path); QProcessEnvironment env = QProcessEnvironment::systemEnvironment(); env.insert(QStringLiteral("TERM"), QStringLiteral("xterm-color")); env.insert(QStringLiteral("ConEmuANSI"), QStringLiteral("ON")); m_heclProc.setProcessEnvironment(env); disconnect(&m_heclProc, qOverload(&QProcess::finished), nullptr, nullptr); connect(&m_heclProc, qOverload(&QProcess::finished), this, &MainWindow::onPackageFinished); const QStringList heclProcArguments{QStringLiteral("package"), QStringLiteral("-y"), QStringLiteral("-g")}; m_heclProc.start(m_heclPath, heclProcArguments, QIODevice::ReadOnly | QIODevice::Unbuffered); m_ui->heclTabs->setCurrentIndex(0); disableOperations(); m_ui->packageBtn->setText(tr("Cancel")); m_ui->packageBtn->setEnabled(true); disconnect(m_ui->packageBtn, &QPushButton::clicked, nullptr, nullptr); connect(m_ui->packageBtn, &QPushButton::clicked, this, &MainWindow::doHECLTerminate); QSize size = QWidget::size(); if (size.width() < 1100) size.setWidth(1100); resize(size); } void MainWindow::onPackageFinished(int returnCode, QProcess::ExitStatus) { m_cursor.movePosition(QTextCursor::End); m_cursor.insertBlock(); disconnect(m_ui->packageBtn, &QPushButton::clicked, nullptr, nullptr); connect(m_ui->packageBtn, &QPushButton::clicked, this, &MainWindow::onPackage); checkDownloadedBinary(); } void MainWindow::onLaunch() { if (m_path.isEmpty()) return; m_ui->processOutput->clear(); KillProcessTree(m_heclProc); m_heclProc.setProcessChannelMode(QProcess::ProcessChannelMode::MergedChannels); m_heclProc.setWorkingDirectory(m_path); QProcessEnvironment env = QProcessEnvironment::systemEnvironment(); env.insert(QStringLiteral("TERM"), QStringLiteral("xterm-color")); env.insert(QStringLiteral("ConEmuANSI"), QStringLiteral("ON")); m_heclProc.setProcessEnvironment(env); disconnect(&m_heclProc, qOverload(&QProcess::finished), nullptr, nullptr); connect(&m_heclProc, qOverload(&QProcess::finished), this, &MainWindow::onLaunchFinished); const auto heclProcArguments = QStringList{m_path + QStringLiteral("/out")} << m_settings.value(QStringLiteral("urde_arguments")) .toStringList() .join(QLatin1Char{' '}) .split(QLatin1Char{' '}); m_heclProc.start(m_urdePath, heclProcArguments, QIODevice::ReadOnly | QIODevice::Unbuffered); m_ui->heclTabs->setCurrentIndex(0); disableOperations(); } void MainWindow::onLaunchFinished(int returnCode, QProcess::ExitStatus) { m_cursor.movePosition(QTextCursor::End); m_cursor.insertBlock(); checkDownloadedBinary(); } void MainWindow::doHECLTerminate() { KillProcessTree(m_heclProc); } void MainWindow::onReturnPressed() { if (sender() == m_ui->pathEdit) setPath(m_ui->pathEdit->text()); } void MainWindow::onIndexDownloaded(const QStringList& index) { int bestVersion = 0; m_ui->binaryComboBox->clear(); for (const QString& str : index) { URDEVersion version(str); if (m_ui->sysReqTable->willRun(version)) bestVersion = m_ui->binaryComboBox->count(); m_ui->binaryComboBox->addItem(version.fileString(false), QVariant::fromValue(version)); } m_ui->binaryComboBox->setCurrentIndex(bestVersion); m_recommendedVersion = m_ui->binaryComboBox->itemData(bestVersion).value(); m_ui->recommendedBinaryLabel->setText(m_recommendedVersion.fileString(false)); m_ui->binaryComboBox->setEnabled(true); if (!m_path.isEmpty()) { checkDownloadedBinary(); m_ui->downloadButton->setEnabled(true); } } void MainWindow::onDownloadPressed() { m_updateURDEButton->hide(); QString filename = m_ui->binaryComboBox->currentData().value().fileString(true); disableOperations(); m_ui->downloadButton->setEnabled(false); m_dlManager.fetchBinary(filename, m_path + QLatin1Char{'/'} + filename); } void MainWindow::onUpdateURDEPressed() { m_ui->heclTabs->setCurrentIndex(1); onDownloadPressed(); } void MainWindow::onBinaryDownloaded(QuaZip& file) { const bool err = !ExtractZip::extractDir(file, m_path); if (err) { m_ui->downloadErrorLabel->setText(tr("Error extracting zip")); } else { m_ui->downloadErrorLabel->setText(tr("Download successful"), true); } m_ui->downloadButton->setEnabled(true); checkDownloadedBinary(); if (!err && m_ui->extractBtn->isEnabled()) { m_ui->downloadErrorLabel->setText(tr("Download successful - Press 'Extract' to continue."), true); } if (!err && !m_ui->sysReqTable->isBlenderVersionOk()) { m_ui->downloadErrorLabel->setText(tr("Blender 2.80+ must be installed. Please download via Steam or blender.org.")); } } void MainWindow::onBinaryFailed() { m_ui->downloadButton->setEnabled(true); checkDownloadedBinary(); } void MainWindow::disableOperations() { m_ui->extractBtn->setEnabled(false); m_ui->packageBtn->setEnabled(false); m_ui->launchBtn->setEnabled(false); m_ui->pathEdit->setEnabled(false); m_ui->browseBtn->setEnabled(false); m_ui->downloadButton->setEnabled(false); } void MainWindow::enableOperations() { disableOperations(); m_ui->pathEdit->setEnabled(true); m_ui->browseBtn->setEnabled(true); if (m_path.isEmpty()) return; m_ui->downloadButton->setEnabled(true); if (m_heclPath.isEmpty()) return; m_ui->extractBtn->setText(tr("Extract")); m_ui->packageBtn->setText(tr("Package")); m_ui->launchBtn->setText(tr("Launch")); m_ui->extractBtn->setEnabled(true); if (QFile::exists(m_path + QStringLiteral("/MP1/!original_ids.yaml"))) { m_ui->packageBtn->setEnabled(true); if (isPackageComplete()) m_ui->launchBtn->setEnabled(true); } if (!m_ui->sysReqTable->isBlenderVersionOk()) { insertContinueNote(tr("Blender 2.80+ must be installed. Please download via Steam or blender.org.")); } else if (m_ui->launchBtn->isEnabled()) { insertContinueNote(tr("Package complete - Press 'Launch' to start URDE.")); } else if (m_ui->packageBtn->isEnabled()) { insertContinueNote(tr("Extract complete - Press 'Package' to continue.")); } else if (m_ui->extractBtn->isEnabled()) { insertContinueNote(tr("Press 'Extract' to begin.")); } } bool MainWindow::isPackageComplete() const { return #if RUNTIME_ORIGINAL_IDS QFile::exists(m_path + QStringLiteral("/out/files/!original_ids.upak")) && #endif QFile::exists(m_path + QStringLiteral("/out/files/AudioGrp.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/GGuiSys.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/Metroid1.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/Metroid2.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/Metroid3.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/Metroid4.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/metroid5.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/Metroid6.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/Metroid7.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/Metroid8.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/MidiData.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/MiscData.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/NoARAM.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/SamGunFx.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/SamusGun.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/SlideShow.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/TestAnim.upak")) && QFile::exists(m_path + QStringLiteral("/out/files/Tweaks.upak")); } static bool GetDLPackage(const QString& path, QString& dlPackage) { QProcess proc; proc.start(path, {QStringLiteral("--dlpackage")}, QIODevice::ReadOnly); if (proc.waitForStarted()) { proc.waitForFinished(); if (proc.exitCode() == 100) dlPackage = QString::fromUtf8(proc.readLine()).trimmed(); return true; } return false; } bool MainWindow::checkDownloadedBinary() { m_updateURDEButton->hide(); m_urdePath = QString(); m_heclPath = QString(); if (m_path.isEmpty()) { m_ui->heclTabs->setCurrentIndex(1); m_ui->downloadErrorLabel->setText(tr("Set working directory to continue."), true); enableOperations(); return false; } #if __APPLE__ QString urdePath = m_path + QStringLiteral("/URDE.app/Contents/MacOS/urde"); QString heclPath = m_path + QStringLiteral("/URDE.app/Contents/MacOS/hecl"); QString visigenPath = m_path + QStringLiteral("/URDE.app/Contents/MacOS/visigen"); #elif _WIN32 QString urdePath = m_path + QStringLiteral("/urde.exe"); QString heclPath = m_path + QStringLiteral("/hecl.exe"); QString visigenPath = m_path + QStringLiteral("/visigen.exe"); #else QString urdePath = m_path + QStringLiteral("/urde"); QString heclPath = m_path + QStringLiteral("/hecl"); QString visigenPath = m_path + QStringLiteral("/visigen"); #endif urdePath = QFileInfo(urdePath).absoluteFilePath(); heclPath = QFileInfo(heclPath).absoluteFilePath(); visigenPath = QFileInfo(visigenPath).absoluteFilePath(); QString urdeDlPackage, heclDlPackage, visigenDlPackage; if (GetDLPackage(urdePath, urdeDlPackage) && GetDLPackage(heclPath, heclDlPackage) && GetDLPackage(visigenPath, visigenDlPackage)) { if (!urdeDlPackage.isEmpty() && urdeDlPackage == heclDlPackage && urdeDlPackage == visigenDlPackage) { URDEVersion v(urdeDlPackage); m_ui->currentBinaryLabel->setText(v.fileString(false)); if (m_recommendedVersion.isValid() && v.isValid() && m_recommendedVersion.getVersion() > v.getVersion()) { m_updateURDEButton->show(); } } else { m_ui->currentBinaryLabel->setText(tr("unknown -- re-download recommended")); } m_urdePath = urdePath; m_heclPath = heclPath; m_ui->downloadErrorLabel->setText({}, true); enableOperations(); return true; } else { m_ui->currentBinaryLabel->setText(tr("none")); m_ui->heclTabs->setCurrentIndex(1); m_ui->downloadErrorLabel->setText(tr("Press 'Download' to fetch latest URDE binary."), true); enableOperations(); } return false; } void MainWindow::setPath(const QString& path) { const QFileInfo finfo(path); QString usePath; if (!path.isEmpty()) { usePath = finfo.absoluteFilePath(); } if (!usePath.isEmpty() && !finfo.exists()) { if (QMessageBox::question(this, tr("Make Directory"), tr("%1 does not exist. Create it now?").arg(usePath)) == QMessageBox::Yes) { QDir().mkpath(usePath); } else { usePath = QString(); } } if (!usePath.isEmpty() && !finfo.isDir()) { QMessageBox::warning(this, tr("Directory Error"), tr("%1 is not a directory").arg(usePath), QMessageBox::Ok, QMessageBox::NoButton); usePath = QString(); } m_path = usePath; m_settings.setValue(QStringLiteral("working_dir"), m_path); if (!m_path.isEmpty()) { m_ui->pathEdit->setText(m_path); m_ui->downloadButton->setToolTip(QString()); m_ui->downloadButton->setEnabled(m_ui->binaryComboBox->isEnabled()); } else { m_ui->downloadButton->setToolTip(tr("Working directory must be set")); m_ui->downloadButton->setEnabled(false); m_ui->currentBinaryLabel->setText(tr("none")); } m_ui->sysReqTable->updateFreeDiskSpace(m_path); checkDownloadedBinary(); } void MainWindow::initSlots() { connect(&m_heclProc, &QProcess::readyRead, [=]() { const QByteArray bytes = m_heclProc.readAll(); setTextTermFormatting(QString::fromUtf8(bytes)); }); connect(m_ui->extractBtn, &QPushButton::clicked, this, &MainWindow::onExtract); connect(m_ui->packageBtn, &QPushButton::clicked, this, &MainWindow::onPackage); connect(m_ui->launchBtn, &QPushButton::clicked, this, &MainWindow::onLaunch); m_ui->launchMenuBtn->setMenu(&m_launchMenu); connect(m_ui->browseBtn, &QPushButton::clicked, [=]() { FileDirDialog dialog(this); dialog.setDirectory(m_path); dialog.setWindowTitle(tr("Select Working Directory")); int res = dialog.exec(); if (res == QFileDialog::Rejected) return; if (dialog.selectedFiles().size() <= 0) return; setPath(dialog.selectedFiles().at(0)); }); connect(m_ui->downloadButton, &QPushButton::clicked, this, &MainWindow::onDownloadPressed); } void MainWindow::setTextTermFormatting(const QString& text) { m_inContinueNote = false; QRegExp const escapeSequenceExpression(QStringLiteral(R"(\x1B\[([\d;\?FA]+)([mlh]?))")); QTextCharFormat defaultTextCharFormat = m_cursor.charFormat(); int offset = escapeSequenceExpression.indexIn(text); ReturnInsert(m_cursor, text.mid(0, offset)); QTextCharFormat textCharFormat = defaultTextCharFormat; while (offset >= 0) { int previousOffset = offset + escapeSequenceExpression.matchedLength(); QStringList captures = escapeSequenceExpression.capturedTexts(); if (captures.size() >= 3 && captures[2] == QStringLiteral("m")) { QStringList capturedTexts = captures[1].split(QLatin1Char{';'}); QListIterator i(capturedTexts); while (i.hasNext()) { bool ok = false; int attribute = i.next().toInt(&ok); Q_ASSERT(ok); ParseEscapeSequence(attribute, i, textCharFormat, defaultTextCharFormat); } } else if (captures.size() >= 2 && (captures[1].endsWith(QLatin1Char{'F'}) || captures[1].endsWith(QLatin1Char{'A'}))) { int lineCount = captures[1].chopped(1).toInt(); if (!lineCount) lineCount = 1; for (int i = 0; i < lineCount; ++i) { m_cursor.movePosition(QTextCursor::PreviousBlock); m_cursor.select(QTextCursor::BlockUnderCursor); m_cursor.removeSelectedText(); m_cursor.insertBlock(); } } offset = escapeSequenceExpression.indexIn(text, previousOffset); if (offset < 0) { ReturnInsert(m_cursor, text.mid(previousOffset), textCharFormat); } else { ReturnInsert(m_cursor, text.mid(previousOffset, offset - previousOffset), textCharFormat); } } m_cursor.setCharFormat(defaultTextCharFormat); m_ui->processOutput->ensureCursorVisible(); } void MainWindow::insertContinueNote(const QString& text) { if (m_inContinueNote) return; m_inContinueNote = true; m_cursor.movePosition(QTextCursor::End); QTextCharFormat textCharFormat = m_cursor.charFormat(); textCharFormat.setForeground(QColor(0, 255, 0)); m_cursor.insertText(text, textCharFormat); m_cursor.insertBlock(); m_ui->processOutput->ensureCursorVisible(); } void MainWindow::onUpdateTrackChanged(int index) { qDebug() << "Track changed from " << m_settings.value(QStringLiteral("update_track")) << " to " << skUpdateTracks[index]; m_settings.setValue(QStringLiteral("update_track"), skUpdateTracks[index]); m_dlManager.fetchIndex(); m_ui->devTrackWarning->setVisible(index == 1); }