lyfz_repos/经典版源码/kernelcode/lyfzIMGSelection/lyfzServer/lyfzServerDlg.cpp

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

#include "stdafx.h"
#include "lyfzServer.h"
#include "lyfzServerDlg.h"
#include "afxdialogex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

#if USE_TRAYICON
#define WM_MY_TRAY_NOTIFICATION WM_USER + 109
const UINT WM_TASKBARCREATED = ::RegisterWindowMessage(_T("lyfzServer"));
#endif

//增加自动启动注册表项
INT AddAutoSysRun(LPBYTE lpExeFile, CONST DWORD &cbExeFile, LPCTSTR lpKeyName, INT iAutoRunSystem)
{
	HKEY hKey = NULL;
	DWORD dwErrorCode = 0;

	//创建注册表，有该键则读取，无则创建
	dwErrorCode = RegCreateKeyEx(HKEY_LOCAL_MACHINE,
		_T("Software\\Microsoft\\Windows\\CurrentVersion\\Run"),
		0,
		NULL,
		REG_OPTION_NON_VOLATILE,
		KEY_ALL_ACCESS,	// 
		//KEY_READ,
		NULL, //Security
		&hKey,
		NULL);
	if (dwErrorCode == ERROR_SUCCESS)
	{
		if (iAutoRunSystem)
		{
			dwErrorCode = RegSetValueEx(hKey, lpKeyName, NULL, REG_SZ, lpExeFile, cbExeFile);
		}
		else
		{
			dwErrorCode = RegDeleteValue(hKey, lpKeyName);
			dwErrorCode = RegDeleteKey(hKey, lpKeyName);
		}
	}

	if (NULL != hKey)
		RegCloseKey(hKey);
	hKey = NULL;

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

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

// 对话框数据
	enum { IDD = IDD_ABOUTBOX };

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

// 实现
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD)
{
}

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

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()


// ClyfzServerDlg 对话框
#if USE_TRAYICON
ClyfzServerDlg::ClyfzServerDlg(CWnd* pParent /*=NULL*/)
	: CDialogEx(ClyfzServerDlg::IDD, pParent), m_trayIcon(IDR_MAINFRAME)
#else
ClyfzServerDlg::ClyfzServerDlg(CWnd* pParent /*=NULL*/)
	: CDialogEx(ClyfzServerDlg::IDD, pParent)
#endif
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
#if USE_TRAYICON
	m_isNotify = TRUE;
	m_bNoticeTray = FALSE;
#endif
}

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

BEGIN_MESSAGE_MAP(ClyfzServerDlg, CDialogEx)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
#if USE_TRAYICON
	ON_MESSAGE(WM_MY_TRAY_NOTIFICATION, OnTrayNotification)
	ON_REGISTERED_MESSAGE(WM_TASKBARCREATED, OnTaskBarCreated)
#endif
	ON_COMMAND(Menu_Exit, &ClyfzServerDlg::OnExit)
	ON_BN_CLICKED(IDOK, &ClyfzServerDlg::OnBnClickedOk)
	ON_BN_CLICKED(IDCANCEL, &ClyfzServerDlg::OnBnClickedCancel)
	ON_WM_TIMER()
END_MESSAGE_MAP()


// ClyfzServerDlg 消息处理程序

BOOL ClyfzServerDlg::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);		// 设置小图标

#if USE_TRAYICON
	InitTrayIcon();
#endif
	// TODO: 在此添加额外的初始化代码

	SetTimer(0, 10, NULL);

	// 启动项;
	TCHAR szDrive[_MAX_DRIVE] = { 0 };
	TCHAR szDir[_MAX_DIR] = { 0 };
	TCHAR szFName[_MAX_FNAME] = { 0 };
	TCHAR szExt[_MAX_EXT] = { 0 };
	TCHAR szModulePath[_MAX_DRIVE] = { 0 };
	::GetModuleFileName(NULL, szModulePath, sizeof(szModulePath) / sizeof(TCHAR));

	_tsplitpath_s(szModulePath, szDrive, szDir, szFName, szExt);
	_tcscpy_s(szModulePath, szDrive);
	_tcscat_s(szModulePath, szDir);
	AfxGetApp()->WriteProfileInt(szFName, _T("Autoruning"), TRUE);
	AddAutoSysRun((LPBYTE)szFName, _tcslen(szFName)*sizeof(TCHAR), szFName, TRUE);

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

