//Copyright (c) 1998 - 1999 Microsoft Corporation
#include "precomp.h"
#include "utils.h"
#include <assert.h>

static DWORD	g_dwAuthRetCode	= ERROR_SUCCESS;

DWORD WINAPI AuthThread(void *pData);


LRW_DLG_INT CALLBACK 
AuthProc(
    IN HWND     hwnd,	
    IN UINT     uMsg,		
    IN WPARAM   wParam,	
    IN LPARAM   lParam 	
    );


LRW_DLG_INT CALLBACK 
WelcomeDlgProc(IN HWND     hwnd,	
			   IN UINT     uMsg,		
			   IN WPARAM   wParam,	
			   IN LPARAM   lParam)
{
	DWORD	dwRetCode = ERROR_SUCCESS;
	DWORD	dwNextPage = 0;	
    BOOL	bStatus = TRUE;

    PageInfo *pi = (PageInfo *)LRW_GETWINDOWLONG( hwnd, LRW_GWL_USERDATA );


    switch (uMsg) 
    {
    case WM_INITDIALOG:
        pi = (PageInfo *)((LPPROPSHEETPAGE)lParam)->lParam;
        LRW_SETWINDOWLONG( hwnd, LRW_GWL_USERDATA, (LRW_LONG_PTR)pi );        

        SetControlFont( pi->hBigBoldFont, hwnd, IDC_BIGBOLDTITLE);	    
		GetGlobalContext()->ClearWizStack();
        break;

    case WM_DESTROY:
        LRW_SETWINDOWLONG( hwnd, LRW_GWL_USERDATA, NULL );
        break;

    case WM_NOTIFY:
        {
            LPNMHDR pnmh = (LPNMHDR)lParam;

            switch( pnmh->code )
            {
            case PSN_SETACTIVE:                
				{
					TCHAR	szMsg[LR_MAX_MSG_TEXT];

					SetDlgItemText(hwnd, IDC_PRODUCTID, GetGlobalContext()->GetLicenseServerID());

					if (GetGlobalContext()->GetWizAction() != WIZACTION_REGISTERLS &&
						GetGlobalContext()->GetWizAction() != WIZACTION_CONTINUEREGISTERLS)
					{
						SetDlgItemText(hwnd, IDC_COMPANYNAME, GetGlobalContext()->GetContactDataObject()->sCompanyName);					

						if (GetGlobalContext()->GetContactDataObject()->sProgramName == PROGRAM_SELECT)
						{
							LoadString(GetInstanceHandle(),IDS_AGREEMENT_SELECT,szMsg,LR_MAX_MSG_TEXT);
						}
						else if (GetGlobalContext()->GetContactDataObject()->sProgramName == PROGRAM_MOLP)
						{
							LoadString(GetInstanceHandle(),IDS_AGREEMENT_MOLP,szMsg,LR_MAX_MSG_TEXT);
						}
						else
						{
							LoadString(GetInstanceHandle(),IDS_AGREEMENT_OTHER,szMsg,LR_MAX_MSG_TEXT);
						}

						SetDlgItemText(hwnd, IDC_LICENSETYPE, szMsg);

						if(GetGlobalContext()->GetActivationMethod() == CONNECTION_INTERNET)
						{
							LoadString(GetInstanceHandle(),IDS_INTERNETMODE,szMsg,LR_MAX_MSG_TEXT);
							SetDlgItemText(hwnd, IDC_ACTIVATIONMETHOD, szMsg);
						}

						if(GetGlobalContext()->GetActivationMethod() == CONNECTION_WWW)
						{
							LoadString(GetInstanceHandle(),IDS_WWWMODE,szMsg,LR_MAX_MSG_TEXT);
							SetDlgItemText(hwnd, IDC_ACTIVATIONMETHOD, szMsg);
						}

						if(GetGlobalContext()->GetActivationMethod() == CONNECTION_PHONE)
						{
							LoadString(GetInstanceHandle(),IDS_TELEPHONEMODE,szMsg,LR_MAX_MSG_TEXT);
							SetDlgItemText(hwnd, IDC_ACTIVATIONMETHOD, szMsg);
						}

					}
					PropSheet_SetWizButtons( GetParent( hwnd ), PSWIZB_NEXT );

					switch (GetGlobalContext()->GetWizAction())
					{
					case WIZACTION_DOWNLOADLKP:
						LoadString(GetInstanceHandle(), IDS_MSG_DOWNLOADLKP, szMsg,LR_MAX_MSG_TEXT);
						break;

					case WIZACTION_REGISTERLS:
						LoadString(GetInstanceHandle(), IDS_MSG_UNREGISTERED, szMsg,LR_MAX_MSG_TEXT);
						break;

					case WIZACTION_CONTINUEREGISTERLS:
						LoadString(GetInstanceHandle(), IDS_MSG_CONTINUEREGISTRATION, szMsg,LR_MAX_MSG_TEXT);
						break;

					case WIZACTION_REREGISTERLS:
						LoadString(GetInstanceHandle(), IDS_MSG_REREGISTRATION, szMsg,LR_MAX_MSG_TEXT);
						break;

					case WIZACTION_UNREGISTERLS:
						LoadString(GetInstanceHandle(), IDS_MSG_UNREGISTRATION, szMsg,LR_MAX_MSG_TEXT);
						break;

					case WIZACTION_DOWNLOADLASTLKP:
						LoadString(GetInstanceHandle(), IDS_MSG_DOWNLOADLASTLKP, szMsg,LR_MAX_MSG_TEXT);
						break;
					}
					SetDlgItemText(hwnd, IDC_BOLDTITLE, szMsg);
				}
                break;

            case PSN_WIZNEXT:
				if (GetGlobalContext()->GetWizAction() == WIZACTION_DOWNLOADLASTLKP)
				{
					if (GetGlobalContext()->GetActivationMethod() == CONNECTION_INTERNET)
					{
						// Solve Bug 610 BEGIN
						dwRetCode = ShowProgressBox(hwnd, AuthThread, 0, 0, 0);
						if(g_dwAuthRetCode == ERROR_SUCCESS)
						{						
							//Previos code BEGIN
							dwRetCode = ShowProgressBox(hwnd, ProcessThread, 0, 0, 0);
							if (dwRetCode == ERROR_SUCCESS)
							{
								dwRetCode = LRGetLastRetCode();
							}

							if (dwRetCode != ERROR_SUCCESS)
							{
								LRMessageBox(hwnd, dwRetCode);
								dwNextPage = IDD_WELCOME;
							}
							else
							{
								dwNextPage = IDD_PROGRESS;
							}
							//Previous code END
						}
						else if (g_dwAuthRetCode == IDS_ERR_SPKBAD ||
								 g_dwAuthRetCode == IDS_ERR_CERTREVOKED)
						{	
							TCHAR	szMsg[LR_MAX_MSG_TEXT];
							TCHAR	szCaption[LR_MAX_MSG_CAPTION];	

							LoadString(GetInstanceHandle(),g_dwAuthRetCode, szMsg,LR_MAX_MSG_TEXT);
							LoadString(GetInstanceHandle(),IDS_TITLE,szCaption,LR_MAX_MSG_CAPTION);

							if (MessageBox(hwnd,szMsg, szCaption, MB_YESNO | MB_ICONQUESTION | MB_DEFBUTTON2) == IDYES)
							{
								SetCursor(LoadCursor(NULL,IDC_WAIT));
								g_dwAuthRetCode = ResetLSSPK();
								SetCursor(LoadCursor(NULL,IDC_ARROW));
								if (g_dwAuthRetCode == ERROR_SUCCESS)
								{
									// Start all over again
									GetGlobalContext()->SetLRState(LRSTATE_NEUTRAL);
									GetGlobalContext()->SetLSStatus(LSERVERSTATUS_UNREGISTER);
									GetGlobalContext()->SetWizAction(WIZACTION_REGISTERLS);
									GetGlobalContext()->ClearWizStack();
									dwNextPage = IDD_WELCOME;
								}
								else
								{
									LRMessageBox(hwnd,g_dwAuthRetCode,LRGetLastError());
									dwNextPage = IDD_WELCOME;
								}
							}
							else
							{
								dwNextPage = IDD_WELCOME;
							}
						}
						else if (g_dwAuthRetCode == IDS_ERR_CERTBAD ||
								 g_dwAuthRetCode == IDS_ERR_CERTEXPIRED)
						{
							TCHAR	szMsg[LR_MAX_MSG_TEXT];
							TCHAR	szCaption[LR_MAX_MSG_CAPTION];	

							LoadString(GetInstanceHandle(), g_dwAuthRetCode, szMsg, LR_MAX_MSG_TEXT);
							LoadString(GetInstanceHandle(), IDS_TITLE, szCaption, LR_MAX_MSG_CAPTION);

							if (MessageBox(hwnd,szMsg, szCaption, MB_YESNO | MB_ICONQUESTION | MB_DEFBUTTON2) == IDYES)
							{
								// Go to Reissue Cert
								GetGlobalContext()->SetLRState(LRSTATE_NEUTRAL);
								GetGlobalContext()->SetLSStatus(LSERVERSTATUS_REGISTER_INTERNET);
								GetGlobalContext()->SetWizAction(WIZACTION_REREGISTERLS);
								dwNextPage = IDD_DLG_CERTLOG_INFO;
							}
							else
							{
								dwNextPage = IDD_WELCOME;
							}
						}
						else
						{
							LRMessageBox(hwnd,g_dwAuthRetCode,LRGetLastError());
							dwNextPage = IDD_WELCOME;
						}
					}
					else
					{
						dwNextPage = GetGlobalContext()->GetEntryPoint();
					}
					// Solve Bug 610 END
				}
				else if (GetGlobalContext()->GetWizAction() == WIZACTION_REGISTERLS)
				{
                    dwNextPage = IDD_DLG_GETREGMODE;
				}
				else if (GetGlobalContext()->GetWizAction() == WIZACTION_DOWNLOADLKP ||
						 GetGlobalContext()->GetWizAction() == WIZACTION_UNREGISTERLS ||
						 GetGlobalContext()->GetWizAction() == WIZACTION_REREGISTERLS)
				{
					if (GetGlobalContext()->GetActivationMethod() == CONNECTION_INTERNET)
					{
						dwRetCode = ShowProgressBox(hwnd, AuthThread, 0, 0, 0);
						if(g_dwAuthRetCode == ERROR_SUCCESS)
						{
							dwNextPage = GetGlobalContext()->GetEntryPoint();
						}
						else if (g_dwAuthRetCode == IDS_ERR_SPKBAD ||
								 g_dwAuthRetCode == IDS_ERR_CERTREVOKED)
						{	
							TCHAR	szMsg[LR_MAX_MSG_TEXT];
							TCHAR	szCaption[LR_MAX_MSG_CAPTION];	

							LoadString(GetInstanceHandle(),g_dwAuthRetCode, szMsg,LR_MAX_MSG_TEXT);
							LoadString(GetInstanceHandle(),IDS_TITLE,szCaption,LR_MAX_MSG_CAPTION);

							if (MessageBox(hwnd,szMsg, szCaption, MB_YESNO | MB_ICONQUESTION | MB_DEFBUTTON2) == IDYES)
							{
								SetCursor(LoadCursor(NULL,IDC_WAIT));
								g_dwAuthRetCode = ResetLSSPK();
								SetCursor(LoadCursor(NULL,IDC_ARROW));
								if (g_dwAuthRetCode == ERROR_SUCCESS)
								{
									// Start all over again
									GetGlobalContext()->SetLRState(LRSTATE_NEUTRAL);
									GetGlobalContext()->SetLSStatus(LSERVERSTATUS_UNREGISTER);
									GetGlobalContext()->SetWizAction(WIZACTION_REGISTERLS);
									GetGlobalContext()->ClearWizStack();
									dwNextPage = IDD_WELCOME;
								}
								else
								{
									LRMessageBox(hwnd,g_dwAuthRetCode,LRGetLastError());
									dwNextPage = IDD_WELCOME;
								}
							}
							else
							{
								dwNextPage = IDD_WELCOME;
							}
						}
						else if (g_dwAuthRetCode == IDS_ERR_CERTBAD ||
								 g_dwAuthRetCode == IDS_ERR_CERTEXPIRED)
						{
							TCHAR	szMsg[LR_MAX_MSG_TEXT];
							TCHAR	szCaption[LR_MAX_MSG_CAPTION];	

							LoadString(GetInstanceHandle(), g_dwAuthRetCode, szMsg, LR_MAX_MSG_TEXT);
							LoadString(GetInstanceHandle(), IDS_TITLE, szCaption, LR_MAX_MSG_CAPTION);

							if (MessageBox(hwnd,szMsg, szCaption, MB_YESNO | MB_ICONQUESTION | MB_DEFBUTTON2) == IDYES)
							{
								// Go to Reissue Cert
								GetGlobalContext()->SetLRState(LRSTATE_NEUTRAL);
								GetGlobalContext()->SetLSStatus(LSERVERSTATUS_REGISTER_INTERNET);
								GetGlobalContext()->SetWizAction(WIZACTION_REREGISTERLS);
								dwNextPage = IDD_DLG_CERTLOG_INFO;
							}
							else
							{
								dwNextPage = IDD_WELCOME;
							}
						}
						else
						{
							LRMessageBox(hwnd,g_dwAuthRetCode,LRGetLastError());
							dwNextPage = IDD_WELCOME;
						}
					}
					else
					{
						dwNextPage = GetGlobalContext()->GetEntryPoint();
					}
				}
				else
				{
					dwNextPage = GetGlobalContext()->GetEntryPoint();
				}

				LRW_SETWINDOWLONG(hwnd,  LRW_DWL_MSGRESULT, dwNextPage);
				if (dwNextPage != IDD_WELCOME)
				{
					LRPush(IDD_WELCOME);
				}
				bStatus = -1;
				break;

            default:
                bStatus = FALSE;
                break;
            }
        }
        break;
	
    default:
        bStatus = FALSE;
        break;
    }
    return bStatus;
}








DWORD WINAPI AuthThread(void *pData)
{
	g_dwAuthRetCode = AuthenticateLS();	

	ExitThread(0);

	return 0;
}