AIO-switch-updater/source/payload_page.cpp

#include "payload_page.hpp"

#include "current_cfw.hpp"
#include "fs.hpp"
#include "utils.hpp"

namespace i18n = brls::i18n;
using namespace i18n::literals;

PayloadPage::PayloadPage() : AppletFrame(true, true)
{
    this->setTitle("menus/payloads/reboot_title"_i18n);
    list = new brls::List();
    label = new brls::Label(
        brls::LabelStyle::DESCRIPTION,
        "menus/payloads/select"_i18n,
        true);
    list->addView(label);
    std::vector<std::string> payloads = util::fetchPayloads();
    for (const auto& payload : payloads) {
        std::string payload_path = payload;
        listItem = new brls::ListItem(payload_path);
        listItem->getClickEvent()->subscribe([&, payload](brls::View* view) {
            util::rebootToPayload(payload);
            brls::Application::popView();
        });
        if (CurrentCfw::running_cfw == CFW::ams) {
            listItem->registerAction("menus/payloads/set_reboot_payload"_i18n, brls::Key::X, [this, payload_path] {
                std::string res1;
                if (fs::copyFile(payload_path, REBOOT_PAYLOAD_PATH)) {
                    res1 += "menus/payloads/copy_success"_i18n + payload_path + "menus/payloads/to"_i18n + std::string(REBOOT_PAYLOAD_PATH) + "'.";
                }
                else {
                    res1 += "Failed.";
                }
                brls::Dialog* dialog = new brls::Dialog(res1);
                brls::GenericEvent::Callback callback = [dialog](brls::View* view) {
                    dialog->close();
                };
                dialog->addButton("menus/common/ok"_i18n, callback);
                dialog->setCancelable(true);
                dialog->open();
                return true;
            });
        }
        listItem->registerAction("menus/payloads/set_update_bin"_i18n, brls::Key::Y, [this, payload] {
            std::string res2;
            if (fs::copyFile(payload, UPDATE_BIN_PATH)) {
                res2 += "menus/payloads/copy_success"_i18n + payload + "menus/payloads/to"_i18n + std::string(UPDATE_BIN_PATH) + "'.";
            }
            else {
                res2 += "Failed.";
            }
            brls::Dialog* dialog = new brls::Dialog(res2);
            brls::GenericEvent::Callback callback = [dialog](brls::View* view) {
                dialog->close();
            };
            dialog->addButton("menus/common/ok"_i18n, callback);
            dialog->setCancelable(true);
            dialog->open();
            return true;
        });
        list->addView(listItem);
    }
    list->addView(new brls::ListItemGroupSpacing(true));

    shutDown = new brls::ListItem("menus/common/shut_down"_i18n);
    shutDown->getClickEvent()->subscribe([](brls::View* view) {
        util::shutDown(false);
        brls::Application::popView();
    });
    list->addView(shutDown);

    reboot = new brls::ListItem("menus/common/reboot"_i18n);
    reboot->getClickEvent()->subscribe([](brls::View* view) {
        util::shutDown(true);
        brls::Application::popView();
    });
    list->addView(reboot);

    this->setContentView(list);
}