#if USE_TRAYICON
void ClyfzServerDlg::InitTrayIcon()
{
	m_trayIcon.SetNotificationWnd(this, WM_MY_TRAY_NOTIFICATION);
	m_trayIcon.SetIcon(IDR_MAINFRAME, _T("选片服务"));
	m_trayIcon.SetDefaultTip(_T("选片服务"));
}

LRESULT ClyfzServerDlg::OnTaskBarCreated(WPARAM wp, LPARAM lp)
{
	Shell_NotifyIcon(NIM_ADD, &m_trayIcon.m_nid);
	return 0;
}

// Handle notification from tray icon: display a message.
LRESULT ClyfzServerDlg::OnTrayNotification(WPARAM uID, LPARAM lEvent)
{
	if (m_isNotify)
	{
		static LPCSTR MouseMessages[] = { "WM_MOUSEMOVE",
			"WM_LBUTTONDOWN", "WM_LBUTTONUP", "WM_LBUTTONDBLCLK",
			"WM_RBUTTONDOWN", "WM_RBUTTONUP", "WM_RBUTTONDBLCLK",
			"WM_MBUTTONDOWN", "WM_MBUTTONUP", "WM_MBUTTONDBLCLK" };
	}

	if (uID != IDR_MAINFRAME)
		return m_trayIcon.OnTrayNotification(uID, lEvent);

	switch (lEvent)
	{
	case WM_RBUTTONUP:
		TrayRight();
		break;
	case WM_LBUTTONUP:
		break;
	case WM_LBUTTONDBLCLK:
		TaskNotifyIcon();
		break;
	default:
		break;
	}
	return 0;
}

void ClyfzServerDlg::TrayRight()
{
	CMenu menu;
	if (!menu.LoadMenu(IDR_MENU1))
		return;

	CMenu* pSubMenu = menu.GetSubMenu(0);
	if (!pSubMenu)
		return;

	// 获取任务栏高度;
	RECT rc;
	INT iHeight;
	HWND hWnd = ::FindWindow(_T("Shell_TrayWnd"), NULL);
	if (hWnd)
	{
		::GetWindowRect(hWnd, &rc);
		iHeight = rc.bottom - rc.top;
	}

	pSubMenu->CheckMenuItem(Menu_Exit, MF_BYCOMMAND | MF_UNCHECKED);

	CPoint point;
	GetCursorPos(&point);
	point.y = rc.top - 1;
	::SetForegroundWindow(m_hWnd);
	pSubMenu->TrackPopupMenu(TPM_LEFTALIGN | TPM_RIGHTBUTTON, point.x - 100, point.y, this);
}

void ClyfzServerDlg::TaskNotifyIcon()
{
	if (!m_bNoticeTray)
	{
		m_trayIcon.SetIcon(IDR_MAINFRAME, _T("选片服务程序"));
		ShowWindow(SW_SHOW);
		m_bNoticeTray = TRUE;
	}
	else
	{
		m_trayIcon.SetIcon(IDR_MAINFRAME, _T("选片服务程序"));
		ShowWindow(SW_HIDE);
		m_bNoticeTray = FALSE;
	}
}
#endif

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

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

void ClyfzServerDlg::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 ClyfzServerDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}


void ClyfzServerDlg::OnExit()
{
	CDialogEx::OnCancel();
}


void ClyfzServerDlg::OnBnClickedOk()
{
	// TODO:  在此添加控件通知处理程序代码
	ShowWindow(SW_HIDE);
	m_bNoticeTray = FALSE;
}


void ClyfzServerDlg::OnBnClickedCancel()
{
	// TODO:  在此添加控件通知处理程序代码
	ShowWindow(SW_HIDE);
	m_bNoticeTray = FALSE;
}


void ClyfzServerDlg::OnTimer(UINT_PTR nIDEvent)
{
	// TODO:  在此添加消息处理程序代码和/或调用默认值
	if ( nIDEvent == 0 )
	{
		KillTimer(nIDEvent);
		ShowWindow(SW_HIDE);
		m_bNoticeTray = FALSE;
	}
	CDialogEx::OnTimer(nIDEvent);
}