#include <stdexcept>
#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);
	if (!m_hModule)
		throw std::invalid_argument("Library not found.");
}

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

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