//---------------------------------------------------------------------------
#include <vcl.h>
#include <tchar.h>
#include <System.IOUtils.hpp>
#include <Winternl.h>
#pragma hdrstop

#include "CommonV8Handler.h"
#include "GlobalUnit.h"

//---------------------------------------------------------------------------
#include <Vcl.Styles.hpp>
#include <Vcl.Themes.hpp>
#pragma package(smart_init)

/////////////////////////////////////////////////////////////////////////////
// 获取父进程函数
typedef struct _PROCESS_BASIC_INFORMATIONWOW64 {
    NTSTATUS ExitStatus;
	ULONG64 PebBaseAddress;
    ULONG64 AffinityMask;
	LONG 	BasePriority;
    ULONG64 UniqueProcessId;
	ULONG64 InheritedFromUniqueProcessId;
} PROCESS_BASIC_INFORMATION_WOW64;

typedef LONG (WINAPI *PNTQUERYINFORMATIONPROCESS)(HANDLE,UINT,PVOID,ULONG,PULONG);

DWORD GetParentProcessId(DWORD pid)//获取指定进程的父进程ID
{
	DWORD dwParentPID = 0;
	HANDLE hProcess = OpenProcess(PROCESS_QUERY_INFORMATION, FALSE, pid);
	if( !hProcess ){
		//ShowMessage( "OpenProcess err" );
		return -1;
	}
	PNTQUERYINFORMATIONPROCESS  NtQueryInformationProcess = NULL;
	BOOL bProcWow = FALSE;
	IsWow64Process(hProcess, &bProcWow);
	if( bProcWow==TRUE ){
		//ShowMessage( "32bit" );
		NtQueryInformationProcess = (PNTQUERYINFORMATIONPROCESS)GetProcAddress(GetModuleHandleA("ntdll.dll"),
														"NtWow64QueryInformationProcess64");
		PROCESS_BASIC_INFORMATION_WOW64 pbi;
		NTSTATUS status = NtQueryInformationProcess(hProcess, ProcessBasicInformation, (PVOID)&pbi,
													sizeof(PROCESS_BASIC_INFORMATION_WOW64), NULL);
		//ShowMessage("NtWow64Query return: " + IntToHex((int)status));
		if( NT_SUCCESS(status) )
			dwParentPID = (DWORD)pbi.InheritedFromUniqueProcessId;
	}
	else{
		//ShowMessage( "64bit" );
		NtQueryInformationProcess = (PNTQUERYINFORMATIONPROCESS)GetProcAddress(GetModuleHandleA("ntdll.dll"),
														"NtQueryInformationProcess");
		PROCESS_BASIC_INFORMATION pbi;
		NTSTATUS status = NtQueryInformationProcess(hProcess, ProcessBasicInformation, (PVOID)&pbi,
													sizeof(PROCESS_BASIC_INFORMATION), NULL);
		//ShowMessage("NtQuery return: " + IntToHex((int)status));
		if( NT_SUCCESS(status) )
			dwParentPID = (DWORD)pbi.Reserved3;
	}
	CloseHandle(hProcess);


	return dwParentPID;
}

/////////////////////////////////////////////////////////////////
// 定义枚举窗口回调函数
BOOL CALLBACK EnumWndProc(HWND hwnd, LPARAM lParam)
{
	HANDLE h = GetProp(hwnd, _TEXT(APPABBR));
	if(h == (HANDLE)SOFTWARE_ID) {
		*(HANDLE*)lParam = hwnd;
		return FALSE;
	}

    return TRUE;
}

/////////////////////////////////////////////////////////////////
// CEF初始化函数

