// ECSERVERDlg.cpp : 實現文件
//

#include "stdafx.h"
#include "ECSERVER.h"
#include "ECSERVERDlg.h"

#include ".\ecserverdlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// 用于應用程序“關于”菜單項的 CAboutDlg 對話框
////////////////////////////////////////////////////////////
//全局或者靜態變量初始化
int CECSERVERDlg::m_conn_num = 0;
CECSERVERDlg* pDlg;//窗口指針，便于線程執行體獲得
bool ThreadRun[MAX_CONNECT_NUM];
class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// 對話框數據
	enum { IDD = IDD_ABOUTBOX };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 實現
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()


// CECSERVERDlg 對話框



CECSERVERDlg::CECSERVERDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CECSERVERDlg::IDD, pParent)
	, m_pSet(NULL)	
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
	
}

void CECSERVERDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_EDIT1, m_message_list);
}

BEGIN_MESSAGE_MAP(CECSERVERDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_WM_CLOSE()
END_MESSAGE_MAP()


// CECSERVERDlg 消息處理程序

BOOL CECSERVERDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// 將\“關于...\”菜單項添加到系統菜單中。

	// IDM_ABOUTBOX 必須在系統命令范圍內。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// 設置此對話框的圖標。當應用程序主窗口不是對話框時，框架將自動
	//  執行此操作
	SetIcon(m_hIcon, TRUE);			// 設置大圖標
	SetIcon(m_hIcon, FALSE);		// 設置小圖標

	// TODO: 在此添加額外的初始化代碼
	pDlg = this;//維護窗口指針
	ServerInit();//服務器初始化

	for(int i=0;i<MAX_CONNECT_NUM;i++)
	{
		ThreadRun[i] = true;
	}

		try
	{
	    CString  sDSN="ODBC;DRIVER=Microsoft Access Driver (*.mdb);DSN='';DBQ=system.mdb";
     
		m_database = new CDatabase;
		if(!m_database->Open(NULL,FALSE,FALSE,sDSN,FALSE))
		{      
			CDialog::EndDialog(0);
			return FALSE;
		}	
        //打開數據庫的BOOK表設置記錄集
		CString strSQL="select * from BOOK";
		m_pSet = new CRecordset(m_database);
		m_pSet->Open(CRecordset::dynaset,strSQL);
	
		if(!m_pSet->Requery())	AfxMessageBox("can not requery");	
		if(m_pSet->IsBOF() && m_pSet->IsEOF())	return -1;	
		
	}
	catch(CDBException* pEx)
	{
		pEx->ReportError();
	}
	return TRUE;  // 除非設置了控件的焦點，否則返回 TRUE
}

void CECSERVERDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// 如果向對話框添加最小化按鈕，則需要下面的代碼
//  來繪制該圖標。對于使用文檔/視圖模型的 MFC 應用程序，
//  這將由框架自動完成。

void CECSERVERDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于繪制的設備上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使圖標在工作矩形中居中
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 繪制圖標
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

//當用戶拖動最小化窗口時系統調用此函數取得光標顯示。
HCURSOR CECSERVERDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

int CECSERVERDlg::ServerInit(void)//服務器初始化
{
     m_psocket = new TCPSocket();
	 m_psocket->Create(PORT_LISTEN);//在端口PORT_LISTEN上創建套接字	
	m_psocket->Listen(MAX_CONNECT_NUM);//設置最大連接數為CONNECT_NUM
		
	//創建接受客戶連接的工作線程
	AfxBeginThread(ThreadAccept,this->GetSafeHwnd(),THREAD_PRIORITY_NORMAL);
    WriteStr("服務器初始化成功!");
	return 0;
}

UINT CECSERVERDlg::ThreadAccept(LPVOID lpram)//接收用戶連接線程
{
	while(m_conn_num<MAX_CONNECT_NUM)
	{
		pDlg->m_psocket->Accept(&(pDlg->clientsocket[m_conn_num]));//接受新的連接，都由clientsocket[i]來處理
		int socketid = m_conn_num;
		AfxBeginThread(ThreadProc,(LPVOID)&socketid,THREAD_PRIORITY_NORMAL);//新創建線程，專門接受發送過來的數據
        m_conn_num++;
	}
	AfxEndThread(0);

return 0;
}

