/*
 * OpenClonk, http://www.openclonk.org
 *
 * Copyright (c) 1998-2000, 2004  Matthes Bender
 * Copyright (c) 2005  Peter Wortmann
 * Copyright (c) 2005  Günther Brammer
 * Copyright (c) 2001-2009, RedWolf Design GmbH, http://www.clonk.de
 *
 * Portions might be copyrighted by other authors who have contributed
 * to OpenClonk.
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 * See isc_license.txt for full license and disclaimer.
 *
 * "Clonk" is a registered trademark of Matthes Bender.
 * See clonk_trademark_license.txt for full license.
 */

/* Some wrappers for easier access to the Windows registry */

#ifndef INC_STDREGISTRY
#define INC_STDREGISTRY

#ifdef _WIN32
#include <windows.h>
#include "StdCompiler.h"
bool DeleteRegistryValue(HKEY hKey, const char *szSubKey,
												 const char *szValueName);
bool DeleteRegistryValue(const char *szSubKey, const char *szValueName);

bool GetRegistryDWord(HKEY hKey, const char *szSubKey,
                      const char *szValueName, DWORD *lpdwValue);
bool GetRegistryDWord(const char *szSubKey, const char *szValueName, DWORD *lpdwValue);


bool SetRegistryDWord(HKEY hKey, const char *szSubKey,
                      const char *szValueName, DWORD dwValue);
bool SetRegistryDWord(const char *szSubKey, const char *szValueName, DWORD dwValue);


bool GetRegistryString(const char *szSubKey, const char *szValueName, char *sValue, DWORD dwValSize);
bool SetRegistryString(const char *szSubKey, const char *szValueName, const char *szValue);

bool DeleteRegistryKey(HKEY hKey, const char *szSubKey);
bool DeleteRegistryKey(const char *szSubKey);

bool SetRegClassesRoot(const char *szSubKey,
                       const char *szValueName,
                       const char *szStringValue);

bool SetRegShell(const char *szClassName,
                 const char *szShellName,
                 const char *szShellCaption,
                 const char *szCommand,
								 bool fMakeDefault = false);

bool RemoveRegShell(const char *szClassName,
										const char *szShellName);

bool SetRegFileClass(const char *szClassRoot,
                     const char *szExtension,
                     const char *szClassName,
                     const char *szIconPath, int iIconNum,
										 const char *szContentType);

bool StoreWindowPosition(HWND hwnd,
												 const char *szWindowName,
												 const char *szSubKey,
												 bool fStoreSize = true);

bool RestoreWindowPosition(HWND hwnd,
													 const char *szWindowName,
													 const char *szSubKey,
													 bool fHidden = false);

// config writer
class StdCompilerConfigWrite : public StdCompiler
{
public:

	// Construct with root key
	StdCompilerConfigWrite(HKEY hRoot, const char *szPath);
	~StdCompilerConfigWrite();

  // Properties
  virtual bool hasNaming() { return true; }
	virtual bool forceWrite() { return true; }

  // Naming
  virtual bool Name(const char *szName);
  virtual void NameEnd(bool fBreak = false);
  virtual bool FollowName(const char *szName);
	virtual bool Default(const char *szName);

  // Seperators
  virtual bool Seperator(Sep eSep);

  // Data writers
  virtual void DWord(int32_t &rInt);
  virtual void DWord(uint32_t &rInt);
  virtual void Word(int16_t &rShort);
  virtual void Word(uint16_t &rShort);
  virtual void Byte(int8_t &rByte);
  virtual void Byte(uint8_t &rByte);
  virtual void Boolean(bool &rBool);
  virtual void Character(char &rChar);
  virtual void String(char *szString, size_t iMaxLength, RawCompileType eType = RCT_Escaped);
  virtual void String(char **pszString, RawCompileType eType = RCT_Escaped);
  virtual void Raw(void *pData, size_t iSize, RawCompileType eType = RCT_Escaped);

  // Passes
  virtual void Begin();
  virtual void End();

private:

	// Key stack
	int iDepth;
	struct Key {
		StdStrBuf Name;
		HKEY Handle;
		Key *Parent;
	} *pKey;

	// Writing
	void CreateKey(HKEY hParent = 0);
	void WriteDWord(uint32_t iVal);
	void WriteString(const char *szStr);

};

// config reader
class StdCompilerConfigRead : public StdCompiler
{
public:

	// Construct with root key
	StdCompilerConfigRead(HKEY hRoot, const char *szPath);
	~StdCompilerConfigRead();

  // Properties
	virtual bool isCompiler() { return true; }
  virtual bool hasNaming() { return true; }

  // Naming
  virtual bool Name(const char *szName);
  virtual void NameEnd(bool fBreak = false);
  virtual bool FollowName(const char *szName);

  // Seperators
  virtual bool Seperator(Sep eSep);

  // Data writers
  virtual void DWord(int32_t &rInt);
  virtual void DWord(uint32_t &rInt);
  virtual void Word(int16_t &rShort);
  virtual void Word(uint16_t &rShort);
  virtual void Byte(int8_t &rByte);
  virtual void Byte(uint8_t &rByte);
  virtual void Boolean(bool &rBool);
  virtual void Character(char &rChar);
  virtual void String(char *szString, size_t iMaxLength, RawCompileType eType = RCT_Escaped);
  virtual void String(char **pszString, RawCompileType eType = RCT_Escaped);
  virtual void Raw(void *pData, size_t iSize, RawCompileType eType = RCT_Escaped);

  // Passes
  virtual void Begin();
  virtual void End();

private:

	// Key stack
	int iDepth;
	struct Key {
		StdStrBuf Name;
		HKEY Handle; // for keys only
		Key *Parent;
		bool Virtual;
		DWORD Type; // for values only
	} *pKey;

	// Reading
	uint32_t ReadDWord();
	StdStrBuf ReadString();

};

#endif

#endif // INC_STDREGISTRY