threeSD/src/frontend/main.cpp

// Copyright 2014 Citra Emulator Project / 2019 threeSD Project
// Licensed under GPLv2 or any later version
// Refer to the license.txt file included.

#include <algorithm>
#include <regex>
#include <string>
#include <QApplication>
#include <QFileDialog>
#include <QMessageBox>
#include <QStorageInfo>
#include <QTreeWidgetItem>
#include <qdevicewatcher.h>
#include "common/assert.h"
#include "common/file_util.h"
#include "frontend/import_dialog.h"
#include "frontend/main.h"
#include "frontend/select_nand_dialog.h"
#include "frontend/utilities.h"
#include "ui_main.h"

#ifdef __APPLE__
#include <unistd.h>
#include "common/common_paths.h"
#endif

#ifdef QT_STATICPLUGIN
#include <QtPlugin>

Q_IMPORT_PLUGIN(QWindowsIntegrationPlugin)
#if QT_VERSION >= QT_VERSION_CHECK(5, 10, 0)
Q_IMPORT_PLUGIN(QWindowsVistaStylePlugin)
#endif
#endif

MainDialog::MainDialog(QWidget* parent)
    : DPIAwareDialog(parent, 640, 256), ui(std::make_unique<Ui::MainDialog>()) {

    ui->setupUi(this);

    ui->buttonBox->button(QDialogButtonBox::StandardButton::Ok)->setEnabled(false);
    ui->buttonBox->button(QDialogButtonBox::StandardButton::Reset)->setText(tr("Refresh"));

    LoadPresetConfig();

    connect(ui->utilitiesButton, &QPushButton::clicked, [this] {
        UtilitiesDialog dialog(this);
        dialog.exec();
    });

    connect(ui->buttonBox, &QDialogButtonBox::clicked, [this](QAbstractButton* button) {
        if (button == ui->buttonBox->button(QDialogButtonBox::StandardButton::Reset)) {
            LoadPresetConfig();
        } else if (button == ui->buttonBox->button(QDialogButtonBox::StandardButton::Ok)) {
            LaunchImportDialog();
        }
    });

    QString destination_text{};
    const auto destination = FileUtil::GetUserPathType();
    if (destination == FileUtil::UserPathType::Normal) {
#ifdef __linux__
        destination_text = tr("Non-Flatpak Citra Install");
#else
        destination_text = tr("User-wide Citra Install");
#endif
    } else if (destination == FileUtil::UserPathType::Portable) {
        destination_text = tr("Portable Citra Install");
    } else if (destination == FileUtil::UserPathType::Flatpak) {
        destination_text = tr("Flatpak Citra Install");
    } else {
        UNREACHABLE();
    }
    ui->importDestination->setText(tr("Import Destination: %1").arg(destination_text));

    connect(ui->main, &QTreeWidget::itemSelectionChanged, [this] {
        ui->buttonBox->button(QDialogButtonBox::StandardButton::Ok)
            ->setEnabled(!ui->main->selectedItems().empty());
    });
    connect(ui->main, &QTreeWidget::itemDoubleClicked, [this] {
        if (!ui->main->selectedItems().empty())
            LaunchImportDialog();
    });

    ui->main->setIndentation(4);

    // Set up device watcher
    auto* device_watcher = new QDeviceWatcher(this);
    device_watcher->start();
    connect(device_watcher, &QDeviceWatcher::deviceAdded, this, &MainDialog::LoadPresetConfig);
    connect(device_watcher, &QDeviceWatcher::deviceChanged, this, &MainDialog::LoadPresetConfig);
    connect(device_watcher, &QDeviceWatcher::deviceRemoved, this, &MainDialog::LoadPresetConfig);
}

MainDialog::~MainDialog() = default;

void MainDialog::SetContentSizes(int previous_width, [[maybe_unused]] int previous_height) {
    const int current_width = width();
    if (previous_width == 0) { // first time
        ui->main->setColumnWidth(0, 0.3 * current_width);
        ui->main->setColumnWidth(1, 0.4 * current_width);
    } else { // proportionally update column widths
        for (int i : {0, 1}) {
            ui->main->setColumnWidth(i, ui->main->columnWidth(i) * current_width / previous_width);
        }
    }
}

static QString GetNANDText(const Core::Config& config, bool has_multiple_sdmc) {
    const bool has_sys = std::any_of(
        config.nands.begin(), config.nands.end(),
        [](const Core::Config::NandConfig& nand) { return nand.nand_name == Core::SysNANDName; });

    if (config.nands.size() > 1) {
        if (has_sys) {
            if (config.nands.size() > 2) {
                return MainDialog::tr("SysNAND, %1 EmuNAND(s)").arg(config.nands.size() - 1);
            } else {
                return MainDialog::tr("SysNAND, EmuNAND");
            }
        } else {
            return MainDialog::tr("%1 EmuNAND(s)").arg(config.nands.size());
        }
    } else if (has_multiple_sdmc) {
        if (has_sys) {
            return MainDialog::tr("SysNAND");
        } else {
            return MainDialog::tr("EmuNAND");
        }
    } else {
        return MainDialog::tr("OK");
    }
}

