#include <Geode/DefaultInclude.hpp>

#ifdef GEODE_IS_MACOS

    #include <Geode/loader/Mod.hpp>
    #include <dlfcn.h>

USE_GEODE_NAMESPACE();

template <typename T>
T findSymbolOrMangled(void* dylib, char const* name, char const* mangled) {
    auto res = reinterpret_cast<T>(dlsym(dylib, name));
    if (!res) {
        res = reinterpret_cast<T>(dlsym(dylib, mangled));
    }
    return res;
}

Result<> Mod::loadPlatformBinary() {
    auto dylib =
        dlopen((this->m_tempDirName / this->m_info.m_binaryName).string().c_str(), RTLD_LAZY);
    if (dylib) {
        this->m_implicitLoadFunc =
            findSymbolOrMangled<geode_load>(dylib, "geode_implicit_load", "_geode_implicit_load");
        this->m_loadFunc = findSymbolOrMangled<geode_load>(dylib, "geode_load", "_geode_load");
        this->m_unloadFunc =
            findSymbolOrMangled<geode_unload>(dylib, "geode_unload", "_geode_unload");
        this->m_enableFunc =
            findSymbolOrMangled<geode_enable>(dylib, "geode_enable", "_geode_enable");
        this->m_disableFunc =
            findSymbolOrMangled<geode_disable>(dylib, "geode_disable", "_geode_disable");
        this->m_saveDataFunc =
            findSymbolOrMangled<geode_save_data>(dylib, "geode_save_data", "_geode_save_data");
        this->m_loadDataFunc =
            findSymbolOrMangled<geode_load_data>(dylib, "geode_load_data", "_geode_load_data");
        this->m_settingUpdatedFunc = findSymbolOrMangled<geode_setting_updated>(
            dylib, "geode_setting_updated", "_geode_setting_updated"
        );

        if (!this->m_implicitLoadFunc && !this->m_loadFunc) {
            return Err(
                "Unable to find mod entry point (lacking both implicit & explicit definition)"
            );
        }

        if (this->m_platformInfo) {
            delete this->m_platformInfo;
        }
        this->m_platformInfo = new PlatformInfo { dylib };

        return Ok();
    }
    std::string err = (char const*)dlerror();
    return Err("Unable to load the DYLIB: dlerror returned (" + err + ")");
}

Result<> Mod::unloadPlatformBinary() {
    auto dylib = this->m_platformInfo->m_dylib;
    delete this->m_platformInfo;
    this->m_platformInfo = nullptr;
    if (dlclose(dylib) == 0) {
        this->m_unloadFunc = nullptr;
        this->m_loadFunc = nullptr;
        this->m_implicitLoadFunc = nullptr;
        this->m_enableFunc = nullptr;
        this->m_disableFunc = nullptr;
        this->m_saveDataFunc = nullptr;
        this->m_loadDataFunc = nullptr;
        this->m_settingUpdatedFunc = nullptr;
        return Ok();
    }
    else {
        return Err("Unable to free library");
    }
}

#endif