resistancemonitor/4.Data/TimeSyncServerDlg.cpp

// TimeSyncServerDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "TimeSyncServer.h"
#include "TimeSyncServerDlg.h"
#include "afxdialogex.h"
#include "Simplelog.h"
#include "MonitorObject.h"
#include "AppService.h"
#include "SuperManager.h"
#include "315ClientManager.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialogEx
{
public:
	CAboutDlg();

// 对话框数据
#ifdef AFX_DESIGN_TIME
	enum { IDD = IDD_ABOUTBOX };
#endif

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

// 实现
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX)
{
}

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

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()


// CTimeSyncServerDlg 对话框



CTimeSyncServerDlg::CTimeSyncServerDlg(CWnd* pParent /*=NULL*/)
	: CDialogEx(IDD_TIMESYNCSERVER_DIALOG, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDI_DATASERVICS);
	m_strSvrPath = CSimpleLog::GetAppDir() + AfxGetAppName() + ".exe";
}

void CTimeSyncServerDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CTimeSyncServerDlg, CDialogEx)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BUTTON1, &CTimeSyncServerDlg::OnBnClickedButton1)
	ON_BN_CLICKED(IDC_BUTTON2, &CTimeSyncServerDlg::OnBnClickedButton2)
	ON_BN_CLICKED(IDC_BUTTON3, &CTimeSyncServerDlg::OnBnClickedButton3)
	ON_BN_CLICKED(IDC_BUTTON4, &CTimeSyncServerDlg::OnBnClickedButton4)
	ON_BN_CLICKED(IDC_BTN_READ, &CTimeSyncServerDlg::OnBnClickedBtnRead)
	ON_MESSAGE(WM_SHOWTASK, OnShowTask)
	ON_COMMAND(ID_POPMENU_SHOW_MAIN_WND, OnTrayMenuShowMainWnd)
	ON_COMMAND(ID_POPMENU_CLOSE_CLIENT, OnTrayMenuCloseClient)
	ON_WM_TIMER()
	ON_MESSAGE(WM_START, &CTimeSyncServerDlg::OnStartapp)
	ON_WM_CLOSE()
	ON_BN_CLICKED(IDC_CHK_DATA_COMPRESSION, &CTimeSyncServerDlg::OnBnClickedChkDataCompression)
END_MESSAGE_MAP()


// CTimeSyncServerDlg 消息处理程序

void CTimeSyncServerDlg::ToTray()
{
	m_nid.cbSize = (DWORD)sizeof(NOTIFYICONDATA);
	m_nid.hWnd = this->m_hWnd;
	m_nid.uID = IDI_DATASERVICS;//图标的ID
	m_nid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
	m_nid.uCallbackMessage = WM_SHOWTASK;//自定义的消息名称，在推盘图标上任意动作就产生此消息
	m_nid.hIcon = LoadIcon(AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_DATASERVICS));
	strcpy_s(m_nid.szTip, "数据服务");    //信息提示条 
	Shell_NotifyIcon(NIM_ADD, &m_nid);    //在托盘区添加图标 
	ShowWindow(SW_HIDE);    //隐藏主窗口
	m_bIsShow = false;
}

void CTimeSyncServerDlg::DeleteTray()
{
	m_nid.cbSize = (DWORD)sizeof(NOTIFYICONDATA);
	m_nid.hWnd = this->m_hWnd;
	m_nid.uID = IDI_DATASERVICS;//删除图标，仅使用ID即可（我们使用的是图标的ID）
	//nid.uFlags=NIF_ICON;//删除图标，仅使用ICON即可
	/*nid.uFlags=NIF_ICON|NIF_MESSAGE|NIF_TIP ;
	nid.uCallbackMessage=WM_SHOWTASK;//自定义的消息名称
	nid.hIcon=LoadIcon(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDR_MAINFRAME));
	strcpy(nid.szTip,"程序名称");    //信息提示条为"计划任务提醒"
	*/
	Shell_NotifyIcon(NIM_DELETE, &m_nid);    //在托盘区删除图标
}

