This commit is contained in:
2013-07-20 23:20:35 +00:00
parent 130d285297
commit b06b1aef3a

590
StockServer/MainFrm.cpp Normal file
View File

@@ -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();
}