// NeL - MMORPG Framework <http://dev.ryzom.com/projects/nel/>
// Copyright (C) 2010  Winch Gate Property Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program 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 Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

#include "stdmisc.h"
#include "nel/misc/system_utils.h"

#ifdef NL_OS_WINDOWS
#define INITGUID
#include <ddraw.h>
#include <windows.h>
#include <string.h>
#include <stdio.h>
#include <dxgi.h>
#include <initguid.h>
#include <CGuid.h>
#	include <ObjBase.h>
#	ifdef _WIN32_WINNT_WIN7
		// only supported by Windows 7 Platform SDK
#		include <ShObjIdl.h>
#		define TASKBAR_PROGRESS 1
#	endif
#elif defined(NL_OS_UNIX) && !defined(NL_OS_MAC)
#include "nel/misc/file.h"
#endif

#ifdef DEBUG_NEW
	#define new DEBUG_NEW
#endif

using namespace std;

// Key in registry
static string RootKey;
static const uint32 KeyMaxLength = 1024;

namespace NLMISC {

nlWindow CSystemUtils::s_window = EmptyWindow;

bool CSystemUtils::init()
{
#ifdef NL_OS_WINDOWS
	// initialize COM
	HRESULT hr = CoInitializeEx(NULL, COINIT_MULTITHREADED);
	if (FAILED(hr)) return false;
#endif

	return true;
}

bool CSystemUtils::uninit()
{
#ifdef NL_OS_WINDOWS
	// uninitialize COM
	CoUninitialize();
#endif

	return true;
}

void CSystemUtils::setWindow(nlWindow window)
{
	s_window = window;
}

bool CSystemUtils::updateProgressBar(uint value, uint total)
{
#ifdef TASKBAR_PROGRESS
	if (s_window == NULL)
	{
		nlwarning("No window has be set with CSystemUtils::setWindow(), progress bar can't be displayed");
		return false;
	}

	ITaskbarList3 *pTaskbarList = NULL;

	// instanciate the taskbar control COM object
	HRESULT hr = CoCreateInstance(CLSID_TaskbarList, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARGS(&pTaskbarList));
	// error can be ignored because Windows versions before Windows 7 doesn't support it
	if (FAILED(hr) || !pTaskbarList) return false;

	if (total)
	{
		// update the taskbar progress
		hr = pTaskbarList->SetProgressValue(s_window, (ULONGLONG)value, (ULONGLONG)total);
	}
	else
	{
		// don't update anymore the progress
		hr = pTaskbarList->SetProgressState(s_window, value == 0 ? TBPF_INDETERMINATE:TBPF_NOPROGRESS);
	}

	// release the interface
	pTaskbarList->Release();

#endif // TASKBAR_PROGRESS

	return true;
}

bool CSystemUtils::copyTextToClipboard(const ucstring &text)
{
	if (!text.size()) return false;

	bool res = false;

#ifdef NL_OS_WINDOWS
	if (OpenClipboard(NULL))
	{
		// check if unicode format is supported by clipboard
		bool isUnicode = (IsClipboardFormatAvailable(CF_UNICODETEXT) == TRUE);

		// allocates a buffer to copy text in global memory
		HGLOBAL mem = GlobalAlloc(GHND|GMEM_DDESHARE, (text.size()+1) * (isUnicode ? 2:1));

		if (mem)
		{
			// create a lock on this buffer
			void *hLock = GlobalLock(mem);

			// copy text to this buffer
			if (isUnicode)
				wcscpy((wchar_t*)hLock, (const wchar_t*)text.c_str());
			else
				strcpy((char*)hLock, text.toString().c_str());

			// unlock buffer
			GlobalUnlock(mem);

			// empty clipboard
			EmptyClipboard();

			// set new data to clipboard in the right format
			SetClipboardData(isUnicode ? CF_UNICODETEXT:CF_TEXT, mem);

			res = true;
		}

		CloseClipboard();
	}
#endif

	return res;
}

bool CSystemUtils::pasteTextFromClipboard(ucstring &text)
{
	bool res = false;

#ifdef NL_OS_WINDOWS
	if (OpenClipboard(NULL))
	{
		// check if unicode format is supported by clipboard
		bool isUnicode = (IsClipboardFormatAvailable(CF_UNICODETEXT) == TRUE);

		// get data from clipboard (if not of this type, they are converted)
		// warning, this code can't be debuggued in VC++ IDE, hObj will be always NULL
		HANDLE hObj = GetClipboardData(isUnicode ? CF_UNICODETEXT:CF_TEXT);

		if (hObj)
		{
			// create a lock on clipboard data
			void *hLock = GlobalLock(hObj);

			if (hLock != NULL)
			{
				// retrieve clipboard data
				if (isUnicode)
					text = (const ucchar*)hLock;
				else
					text = (const char*)hLock;

				// unlock data
				GlobalUnlock(hObj);

				res = true;
			}
		}

		CloseClipboard();
	}
#endif

	return res;
}

bool CSystemUtils::supportUnicode()
{
	static bool init = false;
	static bool unicodeSupported = false;
	if (!init)
	{
		init = true;
#ifdef NL_OS_WINDOWS
		OSVERSIONINFO osvi;
		osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);

		// get Windows version
		if (GetVersionEx(&osvi))
		{
			if (osvi.dwPlatformId == VER_PLATFORM_WIN32_NT)
			{
				// unicode is supported since Windows NT 4.0
				if (osvi.dwMajorVersion >= 4)
				{
					unicodeSupported = true;
				}
			}
		}
#else
		unicodeSupported = true;
#endif
	}
	return unicodeSupported;
}