VOID CTimeSyncServerDlg::OnTrayMenuShowMainWnd()
{
	//ShowWindow(SW_SHOW);
	::SetWindowPos(
		GetSafeHwnd(),
		HWND_TOP,
		0,
		0,
		0,
		0,
		SWP_NOMOVE | SWP_NOSIZE | SWP_NOACTIVATE);
	ShowWindow(SW_RESTORE);

	// 置顶
	SetForegroundWindow();
	m_bIsShow = true;
}

void CTimeSyncServerDlg::OnTrayMenuCloseClient()
{
	DeleteTray();//删除托盘图标
	SPDLOG_ERROR("主动退出程序");
	TerminateProcess(INVALID_HANDLE_VALUE, 0);
	CAppService::Instance()->Stop();
	CTimeSyncServerDlg::OnCancel();
}

BOOL CTimeSyncServerDlg::OnInitDialog()
{
	CDialogEx::OnInitDialog();

	// 将“关于...”菜单项添加到系统菜单中。

	// IDM_ABOUTBOX 必须在系统命令范围内。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

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

	// 设置此对话框的图标。  当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	SetWindowText(AfxGetAppName());

	//托盘图标
	m_nid.cbSize = (DWORD)sizeof(NOTIFYICONDATA);
	m_nid.hWnd = this->m_hWnd;
	m_nid.uID = IDI_DATASERVICS;//图标的ID
	m_nid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
	m_nid.uCallbackMessage = WM_SHOWTASK;//自定义的消息名称，在推盘图标上任意动作就产生此消息
	m_nid.hIcon = LoadIcon(AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_DATASERVICS));
	//strcpy(m_nid.szTip,"数据服务");    //信息提示条 
	strcpy_s(m_nid.szTip, "数据服务");
	Shell_NotifyIcon(NIM_ADD, &m_nid);    //在托盘区添加图标 
	ShowWindow(SW_SHOW);    //隐藏主窗口
	CenterWindow();

	SetTimer(WM_FRESH, 3000, NULL);

	return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

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

	if (nID == SC_MINIMIZE)//点击标题栏的“最小化”
	{
		ToTray(); //最小化到托盘的函数
		return;
	}
	if (nID == SC_CLOSE)
	{
		ToTray(); //最小化到托盘的函数
		return;
	}

	else
	{
		CDialogEx::OnSysCommand(nID, lParam);
	}
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。  对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CTimeSyncServerDlg::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
	{
		CDialogEx::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CTimeSyncServerDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}


LRESULT CTimeSyncServerDlg::OnShowTask(WPARAM wParam, LPARAM lParam)
{
	if (wParam != IDI_DATASERVICS)//不是通过前面注册的图标响应的消息的话，退出
		return 1;

	switch (lParam)
	{
	case WM_RBUTTONDOWN://右键按下时弹出快捷菜单
	{
		POINT point;
		::GetCursorPos(&point);//得到鼠标位置

		CMenu Popupmenu;
		if (!Popupmenu.CreatePopupMenu())
			return 0;
		Popupmenu.AppendMenu(MF_STRING, ID_POPMENU_SHOW_MAIN_WND, _T("显示主界面"));
		Popupmenu.AppendMenu(MF_STRING, ID_POPMENU_CLOSE_CLIENT, _T("退出服务"));
		Popupmenu.TrackPopupMenu(TPM_LEFTBUTTON | TPM_LEFTALIGN, point.x, point.y, this);
		Popupmenu.DestroyMenu();
	}
	break;
	case WM_LBUTTONDBLCLK://双击左键的处理 
	{
		OnTrayMenuShowMainWnd();
		//this->ShowWindow(SW_SHOW);//简单的显示主窗口
		DeleteTray();//删除窗口
	}
	break;
	default:
		break;
	}
	return 0;
}

