#include "InstallerUtil.h"

 #include "InstallerMain.h"

 #include <stdio.h>

 #include <windows.h>

 int tstrcasecmp(const TCHAR *str1, const TCHAR *str2)

 {

 #if defined _UNICODE

 	return _wcsicmp(str1, str2);

 #else

 	return _stricmp(str1, str2);

 #endif

 }

 size_t AnsiToUnicode(const char *str, wchar_t *buffer, size_t maxchars)

 {

 	if (maxchars < 1)

 	{

 		return 0;

 	}

 	size_t total = 

 		(size_t)MultiByteToWideChar(CP_UTF8,

 		0,

 		str,

 		-1,

 		buffer,

 		(int)maxchars);

 	return total;

 }

 bool IsWhiteSpaceA(const char *stream)

 {

 	char c = *stream;

 	if (c & (1<<7))

 	{

 		return false;

 	}

 	else

 	{

 		return isspace(c) != 0;

 	}

 }

 int BreakStringA(const char *str, char *out, size_t maxchars)

 {

 	const char *inptr = str;

 	while (*inptr != '\0' && IsWhiteSpaceA(inptr))

 	{

 		inptr++;

 	}

 	if (*inptr == '\0')

 	{

 		if (maxchars)

 		{

 			*out = '\0';

 		}

 		return -1;

 	}

 	const char *start, *end = NULL;

 	bool quoted = (*inptr == '"');

 	if (quoted)

 	{

 		inptr++;

 		start = inptr;

 		/* Read input until we reach a quote. */

 		while (*inptr != '\0' && *inptr != '"')

 		{

 			/* Update the end point, increment the stream. */

 			end = inptr++;

 		}

 		/* Read one more token if we reached an end quote */

 		if (*inptr == '"')

 		{

 			inptr++;

 		}

 	}

 	else

 	{

 		start = inptr;

 		/* Read input until we reach a space */

 		while (*inptr != '\0' && !IsWhiteSpaceA(inptr))

 		{

 			/* Update the end point, increment the stream. */

 			end = inptr++;

 		}

 	}

 	/* Copy the string we found, if necessary */

 	if (end == NULL)

 	{

 		if (maxchars)

 		{

 			*out = '\0';

 		}

 	}

 	else if (maxchars)

 	{

 		char *outptr = out;

 		maxchars--;

 		for (const char *ptr=start;

 			(ptr <= end) && ((unsigned)(outptr - out) < (maxchars));

 			ptr++, outptr++)

 			{

 			*outptr = *ptr;

 		}

 		*outptr = '\0';

 	}

 	/* Consume more of the string until we reach non-whitespace */

 	while (*inptr != '\0' && IsWhiteSpaceA(inptr))

 	{

 		inptr++;

 	}

 	return (int)(inptr - str);

 }

 size_t UTIL_Format(TCHAR *buffer, size_t count, const TCHAR *fmt, ...)

 {

 	va_list ap;

 	size_t len;

 	va_start(ap, fmt);

 	len = UTIL_FormatArgs(buffer, count, fmt, ap);

 	va_end(ap);

 	if (len >= count)

 	{

 		len = count - 1;

 		buffer[len] = '\0';

 	}

 	return len;

 }

 size_t UTIL_FormatArgs(TCHAR *buffer, size_t count, const TCHAR *fmt, va_list ap)

 {

 	size_t len = _vsntprintf(buffer, count, fmt, ap);

 	if (len >= count)

 	{

 		len = count - 1;

 		buffer[len] = '\0';

 	}

 	return len;

 }

 size_t UTIL_PathFormat(TCHAR *buffer, size_t count, const TCHAR *fmt, ...)

 {

 	va_list ap;

 	size_t len;

 	va_start(ap, fmt);

 	len = UTIL_FormatArgs(buffer, count, fmt, ap);

 	va_end(ap);

 	for (size_t i = 0; i < len; i++)

 	{

 		if (buffer[i] == '/')

 		{

 			buffer[i] = '\\';

 		}

 	}

 	return len;

 }

 const TCHAR *GetFileFromPath(const TCHAR *path)

 {

 	size_t len = _tcslen(path);

 	for (size_t i = len - 1;

 		 i >= 0 && i < len;

 		 i--)

 	{

 		if (path[i] == '\\' || path[i] == '/')

 		{

 			return &path[i+1];

 		}

 	}

 	return NULL;

 }

 void GenerateErrorMessage(DWORD err, TCHAR *buffer, size_t maxchars)

 {

 	if (FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM,

 		NULL,

 		err,

 		MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),

 		buffer,

 		(DWORD)maxchars,

 		NULL) == 0)

 	{

 		UTIL_Format(buffer, maxchars, _T("Unknown error"));

 	}

 }

 INT_PTR AskToExit(HWND hWnd)

 {

 	TCHAR verify_exit[100];

 	if (LoadString(g_hInstance,

 		IDS_VERIFY_EXIT,

 		verify_exit,

 		sizeof(verify_exit) / sizeof(TCHAR)

 		) == 0)

 	{

 		return (INT_PTR)FALSE;

 	}

 	int val = MessageBox(

 		hWnd, 

 		_T("Are you sure you want to exit?"), 

 		_T("SourceMod Installer"), 

 		MB_YESNO|MB_ICONQUESTION);

 	if (val == 0 || val == IDYES)

 	{

 		UpdateGlobalPosition(hWnd);

 		EndDialog(hWnd, NULL);

 		return (INT_PTR)TRUE;

 	}

 	return (INT_PTR)FALSE;

 }

 size_t UTIL_GetFileSize(const TCHAR *file_path)

 {

 	HANDLE hFile;

 	if ((hFile = CreateFile(file_path, 

 		GENERIC_READ,

 		FILE_SHARE_READ,

 		NULL,

 		OPEN_EXISTING,

 		FILE_ATTRIBUTE_NORMAL,

 		NULL)) 

 		!= INVALID_HANDLE_VALUE)

 	{

 		LARGE_INTEGER size;

 		if (GetFileSizeEx(hFile, &size))

 		{

 			CloseHandle(hFile);

 			return (size_t)size.QuadPart;

 		}

 		CloseHandle(hFile);

 	}

 	return 0;

 }

 #if 0

 size_t UTIL_GetFolderSize(const TCHAR *basepath)

 {

 	HANDLE hFind;

 	WIN32_FIND_DATA fd;

 	TCHAR search_path[MAX_PATH];

 	size_t total = 0;

 	UTIL_PathFormat(search_path,

 		sizeof(search_path) / sizeof(TCHAR),

 		_T("%s\\*.*"),

 		basepath);

 	if ((hFind = FindFirstFile(search_path, &fd)) == INVALID_HANDLE_VALUE)

 	{

 		return 0;

 	}

 	do

 	{

 		if (tstrcasecmp(fd.cFileName, _T(".")) == 0

 			|| tstrcasecmp(fd.cFileName, _T("..")) == 0)

 		{

 			continue;

 		}

 		if ((fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY)

 		{

 			UTIL_PathFormat(search_path,

 				sizeof(search_path) / sizeof(TCHAR),

 				_T("%s\\%s"),

 				basepath,

 				fd.cFileName);

 			total += UTIL_GetFolderSize(search_path);

 		}

 		else

 		{

 			UTIL_PathFormat(search_path,

 				sizeof(search_path) / sizeof(TCHAR),

 				_T("%s\\%s"),

 				basepath,

 				fd.cFileName);

 			total += UTIL_GetFileSize(search_path);

 		}

 	} while (FindNextFile(hFind, &fd));

 	FindClose(hFind);

 	return total;

 }

 #endif