#include "payload_page.hpp"
#include "utils.hpp"
#include "reboot_payload.h"
#include "current_cfw.hpp"
#include "utils.hpp"
#include "fs.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) {
            reboot_to_payload(payload.c_str());
            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.c_str(), 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_reboot_payload"_i18n, brls::Key::Y, [this, payload] { 
                std::string res2;
                if(fs::copyFile(payload.c_str(), 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/payloads/reboot"_i18n);
    reboot->getClickEvent()->subscribe([](brls::View* view) {
        util::shutDown(true);
        brls::Application::popView();
    });
    list->addView(reboot);

    this->setContentView(list);

}