lyfz_repos/经典版源码/lyfzInternalSms/branches/wjf/lyfzSMSHandle/lyfzSMSHandleDlg.cpp

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

#include "stdafx.h"
#include "lyfzSMSHandle.h"
#include "lyfzSMSHandleDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

#if USE_TRAYICON
#define WM_MY_TRAY_NOTIFICATION WM_USER + 108
const UINT WM_TASKBARCREATED = ::RegisterWindowMessage(_T("lyfzSMSHandle"));
#endif
// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

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()


// ClyfzSMSHandleDlg 对话框
#if USE_TRAYICON
ClyfzSMSHandleDlg::ClyfzSMSHandleDlg(CWnd* pParent /*=NULL*/)
	: CDialog(ClyfzSMSHandleDlg::IDD, pParent), m_trayIcon(IDR_MAINFRAME)
#else
ClyfzSMSHandleDlg::ClyfzSMSHandleDlg(CWnd* pParent /*=NULL*/)
	: CDialog(ClyfzSMSHandleDlg::IDD, pParent)
#endif
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);

	m_bAutoRuning = AfxGetApp()->GetProfileInt(_T("lyfzSMSHandle"), _T("Autoruning"), 0);
#if USE_TRAYICON
	m_isNotify = TRUE;
	m_bNoticeTray = TRUE;
#endif
}

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

BEGIN_MESSAGE_MAP(ClyfzSMSHandleDlg, CDialog)
	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
	//}}AFX_MSG_MAP
	ON_COMMAND(Menu_AutoRun, &ClyfzSMSHandleDlg::OnAutorun)
	ON_COMMAND(Menu_Exit, &ClyfzSMSHandleDlg::OnExit)
	ON_BN_CLICKED(BTN_REFRESH, &ClyfzSMSHandleDlg::OnBnClickedRefresh)
	ON_WM_TIMER()
END_MESSAGE_MAP()


// ClyfzSMSHandleDlg 消息处理程序
#if USE_TRAYICON
void ClyfzSMSHandleDlg::InitTrayIcon()
{
	m_trayIcon.SetNotificationWnd(this, WM_MY_TRAY_NOTIFICATION);
	m_trayIcon.SetIcon(IDR_MAINFRAME, _T("短信处理程序"));
	m_trayIcon.SetIconInfos(IDI_ITRAY, IDI_ITRAY, IDI_ITRAY);
	m_trayIcon.SetDefaultTip(_T("短信处理程序"));
}

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

// Handle notification from tray icon: display a message.
LRESULT ClyfzSMSHandleDlg::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 ClyfzSMSHandleDlg::TrayRight()
{
	CMenu menu;
	if (!menu.LoadMenu(IDR_MENU1))
		return;

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

	if (m_bAutoRuning)
		pSubMenu->CheckMenuItem(Menu_AutoRun, MF_BYCOMMAND | MF_CHECKED);
	else
		pSubMenu->CheckMenuItem(Menu_AutoRun, MF_BYCOMMAND | MF_UNCHECKED);

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

void ClyfzSMSHandleDlg::TaskNotifyIcon()
{
	if (!m_bNoticeTray)
	{
		m_trayIcon.SetIcon(IDR_MAINFRAME, _T("短信处理程序"));
		ShowWindow(SW_SHOWNORMAL);
		m_bNoticeTray = TRUE;
	}
	else
	{
		m_trayIcon.SetIcon(IDR_MAINFRAME, _T("短信处理程序"));
		ShowWindow(SW_HIDE);
		m_bNoticeTray = FALSE;
	}
}
#endif

BOOL ClyfzSMSHandleDlg::OnInitDialog()
{
	SetWindowPos(&wndNoTopMost, 0, 0, 0, 0, SWP_HIDEWINDOW);
	ModifyStyleEx(WS_EX_APPWINDOW, WS_EX_TOOLWINDOW);

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

	// TODO: 在此添加额外的初始化代码
	if (CDBInterface::GetInstance()->OpenDatabase() == FALSE)
	{
		WriteClientLog(_T("首次数据库连接失败,请检查配置信息或数据库服务是否启动！"));
		_exit(0);	// 首次数据库连接失败,直接退出;
		//CDialog::OnOK();
		return FALSE;
	}

	m_SMSHandle.GetMainInfo();

	SetDlgItemText(STC_ACCOUNT, m_SMSHandle.s_SMSInfo.msgaccount);
	SetDlgItemText(STC_REMAINING, m_SMSHandle.s_SMSInfo.msgbalance);
	SetDlgItemText(STC_HISTORYSEND, m_SMSHandle.s_SMSInfo.msgused);

	//SetTimer(0, 5000, NULL);
#if USE_TRAYICON
	InitTrayIcon();
#endif

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

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

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。
void ClyfzSMSHandleDlg::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 ClyfzSMSHandleDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

void ClyfzSMSHandleDlg::OnAutorun()
{
	m_bAutoRuning = !m_bAutoRuning;
	AfxGetApp()->WriteProfileInt(_T("lyfzSMSHandle"), _T("Autoruning"), m_bAutoRuning);

	//BYTE szData[1024] = { 0 };
	//::GetModuleFileNameA(NULL, (LPSTR)szData, sizeof(szData));
	AddAutoSysRun((LPBYTE)g_szModuleFileName,_tcslen(g_szModuleFileName)*sizeof(TCHAR), _T("lyfzSMSHandle"), m_bAutoRuning);
}

void ClyfzSMSHandleDlg::OnExit()
{
	CDialog::OnOK();
}

void ClyfzSMSHandleDlg::OnCancel()
{
	ShowWindow(SW_HIDE);
	m_bNoticeTray = FALSE;
}

void ClyfzSMSHandleDlg::OnOK()
{
	ShowWindow(SW_HIDE);
	m_bNoticeTray = FALSE;
}


void ClyfzSMSHandleDlg::OnBnClickedRefresh()
{
	QuerAccountInfo tAccountInfo;
	m_SMSHandle.GetSMSBlance(tAccountInfo);

	SetDlgItemText(STC_REMAINING, tAccountInfo.strBalance);
	SetDlgItemText(STC_SIGNATURE, tAccountInfo.strSignatures);
	UpdateData();
}


void ClyfzSMSHandleDlg::OnTimer(UINT_PTR nIDEvent)
{
	if (nIDEvent == 0)
	{
		if (CDBInterface::GetInstance()->IsOpen() == FALSE)
		{
			SetDlgItemText(STC_DBCONNECTSTATUS, _T("失败"));
			SetDlgItemText(STC_DBFAILDREASON, CDBInterface::GetInstance()->GetConnectErrorDescriptor());
		}
		else
		{
			SetDlgItemText(STC_DBCONNECTSTATUS, _T("正常"));
			SetDlgItemText(STC_DBFAILDREASON, _T("无"));
		}

		if (m_SMSHandle.GetHTTPErrorInf() == _T(""))
		{
			SetDlgItemText(STC_SENDSMSSTATUS, _T("正常"));
			SetDlgItemText(STC_SENDFAILDREASON, _T("无"));
		}
		else
		{
			SetDlgItemText(STC_SENDSMSSTATUS, _T("失败"));
			SetDlgItemText(STC_SENDFAILDREASON, m_SMSHandle.GetHTTPErrorInf());
		}
	}

	CDialog::OnTimer(nIDEvent);
}