#pragma once
#include <cstdint>

namespace rage {

class pgBase {
public:
    virtual void ShutdownClass() = 0;
    virtual void SetHandleIndex(uint32_t index) = 0;
    virtual void Validate(uint32_t arg) = 0;
    virtual void InitClass(int arg) = 0;
    virtual uint32_t GetHandleIndex() const { return 0; }

    uintptr_t LookupMetaData(uint32_t index) const {
        uintptr_t current = reinterpret_cast<uintptr_t>(this);

        uintptr_t nextPage = *reinterpret_cast<uintptr_t*>(current + 8);
        if (nextPage == 0)
            return 0;

        uint8_t pageStatus = *reinterpret_cast<uint8_t*>(nextPage + 11);
        if (pageStatus == 0)
            return 0;

        uintptr_t* metadataStart = reinterpret_cast<uintptr_t*>(nextPage + 16);
        uintptr_t metadataCount = *reinterpret_cast<uintptr_t*>(nextPage + 24);

        for (uintptr_t i = 0; i < metadataCount; ++i) {
            uintptr_t currentMeta = *(metadataStart + i);

            uint32_t metaIndex = *reinterpret_cast<uint32_t*>(currentMeta + 8);
            if (metaIndex == index) {
                return currentMeta;
            }
        }

        return 0;
    }

    bool HasPageMap() const {
        uintptr_t nextPage = *reinterpret_cast<uintptr_t*>(reinterpret_cast<uintptr_t>(this) + 8);
        if (nextPage == 0)
            return false;

        uint8_t pageStatus = *reinterpret_cast<uint8_t*>(nextPage + 11);
        if (pageStatus == 0)
            return false;

        nextPage = 0;
        bool hasPageMap = (nextPage != 0);

        return hasPageMap;
    }

    bool MapContainsPointer(void* ptr) const {
        uintptr_t basePage = reinterpret_cast<uintptr_t>(this);
        uintptr_t nextPage = *reinterpret_cast<uintptr_t*>(basePage + 8);
        if (nextPage == 0)
            return false;

        uint8_t pageStatus = *reinterpret_cast<uint8_t*>(nextPage + 11);
        if (pageStatus == 0)
            return false;

        uintptr_t* data = reinterpret_cast<uintptr_t*>(nextPage + 16);
        uintptr_t count = *reinterpret_cast<uintptr_t*>(nextPage + 8);

        for (uintptr_t i = 0; i < count; ++i) {
            uintptr_t elementPage = *(data + i);
            if (elementPage >= reinterpret_cast<uintptr_t>(ptr) && elementPage < reinterpret_cast<uintptr_t>(ptr) + 24) {
                return true;
            }
        }

        return false;
    }

private:
    uint32_t m_MapSize;
};

class pgBaseRefCounted : public pgBase {
public:
    virtual ~pgBaseRefCounted() = default;
};

class pgBaseMetaDataType {
public:
    static pgBaseMetaDataType* m_First;

    pgBaseMetaDataType(uint32_t id, uintptr_t value)
        : m_Id(id), m_Value(value), m_Next(nullptr) {
        if (!m_First) {
            m_First = this;
        }
    }

    ~pgBaseMetaDataType() {
        if (m_First == this) {
            m_First = m_Next;
        } else {
            pgBaseMetaDataType* current = m_First;
            while (current && current->m_Next != this) {
                current = current->m_Next;
            }
            if (current) {
                current->m_Next = m_Next;
            }
        }
    }

    static pgBaseMetaDataType* Lookup(uint32_t id) {
        pgBaseMetaDataType* current = m_First;
        while (current != nullptr) {
            if (current->m_Id == id) {
                return current;
            }
            current = current->m_Next;
        }
        return nullptr;
    }

private:
    uint32_t m_Id;
    uintptr_t m_Value;
    pgBaseMetaDataType* m_Next;
};

}