geode/loader/src/hooks/LoadingLayer.cpp

#include <Geode/modify/LoadingLayer.hpp>
#include <array>
#include <InternalLoader.hpp>

USE_GEODE_NAMESPACE();

class $modify(CustomLoadingLayer, LoadingLayer) {
    CCScale9Sprite* m_updatingResourcesBG;
    CCLabelBMFont* m_updatingResources;

    CustomLoadingLayer() : m_updatingResources(nullptr) {}

    bool init(bool fromReload) {
        if (!LoadingLayer::init(fromReload))
            return false;

        auto winSize = CCDirector::sharedDirector()->getWinSize();

        auto count = Loader::get()->getAllMods().size();

        auto label = CCLabelBMFont::create(
            CCString::createWithFormat(
                "Geode: Loaded %lu mods",
                static_cast<unsigned long>(count)
            )->getCString(),
            "goldFont.fnt"
        );
        label->setPosition(winSize.width / 2, 30.f);
        label->setScale(.45f);
        label->setID("geode-loaded-info");
        this->addChild(label);

        // verify loader resources
        if (!InternalLoader::get()->verifyLoaderResources(
            std::bind(
                &CustomLoadingLayer::updateResourcesProgress, this,
                std::placeholders::_1, std::placeholders::_2,
                std::placeholders::_3
            )
        )) {
            auto bg = CCScale9Sprite::create(
                "square02b_001.png", { 0.0f, 0.0f, 80.0f, 80.0f }
            );
            bg->setScale(.6f);
            bg->setColor({ 0, 0, 0 });
            bg->setOpacity(150);
            bg->setPosition(winSize / 2);
            this->addChild(bg);

            m_fields->m_updatingResourcesBG = bg;

            auto label = CCLabelBMFont::create("", "goldFont.fnt");
            label->setScale(1.1f);
            bg->addChild(label);

            m_fields->m_updatingResources = label;

            this->setUpdateText("Downloading Resources");
        }

        return true;
    }

    void setUpdateText(std::string const& text) {
        m_fields->m_updatingResources->setString(text.c_str());
        m_fields->m_updatingResourcesBG->setContentSize({
            m_fields->m_updatingResources->getScaledContentSize().width + 30.f,
            50.f
        });
        m_fields->m_updatingResources->setPosition(
            m_fields->m_updatingResourcesBG->getContentSize() / 2
        );
    }

    void updateResourcesProgress(
        UpdateStatus status,
        std::string const& info,
        uint8_t progress
    ) {
        switch (status) {
            case UpdateStatus::Progress: {
                this->setUpdateText(
                    "Downloading Resources: " +
                    std::to_string(progress) + "%"
                );
            } break;

            case UpdateStatus::Finished: {
                this->setUpdateText("Resources Downloaded");
                m_fields->m_updatingResourcesBG->runAction(CCSequence::create(
                    CCDelayTime::create(1.f),
                    CCRemoveSelf::create(),
                    nullptr
                ));
                m_fields->m_updatingResourcesBG = nullptr;
                m_fields->m_updatingResources = nullptr;
                this->loadAssets();
            } break;

            case UpdateStatus::Failed: {
                InternalLoader::platformMessageBox(
                    "Error updating resources",
                    "Unable to update Geode resources: " + info + ".\n"
                    "The game will be loaded as normal, but please be aware "
                    "that it may very likely crash."
                );
                this->setUpdateText("Resource Download Failed");
                m_fields->m_updatingResourcesBG->runAction(CCSequence::create(
                    CCDelayTime::create(1.f),
                    CCRemoveSelf::create(),
                    nullptr
                ));
                m_fields->m_updatingResourcesBG = nullptr;
                m_fields->m_updatingResources = nullptr;
                this->loadAssets();
            } break;
        }
    }

	void loadAssets() {
		if (m_fields->m_updatingResources) {
            return;
		}
		LoadingLayer::loadAssets();
	}
};