100 lines
2.3 KiB
C++
100 lines
2.3 KiB
C++
#pragma once
|
|
|
|
#include <functional>
|
|
#include <string>
|
|
#include <vector>
|
|
#include <hex/helpers/utils.hpp>
|
|
|
|
#if defined(OS_WINDOWS)
|
|
#include <Windows.h>
|
|
#else
|
|
#include <dlfcn.h>
|
|
#endif
|
|
|
|
namespace hex::script::loader {
|
|
|
|
class ScriptLoader;
|
|
|
|
struct Script {
|
|
std::string name;
|
|
std::fs::path path;
|
|
|
|
bool background;
|
|
std::function<void()> entryPoint;
|
|
const ScriptLoader *loader;
|
|
};
|
|
|
|
class ScriptLoader {
|
|
public:
|
|
ScriptLoader(std::string typeName) : m_typeName(std::move(typeName)) {}
|
|
virtual ~ScriptLoader() = default;
|
|
|
|
virtual bool initialize() = 0;
|
|
virtual bool loadAll() = 0;
|
|
virtual void clearScripts() = 0;
|
|
|
|
void addScript(std::string name, std::fs::path path, bool background, std::function<void()> entryPoint) {
|
|
m_scripts.emplace_back(std::move(name), std::move(path), background, std::move(entryPoint), this);
|
|
}
|
|
|
|
const auto& getScripts() const {
|
|
return m_scripts;
|
|
}
|
|
|
|
const std::string& getTypeName() const {
|
|
return m_typeName;
|
|
}
|
|
|
|
protected:
|
|
auto& getScripts() {
|
|
return m_scripts;
|
|
}
|
|
|
|
private:
|
|
std::vector<Script> m_scripts;
|
|
std::string m_typeName;
|
|
};
|
|
|
|
#if defined(OS_WINDOWS)
|
|
inline void *loadLibrary(const wchar_t *path) {
|
|
try {
|
|
HMODULE h = ::LoadLibraryW(path);
|
|
return h;
|
|
} catch (...) {
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
inline void *loadLibrary(const char *path) {
|
|
try {
|
|
auto utf16Path = hex::utf8ToUtf16(path);
|
|
HMODULE h = ::LoadLibraryW(utf16Path.c_str());
|
|
return h;
|
|
} catch (...) {
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
template<typename T>
|
|
T getExport(void *h, const char *name) {
|
|
try {
|
|
FARPROC f = ::GetProcAddress(static_cast<HMODULE>(h), name);
|
|
return reinterpret_cast<T>(reinterpret_cast<uintptr_t>(f));
|
|
} catch (...) {
|
|
return nullptr;
|
|
}
|
|
}
|
|
#else
|
|
inline void *loadLibrary(const char *path) {
|
|
void *h = dlopen(path, RTLD_LAZY);
|
|
return h;
|
|
}
|
|
|
|
template<typename T>
|
|
T getExport(void *h, const char *name) {
|
|
void *f = dlsym(h, name);
|
|
return reinterpret_cast<T>(f);
|
|
}
|
|
#endif
|
|
|
|
} |