bool CSystemUtils::isAzertyKeyboard()
{
#ifdef NL_OS_WINDOWS
	uint16 klId = uint16((uintptr_t)GetKeyboardLayout(0) & 0xFFFF);
	// 0x040c is French, 0x080c is Belgian
	if (klId == 0x040c || klId == 0x080c)
		return true;
#endif
	return false;
}

bool CSystemUtils::isScreensaverEnabled()
{
	bool res = false;
#ifdef NL_OS_WINDOWS
//	old code, is not working anymore
//	BOOL bRetValue;
//	SystemParametersInfoA(SPI_GETSCREENSAVEACTIVE, 0, &bRetValue, 0);
//	res = (bRetValue == TRUE);
	HKEY hKeyScreenSaver = NULL;
	LSTATUS lReturn = RegOpenKeyExA(HKEY_CURRENT_USER, TEXT("Control Panel\\Desktop"), 0, KEY_QUERY_VALUE, &hKeyScreenSaver);
	if (lReturn == ERROR_SUCCESS)
	{
		DWORD dwType = 0L;
		DWORD dwSize = KeyMaxLength;
		unsigned char Buffer[KeyMaxLength] = {0};

		lReturn = RegQueryValueExA(hKeyScreenSaver, TEXT("SCRNSAVE.EXE"), NULL, &dwType, NULL, &dwSize);
		// if SCRNSAVE.EXE is present, check also if it's empty
		if (lReturn == ERROR_SUCCESS)
			res = (Buffer[0] != '\0');
	}
	RegCloseKey(hKeyScreenSaver);
#endif
	return res;
}

bool CSystemUtils::enableScreensaver(bool screensaver)
{
	bool res = false;
#ifdef NL_OS_WINDOWS
	res = (SystemParametersInfoA(SPI_SETSCREENSAVEACTIVE, screensaver ? TRUE:FALSE, NULL, 0) == TRUE);
#endif
	return res;
}

std::string CSystemUtils::getRootKey()
{
	return RootKey;
}

void CSystemUtils::setRootKey(const std::string &root)
{
	RootKey = root;
}

string CSystemUtils::getRegKey(const string &Entry)
{
	string ret;
#ifdef NL_OS_WINDOWS
	HKEY hkey;

	if(RegOpenKeyEx(HKEY_CURRENT_USER, RootKey.c_str(), 0, KEY_READ, &hkey) == ERROR_SUCCESS)
	{
		DWORD	dwType	= 0L;
		DWORD	dwSize	= KeyMaxLength;
		unsigned char	Buffer[KeyMaxLength];

		if(RegQueryValueEx(hkey, Entry.c_str(), NULL, &dwType, Buffer, &dwSize) != ERROR_SUCCESS)
		{
			nlwarning("Can't get the reg key '%s'", Entry.c_str());
		}
		else
		{
			ret = (char*)Buffer;
		}
		RegCloseKey(hkey);
	}
	else
	{
		nlwarning("Can't get the reg key '%s'", Entry.c_str());
	}
#endif
	return ret;
}

