/******************************************************************************
	File modified by Yohann NICOLAS.

	NAME
		INIfile.cpp

	DESCRIPTION
		Memory cached INI file read/write class to replace legacy MS code

	COPYRIGHT
		©1999-2004 Ultrafunk (www.ultrafunk.com) - info@ultrafunk.com

******************************************************************************/

#include <stdio.h>
#include "INIfile.h"

#include "d2functions.h"
#include "error.h"

/*
http://msdn.microsoft.com/library/default.asp?url=/library/en-us/sysinfo/base/getprivateprofilestring.asp
*/


/*****************************************************************************/
// First a few support routines

static char *strstri(char *text, char *string)
{
	char *found = NULL;

	if(text && string)
	{
		int len = strlen(string);
		if(len)
		{
			while(*text)
			{
				if(strnicmp(string, text, len) == 0)
				{
					found = text;
					break;
				}

				text++;
			}
		}
	}

	return found;
}


static BOOL chrcmp(char c, char *string)
{
	for(unsigned int i=0; i<strlen(string); i++)
	{
		if(c == string[i])
			return TRUE;
	}

	return FALSE;
}


static BOOL FileExists(char *name)
{
	DWORD attr = GetFileAttributes(name);
	return ( (attr != (DWORD)-1) && !(attr & FILE_ATTRIBUTE_DIRECTORY) );
}


/*****************************************************************************/


INIFile::INIFile()
{
	m_readWrite		= 0;
	m_file			= NULL;
	m_cache			= NULL;
	m_cacheWritePos	= 0;
	m_sectionStart	= NULL;
	m_sectionEnd	= NULL;
	 
	ZeroMemory((void *)m_path, sizeof(m_path));
	ZeroMemory((void *)m_currentSection, sizeof(m_currentSection));
}


INIFile::~INIFile()
{
	if(m_file)
		CloseHandle(m_file);

	if(m_cache)
		delete [] m_cache;
}


/*****************************************************************************/



BOOL INIFile::InitReadWrite(char *path, int readWrite, DWORD writeCacheSize)
{
	if(path)
	{
		m_readWrite		= readWrite;
		m_cacheWritePos	= 0;
		m_sectionStart	= NULL;
		m_sectionEnd	= NULL;

		ZeroMemory((void *)m_path, sizeof(m_path));
		ZeroMemory((void *)m_currentSection, sizeof(m_currentSection));

		strncpy(m_path, path, sizeof(m_path));

		if(m_file)
		{
			CloseHandle(m_file);
			m_file = NULL;
		}

		if(m_cache)
		{
			delete [] m_cache;
			m_cache	= NULL;
		}

		// If read, the open file and cache content in memory
		if(m_readWrite == INIFILE_MPQREAD)
		{
			void* refFile;
			if ( D2MPQOpenFile(m_path,&refFile) )
			{
				DWORD fileSize = D2MPQGetSizeFile(refFile, NULL);
				m_cache = new char[fileSize + 1];
				if(m_cache)
				{
					DWORD read;
					if(D2MPQReadFile(refFile, (BYTE*)m_cache, fileSize, &read, NULL, NULL, NULL) == 0)
					{
						delete [] m_cache;
						m_cache = NULL;
					}
					else
						m_cache[fileSize] = 0;
				}
				D2MPQCloseFile(refFile);
				if(m_cache)
					return TRUE;
			}
		}

//		if((m_readWrite == INIFILE_MPQREAD) || (m_readWrite == INIFILE_READ))
		if(m_readWrite == INIFILE_READ)
		{
			if(FileExists(m_path))
			{
				m_file = CreateFile(m_path,
									GENERIC_READ,
									0,
									NULL,
									OPEN_EXISTING,
									FILE_ATTRIBUTE_NORMAL,
									NULL);

				if(m_file != INVALID_HANDLE_VALUE)
				{
					DWORD fileSize = GetFileSize(m_file, NULL);

					m_cache = new char[fileSize + 1];
					if(m_cache)
					{
						DWORD read;

						if(ReadFile(m_file, m_cache, fileSize, &read, NULL) == 0)
						{
							delete [] m_cache;
							m_cache = NULL;
						}
						else
							m_cache[fileSize] = 0;
					}

					CloseHandle(m_file);
					m_file = NULL;

					if(m_cache)
						return TRUE;
				}
			}
		}

		// If write, then create write cache memory buffer
		if(m_readWrite == INIFILE_WRITE)
		{
			m_cache = new char[writeCacheSize];

			if(m_cache)
				return TRUE;
		}
	}

	return FALSE;
}