bool InitCef()
{
	GlobalCEFApp = new TCefApplication;

	GlobalCEFApp->Cache = String(g_AppPath) + "cache";
	GlobalCEFApp->Locale = "zh-CN";
	GlobalCEFApp->AcceptLanguageList = "zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6";

	if(g_bTesting){
		GlobalCEFApp->LogFile = "debug-" + IntToStr(DateTimeToUnix(Now(), false)) + ".log";
		GlobalCEFApp->LogSeverity = LOGSEVERITY_INFO; //LOGSEVERITY_VERBOSE; //
	}
	else{
		GlobalCEFApp->LogSeverity = LOGSEVERITY_DISABLE;
	}

	GlobalCEFApp->CheckCEFFiles = false;
//	GlobalCEFApp->PackLoadingDisabled = false;
	GlobalCEFApp->PersistSessionCookies = true;
//	GlobalCEFApp->PersistUserPreferences = true;
	GlobalCEFApp->IgnoreCertificateErrors = true;
	GlobalCEFApp->NoSandbox = true;
	GlobalCEFApp->MuteAudio = true;
	GlobalCEFApp->EnableGPU = true;
	GlobalCEFApp->SitePerProcess = true;              //78.0以上iframe运行js代码必须设置
	GlobalCEFApp->DisableWebSecurity = true;           //78.0以上iframe运行js代码必须设置
	GlobalCEFApp->DisableSiteIsolationTrials = true;   //78.0以上iframe运行js代码必须设置
	GlobalCEFApp->DisableChromeLoginPrompt = true;
	GlobalCEFApp->EnableMediaStream = false; //true; //
	GlobalCEFApp->LogProcessInfo = false; //true; //
	GlobalCEFApp->SingleProcess = false; //true; //
	GlobalCEFApp->MultiThreadedMessageLoop = true;
	GlobalCEFApp->WindowlessRenderingEnabled = true;
	GlobalCEFApp->DisableImageLoading = true;

	GlobalCEFApp->OnWebKitInitialized = _di_TOnWebKitInitializedEvent(new TWebKitInitializeEvent());

	if( !GlobalCEFApp->StartMainProcess() )
		return false;

	return true;
}

/////////////////////////////////////////////////////////////////
// 子进程判断函数
bool IsRestart()
{
    HWND hWnd = NULL;
    if(SINGLE_INSTANCE){
	    EnumWindows(EnumWndProc,(LPARAM)&hWnd);    //枚举所有运行的窗口
    }
    else{
		String strFullName = TPath::GetFullPath(Application->ExeName);
		HWND hWnd = FindWindow( (LPCTSTR)strFullName.c_str(), NULL );
    }

	if( hWnd ){
		//ShowMessage("restart");

		DWORD wndProcess_id = 0;
		DWORD parent_id=GetParentProcessId(GetCurrentProcessId());
		GetWindowThreadProcessId( hWnd, &wndProcess_id );

		if( wndProcess_id==parent_id ){
			//ShowMessage("SubProcess");
			InitCef();
			return true;
		}
        else{
            SendMessage( hWnd, UM_RESTORE, 0, 0 );
            return true;
		}
	}

	return false;
}
//---------------------------------------------------------------------------

/////////////////////////////////////////////////////////////////
// 程序自动升级函数
void StartUpgrade( int mode )
{
	HANDLE hToken = NULL,
		hTokenDup = NULL;
	if(OpenProcessToken(GetCurrentProcess(),TOKEN_ALL_ACCESS,&hToken)){
		if(DuplicateTokenEx(hToken, TOKEN_ALL_ACCESS,NULL, SecurityIdentification,
													TokenPrimary, &hTokenDup)){
			STARTUPINFO si;
			PROCESS_INFORMATION pi;

			ZeroMemory(&si,sizeof(STARTUPINFO));
			ZeroMemory(&pi,sizeof(PROCESS_INFORMATION));
			si.cb = sizeof(STARTUPINFO);
			si.lpDesktop = NULL;
			si.lpReserved = NULL;
			si.lpTitle = NULL;
			si.cbReserved2 = NULL;
			si.lpReserved2 = NULL;
			si.dwFlags = STARTF_USESHOWWINDOW;
			si.wShowWindow = mode==2 ? SW_SHOWNORMAL //显示进程窗口
									: SW_HIDE;    	//隐藏进程窗口

			String command = "\"" + (String)g_AppPath + UPDATE_APP_NAME + "\"";
			if( mode==3  )
				command += " move ";
			else if( mode==2 )
				command += " show ";
			else
				command += " hide ";

			command += ExtractFileName(Application->ExeName);
			command += " " + IntToStr(SOFTWARE_ID);
            command += " " + String(GLOBAL_VAR_FILE);

			CreateProcessAsUser(hTokenDup,NULL,command.c_str(),NULL,NULL,FALSE,0,NULL,NULL,&si,&pi);
		}
	}
}
//---------------------------------------------------------------------------
extern bool g_bBacData;