UINT CECSERVERDlg::ThreadProc(LPVOID lpram)
{	
	char* buf = new char[DATAPACK_SIZE];
	DataPack* m_pdatapack = new DataPack();
    int* sockid =(int*)lpram ;	
	
	while(ThreadRun[*sockid])//運行參數
	{	
		memset(buf,0,DATAPACK_SIZE);
		memset(m_pdatapack,0,DATAPACK_SIZE);

		pDlg->clientsocket[*sockid].Recv(buf,DATAPACK_SIZE);
 
		m_pdatapack = (DataPack*)buf;

		switch(m_pdatapack->flag)
		{
		case LOGIN://用戶登陸處理
			{
                 //查找用戶
				if((pDlg->GetUser(m_pdatapack->DataInfo.user))==-1)					
					break;
				else
				{
					CString str;
					str.Format("用戶:%d登陸成功!",m_pdatapack->DataInfo.user.id);
					pDlg->WriteStr(str);
					//通知用戶登陸成功,并把個人信息發送給用戶
					m_pdatapack->flag = LOGINSUCCESS;
					pDlg->clientsocket[*sockid].Send((const char*)m_pdatapack,DATAPACK_SIZE);
					//把書目信息發送給用戶
					pDlg->SendBookInfo(*sockid);
					//把該用戶的購書信息發送給用戶
					pDlg->SendBuyBookInfo(*sockid,m_pdatapack->DataInfo.user.id);

				}	
				break;
			}
		case BUYBOOK://用戶購書
			{
				CString str;
				str.Format("用戶:%d購書成功",m_pdatapack->DataInfo.buyitem.user.id);
				pDlg->WriteStr(str);
				//用戶購書處理
				pDlg->BuyBook(m_pdatapack->DataInfo.buyitem);
                //通知用戶購書成功          
                m_pdatapack->flag = BUYBOOK_SUCCESS;
                pDlg->clientsocket[*sockid].Send((const char*)m_pdatapack,DATAPACK_SIZE);
				break;
			}
		case DROPBOOK://用戶退書
			{
				CString str;
				str.Format("用戶:%d退購成功",m_pdatapack->DataInfo.buyitem.user.id);
				pDlg->WriteStr(str);
				//用戶退購處理
				if(pDlg->DropBook(m_pdatapack->DataInfo.buyitem)==0)
				{
					m_pdatapack->flag = DROPBOOK_SUCCESS;
					pDlg->clientsocket[*sockid].Send((const char*)m_pdatapack,DATAPACK_SIZE);
				}					
				break;
			}
		case LOGOUT:
			{//用戶退出
				CString str;
				str.Format("用戶:%d退出",m_pdatapack->DataInfo.buyitem.user.id);
				pDlg->WriteStr(str);
				pDlg->clientsocket[*sockid].Close; 
				ThreadRun[*sockid] = false;
				break;
			}
		}


	}

	AfxEndThread(0);//結束線程

return 0;
}
void CECSERVERDlg::OnClose()
{
	// TODO: 在此添加消息處理程序代碼和/或調用默認值
	CDialog::OnClose();
}

//本函數用于從數據庫中根據用戶ID:userin.id查找用戶user,找到的話把其余信息放在user中,否則直接注冊
int CECSERVERDlg::GetUser(User& user)//
{
	CString m_errmsg;
	CDBVariant m_value;
    //安全性處理
	if(user.id==0||strcmp(user.name,"")==0||strcmp(user.password,"")==0)
	{
		return -1;
	}

	TRY
	{	

		CString strSQL; 
		strSQL.Format("select * from User where USER_ID=%d AND USER_PASSWORD='%s'",user.id,user.password);
		CRecordset* m_pSet_tmp = new CRecordset(m_database);
		m_pSet_tmp->Open(CRecordset::dynaset,strSQL);
    
	if(m_pSet_tmp->IsBOF())//沒有這個id,插入新紀錄注冊這個用戶
	{
		CString strSQL ;
		strSQL.Format("insert into User(USER_ID,USER_NAME,USER_PASSWORD) values(%d,'%s',%s)",user.id,user.name,user.password);
		m_database->ExecuteSQL(strSQL);
		return 0;
	}	 
    //得到對應id的用戶名
	 m_pSet_tmp->GetFieldValue("USER_NAME",m_value);	 
	 strcpy(user.name,m_value.m_pstring->GetBuffer(0)); 
	 m_pSet_tmp->Close();

	}
	CATCH_ALL(e)
	{
		e->GetErrorMessage(m_errmsg.GetBuffer(128),128);
		AfxMessageBox(m_errmsg);
		return -1;
	}
	END_CATCH_ALL
	return 0;
}
int CECSERVERDlg::SendBookInfo(int socketid)//向用戶發送書目信息
{
	CString temp;	
	DataPack datapack;
	memset(&datapack,0,DATAPACK_SIZE);
	datapack.flag = BOOK;
    //依次把書目信息發送給用戶
    m_pSet->MoveFirst();
	while( !m_pSet->IsEOF() )
	{
		m_pSet->GetFieldValue("BOOK_ID",temp);
        datapack.DataInfo.book.book_id = atoi(temp.GetBuffer(0));

		m_pSet->GetFieldValue("BOOK_NAME",temp);
		strcpy(datapack.DataInfo.book.book_name,temp.GetBuffer(0));

		m_pSet->GetFieldValue("BOOK_AUTHOR",temp);
		strcpy(datapack.DataInfo.book.book_author,temp.GetBuffer(0));

		m_pSet->GetFieldValue("PUBLISHER",temp);
		strcpy(datapack.DataInfo.book.publisher,temp.GetBuffer(0));

		m_pSet->GetFieldValue("PUBLISH_TIME",temp);
		strcpy(datapack.DataInfo.book.publish_time,temp.GetBuffer(0));

		m_pSet->GetFieldValue("PRICE",temp);
		strcpy(datapack.DataInfo.book.price,temp.GetBuffer(0));

		m_pSet->GetFieldValue("LEFT_NUM",temp);
		datapack.DataInfo.book.left_num = atoi(temp.GetBuffer(0));

		m_pSet->GetFieldValue("PRIVIEW",temp);
		strcpy(datapack.DataInfo.book.priview,temp.GetBuffer(0));
        //向用戶發送書目信息
		clientsocket[socketid].Send((const char*)&datapack,DATAPACK_SIZE);

		m_pSet->MoveNext();
	}

	return 0;
}