BOOL INIFile::close()
{
	BOOL success = FALSE;

	// If read, then kill the memory cached settings
	if(m_readWrite == INIFILE_READ)
	{
		// Just to be sure, close the settings file again if it's still open for some reason
		if(m_file)
		{
			CloseHandle(m_file);
			m_file = NULL;
		}

		if(m_cache)
		{
			delete [] m_cache;
			m_cache	= NULL;

			success = TRUE;
		}

		// Just to be sure, reset section position markers
		m_sectionStart = NULL;
		m_sectionEnd   = NULL;
	}

	// If write, then commit (write) cached settings file to disk
	if(m_readWrite == INIFILE_WRITE)
	{
		if(m_path && m_cache)
		{
			m_file = CreateFile(m_path,
								GENERIC_WRITE,
								0,
								NULL,
								CREATE_ALWAYS,
								FILE_ATTRIBUTE_NORMAL,
								NULL);

			if(m_file != INVALID_HANDLE_VALUE)
			{
				DWORD written;
				WriteFile(m_file, m_cache, m_cacheWritePos, &written, NULL);

				CloseHandle(m_file);
				m_file = NULL;

				if(written == m_cacheWritePos)
					success = TRUE;
			}

			delete [] m_cache;
			m_cache	= NULL;
		}
	}

	return success;
}


/*****************************************************************************/


static char delimiterEnd[] = "\r\n\0";


int INIFile::GetPrivateProfileString(const char *section, const char *key, const char *def, char *dest, DWORD size)
{
	DWORD copied = 0;

	if((m_cache) && (section && key && dest && size))
	{
		if(stricmp(section, m_currentSection) != 0)
		{
			strncpy(m_currentSection, section, MAX_SECTIONNAME_LENGTH);

			char headerString[MAX_SECTIONNAME_LENGTH];
			ZeroMemory((void *)headerString, sizeof(headerString));
//			_snprintf(headerString, MAX_SECTIONNAME_LENGTH, "[%s]\r\n", section);
			_snprintf(headerString, MAX_SECTIONNAME_LENGTH, "[%s]", section);

			// Locate section start in buffer
			m_sectionStart = strstri(m_cache, headerString);

			// We found the start section
			if(m_sectionStart)
			{
				// Keys start after the "[" + section + "]" header
				m_sectionStart += strlen(section) + 2;

				// Locate section end in buffer
				m_sectionEnd = strstr(m_sectionStart, "\r\n[");

				// If no next section is found, then use end of buffer
				if(m_sectionEnd == NULL)
					m_sectionEnd = m_cache + strlen(m_cache);
			}
		}

		if(m_sectionStart && m_sectionEnd)
		{
			char keyString[MAX_KEYNAME_LENGTH];
			ZeroMemory((void *)keyString, sizeof(keyString));
			_snprintf(keyString, MAX_KEYNAME_LENGTH, "\r\n%s=", key);

			if(char *s = strstri(m_sectionStart, keyString))
			{
				if(s < m_sectionEnd)
				{
					s += strlen(keyString);

					while(*s && (copied < size) && (chrcmp(*s, delimiterEnd) == FALSE))
					{
						*dest++ = *s++;
						copied++;
					}

					*dest = 0;
					copied++;
				}
			}
		}
	}

	if(!copied && def)
		strncpy(dest, def, size);

	return copied; // Do not include the terminating null character
}


BOOL INIFile::WritePrivateProfileString(char *section, char *key, char *string)
{
	if(!section || !key || !string) return false;

	if(stricmp(section, m_currentSection) != 0)
	{
		if(m_cacheWritePos == 0)
			m_cacheWritePos += sprintf((m_cache + m_cacheWritePos), "[%s]\r\n", section);
		else
			m_cacheWritePos += sprintf((m_cache + m_cacheWritePos), "\r\n[%s]\r\n", section);

		strncpy(m_currentSection, section, MAX_SECTIONNAME_LENGTH);
	}
	m_cacheWritePos += sprintf((m_cache + m_cacheWritePos), "%s=%s\r\n", key, string);
	return true;
}

/*================================= END OF FILE =================================*/