#pragma once

#include "gc_cause.hpp"
#include "collector_type.hpp"
#include "art/thread.hpp"
#include "common.hpp"


namespace lsplant::art::gc {

class GCCriticalSection {
private:
    [[maybe_unused]] void *self_;
    [[maybe_unused]] const char *section_name_;
};

class ScopedGCCriticalSection {
    CREATE_MEM_FUNC_SYMBOL_ENTRY(void, constructor, ScopedGCCriticalSection *thiz, Thread *self,
                                 GcCause cause, CollectorType collector_type) {
        if (thiz == nullptr) [[unlikely]] return;
        if (constructorSym) [[likely]]
            return constructorSym(thiz, self, cause, collector_type);
    }

    CREATE_MEM_FUNC_SYMBOL_ENTRY(void, destructor, ScopedGCCriticalSection *thiz) {
        if (thiz == nullptr) [[unlikely]] return;
        if (destructorSym) [[likely]]
            return destructorSym(thiz);
    }

public:
    ScopedGCCriticalSection(Thread *self, GcCause cause, CollectorType collector_type) {
        constructor(this, self, cause, collector_type);
    }

    ~ScopedGCCriticalSection() {
        destructor(this);
    }

    static bool Init(const HookHandler &handler) {
        if (!RETRIEVE_MEM_FUNC_SYMBOL(constructor,
                                      "_ZN3art2gc23ScopedGCCriticalSectionC2EPNS_6ThreadENS0_7GcCauseENS0_13CollectorTypeE")) {
            return false;
        }
        if (!RETRIEVE_MEM_FUNC_SYMBOL(destructor, "_ZN3art2gc23ScopedGCCriticalSectionD2Ev")) {
            return false;
        }
        return true;
    }

private:
    [[maybe_unused]] GCCriticalSection critical_section_;
    [[maybe_unused]] const char *old_no_suspend_reason_;
};
}