#include <memory>
#include <windows.h>
#include <SWI-Prolog.h>

#include "resource.h"
#include "defs.h"

/* Convert normal string to TSTR using given codepage. */
TCHAR *TszFromSz(const char *sz, int iCp)
{
	TCHAR *tsz;

#ifdef UNICODE
	int cbMultiByte, cchWideChar;

	cbMultiByte = strlen(sz)+1;
	cchWideChar = MultiByteToWideChar(iCp, 0, sz, cbMultiByte, NULL, 0);
	tsz = (TCHAR *)malloc(cchWideChar*sizeof(WCHAR));
	if (!tsz) return NULL;
	if (!MultiByteToWideChar(iCp, 0, sz, cbMultiByte, tsz, cchWideChar))
		return NULL;
#else
	tsz = malloc(strlen(sz)+1);
	if (!tsz) return NULL;
	strcpy(tsz, sz);
#endif

	return tsz;
}

Library::Library(const TCHAR *tszLibrary)
{
	m_hModule = LoadLibrary(tszLibrary);
}

Library::~Library()
{
	if (m_hModule)
		FreeLibrary(m_hModule);
}

FARPROC Library::GetProcAddress(const char *szProc)
{
	return m_hModule? ::GetProcAddress(m_hModule, szProc): NULL;
}

std::unique_ptr<Library> Library::Load(const TCHAR *tszLibrary)
{
	auto upLib = std::make_unique<Library>(tszLibrary);
	if (!upLib->m_hModule) upLib = nullptr;
	return upLib;
}