// ManageRegistry.cpp: implementation of the CManageRegistry class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "cubridtray.h"
#include "ManageRegistry.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

#define	REGKEY_ROOT					"SOFTWARE\\CUBRID"
#define	REGKEY_PROD_ROOT			"RootPath"
#define REGKEY_JAVA					"Software\\JavaSoft\\Java Runtime Environment"


//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CManageRegistry::CManageRegistry( char* sprodname )
{
	hKeyReg     = NULL;
	sResultName = NULL;

	if( sprodname )
	{
		memset( sKeyPath, 0x00, sizeof( sKeyPath ) );
		sprintf( sKeyPath, "%s\\%s", REGKEY_ROOT, sprodname );
		bOpenRegistry();
	}
}

CManageRegistry::~CManageRegistry()
{
	if( hKeyReg  ) RegCloseKey( hKeyReg );
	if( sResultName ) delete sResultName;

	hKeyReg  = NULL;
	sResultName = NULL;
}



bool CManageRegistry::bOpenRegistry()
{
	DWORD   rc;

	if( !sKeyPath )
		return false;

	rc = RegOpenKeyEx( HKEY_LOCAL_MACHINE,
						sKeyPath,
						0,
						KEY_QUERY_VALUE,
						&hKeyReg );

	if (rc != ERROR_SUCCESS) return false;

	return true;
}



char* CManageRegistry::sGetItem( char* sItemName )
{
	char* sValue;
	char* sBuff = new char[2048];

	DWORD   rc;
	DWORD   len;
	DWORD   dwType;

	if( !sItemName || !hKeyReg || !sKeyPath ) return NULL;

	memset(sBuff, 0x00, 2048);

	rc = RegQueryValueEx( hKeyReg, sItemName, 0, &dwType, (LPBYTE)sBuff, &len );

	if (rc != ERROR_SUCCESS) return NULL;
	if( !sBuff || strlen( sBuff ) <= 0 ) return NULL;

	sValue = new char[strlen( sBuff ) + 1];

	memset( sValue, 0x00, strlen( sBuff ) + 1 );
	strcpy( sValue, sBuff );
	delete sBuff;

	return sValue;
}




char* CManageRegistry::sGetString( char* sItemName, char* sCmdName )
{
	char* sPath = sGetItem( sItemName );

	if( !sPath ) return NULL;

	int dSize = strlen( sPath ) + strlen( sCmdName );
	char* sFullName = new char[ dSize + 5 ];
	memset( sFullName, 0x00, dSize + 5 );
	sprintf( sFullName, "%s\\%s", sPath, sCmdName );

	return sFullName;
}



int CManageRegistry::dGetString( char* sItemName, char* sCmdName )
{
	char* sPath = sGetItem( sItemName );

	if( !sPath ) return 0;

	int dSize = strlen( sPath ) + strlen( sCmdName );
	sResultName = new char[ dSize + 5 ];
	memset( sResultName, 0x00, dSize + 5 );
	sprintf( sResultName, "%s\\%s", sPath, sCmdName );

	return strlen( sResultName );
}


bool CManageRegistry::bGetString( char* sString )
{
	strcpy( sString, sResultName );
	return true;
}



bool CManageRegistry::GetJavaRootPath(char *path)
{
	DWORD   rc;
	DWORD   len;
	DWORD   dwType;

	char currentVersion[16];
	char regkey_java_current_version[2048];
	char java_root_path[2048];

	if (!path) return false;

	rc = RegOpenKeyEx(  HKEY_LOCAL_MACHINE,
						REGKEY_JAVA,
						0,
						KEY_QUERY_VALUE,
						&hKeyReg );

	if (rc != ERROR_SUCCESS) return false;

	len =  sizeof(currentVersion);
	rc = RegQueryValueEx( hKeyReg, "CurrentVersion", 0, &dwType, (LPBYTE)currentVersion, &len );

	if( hKeyReg  ) RegCloseKey( hKeyReg );

	if (rc != ERROR_SUCCESS) return false;

	hKeyReg = NULL;

	sprintf(regkey_java_current_version, "%s\\%s", REGKEY_JAVA, currentVersion);
	rc = RegOpenKeyEx(  HKEY_LOCAL_MACHINE,
						regkey_java_current_version,
						0,
						KEY_QUERY_VALUE,
						&hKeyReg );

	if (rc != ERROR_SUCCESS) return false;
	len = sizeof(java_root_path);
	rc = RegQueryValueEx( hKeyReg, "JavaHome", 0, &dwType, (LPBYTE)java_root_path, &len );

	if (rc != ERROR_SUCCESS) return false;

	strcpy(path, java_root_path);

	return true;
}