bool CSystemUtils::setRegKey(const string &ValueName, const string &Value)
{
	bool res = false;
#ifdef NL_OS_WINDOWS
	HKEY hkey;
	DWORD dwDisp;

	char nstr[] = { 0x00 };
	if (RegCreateKeyExA(HKEY_CURRENT_USER, RootKey.c_str(), 0, nstr, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hkey, &dwDisp) == ERROR_SUCCESS)
	{
		if (RegSetValueExA(hkey, ValueName.c_str(), 0L, REG_SZ, (const BYTE *)Value.c_str(), (DWORD)(Value.size())+1) == ERROR_SUCCESS)
			res = true;
		RegCloseKey(hkey);
	}
	else
	{
		nlwarning("Can't set the reg key '%s' '%s'", ValueName.c_str(), Value.c_str());
	}
#endif

	return res;
}

uint CSystemUtils::getCurrentColorDepth()
{
	uint depth = 0;

#ifdef NL_OS_WINDOWS
	HWND desktopWnd = GetDesktopWindow();
	if (desktopWnd)
	{
		HDC desktopDC = GetWindowDC(desktopWnd);
		if (desktopDC)
		{
			depth = (uint) GetDeviceCaps(desktopDC, BITSPIXEL);
			ReleaseDC(desktopWnd, desktopDC);
		}
	}
#else
	depth = 24; // temporary fix for compilation under Linux
/*
	Display *display = XOpenDisplay(NULL);
	if (display)
	{
		depth = (uint) DefaultDepth(display, DefaultScreen(display));
		XCloseDisplay(display);
	}
*/
#endif
	return depth;
}

/// Detect whether the current process is a windowed application. Return true if definitely yes, false if unknown
bool CSystemUtils::detectWindowedApplication()
{
#ifdef NL_OS_WINDOWS
	if (GetConsoleWindow() == NULL)
		return true;
#endif
	return false;
}

#ifdef NL_OS_WINDOWS
#ifndef SAFE_RELEASE
#define SAFE_RELEASE(p)      { if (p) { (p)->Release(); (p) = NULL; } }
#endif

typedef HRESULT (WINAPI* LPDIRECTDRAWCREATE)(GUID FAR *lpGUID, LPDIRECTDRAW FAR *lplpDD, IUnknown FAR *pUnkOuter);
typedef HRESULT (WINAPI* LPCREATEDXGIFACTORY)(REFIID, void**);

static std::string FormatError(HRESULT hr)
{
	std::string res;

	LPTSTR errorText = NULL;

	FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_IGNORE_INSERTS,
		NULL, hr, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR)&errorText, 0, NULL);

	if (errorText)
	{
		res = NLMISC::toString("%s (0x%x)", errorText, hr);
		LocalFree(errorText);
		errorText = NULL;
	}

	return res;
}

struct SAdapter
{
	uint id;
	std::string name;
	sint memory;
	GUID guid;
	HMONITOR hMonitor;
	bool found;

	SAdapter()
	{
		id = 0;
		memory = -1;
		guid = GUID_NULL;
		hMonitor = NULL;
		found = false;
	}
};

static std::list<SAdapter> s_dxgiAdapters;

static void EnumerateUsingDXGI(IDXGIFactory *pDXGIFactory)
{
	nlassert(pDXGIFactory != NULL);

	for(uint index = 0; ; ++index)
	{
		IDXGIAdapter *pAdapter = NULL;
		HRESULT hr = pDXGIFactory->EnumAdapters(index, &pAdapter);
		// DXGIERR_NOT_FOUND is expected when the end of the list is hit
		if (FAILED(hr)) break;

		DXGI_ADAPTER_DESC desc;
		memset(&desc, 0, sizeof(DXGI_ADAPTER_DESC));

		if (SUCCEEDED(pAdapter->GetDesc(&desc)))
		{
			SAdapter adapter;
			adapter.id = index;
			adapter.name = ucstring((ucchar*)desc.Description).toUtf8();
			adapter.memory = desc.DedicatedVideoMemory / 1024;
			adapter.found = true;

			nldebug("DXGI Adapter: %u - %s - DedicatedVideoMemory: %d KiB", index, adapter.name.c_str(), adapter.memory);

			s_dxgiAdapters.push_back(adapter);
		}

		SAFE_RELEASE(pAdapter);
	}
}

BOOL WINAPI DDEnumCallbackEx(GUID FAR* lpGUID, LPSTR lpDriverDescription, LPSTR lpDriverName, LPVOID lpContext, HMONITOR hm)
{
	SAdapter * pAdapter = (SAdapter*)lpContext;

	if (pAdapter->hMonitor == hm)
	{
		pAdapter->name = lpDriverDescription;
		pAdapter->guid = *lpGUID;
		pAdapter->found = true;
	}

	return TRUE;
}