void InitUserPath()
{
	String sAppPath = g_AppPath;
	String sUserPath = sAppPath + "user\\";
	if(!DirectoryExists( sUserPath, false)){
		ForceDirectories( sUserPath );
	}

	if(g_bBacData){
		String sDataPath = sAppPath + "data\\";
		if(!DirectoryExists( sDataPath, false)){
			ForceDirectories( sDataPath );
		}
	}
}
//---------------------------------------------------------------------------

USEFORM("RBmasterMain.cpp", MainForm);
//---------------------------------------------------------------------------
bool Startup()
{
	String strAppPath = ExtractFileDir(ParamStr(0)) + "\\";
	strcpy( g_AppPath, ((AnsiString)strAppPath).c_str() );

	for(int i=1; i<=ParamCount(); i++){
		String sParam = LowerCase(ParamStr(i));
		if (sParam == "-d")
			g_bBacData = true;
		else if (sParam == "-testing")
			g_bTesting = true;
	}

	if( IsRestart() ){
		return false;
	}

	try{
		Application->Initialize();

		int tag = 0;
		if( !InitCef() ){
			tag = -5;
		}
		else{
			InitUserPath();
			Application->Title = APPNAME;
			Application->CreateForm(__classid(TMainForm), &MainForm);
			Application->MainFormOnTaskBar = true;
			tag = Application->MainForm->Tag;
		}

		bool bRun(0);
		if( tag==0 ){
			bRun = true;
		}
		else if( tag==-1 ){
			ShowMessage( "加载应用模块失败，退出程序" );
		}
		else if( tag==-3 ){
			ShowMessage( "连接服务器失败，退出程序" );
		}
		else if( tag==-4 ){
			ShowMessage( "软件已停止运行，退出程序" );
		}
		else if( tag==-5 ){
			ShowMessage( "初始化网络模块失败，退出程序" );
			return false;
		}
		else{
			int mode = tag;
			if( tag==4 ){
				int mrRet = MessageBox(NULL, L"检测到有新的版本，是否立即升级？",
										APPNAME, MB_ICONQUESTION | MB_YESNO);
				if( mrRet==mrYes )
					mode = 2;
				else
					mode = 0;
			}
			else if( tag==2 ){
				ShowMessage( "软件运行需更新相关组件，即将启动更新程序" );
			}

			bRun = mode<=1;

			if( mode>0 )
				StartUpgrade( mode );
		}

		if(bRun){
			if(!g_bTesting)
				SetProp(Application->MainFormHandle, _TEXT(APPABBR), (HANDLE)SOFTWARE_ID);

			Application->HintHidePause = 10000;
			Application->MainForm->Show();

			Application->Run();

			if(!g_bTesting)
				RemoveProp(Application->MainFormHandle, _TEXT(APPABBR));
		}

		DestroyGlobalCEFApp();
	}
	catch (Exception &exception)
	{
		Application->ShowException(&exception);
	}
	catch (...)
	{
		try
		{
			throw Exception("");
		}
		catch (Exception &exception)
		{
			Application->ShowException(&exception);
		}
	}

	return true;
}
//---------------------------------------------------------------------------

int WINAPI _tWinMain(HINSTANCE, HINSTANCE, LPTSTR, int)
{
	if( !Startup() )
		return -1;

	return 0;
}
//---------------------------------------------------------------------------