void CTimeSyncServerDlg::OnBnClickedButton1()
{
	UpdateData(TRUE);//读取编辑框变量m_strSvrPath的值

	if (InstallService(m_strSvrPath))
	{
		//printf("/n/nService Installed Sucessfully/n");
		MessageBox("Service Installed Sucessfully", "note", MB_OK | MB_ICONINFORMATION);
	}
	else
	{
		//printf("/n/nError Installing Service/n");
		MessageBox("Error Installing Service", "note", MB_OK | MB_ICONERROR);
	}
}


void CTimeSyncServerDlg::OnBnClickedButton2()
{
	if (DeleteService())
	{
		//printf("/n/nService UnInstalled Sucessfully/n");
		MessageBox("Service UnInstalled Sucessfully", "note", MB_OK | MB_ICONINFORMATION);
	}
	else
	{
		//printf("/n/nError UnInstalling Service/n");
		MessageBox("Error UnInstalling Service", "note", MB_OK | MB_ICONWARNING);
	}
}


void CTimeSyncServerDlg::OnBnClickedButton3()
{
	//打开服务管理器
	::ShellExecute(NULL, "open", "cmd.exe", "/c services.msc", NULL, SW_HIDE);
}

void CTimeSyncServerDlg::OnBnClickedButton4()
{
	//打开启动项
	::ShellExecute(NULL, "open", "cmd.exe", "/c msconfig", NULL, SW_HIDE);
}


void CTimeSyncServerDlg::OnBnClickedBtnRead()
{
	CMonitorObjectMng::Instance()->LoadMonitorTree();
	CMonitorObjectMng::Instance()->LoadHistoryData();
}


void CTimeSyncServerDlg::OnTimer(UINT_PTR nIDEvent)
{
	switch (nIDEvent)
	{
	case WM_FRESH:
	if (IsWindowVisible())
	{
		((CButton*)GetDlgItem(IDC_CHK_DATA_COMPRESSION))->SetCheck(g_bDataCompression);
		auto pSuper = CSuperManager::Instance();
		if (pSuper->GetSuperNum() == 0)
			((CButton*)GetDlgItem(IDC_CHK_SWITCH))->SetCheck(FALSE);
		else
		{
			((CButton*)GetDlgItem(IDC_CHK_SWITCH))->SetCheck(TRUE);
			SetDlgItemText(IDC_STC_SEND_NUM, pSuper->SendDesc().c_str());
			SetDlgItemText(IDC_STC_SEND_FAIL, pSuper->FailDesc().c_str());
			SetDlgItemText(IDC_STC_RECV_NUM, pSuper->recvDesc().c_str());
			if (g_b315) ((CButton*)GetDlgItem(IDC_CHK_315))->SetCheck(true);
			else ((CButton*)GetDlgItem(IDC_CHK_315))->SetCheck(false);
		}
	}
	break;
	default:
		break;
	}
	CDialogEx::OnTimer(nIDEvent);
}


afx_msg LRESULT CTimeSyncServerDlg::OnStartapp(WPARAM wParam, LPARAM lParam)
{
	SetDlgItemText(IDC_STC_VERSION, g_strVersion.c_str());
	auto ss = CTime(g_stStart).Format("%Y-%m-%d %H:%M:%S %A");
	SetDlgItemText(IDC_STC_START, ss);

	return 0;
}


void CTimeSyncServerDlg::OnClose()
{

	CDialogEx::OnClose();
}


void CTimeSyncServerDlg::OnBnClickedChkDataCompression()
{
	auto ret = ((CButton*)GetDlgItem(IDC_CHK_DATA_COMPRESSION))->GetCheck() == BST_CHECKED;
	g_bDataCompression = ret;
	CString m_strIniPath = CSimpleLog::GetAppDir() + "DataServices.ini";
	::WritePrivateProfileString("SET", "DataCompression", to_string(g_bDataCompression).c_str(), m_strIniPath);
}