#endif

sint CSystemUtils::getTotalVideoMemory()
{
	sint res = -1;

#if defined(NL_OS_WINDOWS)
	// using DXGI
	HINSTANCE hDXGI = LoadLibraryA("dxgi.dll");

	if (hDXGI)
	{

		// We prefer the use of DXGI 1.1
		LPCREATEDXGIFACTORY pCreateDXGIFactory = (LPCREATEDXGIFACTORY)GetProcAddress(hDXGI, "CreateDXGIFactory1");

		if (!pCreateDXGIFactory)
		{
			pCreateDXGIFactory = (LPCREATEDXGIFACTORY)GetProcAddress(hDXGI, "CreateDXGIFactory");
		}

		if (pCreateDXGIFactory)
		{
			IDXGIFactory *pDXGIFactory = NULL;
			HRESULT hr = pCreateDXGIFactory(__uuidof(IDXGIFactory), (LPVOID*)&pDXGIFactory);

			if (SUCCEEDED(hr))
			{
				EnumerateUsingDXGI(pDXGIFactory);

				SAFE_RELEASE(pDXGIFactory);

				if (!s_dxgiAdapters.empty())
				{
					// TODO: determine what adapter is used by NeL
					res = s_dxgiAdapters.front().memory;
				}
				else
				{
					nlwarning("Unable to find an DXGI adapter");
				}
			}
			else
			{
				nlwarning("Unable to create DXGI factory");
			}
		}
		else
		{
			nlwarning("dxgi.dll missing entry-point");
		}

		FreeLibrary(hDXGI);
	}

	if (res == -1)
	{
		// using DirectDraw
		HMODULE hInstDDraw = LoadLibraryA("ddraw.dll");

		if (hInstDDraw)
		{
			SAdapter adapter;
			adapter.hMonitor = MonitorFromWindow(s_window, MONITOR_DEFAULTTONULL);

			LPDIRECTDRAWENUMERATEEXA pDirectDrawEnumerateEx = (LPDIRECTDRAWENUMERATEEXA)GetProcAddress(hInstDDraw, "DirectDrawEnumerateExA");
			LPDIRECTDRAWCREATE pDDCreate = (LPDIRECTDRAWCREATE)GetProcAddress(hInstDDraw, "DirectDrawCreate");

			if (pDirectDrawEnumerateEx && pDDCreate)
			{
				HRESULT hr = pDirectDrawEnumerateEx(DDEnumCallbackEx, (VOID*)&adapter, DDENUM_ATTACHEDSECONDARYDEVICES);

				if (SUCCEEDED(hr) && adapter.found)
				{
					LPDIRECTDRAW pDDraw = NULL;
					hr = pDDCreate(&adapter.guid, &pDDraw, NULL);

					if (SUCCEEDED(hr))
					{
						LPDIRECTDRAW7 pDDraw7 = NULL;
						hr = pDDraw->QueryInterface(IID_IDirectDraw7, (VOID**)&pDDraw7);

						if (SUCCEEDED(hr))
						{
							DDSCAPS2 ddscaps;
							memset(&ddscaps, 0, sizeof(DDSCAPS2));
							ddscaps.dwCaps = DDSCAPS_VIDEOMEMORY | DDSCAPS_LOCALVIDMEM;

							DWORD pdwAvailableVidMem;
							hr = pDDraw7->GetAvailableVidMem(&ddscaps, &pdwAvailableVidMem, NULL);

 							if (SUCCEEDED(hr))
							{
								res = (sint)pdwAvailableVidMem / 1024;
								nlinfo("DirectDraw Adapter: %s - DedicatedVideoMemory: %d KiB", adapter.name.c_str(), adapter.memory);
							}
							else
							{
								nlwarning("Unable to get DirectDraw available video memory: %s", FormatError(hr).c_str());
							}

							SAFE_RELEASE(pDDraw7);
						}
						else
						{
							nlwarning("Unable to query IDirectDraw7 interface: %s", FormatError(hr).c_str());
						}
					}
					else
					{
						nlwarning("Unable to call DirectDrawCreate: %s", FormatError(hr).c_str());
					}
				}
				else
				{
					nlwarning("Unable to enumerate DirectDraw adapters (%s): %s", (adapter.found ? "found":"not found"), FormatError(hr).c_str());
				}
			}
			else
			{
				nlwarning("Unable to get pointer on DirectDraw functions (DirectDrawEnumerateExA %p, DirectDrawCreate %p)", pDirectDrawEnumerateEx, pDDCreate);
			}

			FreeLibrary(hInstDDraw);
		}
		else
		{
			nlwarning("Unable to load ddraw.dll");
		}
	}
#elif defined(NL_OS_MAC)
	// the right method is using OpenGL
#else
	// under Linux, no method is really reliable...
	NLMISC::CIFile file;
	
	std::string logFile = "/var/log/Xorg.0.log";

	// parse last Xorg.0.log
	if (file.open(logFile, true))
	{
		char buffer[256];

		while(!file.eof())
		{
			file.getline(buffer, 256);
			
			if (buffer[0] == '\0') break;

			std::string line(buffer);

			// nvidia driver
			std::string::size_type pos = line.find(") NVIDIA(");

			if (pos != std::string::npos)
			{
				// [    20.883] (--) NVIDIA(0): Memory: 2097152 kBytes
				pos = line.find("Memory: ", pos);

				// found memory line
				if (pos == std::string::npos) continue;
				pos += 8;

				std::string::size_type posUnits = line.find(" kBytes", pos);

				// found units in KiB
				if (posUnits == std::string::npos) continue;

				std::string videoMemory = line.substr(pos, posUnits-pos);
					
				if (!NLMISC::fromString(videoMemory, res)) continue;

				nlinfo("Xorg NVIDIA driver reported %d KiB of video memory", res);
				break;
			}

			// intel driver
			pos = line.find(") intel(");

			if (pos != std::string::npos)
			{
				// (**) intel(0): VideoRam: 131072 KB
				pos = line.find("VideoRam: ", pos);

				// found memory line
				if (pos == std::string::npos) continue;
				pos += 10;

				std::string::size_type posUnits = line.find(" KB", pos);

				// found units in KiB
				if (posUnits == std::string::npos) continue;

				std::string videoMemory = line.substr(pos, posUnits-pos);
					
				if (!NLMISC::fromString(videoMemory, res)) continue;

				nlinfo("Xorg Intel driver reported %d KiB of video memory", res);
				break;
			}

			// TODO: other drivers: nv, fglrx (ATI), radeon (ATI)
		}

		file.close();
	}

	if (res == -1)
	{
		// use lspci
		std::string command = "lspci";

		std::string out = getCommandOutput(command);

		if (out.empty())
		{
			nlwarning("Unable to launch %s", command.c_str());
		}
		else
		{
			std::vector<std::string> lines;
			std::string deviceId;

			explode(out, std::string("\n"), lines, true);
	
			// process each line
			for(uint i = 0; i < lines.size(); ++i)
			{
				std::string line = lines[i];
	
				if (line.find("VGA") == std::string::npos &&
					line.find("3D") == std::string::npos &&
					line.find("2D") == std::string::npos)
					continue;

				std::string::size_type pos = line.find(' ');
			
				if (pos == std::string::npos) continue;

				// found device ID
				deviceId = line.substr(0, pos);
				break;
			}

			if (deviceId.empty())
			{
				nlwarning("Unable to find a 3D device with lspci");
			}
			else
			{
				command = "lspci -v -s " + deviceId;

				out = getCommandOutput(command);

				if (out.empty())
				{
					nlwarning("Unable to launch %s", command.c_str());
				}
				else
				{
					explode(out, std::string("\n"), lines, true);

					// process each line
					for(uint i = 0; i < lines.size(); ++i)
					{
						std::string line = lines[i];
	
						// look for a size
						std::string::size_type pos0 = line.find("[size=");
						if (pos0 == std::string::npos) continue;

						// move to first digit
						pos0 += 6;

						// end of the size
						std::string::size_type pos1 = line.find("]", pos0);
						if (pos1 == std::string::npos) continue;

						sint units;

						if (line.substr(pos1-1, 1) == "M")
						{
							// size in MiB
							units = 1024;
							--pos1;
						}
						else if (line.substr(pos1-1, 1) == "K")
						{
							// size in KiB
							units = 1;
							--pos1;
						}
						else
						{
							// size in B
							units = 0;
						}

						// extract the size
						std::string sizeStr = line.substr(pos0, pos1-pos0);

						// convert size to integer with right units
						sint tmpSize;
						if (!NLMISC::fromString(sizeStr, tmpSize)) continue;

						tmpSize *= units;

						// take the higher size (up to 256 MiB apparently)
						if (tmpSize > res) res = tmpSize;
					}

					nlinfo("lspci reported %d KiB of video memory", res);
				}
			}
		}
	}
#endif

	return res;
}

} // NLMISC