skybotdlg.cpp

Visual C++自动查询和智能代理程序设计书籍源码-SkyBot

// SkyBotDlg.cpp : implementation file
//

#include "stdafx.h"
#include "SkyBot.h"
#include "SkyBotDlg.h"
#include "CRobot.h"
#include "CRobotInternet.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

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

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CSkyBotDlg dialog

CSkyBotDlg::CSkyBotDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CSkyBotDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CSkyBotDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CSkyBotDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CSkyBotDlg)
		// NOTE: the ClassWizard will add DDX and DDV calls here
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CSkyBotDlg, CDialog)
	//{{AFX_MSG_MAP(CSkyBotDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_WM_TIMER()
	//}}AFX_MSG_MAP
	ON_MESSAGE(WM_TRAY_NOTIFY, OnTrayNotify)
	ON_COMMAND(IDC_UPDATE, CSkyBotDlg::Update)
	ON_COMMAND(IDC_WEB, CSkyBotDlg::OnMenuWeb)
	ON_COMMAND(IDC_EXIT, CSkyBotDlg::OnMenuExit)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CSkyBotDlg message handlers

BOOL CSkyBotDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

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

	// Set the icon for this dialog.
	// The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon
	
	// Define terms.

	m_nTerms = 9;
	m_sTerm[0] = "SUNNY";		m_nTermIcon[0] = IDI_SUNNY;
	m_sTerm[1] = "CLOUDY";		m_nTermIcon[1] = IDI_CLOUDY;
	m_sTerm[2] = "RAIN";		m_nTermIcon[2] = IDI_RAIN;
	m_sTerm[3] = "SNOW";		m_nTermIcon[3] = IDI_SNOW;
	m_sTerm[4] = "OVERCAST";	m_nTermIcon[4] = IDI_CLOUDY;
	m_sTerm[5] = "FOGGY";		m_nTermIcon[5] = IDI_CLOUDY;
	m_sTerm[6] = "PARTLY";		m_nTermIcon[6] = IDI_CLOUDY;
	m_sTerm[7] = "SLEET";		m_nTermIcon[7] = IDI_RAIN;
	m_sTerm[8] = "SHOWERS";		m_nTermIcon[8] = IDI_RAIN;

	// Put agent in the Windows system tray.

	ShowWindow(SW_MINIMIZE);
	m_nTrayData.cbSize = sizeof(NOTIFYICONDATA) ;
	m_nTrayData.hWnd = m_hWnd;
	m_nTrayData.uID = 0 ;
	m_nTrayData.hIcon = LoadIcon (AfxGetInstanceHandle(),
								  MAKEINTRESOURCE(IDI_UNKNOWN));
	m_nTrayData.uCallbackMessage = WM_TRAY_NOTIFY;
	strcpy (m_nTrayData.szTip, "SkyBot (initializing)");
	m_nTrayData.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
	Shell_NotifyIcon (NIM_ADD, &m_nTrayData);

	m_bHidden = false;
	m_bUpdating = false;

	SetTimer(1, 1000, NULL);
	
	return TRUE; // return TRUE unless you set the focus to a control
}

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

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CSkyBotDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		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;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CSkyBotDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

// ******************
// *                *
// *  OnTrayNotify  *
// *                *
// ******************
// Tray notification handler

LRESULT CSkyBotDlg::OnTrayNotify(WPARAM wParam, LPARAM lParam)
{
	// ---- Right button down: Pop up menu ----
	if (lParam == WM_RBUTTONDOWN)
	{
		CMenu menu;
		VERIFY(menu.LoadMenu(IDR_MENU1));
		CMenu* pPopup = menu.GetSubMenu(0);
		ASSERT(pPopup != NULL); 
		CRect screen;
		GetDesktopWindow()->GetWindowRect(screen);
		pPopup->TrackPopupMenu(TPM_LEFTALIGN | TPM_RIGHTBUTTON,
							   screen.right,
							   screen.bottom,
							   AfxGetMainWnd());
	} // end End if
	return LRESULT(0);
}

void CSkyBotDlg::OnTimer(UINT nIDEvent) 
{
	KillTimer(1);

	if (!m_bHidden)
	{
		m_bHidden = true;
		ShowWindow(SW_HIDE);
	} // end End if

	Update();

	SetTimer(1, INTERVAL * 60 *1000, NULL);
	
	CDialog::OnTimer(nIDEvent);
}

