/**
 *    Copyright (C) 2016-2023 Grok Image Compression Inc.
 *
 *    This source code is free software: you can redistribute it and/or  modify
 *    it under the terms of the GNU Affero General Public License, version 3,
 *    as published by the Free Software Foundation.
 *
 *    This source code is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU Affero General Public License for more details.
 *
 *    You should have received a copy of the GNU Affero General Public License
 *    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#include <plugin/minpf_dynamic_library.h>
#include <stdio.h>
#include <cstring>

#ifdef _WIN32
#include <string>
// Returns the last Win32 error, in string format. Returns an empty string if there is no error.
static std::string GetLastErrorAsString()
{
	// Get the error message, if any.
	DWORD errorMessageID = ::GetLastError();
	if(errorMessageID == 0)
		return std::string(); // No error message has been recorded

	LPSTR messageBuffer = nullptr;
	size_t size = FormatMessageA(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM |
									 FORMAT_MESSAGE_IGNORE_INSERTS,
								 nullptr, errorMessageID, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
								 (LPSTR)&messageBuffer, 0, nullptr);

	std::string message(messageBuffer, size);

	// Free the buffer.
	LocalFree(messageBuffer);

	return message;
}

#endif

namespace grk
{
bool minpf_get_full_path([[maybe_unused]] const char* path, [[maybe_unused]] const void* addr,
						 [[maybe_unused]] dynamic_handle_t handle, [[maybe_unused]] char* fullPath,
						 [[maybe_unused]] size_t fullPathLen)
{
#ifdef GRK_BUILD_PLUGIN_LOADER
	if(!path || !addr || !fullPath || !fullPathLen)
	{
		return false;
	}

#ifdef _WIN32
	auto dwLength = GetModuleFileName(handle, fullPath, (DWORD)fullPathLen);
	if(dwLength == 0)
	{
		auto error = GetLastErrorAsString();
		error = "GetModuleFileName failed: " + error + "\n";
		fprintf(stderr, "%s", error.c_str());
		return false;
	}
	return true;
#else
	Dl_info dl_info;
	memset(&dl_info, 0, sizeof(dl_info));
	if(dladdr(addr, &dl_info) && strlen(dl_info.dli_fname) < fullPathLen)
	{
		strcpy(fullPath, dl_info.dli_fname);
		return true;
	}
	fprintf(stderr, "dladdr failed\n");
	return false;

#endif

#else
	return false;
#endif
}

bool minpf_unload_dynamic_library([[maybe_unused]] minpf_dynamic_library* library)
{
#ifdef GRK_BUILD_PLUGIN_LOADER
	if(!library)
		return true;
	bool rc = false;
#ifdef _WIN32
	rc = FreeLibrary(library->handle) ? true : false;
#else
	dlclose(library->handle);
	rc = true;
#endif
	free(library);
	return rc;
#else
	return false;
#endif
}

minpf_dynamic_library* minpf_load_dynamic_library([[maybe_unused]] const char* path,
												  [[maybe_unused]] char* error)
{
#ifdef GRK_BUILD_PLUGIN_LOADER
	minpf_dynamic_library* lib = nullptr;
	dynamic_handle_t handle = nullptr;

	if(!path)
		return nullptr;

#ifdef _WIN32
	handle = LoadLibrary(path);
	if(handle == nullptr)
	{
		// TODO report error
		return nullptr;
	}
#else
	handle = dlopen(path, RTLD_NOW);
	if(!handle)
	{
		// TODO report error
		return nullptr;
	}

#endif

	lib = (minpf_dynamic_library*)calloc(1, sizeof(minpf_dynamic_library));
	if(!lib)
	{
#ifdef _WIN32
		FreeLibrary(handle);
#else
		dlclose(handle);
#endif
		return nullptr;
	}
	strcpy(lib->path, path);
	lib->handle = handle;
	return lib;
#else
	return nullptr;
#endif
}

void* minpf_get_symbol([[maybe_unused]] minpf_dynamic_library* library,
					   [[maybe_unused]] const char* symbol)
{
#ifdef GRK_BUILD_PLUGIN_LOADER
	if(!library || !library->handle)
		return nullptr;

	void* rc = nullptr;

#ifdef _WIN32
	rc = GetProcAddress((HMODULE)library->handle, symbol);
	if(!rc)
		Logger::logger_.error("Error getting symbol : %d", GetLastError());
#else
	rc = dlsym(library->handle, symbol);
#endif

	return rc;
#else
	return nullptr;
#endif
}

} // namespace grk
