geode/loader/include/Geode/utils/NodeIDs.hpp

#pragma once
#include <Geode/modify/Modify.hpp>
#include "cocos.hpp"

namespace geode::node_ids {
    using namespace cocos2d;

    static constexpr int32_t GEODE_ID_PRIORITY = Priority::VeryEarlyPost;

    template <class T = CCNode>
        requires std::is_base_of_v<CCNode, T>
    T* setIDSafe(CCNode* node, int index, char const* id) {
        if constexpr (std::is_same_v<CCNode, T>) {
            if (auto child = cocos::getChild(node, index)) {
                child->setID(id);
                return child;
            }
        } else {
            if (auto child = node->getChildByType<T>(index)) {
                child->setID(id);
                return child;
            }
        }
        return nullptr;
    }

    template <typename ...Args>
    void setIDs(CCNode* node, int startIndex, Args... args) {
        for (auto i : { args... }) {
            setIDSafe(node, startIndex, i);
            ++startIndex;
        }
    }

    template <typename ...Args>
    void setIDs(CCNode* node, int* startIndex, Args... args) {
        for (auto i : { args... }) {
            if (setIDSafe(node, *startIndex, i)) {
                *startIndex += 1;
            }
        }
    }

    static void switchToMenu(CCNode* node, CCNode* menu) {
        if (!node || !menu) return;
        
        auto worldPos = node->getParent() ?
            node->getParent()->convertToWorldSpace(node->getPosition()) : 
            node->getPosition();

        node->retain();
        node->removeFromParent();
        node->setZOrder(0);

        menu->addChild(node);
        node->setPosition(menu->convertToNodeSpace(worldPos));
    }

    static void switchChildToMenu(CCNode* parent, int idx, CCNode* menu) {
        switchToMenu(static_cast<CCNode*>(parent->getChildren()->objectAtIndex(idx)), menu);
    }

    template <typename ...Args>
    static void switchChildrenToMenu(CCNode* parent, CCNode* menu, Args... args) {
        for (auto i : { args... }) {
            switchChildToMenu(parent, i, menu);
        }
    }

    template <class N, typename T, typename... Args>
    static N* detachAndCreateNode(
        CCNode* parent, char const* menuID, Layout* layout, T first, Args... args
    ) {
        if (!first) {
            auto menu = N::create();
            menu->setID(menuID);
            menu->setLayout(layout);
            parent->addChild(menu);
            return menu;
        }

        auto oldMenu = first->getParent();

        first->retain();
        first->removeFromParent();

        auto newMenu = N::create();
        newMenu->setPosition(parent->convertToNodeSpace(oldMenu->convertToWorldSpace(first->getPosition())));
        newMenu->setID(menuID);
        newMenu->setZOrder(oldMenu->getZOrder());
        parent->addChild(newMenu);

        first->setPosition(0, 0);
        first->setZOrder(0);
        newMenu->addChild(first);
        first->release();

        (switchToMenu(args, newMenu), ...);
        
        newMenu->setLayout(layout);

        return newMenu;
    }

    template <typename T, typename... Args>
    static CCMenu* detachAndCreateMenu(
        CCNode* parent, char const* menuID, Layout* layout, T first, Args... args
    ) {
        return detachAndCreateNode<CCMenu>(parent, menuID, layout, first, args...);
    }

    static CCSize getSizeSafe(CCNode* node) {
        if (node) {
            return node->getScaledContentSize();
        }
        else {
            return CCSizeZero;
        }
    }
}