From b06b1aef3ad8430e1895353b70722043a0400d3a Mon Sep 17 00:00:00 2001 From: mjjo Date: Sat, 20 Jul 2013 23:20:35 +0000 Subject: [PATCH] --- StockServer/MainFrm.cpp | 590 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 590 insertions(+) create mode 100644 StockServer/MainFrm.cpp diff --git a/StockServer/MainFrm.cpp b/StockServer/MainFrm.cpp new file mode 100644 index 0000000..7a767bf --- /dev/null +++ b/StockServer/MainFrm.cpp @@ -0,0 +1,590 @@ + // MainFrm.cpp : implementation of the CMainFrame class +// + +#include "stdafx.h" +#include "StockServer.h" + +#include "MainFrm.h" +#include "UserListView.h" +#include "MessagesView.h" +#include "ServerSocket.h" +#include "..\\Common\\IPInfos.h" + +#ifdef _DEBUG +#define new DEBUG_NEW +#undef THIS_FILE +static char THIS_FILE[] = __FILE__; +#endif + +///////////////////////////////////////////////////////////////////////////// +// CMainFrame + +IMPLEMENT_DYNCREATE(CMainFrame, CMainTray) + +BEGIN_MESSAGE_MAP(CMainFrame, CMainTray) + //{{AFX_MSG_MAP(CMainFrame) + ON_WM_CREATE() + ON_WM_DESTROY() + ON_WM_GETMINMAXINFO() + ON_WM_CLOSE() + ON_UPDATE_COMMAND_UI(ID_INDICATOR_CONNECTION_STATUS, OnUpdateConnectionStatus) + ON_UPDATE_COMMAND_UI(ID_INDICATOR_CONNECTION_CLIENTS, OnUpdateConnectionClients) + ON_UPDATE_COMMAND_UI(ID_INDICATOR_NBMESSAGES, OnUpdateNbMessages) + ON_WM_TIMER() + ON_COMMAND(ID_VIEW_PARAMETERS, OnViewParameters) + ON_UPDATE_COMMAND_UI(ID_VIEW_PARAMETERS, OnUpdateViewParameters) + ON_COMMAND(ID_OPTIONS_ALWAYSVISIBLE, OnOptionsAlwaysvisible) + ON_UPDATE_COMMAND_UI(ID_OPTIONS_ALWAYSVISIBLE, OnUpdateOptionsAlwaysvisible) + ON_COMMAND(ID_FILE_SAVE_AS, OnFileSaveAs) + ON_UPDATE_COMMAND_UI(ID_FILE_SAVE_AS, OnUpdateFileSaveAs) + ON_COMMAND(ID_OPERATIONS_STARTSERVER, OnOperationsStartserver) + ON_UPDATE_COMMAND_UI(ID_OPERATIONS_STARTSERVER, OnUpdateOperationsStartserver) + ON_COMMAND(ID_OPERATIONS_STOPSERVER, OnOperationsStopserver) + ON_UPDATE_COMMAND_UI(ID_OPERATIONS_STOPSERVER, OnUpdateOperationsStopserver) + ON_WM_MENUCHAR() + ON_WM_INITMENUPOPUP() + ON_COMMAND(ID_TRAYMENU_SHOWSTOCKSERVER, OnTraymenuShowStockServer) + //}}AFX_MSG_MAP +END_MESSAGE_MAP() + +static UINT indicators[] = +{ + ID_SEPARATOR, // status line indicator + ID_INDICATOR_CONNECTION_STATUS, + ID_INDICATOR_CONNECTION_CLIENTS, + ID_INDICATOR_NBMESSAGES, + ID_INDICATOR_TIME, +}; + +///////////////////////////////////////////////////////////////////////////// +// CMainFrame construction/destruction + +CMainFrame::CMainFrame() +{ + m_pApp = (CStockServerApp*)AfxGetApp(); + + m_pUserListView = NULL; + m_pMessagesView = NULL; + + m_pServerSocket = NULL; + m_bServerStarted = FALSE; + m_bWindowOnTop = m_pApp->GetProfileInt("Settings", "OnTop", FALSE); +} + +CMainFrame::~CMainFrame() +{ +} + +BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs) +{ + if( !CMainTray::PreCreateWindow(cs) ) + return FALSE; + // TODO: Modify the Window class or styles here by modifying + // the CREATESTRUCT cs + cs.style = WS_OVERLAPPEDWINDOW; + + return TRUE; +} + +int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct) +{ + if (CMainTray::OnCreate(lpCreateStruct) == -1) + return -1; + + SetMessageText("프로그램 준비중..."); + + if (!m_wndToolBar.CreateEx(this, TBSTYLE_FLAT, WS_CHILD | WS_VISIBLE | CBRS_TOP + | CBRS_GRIPPER | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC) || + !m_wndToolBar.LoadToolBar(IDR_MAINFRAME)) + { + TRACE0("Failed to create toolbar\n"); + return -1; // fail to create + } + + if (!m_DlgBarUtil.Create(this, IDD_DIALOGBAR_UTIL, CBRS_GRIPPER|CBRS_TOOLTIPS| + CBRS_FLYBY|CBRS_SIZE_DYNAMIC|TBSTYLE_FLAT|CBRS_TOP, IDD_DIALOGBAR_UTIL)) + { + TRACE0("Failed to create dialog bar from CDialogBarUtil class\n"); + return -1; // fail to create + } + + if (!m_wndStatusBar.Create(this) || + !m_wndStatusBar.SetIndicators(indicators, + sizeof(indicators)/sizeof(UINT))) + { + TRACE0("Failed to create status bar\n"); + return -1; // fail to create + } + m_wndStatusBar.SetPaneInfo(m_wndStatusBar.CommandToIndex(ID_INDICATOR_CONNECTION_STATUS), + ID_INDICATOR_CONNECTION_STATUS, + SBPS_POPOUT, 40); + + // TODO: Delete these three lines if you don't want the toolbar to + // be dockable + EnableDocking(CBRS_ALIGN_ANY); + + m_DlgBarUtil.SetWindowText("Util bar"); + m_DlgBarUtil.EnableDocking(CBRS_ALIGN_ANY); + DockControlBar(&m_DlgBarUtil); + RecalcLayout(); + + m_wndToolBar.SetWindowText("Main bar"); + m_wndToolBar.EnableDocking(CBRS_ALIGN_ANY); + CRect rc; + m_DlgBarUtil.GetWindowRect(rc); + DockControlBar(&m_wndToolBar, AFX_IDW_DOCKBAR_TOP, rc); + RecalcLayout(); + + CString szBuf; + szBuf.LoadString(AFX_IDS_APP_TITLE); + + //Change title caption + SetWindowText(szBuf); + + //Set window placement + RestoreWindowPlacement(); + + //Apply App Language +// SetLanguage(m_bEnglishLanguage); + UpdateAllView(); + UpdateMenu(); + + LoadBarState("ControlsPos"); + + // System Tray + TraySetIcon(IDR_MAINFRAME); + TraySetToolTip(szBuf); + TraySetMenu(IDR_TRAY_MENU, IDR_BITMAP_MENU); + TraySetMinimizeToTray(m_pApp->GetProfileInt("Options", "MinimizeToTray", TRUE)); + + m_bWindowOnTop = !m_bWindowOnTop; + OnOptionsAlwaysvisible(); + + SetTimer(ID_TIMER_TIME, TIMER_TIME_VALUE, NULL); + + return 0; +} + +BOOL CMainFrame::OnCreateClient(LPCREATESTRUCT /*lpcs*/, + CCreateContext* pContext) +{ + int cxCur[2]={0,}; + cxCur[0] = m_pApp->GetProfileInt("Settings", "SplitWnd0", 100); + cxCur[1] = m_pApp->GetProfileInt("Settings", "SplitWnd1", 100); + + // create splitter window + if (!m_wndSplitter.CreateStatic(this, 1, 2)) + return FALSE; + + if (!m_wndSplitter.CreateView(0, 0, RUNTIME_CLASS(CUserListView), CSize(cxCur[0], 100), pContext) || + !m_wndSplitter.CreateView(0, 1, RUNTIME_CLASS(CMessagesView), CSize(cxCur[1], 100), pContext)) + { + m_wndSplitter.DestroyWindow(); + return FALSE; + } + + m_pUserListView = GetUserListView(); + m_pMessagesView = GetMessagesView(); + return TRUE; +} + +void CMainFrame::OnClose() +{ + if (!DisconnectServer()) + return; + + KillTimer(ID_TIMER_TIME); +// SetLanguage(DEFAULT_LANGUAGE_ENGLISH, FALSE); + SaveWindowPlacement(); + SaveBarState("ControlsPos"); + m_pApp->WriteProfileInt("Settings", "OnTop", m_bWindowOnTop); + + // information about a specific row or column + int cxCur[2], cxMin[2]; + m_wndSplitter.GetColumnInfo( 0, cxCur[0], cxMin[0] ); + m_wndSplitter.GetColumnInfo( 1, cxCur[1], cxMin[1] ); + m_pApp->WriteProfileInt("Settings", "SplitWnd0", cxCur[0]); + m_pApp->WriteProfileInt("Settings", "SplitWnd1", cxCur[1]); + + CMainTray::OnClose(); +} + +void CMainFrame::OnDestroy() +{ + CMainTray::OnDestroy(); + +} + +void CMainFrame::UpdateMenu() +{ + CMenu* pMenu = GetMenu(); + if (pMenu) + pMenu->DestroyMenu(); + + m_menu.LoadMenu(IDR_MAINFRAME); + m_menu.LoadToolbar(IDR_BITMAP_MENU); + HMENU hMenu = m_menu.Detach(); + ::SetMenu(GetSafeHwnd(), hMenu); + m_hMenuDefault = hMenu; + + BCMenu::SetMenuDrawMode(1); // Original or XP style +} + +void CMainFrame::SaveWindowPlacement() +{ + WINDOWPLACEMENT wp; + GetWindowPlacement(&wp); + m_pApp->WriteProfileBinary("Settings", "WindowPos", (LPBYTE)&wp, sizeof(wp)); +} + +void CMainFrame::RestoreWindowPlacement() +{ + WINDOWPLACEMENT *lwp; + UINT nl; + + if(m_pApp->GetProfileBinary("Settings", "WindowPos", (LPBYTE*)&lwp, &nl)) + { + SetWindowPlacement(lwp); + delete [] lwp; + } + else { + CenterWindow(GetDesktopWindow()); + } +} + +void CMainFrame::UpdateAllView() +{ + m_pUserListView->UpdateCols(); + m_pMessagesView->UpdateCols(); +} + +void CMainFrame::OnGetMinMaxInfo(MINMAXINFO FAR* lpMMI) +{ + CMainTray::OnGetMinMaxInfo(lpMMI); + + CPoint minSize(MIN_FRAME_WITH, MIN_FRAME_HEIGTH); +// CPoint maxSize(640, 480); + lpMMI->ptMinTrackSize.x = minSize.x; + lpMMI->ptMinTrackSize.y = minSize.y; +// lpMMI->ptMaxTrackSize.x = maxSize.x; +// lpMMI->ptMaxTrackSize.y = maxSize.y; +} + + +/*============================================================================ + +Description: Update the status bar connection status indicator + +Return: - + +============================================================================*/ +void CMainFrame::OnUpdateConnectionStatus(CCmdUI* pCmdUI) +{ + m_wndStatusBar.SetPaneStyle(m_wndStatusBar.CommandToIndex(ID_INDICATOR_CONNECTION_STATUS), + m_bServerStarted ? SBPS_NORMAL : SBPS_POPOUT); + + pCmdUI->SetText(m_bServerStarted ? CString((LPCTSTR)IDS_SERVER_CONNECTED) : CString((LPCTSTR)IDS_SERVER_DISCONNECTED)); +} + +/*============================================================================ + +Description: Update the status bar connection status indicator + +Return: - + +============================================================================*/ +void CMainFrame::OnUpdateConnectionClients(CCmdUI* pCmdUI) +{ + CString szBuf; + int nClients = m_pServerSocket ? m_pServerSocket->GetClientsConnectedCount() : 0; + szBuf.Format(IDS_INDICATOR_CONNECTION_CLIENTS, nClients); + + pCmdUI->SetText(szBuf); +} + +/*============================================================================ + +Description: Update the status bar connection status indicator + +Return: - + +============================================================================*/ +void CMainFrame::OnUpdateNbMessages(CCmdUI* pCmdUI) +{ + CString szBuf; + int nClients = m_pServerSocket ? m_pServerSocket->GetTotalMessagesCount() : 0; + + szBuf.Format(IDS_INDICATOR_NBMESSAGES, nClients); + + pCmdUI->SetText(szBuf); +} + +void CMainFrame::OnTimer(UINT nIDEvent) +{ + switch (nIDEvent) { + + case ID_TIMER_TIME: + m_wndStatusBar.SetPaneText(m_wndStatusBar.CommandToIndex(ID_INDICATOR_TIME), + CTime::GetCurrentTime().Format("%m/%d/%Y %H:%M:%S") ); + + TraySetMinimizeToTray(m_pApp->GetProfileInt("Options", "MinimizeToTray", TRUE)); + break; + + default: + CMainTray::OnTimer(nIDEvent); + break; + } +} + +///////////////////////////////////////////////////////////////////////////// +// CMainFrame diagnostics + +#ifdef _DEBUG +void CMainFrame::AssertValid() const +{ + CMainTray::AssertValid(); +} + +void CMainFrame::Dump(CDumpContext& dc) const +{ + CMainTray::Dump(dc); +} + +#endif //_DEBUG + +///////////////////////////////////////////////////////////////////////////// +// CMainFrame message handlers + +CUserListView* CMainFrame::GetUserListView() +{ + CWnd* pWnd = m_wndSplitter.GetPane(0, 0); + CUserListView* pView = DYNAMIC_DOWNCAST(CUserListView, pWnd); + return pView; +} + +CMessagesView* CMainFrame::GetMessagesView() +{ + CWnd* pWnd = m_wndSplitter.GetPane(0, 1); + CMessagesView* pView = DYNAMIC_DOWNCAST(CMessagesView, pWnd); + return pView; +} + +//This handler ensures that keyboard shortcuts work +LRESULT CMainFrame::OnMenuChar(UINT nChar, UINT nFlags, CMenu* pMenu) +{ + LRESULT lresult; + + if (m_menu.IsMenu(pMenu)) + lresult = BCMenu::FindKeyboardShortcut(nChar, nFlags, pMenu); + else + lresult = CMainTray::OnMenuChar(nChar, nFlags, pMenu); + + return(lresult); +} + +//This handler updates the menus from time to time +void CMainFrame::OnInitMenuPopup(CMenu* pPopupMenu, UINT nIndex, BOOL bSysMenu) +{ + CMainTray::OnInitMenuPopup(pPopupMenu, nIndex, bSysMenu); + + if (!bSysMenu) { + if (m_menu.IsMenu(pPopupMenu)) + BCMenu::UpdateMenu(pPopupMenu); + } +} + +void CMainFrame::OnViewParameters() +{ + ShowControlBar(&m_DlgBarUtil, !m_DlgBarUtil.IsVisible(), FALSE); +} + +void CMainFrame::OnUpdateViewParameters(CCmdUI* pCmdUI) +{ + pCmdUI->SetCheck(m_DlgBarUtil.IsVisible()); +} + +void CMainFrame::OnTraymenuShowStockServer() +{ + TrayShowMainWindow(); +} + +void CMainFrame::OnOptionsAlwaysvisible() +{ + if (!m_bWindowOnTop) { + SetWindowPos(&wndTopMost, 0, 0, 0, 0, SWP_NOMOVE|SWP_NOSIZE); + } + else { + SetWindowPos(&wndNoTopMost, 0, 0, 0, 0, SWP_NOMOVE|SWP_NOSIZE); + } + m_bWindowOnTop = !m_bWindowOnTop; +} + +void CMainFrame::OnUpdateOptionsAlwaysvisible(CCmdUI* pCmdUI) +{ + pCmdUI->SetCheck(m_bWindowOnTop); +} + +CPtrList* CMainFrame::GetClientsList() +{ + if (!m_pServerSocket) + return NULL; + + return m_pServerSocket->GetClientsList(); +} + +void CMainFrame::UpdateClientsList(CPtrList* pClientstList) +{ + m_pUserListView->UpdateClientsList(pClientstList); +} + +void CMainFrame::OnFileSaveAs() +{ + CStdioFile file; + static CString szFileName; + CString szFileExt; + CString szFileFilter; + CString szMessage; + CString szBuf1; + CString szBuf2; + CString szBuf3; + BOOL bOpen; + + if (m_pMessagesView->ListIsEmpty()) { + AfxMessageBox(IDS_SAVE_EMPTY_LIST); + return; + } + + szFileExt.LoadString(IDS_TXT_FILE_EXT); + szFileFilter.LoadString(IDS_TXT_FILE_FLT); + + CFileDialog Dlg(FALSE, szFileExt, szFileName, OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT, szFileFilter); + + if (Dlg.DoModal() == IDOK) { + szFileName = Dlg.GetPathName(); + + try { + bOpen = file.Open(szFileName, CFile::modeWrite | CFile::modeCreate); + if (bOpen) { + + CListCtrl& lc = m_pMessagesView->GetListCtrl(); + szBuf1.LoadString(IDR_MAINFRAME); + /* szBuf2.LoadString(IDS_TAB_CLIENTS); + szBuf3.LoadString(IDS_FILE_LOG_HEADER); + szMessage.Format("%s - %s\n%s", szBuf1, szBuf2, szBuf3); + file.WriteString(szMessage); + + szBuf1.LoadString(IDS_COL_USERS); + szMessage.Format(_T("%s\n"), szBuf1); + file.WriteString(szMessage); + szBuf1.LoadString(IDS_FILE_LOG_HEADER); + file.WriteString(szBuf1); + + for (int i = 0; i < lc.GetItemCount(); i++) { + szMessage.Format(_T("%s\n"), + lc.GetItemText(i, CLIENTSLISTVIEW_COL_USER)); + file.WriteString(szMessage); + } + */ + } + } + catch (CException *e) { + e->Delete(); + } + if (bOpen) { + file.Close(); + } + else { + szBuf1.LoadString(IDS_MSG_SAVE_AS_FAILED); + szMessage.Format(szBuf1, szFileName); + AfxMessageBox(szMessage, MB_ICONSTOP); + } + } +} + +void CMainFrame::OnUpdateFileSaveAs(CCmdUI* pCmdUI) +{ + pCmdUI->Enable(!m_pMessagesView->ListIsEmpty()); +} + +void CMainFrame::OnOperationsStartserver() +{ + CString szBuf; + + UINT uPort = m_DlgBarUtil.GetPort(); + if (uPort < DEFAULT_MIN_PORT_NUMBER || uPort > DEFAULT_MAX_PORT_NUMBER) { + szBuf.Format(IDS_INVALID_PORT_NUMBER, DEFAULT_MIN_PORT_NUMBER, + DEFAULT_MAX_PORT_NUMBER); + AfxMessageBox(szBuf); + return; + } + + m_pUserListView->ClearView(); // ClearAllViews(); + + m_pServerSocket = new CServerSocket(this); + if (m_pServerSocket->BeginListening(uPort, 100)) { + + szBuf.Format("Server started and online on port %d.", uPort); + AddMessage( "Server", szBuf ); + + m_DlgBarUtil.BlockConrols(); + m_bServerStarted = TRUE; + } + else { + DisconnectServer(); + } +} + +void CMainFrame::OnUpdateOperationsStartserver(CCmdUI* pCmdUI) +{ + pCmdUI->Enable(!m_bServerStarted); +} + +void CMainFrame::OnOperationsStopserver() +{ + if (DisconnectServer()) + m_DlgBarUtil.BlockConrols(FALSE); +} + +void CMainFrame::OnUpdateOperationsStopserver(CCmdUI* pCmdUI) +{ + pCmdUI->Enable(m_bServerStarted); +} + +BOOL CMainFrame::DisconnectServer() +{ + if (!m_pServerSocket) + return TRUE; + + int nClients = m_pServerSocket->GetClientsConnectedCount(); + if (nClients > 0) { + CString szBuf; + szBuf.Format(IDS_STOP_SERVER, nClients); + int nRet = MessageBox(szBuf, CString((LPCTSTR)IDR_MAINFRAME), MB_ICONQUESTION|MB_YESNO); + if (nRet == IDNO) + return FALSE; + } + + AddMessage( "Server", "Server shutdown and offline!" ); + + if (m_pServerSocket) { + delete m_pServerSocket; + m_pServerSocket = NULL; + } + + m_bServerStarted = FALSE; + + return TRUE; +} + +/////////////////////////////////////////////////////////////////////////// +// +void CMainFrame::AddMessage(CString szName, CString szMessage) +{ + m_pMessagesView->AddMessage(GlobalHelper::GetLocalTime(), szName, szMessage); +} + +void CMainFrame::ClearAllViews() +{ + m_pMessagesView->ClearView(); + m_pUserListView->ClearView(); +}