// ************
// *          *
// *  Update  *
// *          *
// ************
// Description:	Retrieve latest weather page, scan it,
//				and update icon.

void CSkyBotDlg::Update()
{
	if (m_bUpdating)
		return;

	m_bUpdating = true;
	CRobotInternet internet;
	CString sHtml;
	int nResult;
	CString sErrMsg;
	BOOL bHaveData = false;
	m_bMatch = false;

	CString sURL = WEATHER_SOURCE;
	if (sURL.Left(5) == "file:")
	{
		CStdioFile file;
		CString sLine;
		if (file.Open(sURL.Mid(5), CFile::modeRead))
		{
			sHtml = "";
			while (file.ReadString(sLine))
			{
				sHtml += sLine + "\n";
			} // End while
			file.Close();
			bHaveData = true;
		} // End if
	} // End if
	else
		if (internet.httpGet(WEATHER_SOURCE,
							 sHtml,
							 nResult,
							 sErrMsg))
			bHaveData = true;

	if (bHaveData)
	{
		sHtml.MakeUpper();
		ParseHtml(sHtml);
	} // End if bHaveData

	if (!m_bMatch)
	{
		m_nIcon = IDI_UNKNOWN;
		m_sConditions = "UNKNOWN (Internet error)";
	} // End if

	m_nTrayData.hIcon = LoadIcon (AfxGetInstanceHandle(),
								  MAKEINTRESOURCE(m_nIcon));
	CString sForecast = "SkyBot - Forecast: " + m_sConditions;
	sForecast = sForecast.Left(63);
	strcpy (m_nTrayData.szTip, sForecast);
	m_nTrayData.uFlags = NIF_ICON | NIF_TIP;
	Shell_NotifyIcon (NIM_MODIFY, &m_nTrayData);

	m_bUpdating = false;
}

// ***************
// *             *
// *  ParseHtml  *
// *             *
// ***************
// Description:	Separate HTML stream into elements
//				and check for weather terms

void CSkyBotDlg::ParseHtml(const CString& sHTML)
{
	CString sItem = "";
	CString sChar;
	BOOL bInTag = false;
	BOOL bInQuote = false;
	BOOL bInSection = false;

	int nPos = 0;
	int nLen = sHTML.GetLength();
	for (nPos = 0; nPos < nLen; nPos++)
	{
		sChar = sHTML.Mid(nPos, 1);
		if (sChar == "\"")
		{
			sItem += sChar;
			bInQuote = !bInQuote;
		} // End if
		else
		{
			if (sChar == "<" && !bInQuote)
			{
				sItem.TrimLeft();
				sItem.TrimRight();
				if (sItem != "")
				{
					CString sTemp;

					if (!bInSection && sItem.Left(1) != "<")
					{
						for (int t = 0; t < m_nTerms; t++)
						{
							if (sItem.Find(m_sTerm[t]) != -1)
							{
								m_nIcon = m_nTermIcon[t];
								m_sConditions = sItem;
								m_bMatch = true;
								t = m_nTerms;
								return;
							} // End if
						} // End for t
					} // End if
					else
					{
						if (sItem.Left(7) == "<SELECT")
							bInSection = true;
						if (sItem.Left(8) == "</SELECT")
							bInSection = false;
					} // End else
					sItem = "";
				} // End if
			bInTag = true;
			sItem = sChar;
			} // End if
			else
				sItem += sChar;
			if (sChar == ">")
			{
				if (sItem.Left(7) == "<SELECT")
					bInSection = true;
				if (sItem.Left(8) == "</SELECT")
					bInSection = false;
				if (bInTag)
				{
					sItem = "";
					bInTag = false;
					bInQuote = false;
				} // End if
			} // End if
		} // End else
	} // End for nPos
}

void CSkyBotDlg::OnMenuWeb()
{
	ShellExecute(NULL,
				 "open",
				 WEATHER_SOURCE,
				 NULL,
				 NULL,
				 SW_SHOWNORMAL);
}

void CSkyBotDlg::OnMenuExit()
{
	Shell_NotifyIcon(NIM_DELETE, &m_nTrayData);
	EndDialog(0);
}