/**
 *	KeyMap.cpp
 *	Copyright (C) 2003-2004 Nate
 *
 *	This file is part of DigitalWatch, a free DTV watching and recording
 *	program for the VisionPlus DVB-T.
 *
 *	DigitalWatch is free software; you can redistribute it and/or modify
 *	it under the terms of the GNU General Public License as published by
 *	the Free Software Foundation; either version 2 of the License, or
 *	(at your option) any later version.
 *
 *	DigitalWatch 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 General Public License for more details.
 *
 *	You should have received a copy of the GNU General Public License
 *	along with DigitalWatch; if not, write to the Free Software
 *	Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include "KeyMap.h"
#include "ParseLine.h"
#include "GlobalFunctions.h"
#include <stdio.h>

KeyMap::KeyMap(void)
{
}

KeyMap::~KeyMap(void)
{
	std::vector<KeyMapEntry>::iterator it = m_keyMaps.begin();
	for ( ; it != m_keyMaps.end() ; it++ )
	{
		KeyMapEntry keyMap = *it;
		delete[] keyMap.Function;
	}
}

HRESULT KeyMap::GetFunction(int keycode, BOOL shift, BOOL ctrl, BOOL alt, LPWSTR *function)
{
	CAutoLock keyMapsLock(&m_keyMapsLock);

	std::vector<KeyMapEntry>::iterator it = m_keyMaps.begin();
	for ( ; it != m_keyMaps.end() ; it++ )
	{
		KeyMapEntry keyMap = *it;
		if ((keyMap.Keycode == keycode) &&
			(keyMap.Shift == shift) &&
			(keyMap.Ctrl == ctrl) &&
			(keyMap.Alt == alt))
		{
			strCopy(*function, keyMap.Function);
			return S_OK;
		}
	}
	return E_FAIL;
}

HRESULT KeyMap::LoadFromFile(LPWSTR filename)
{
	(log << "Loading Keys file: " << filename << "\n").Write();
	LogMessageIndent indent(&log);

	HRESULT hr;

	XMLDocument file;
	file.SetLogCallback(m_pLogCallback);
	if (hr = file.Load(filename) != S_OK)
	{
		return (log << "Could not load keys file: " << filename << "\n").Show(hr);
	}

	hr = LoadFromXML(&file.Elements);
	if FAILED(hr)
		return hr;

	indent.Release();
	(log << "Finished loading keys file : " << hr << "\n").Write();

	return hr;
}

HRESULT KeyMap::LoadFromXML(XMLElementList* elementList)
{
	CAutoLock keyMapsLock(&m_keyMapsLock);

	(log << "Loading Keys from XML\n").Write();
	LogMessageIndent indent(&log);

	XMLElement *element;
	XMLAttribute *attr;

	int elementCount = elementList->Count();
	for ( int item=0 ; item<elementCount ; item++ )
	{
		element = elementList->Item(item);
		if ((_wcsicmp(element->name, L"Key") == 0) || (_wcsicmp(element->name, L"Mouse") == 0))
		{
			KeyMapEntry newKeyMapEntry;
			ZeroMemory(&newKeyMapEntry, sizeof(KeyMapEntry));

			attr = element->Attributes.Item(L"code");
			if (attr == NULL)
				continue;

			if (_wcsicmp(element->name, L"Key") == 0)
			{
				if ((attr->value[0] == '\'') && (attr->value[2] == '\''))
					newKeyMapEntry.Keycode = attr->value[1];
				else
					newKeyMapEntry.Keycode = _wtoi(attr->value);
			}
			else
			{
				newKeyMapEntry.Keycode = -1 * _wtoi(attr->value);
			}

			attr = element->Attributes.Item(L"shift");
			newKeyMapEntry.Shift = (attr) && (attr->value[0] != '0');

			attr = element->Attributes.Item(L"ctrl");
			newKeyMapEntry.Ctrl = (attr) && (attr->value[0] != '0');

			attr = element->Attributes.Item(L"alt");
			newKeyMapEntry.Alt = (attr) && (attr->value[0] != '0');

			strCopy(newKeyMapEntry.Function, element->value);

			m_keyMaps.push_back(newKeyMapEntry);
			continue;
		}
	}
	(log << "Loaded " << (long)m_keyMaps.size() << " key mappings\n").Write();

	indent.Release();
	(log << "Finished loading keys from XML\n").Write();

	return S_OK;
}