/*
 *      ========================================================
 *
 *      hCore
 *      Copyright 2024 Mahrouss Logic, all rights reserved.
 *
 *      ========================================================
 */

#ifndef __KERNELKIT_SHARED_OBJECT_CORE_HXX__
#define __KERNELKIT_SHARED_OBJECT_CORE_HXX__

#include "CodeManager.hpp"
#include "PEF.hpp"
#include <KernelKit/Loader.hpp>
#include <NewKit/Defines.hpp>

namespace hCore
{
class SharedObject final
{
  public:
    struct SharedObjectTraits final
    {
        VoidPtr fImageObject;
        VoidPtr fImageEntrypointOffset;
    };

  public:
    explicit SharedObject() = default;
    ~SharedObject() = default;

  public:
    HCORE_COPY_DEFAULT(SharedObject);

  private:
    SharedObjectTraits *fMounted{nullptr};

  public:
    SharedObjectTraits **GetAddressOf()
    {
        return &fMounted;
    }
    SharedObjectTraits *Get()
    {
        return fMounted;
    }

  public:
    void Mount(SharedObjectTraits *to_mount)
    {
        fMounted = to_mount;

        if (fLoader && to_mount)
        {
            delete fLoader;
            fLoader = nullptr;
        }

        if (!fLoader)
        {
            fLoader = new PEFLoader(fMounted->fImageObject);
        }
    }
    void Unmount()
    {
        if (fMounted)
            fMounted = nullptr;
    };

    template <typename SymbolType> SymbolType Load(const char *symbol_name)
    {
        auto ret = reinterpret_cast<SymbolType>(fLoader->FindSymbol(symbol_name, kPefCode));

        if (!ret)
            ret = reinterpret_cast<SymbolType>(fLoader->FindSymbol(symbol_name, kPefData));

        if (!ret)
            ret = reinterpret_cast<SymbolType>(fLoader->FindSymbol(symbol_name, kPefZero));

        return ret;
    }

  private:
    PEFLoader *fLoader{nullptr};
};

inline void hcore_pure_call(void)
{
    // virtual placeholder.
    return;
}
} // namespace hCore

#endif /* ifndef __KERNELKIT_SHARED_OBJECT_CORE_HXX__ */