int CECSERVERDlg::BuyBook(BuyItem& buyitem)//購書處理
{
	CString m_errmsg;	

	TRY
	{//向表BUY_ITEM插入購書記錄
		CString strSQL ;
		strSQL.Format("insert into BUY_ITEM(BOOK_ID,BOOK_NAME,USER_ID,BUY_NUM) values(%d,'%s',%d,%d)",buyitem.book.book_id,buyitem.book.book_name,buyitem.user.id,buyitem.buy_num);
		m_database->ExecuteSQL(strSQL);
	}
	CATCH_ALL(e)
	{
		e->GetErrorMessage(m_errmsg.GetBuffer(128),128);
		AfxMessageBox(m_errmsg);
		return -1;
	}
	END_CATCH_ALL
	
	return 0;
}

int CECSERVERDlg::SendBuyBookInfo(int sockid, int user_id)//發送用戶的購書信息
{
	CString m_errmsg;
	CString temp;

	DataPack datapack;
    memset(&datapack,0,DATAPACK_SIZE);

	datapack.flag = BUYBOOK_SUCCESS;

	TRY
	{	
       //從表BUY_ITEM中查找用戶的購書記錄
		CString strSQL; 
		strSQL.Format("select * from BUY_ITEM where USER_ID=%d",user_id);
		CRecordset* m_pSet_tmp = new CRecordset(m_database);
		m_pSet_tmp->Open(CRecordset::dynaset,strSQL);
    
	if(m_pSet_tmp->IsBOF())
	return -1; 
    //發送用戶的購書記錄
	while(!m_pSet_tmp->IsEOF())
	{
		m_pSet_tmp->GetFieldValue("BOOK_ID",temp);
		datapack.DataInfo.buyitem.book.book_id = atoi(temp.GetBuffer(0));
		m_pSet_tmp->GetFieldValue("BOOK_NAME",temp);		
		strcpy(datapack.DataInfo.buyitem.book.book_name,temp);
		m_pSet_tmp->GetFieldValue("USER_ID",temp);
		datapack.DataInfo.buyitem.user.id = atoi(temp.GetBuffer(0));
		m_pSet_tmp->GetFieldValue("BUY_NUM",temp);
		datapack.DataInfo.buyitem.buy_num = atoi(temp.GetBuffer(0));
		clientsocket[sockid].Send((const char*)&datapack,DATAPACK_SIZE);
		m_pSet_tmp->MoveNext();
	} 

	}
	CATCH_ALL(e)
	{
		e->GetErrorMessage(m_errmsg.GetBuffer(128),128);
		AfxMessageBox(m_errmsg);
		return -1;
	}
	END_CATCH_ALL

	return 0;
}

int CECSERVERDlg::DropBook(BuyItem& buyitem)//退購處理
{
	CString m_errmsg;	

	TRY
	{//刪除用戶的購書記錄
		CString strSQL ;
		strSQL.Format("delete * from BUY_ITEM where USER_ID = %d AND BOOK_ID = %d",buyitem.user.id,buyitem.book.book_id);
		m_database->ExecuteSQL(strSQL);
	}
	CATCH_ALL(e)
	{
		e->GetErrorMessage(m_errmsg.GetBuffer(128),128);
		AfxMessageBox(m_errmsg);
		return -1;
	}
	END_CATCH_ALL
	return 0;
}

int CECSERVERDlg::WriteStr(CString str)//向界面上輸出狀態信息
{
	CString strTemp = str;	
	strTemp += _T("\r\n");	
	m_message_list.ReplaceSel(strTemp);
	return 0;
}