void MainDialog::LoadPresetConfig() {
    ui->main->clear();
    preset_config_list.clear();

    for (const auto& storage : QStorageInfo::mountedVolumes()) {
        if (!storage.isValid() || !storage.isReady()) {
            continue;
        }

        auto list = Core::LoadPresetConfig(storage.rootPath().toStdString());
        for (std::size_t i = 0; i < list.size(); ++i) {
            preset_config_list.emplace_back(list[i]);

            QString path = storage.rootPath();
            if (path.endsWith(QLatin1Char{'/'}) || path.endsWith(QLatin1Char{'\\'})) {
                path.remove(path.size() - 1, 1);
            }

            // Get ID0
            QString id0 = QString::fromStdString(list[i].id0);

            // Get status
            QString status = GetNANDText(list[i], list.size() > 1);
            if (list[i].version != Core::CurrentDumperVersion) {
                status = tr("Version Mismatch");
            } else if (!IsConfigGood(list[i])) {
                status = tr("No Configuration Found");
            } else if (!IsConfigComplete(list[i])) {
                status = tr("Missing System Files");
            }

            auto* item = new QTreeWidgetItem{{path, id0, status}};
            ui->main->invisibleRootItem()->addChild(item);
        }
    }

    auto* item = new QTreeWidgetItem{{tr("Browse SD Card Root...")}};
    item->setFirstColumnSpanned(true);
    ui->main->invisibleRootItem()->addChild(item);
}

void MainDialog::LaunchImportDialog() {
    auto* item = ui->main->currentItem();
    if (!item) {
        return;
    }

    Core::Config config;
    const auto index = ui->main->invisibleRootItem()->indexOfChild(item);
    if (index == ui->main->invisibleRootItem()->childCount() - 1) {
        const QString path = QFileDialog::getExistingDirectory(this, tr("Select SD Card Root"));
        if (path.isEmpty()) {
            return;
        }

        const auto& list = Core::LoadPresetConfig(path.toStdString());
        if (list.size() > 1) {
            QMessageBox::information(
                this, tr("threeSD"),
                tr("You have more than one 3DS data on your SD Card.\nthreeSD will "
                   "select the first one for you."));
        } else if (list.empty() || !IsConfigGood(list[0])) {
            QMessageBox::critical(
                this, tr("Error"),
                tr("Could not load configuration.<br>Please check if you have followed the <a "
                   "href='https://github.com/zhaowenlan1779/threeSD/wiki/Quickstart-Guide'>"
                   "guide</a> correctly."));
            return;
        }

        config = list[0];
    } else {
        config = preset_config_list.at(index);
    }

    // Display info regarding status
    if (config.version != Core::CurrentDumperVersion) {
        QMessageBox::critical(this, tr("Version Mismatch"),
                              tr("You are using an unsupported version of threeSDumper.<br>Please "
                                 "ensure that you are using the most recent version of both "
                                 "threeSD and threeSDumper and try again."));
        return;
    }
    if (!IsConfigGood(config)) {
        QMessageBox::critical(
            this, tr("Error"),
            tr("Could not load configuration from this SD card. You need to prepare your SD card "
               "before using threeSD.<br>Please check if you have followed the <a "
               "href='https://github.com/zhaowenlan1779/threeSD/wiki/Quickstart-Guide'>"
               "guide</a> correctly."));
        return;
    }
    if (!IsConfigComplete(config)) {
        QMessageBox::warning(
            this, tr("Warning"),
            tr("Certain system files are missing from your configuration.<br>Some contents "
               "may not be importable, or may not run.<br>Please check if you have followed the <a "
               "href='https://github.com/zhaowenlan1779/threeSD/wiki/Quickstart-Guide'>guide</a> "
               "correctly."));
    }

    if (config.nands.size() > 1) {
        SelectNandDialog dialog(this, config.nands);
        if (dialog.exec() != QDialog::Accepted) {
            return;
        }
        // Swap the selected NAND to the front
        std::swap(config.nands[0], config.nands[dialog.GetResult()]);
    }

    ImportDialog dialog(this, config);
    dialog.exec();
}

int main(int argc, char* argv[]) {
    // Init settings params
    QCoreApplication::setOrganizationName(QStringLiteral("zhaowenlan1779"));
    QCoreApplication::setApplicationName(QStringLiteral("threeSD"));

    Common::Logging::InitializeLogging();

#ifdef __APPLE__
    std::string bin_path = FileUtil::GetBundleDirectory() + DIR_SEP + "..";
    chdir(bin_path.c_str());
#endif

    QApplication app(argc, argv);

    QIcon::setThemeSearchPaths(QStringList(QStringLiteral(":/icons/default")));
    QIcon::setThemeName(QStringLiteral(":/icons/default"));

    MainDialog main_dialog;
    main_dialog.show();

    return app.exec();
}