#include <windows.h>
#include "datas.h"
#include "AdoConnect.h"
#include "resource.h"

static USERINFO * reguser;
void Register(HWND, USERINFO *);
LRESULT CALLBACK dpRegister(HWND, UINT, WPARAM, LPARAM);

void Register(HWND hWnd, USERINFO * ui)
{
	reguser = ui;
	DialogBox((HINSTANCE)GetWindowLong(hWnd, GWL_HINSTANCE), 
		(LPCTSTR)IDD_REGISTER, hWnd, (DLGPROC)dpRegister);
}

LRESULT CALLBACK dpRegister(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
	AdoConnect ado;
	TCHAR password[MAX_STRING];
	TCHAR lisence[MAX_STRING]; 
  	_bstr_t bstr;
	_RecordsetPtr pr;

	switch (message)
	{
		case WM_COMMAND:
			if (LOWORD(wParam) == IDOK)
			{			
				/*開始登錄*/ 
				SendMessage(GetDlgItem(hDlg, IDC_RE_LISENCE), WM_GETTEXT, MAX_STRING-1, (LPARAM)lisence);
				SendMessage(GetDlgItem(hDlg, IDC_RE_PASSWORD), WM_GETTEXT, MAX_STRING-1, (LPARAM)password);
				if(0 == lstrcmp(lisence, "") || 0 == lstrcmp(password, ""))
				{
					MessageBox(hDlg, "證號和密碼不能為空!", "Fail!", MB_OK);
					return TRUE;
				}
				else
				{
					BOOL successed = FALSE;
					ado.OnInitDBConnect();

					/*讀者登錄*/
					if(SendMessage(GetDlgItem(hDlg, IDC_RE_READER), BM_GETCHECK, 0, 0))
					{
						UniteBSTR(bstr, var "select * from reader where lisence='", var lisence, var "' and password='",
							var password, var "';");
						pr = ado.GetRecordset(bstr);

						if(!pr->adoBOF && !pr->adoEOF)
						{
							reguser->ut = UT_READER;
							reguser->id = atoi(lpstr pr->GetCollect("id"));	
							successed = TRUE;
						}
					}
					
					/*普管登錄*/
					if(SendMessage(GetDlgItem(hDlg, IDC_RE_COMADM), BM_GETCHECK, 0, 0))
					{
						UniteBSTR(bstr, var "select * from comadm where lisence='", var lisence, var "' and password='",
							var password, var "';");
						pr = ado.GetRecordset(bstr);

						if(!pr->adoBOF && !pr->adoEOF)
						{
							reguser->ut = UT_COMADM;
							reguser->id = atoi(lpstr pr->GetCollect("id"));	
							successed = TRUE;
						}						
					}
					
					/*系管登錄*/
					if(SendMessage(GetDlgItem(hDlg, IDC_RE_SYSADM), BM_GETCHECK, 0, 0))
					{
						UniteBSTR(bstr, var "select * from sysadm where lisence='", var lisence, var "' and password='",
							var password, var "';");
						pr = ado.GetRecordset(bstr);

						if(!pr->adoBOF && !pr->adoEOF)
						{
							reguser->ut = UT_SYSADM;
							reguser->id = atoi(lpstr pr->GetCollect("id"));	
							successed = TRUE;
						}
					}

					ado.ExitConnect();

					if(successed == FALSE)
					{
 						MessageBox(GetParent(hDlg), "用戶不存在!", "Fail!", MB_ICONSTOP|MB_OK);
						return TRUE;
					}
				}
				EndDialog(hDlg, LOWORD(wParam));
			}
			
			if(LOWORD(wParam) == IDCANCEL) 
			{
				EndDialog(hDlg, LOWORD(wParam));
				SendMessage(GetParent(hDlg), WM_CLOSE, 0, 0);
				return TRUE;
			}
			return TRUE;//WM_COMMAND end

		default:
			return FALSE;
	}
}