1 files changed, 342 insertions, 0 deletions
diff --git a/Src/auth/Loginbox/pageAddress.cpp b/Src/auth/Loginbox/pageAddress.cpp
new file mode 100644
index 00000000..6aaf9b79
--- /dev/null
+++ b/Src/auth/Loginbox/pageAddress.cpp
@@ -0,0 +1,342 @@
+#include "./pageAddress.h"
+#include "./dataAddress.h"
+#include "./common.h"
+#include "./addressEncoder.h"
+#include "./addressEditbox.h"
+#include "./loginGui.h"
+
+#include "../resource.h"
+
+#include <shlwapi.h>
+#include <wininet.h>
+
+
+
+static HRESULT CALLBACK LoginPageAddress_CreateInstance(HWND hwnd, HWND hLoginbox, LoginPage **instance)
+{
+	if (NULL == instance) return E_POINTER;
+	if (NULL == hwnd || NULL == hLoginbox) return E_INVALIDARG;
+
+	*instance = new LoginPageAddress(hwnd, hLoginbox);
+	if (NULL == instance) return E_OUTOFMEMORY;
+
+	return S_OK;
+}
+
+LoginPageAddress::LoginPageAddress(HWND hwnd, HWND hLoginbox)
+	: LoginPage(hwnd, hLoginbox)
+{
+	HWND hEdit = GetDlgItem(hwnd, IDC_ADDRESS_EDIT);
+	if (NULL != hEdit)
+		AddressEditbox_AttachWindow(hEdit);
+
+}
+
+LoginPageAddress::~LoginPageAddress()
+{
+}
+
+HWND LoginPageAddress::CreatePage(HWND hLoginbox, HWND hParent)
+{
+	return LoginPage::CreatePage(hLoginbox, MAKEINTRESOURCE(IDD_PAGE_ADDRESS), 
+						hParent, NULL, LoginPageAddress_CreateInstance);
+}
+
+BOOL LoginPageAddress::OnInitDialog(HWND hFocus, LPARAM param)
+{
+	HFONT fontEdit = NULL, fontLabel = NULL;
+	LoginGuiObject *loginGui;
+	if (SUCCEEDED(LoginGuiObject::QueryInstance(&loginGui)))
+	{
+		fontLabel = loginGui->GetTextFont();
+		fontEdit = loginGui->GetEditorFont();
+		loginGui->Release();
+	}
+
+	if (NULL != fontLabel)
+	{
+		HWND hLabel = GetDlgItem(hwnd, IDC_ADDRESS_LABEL);
+		if (NULL != hLabel)
+			SendMessage(hLabel, WM_SETFONT, (WPARAM)fontLabel, 0L);
+
+		hLabel = GetDlgItem(hwnd, IDC_MESSAGE);
+		if (NULL != hLabel)
+			SendMessage(hLabel, WM_SETFONT, (WPARAM)fontLabel, 0L);
+	}
+
+	if (NULL != fontEdit)
+	{
+		HWND hEdit = GetDlgItem(hwnd, IDC_ADDRESS_EDIT);
+		if (NULL != hEdit)
+			SendMessage(hEdit, WM_SETFONT, (WPARAM)fontEdit, 0L);
+	}
+
+	LoginPage::OnInitDialog(hFocus, param);
+	return FALSE;
+}
+
+void LoginPageAddress::OnDestroy()
+{
+	LoginPage::OnDestroy();
+}
+
+void LoginPageAddress::UpdateLayout(BOOL fRedraw)
+{
+	LoginPage::UpdateLayout(fRedraw);
+
+	RECT pageRect;
+	GetPageRect(&pageRect);
+	
+	UINT flags = SWP_NOACTIVATE | SWP_NOZORDER | SWP_NOREDRAW;
+	
+	const INT szControls[] = { IDC_ADDRESS_LABEL, IDC_ADDRESS_EDIT, IDC_MESSAGE, };
+	HDWP hdwp = BeginDeferWindowPos(ARRAYSIZE(szControls));
+
+	INT baseunitX, baseunitY;
+	if (FALSE == LoginBox_GetWindowBaseUnits(hwnd, &baseunitX, &baseunitY))
+	{
+		baseunitY = 13;
+	}
+
+	HRGN invalidRegion = CreateRectRgn(0, 0, 0, 0);
+	HRGN tempRegion = CreateRectRgn(0, 0, 0, 0);
+	HWND hControl;
+
+	INT nextTop = pageRect.top;
+	INT x, y, cx, cy;
+
+	RECT rect;
+
+	INT maxWidth = pageRect.right - pageRect.left;
+	hControl = GetDlgItem(hwnd, IDC_ADDRESS_EDIT);
+	if (NULL != hControl)
+	{
+		HDC hdc = GetDCEx(hControl, NULL, DCX_CACHE | DCX_NORESETATTRS);
+		if (NULL != hdc)
+		{
+			HFONT fontControl = (HFONT)SendMessage(hControl, WM_GETFONT, 0, 0L);
+			HFONT fontOrig = (HFONT)SelectObject(hdc, fontControl);
+			
+			maxWidth = LoginBox_GetAveStrWidth(hdc, 80);
+			if(maxWidth > (pageRect.right - pageRect.left))
+				maxWidth = pageRect.right - pageRect.left;
+
+			SelectObject(hdc, fontOrig);
+			ReleaseDC(hControl, hdc);
+		}
+	}
+
+	for (INT i = 0; i < ARRAYSIZE(szControls); i++)
+	{
+		hControl = GetDlgItem(hwnd, szControls[i]);
+		if (NULL == hControl || FALSE == GetWindowRect(hControl, &rect)) continue;
+		MapWindowPoints(HWND_DESKTOP, hwnd, (POINT*)&rect, 2);
+		x = rect.left;
+		y = rect.top;
+		cx = rect.right - rect.left;
+		cy = rect.bottom - rect.top;
+		switch(szControls[i])
+		{
+			case IDC_ADDRESS_LABEL:
+			case IDC_MESSAGE:
+				x = pageRect.left;
+				y = nextTop;
+				LoginBox_GetWindowTextSize(hControl, maxWidth, &cx, &cy);
+				nextTop += (cy + MulDiv(2, baseunitY, 8));
+				break;
+			case IDC_ADDRESS_EDIT:
+				x = pageRect.left;
+				y = nextTop;
+				cy = LoginBox_GetWindowTextHeight(hControl, 0);
+				{
+					RECT r1, r2;
+					GetWindowRect(hControl, &r1);
+					SendMessage(hControl, EM_GETRECT, 0, (LPARAM)&r2);
+					INT t = (r1.bottom - r1.top) - (r2.bottom - r2.top);
+					cy += t;
+				}
+				cx = maxWidth;
+				nextTop += (cy + MulDiv(7, baseunitY, 8));
+				break;
+		}
+
+		if (rect.left != x || rect.top != y ||
+			(rect.right - rect.left) != cx || (rect.bottom - rect.top) != cy)
+		{
+			hdwp = DeferWindowPos(hdwp, hControl, NULL, x, y, cx, cy, flags);
+			if (NULL == hdwp) break;
+
+			if (FALSE != fRedraw)
+			{
+				SetRectRgn(tempRegion, rect.left, rect.top, rect.right, rect.bottom);
+				CombineRgn(invalidRegion, invalidRegion, tempRegion, RGN_OR);
+
+				SetRectRgn(tempRegion, x, y , x + cx, y + cy);
+				CombineRgn(invalidRegion, invalidRegion, tempRegion, RGN_OR);
+			}
+		}
+	}
+
+	if (NULL != hdwp)
+		EndDeferWindowPos(hdwp);
+
+	if (FALSE != fRedraw)
+	{
+		RedrawWindow(hwnd, NULL, invalidRegion, 
+			RDW_INVALIDATE | RDW_UPDATENOW | RDW_ERASE | RDW_ERASENOW | RDW_ALLCHILDREN | RDW_VALIDATE);
+	}
+
+	if (NULL != invalidRegion)
+		DeleteObject(invalidRegion);
+	if (NULL != tempRegion)
+		DeleteObject(tempRegion);
+}
+
+BOOL LoginPageAddress::OnGetLoginData(LoginData **ppLoginData)
+{
+	if (NULL == ppLoginData)
+		return FALSE;
+
+	HWND hEdit;
+	LPWSTR address;
+
+	hEdit  = GetDlgItem(hwnd, IDC_ADDRESS_EDIT);
+	if (NULL == hEdit || FAILED(LoginBox_GetWindowText(hEdit, &address, NULL)))
+		address = NULL;
+	else
+	{
+		size_t encodedMax = lstrlen(address)*2;
+		LPWSTR addressEncoded = LoginBox_MallocString(encodedMax);
+
+		for(;;)
+		{
+			if (FALSE == InternetCanonicalizeUrl(address, addressEncoded, (DWORD*)&encodedMax, ICU_ENCODE_PERCENT | ICU_NO_META))
+			{
+				if (ERROR_INSUFFICIENT_BUFFER == GetLastError())
+				{
+					LoginBox_FreeString(addressEncoded);
+					addressEncoded = LoginBox_MallocString(encodedMax);
+					if (NULL != addressEncoded)
+						continue;
+				}
+			}
+			else
+			{
+				LoginBox_FreeString(address);
+				address = LoginBox_CopyString(addressEncoded);
+			}
+			break;
+		}
+
+		HRESULT hr;
+		for(;;)
+		{
+			hr = AddressEncoder_EncodeString(address, addressEncoded, &encodedMax, ICU_BROWSER_MODE);
+			if (ENC_E_INSUFFICIENT_BUFFER == hr)
+			{
+				LoginBox_FreeString(addressEncoded);
+				addressEncoded = LoginBox_MallocString(encodedMax);
+				if (NULL == addressEncoded)
+					hr = E_OUTOFMEMORY;
+				else
+					continue;
+			}
+			break;
+		} 
+
+		if (SUCCEEDED(hr))
+		{
+			LoginBox_FreeString(address);
+			address = addressEncoded;
+			addressEncoded = NULL;
+		}
+		else
+			LoginBox_FreeString(addressEncoded);
+	}
+
+	HRESULT hr = LoginDataAddress::CreateInstance(NULL, hwnd, hLoginbox, 
+					address, (LoginDataAddress**)ppLoginData);
+
+	LoginBox_FreeString(address);
+	return SUCCEEDED(hr);
+}
+
+BOOL LoginPageAddress::OnSetUsername(LPCWSTR pszUsername)
+{
+	HWND hEdit = GetDlgItem(hwnd, IDC_ADDRESS_EDIT);
+	if (NULL == hEdit) return FALSE;
+
+	if (NULL == pszUsername || L'\0' == *pszUsername)
+		return FALSE;
+
+	INT cchLen = lstrlen(pszUsername);
+
+	INT f, l;
+	SNDMSG(hEdit, EM_GETSEL, (WPARAM)&f, (LPARAM)&l);
+	if (f == l) return FALSE;
+	
+	SNDMSG(hEdit, EM_REPLACESEL, FALSE, (LPARAM)pszUsername);
+	SNDMSG(hEdit, EM_SETSEL, (WPARAM)f, (LPARAM)f + cchLen);
+
+	return TRUE;
+
+}
+HBRUSH LoginPageAddress::OnGetStaticColor(HDC hdc, HWND hControl)
+{
+	INT_PTR controlId = (INT_PTR)GetWindowLongPtr(hControl, GWLP_ID);
+	switch(controlId)
+	{
+		case IDC_MESSAGE:
+			SetTextColor(hdc, rgbSecondaryText);
+			SetBkColor(hdc, rgbBack);
+			return hbrBack;
+	}
+
+	return LoginPage::OnGetStaticColor(hdc, hControl);
+}
+
+BOOL LoginPageAddress::OnSetAddress(LPCWSTR pszAddress, BOOL replaceUsername)
+{
+	HWND hEdit = GetDlgItem(hwnd, IDC_ADDRESS_EDIT);
+	if (NULL == hEdit) return FALSE;
+	
+	BOOL succeeded;
+	LPWSTR addressDecoded;
+
+	if (SUCCEEDED(AddressEncoder_DecodeString(pszAddress, &addressDecoded)))
+	{
+		succeeded = (BOOL)SNDMSG(hEdit, WM_SETTEXT, 0, (LPARAM)addressDecoded);
+		LoginBox_FreeString(addressDecoded);
+	}
+	else
+		succeeded = (BOOL)SNDMSG(hEdit, WM_SETTEXT, 0, (LPARAM)pszAddress);
+
+
+	return succeeded;
+}
+
+BOOL LoginPageAddress::OnSetAddressTitle(LPCWSTR pszTitle)
+{
+	return SetLabelText(IDC_ADDRESS_LABEL, pszTitle);
+}
+
+BOOL LoginPageAddress::OnSetMessage(LPCWSTR pszMessage)
+{
+	HWND hLabel = GetDlgItem(hwnd, IDC_MESSAGE);
+	if (NULL == hLabel) return FALSE;
+	
+	return SetWindowText(hLabel, pszMessage);
+}
+
+INT_PTR LoginPageAddress::DialogProc(UINT uMsg, WPARAM wParam, LPARAM lParam)
+{
+	switch(uMsg)
+	{
+		case NLPAM_SETADDRESS:		MSGRESULT(hwnd, OnSetAddress((LPCWSTR)lParam, (BOOL)wParam));
+		case NLPAM_SETADDRESSTITLE:	MSGRESULT(hwnd, OnSetAddressTitle((LPCWSTR)lParam));
+		case NLPAM_SETMESSAGE:		MSGRESULT(hwnd, OnSetMessage((LPCWSTR)lParam));
+
+	}
+	return __super::DialogProc(uMsg, wParam, lParam);
+}
+