updatebar.h.svn-base

wince c++ 下 开发的 rss 阅读器源代码

/**
 *  UpdateBar.h
 *
 *  Copyright (C) 2008  David Andrs <[e-mail]>
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#if !defined(AFX_UPDATEBAR__INCLUDED_)
#define AFX_UPDATEBAR__INCLUDED_

#if _MSC_VER >= 1000
#pragma once
#endif // _MSC_VER >= 1000

#include "ctrls/CeToolBar.h"
#include "ctrls/TextProgressCtrl.h"

#include "net/Download.h"
#include "misc.h"

class CSiteItem;
class CFeedItem;
class CFeed;

#define DOWNLOAD_TRIES					3

class CUpdateItem {
public:
	CUpdateItem(CSiteItem *siteItem, BOOL updateOnly = FALSE) {
		SiteItem = siteItem;
		UpdateOnly = updateOnly;
	}

	BOOL UpdateOnly;
	CSiteItem *SiteItem;
};

/////////////////////////////////////////////////////////////////////////////
// CDownloadQueue class

class CDownloadItem {
public:
	EFileType Type;
	CString URL;
	CString FileName;
	int Try;
	int SiteIdx;
};

//
// CDownloadQueue class
//

class CDownloadQueue {
public:
	CDownloadQueue();
	virtual ~CDownloadQueue();

	BOOL IsEmpty();
	int GetCount();

	void Enqueue(CDownloadItem *item);
	CDownloadItem *Dequeue();

protected:
	CList<CDownloadItem*, CDownloadItem*> Items;
	CRITICAL_SECTION CS;
	int FinishedItems;

	friend class CUpdateBar;
};

//
// CUpdateBar class
//
class CUpdateBar : public CControlBar {
// Construction
public:
	CUpdateBar();
	virtual ~CUpdateBar();

	BOOL Create(CWnd *pParentWnd);

	virtual void OnUpdateCmdUI(CFrameWnd *pTarget, BOOL bDisableIfNoHndler);
	virtual CSize CalcFixedLayout(BOOL bStretch, BOOL bHorz);

	void Start();
	void EnqueueSite(CSiteItem *site, BOOL updateOnly = FALSE);
	void EnqueueSites(CList<CSiteItem *, CSiteItem *> &sites, BOOL updateOnly = FALSE);
	void EnqueueImages(CArray<CFeedItem *, CFeedItem *> &items);
	void EnqueueHtmls(CArray<CFeedItem *, CFeedItem *> &items);
	void EnqueueHtml(const CString &url, CSiteItem  *item);
	void EnqueueEnclosures(CArray<CFeedItem *, CFeedItem *> &items, DWORD limit = 0);
	void EnqueueItem(const CString &strUrl, EFileType type);

	BOOL IsUpdating() { return HUpdateThread != NULL; }

	void Redraw();

	void ShowErrorCount();
	void ShowError(UINT nID);
	void ShowError(const CString &str);

// Implementation
public:

protected:
	CTextProgressCtrl m_ctlProgress;
	CBitmapButton m_ctlStopBtn;
	CStatic m_ctlText;

	HANDLE HUpdateThread;
	CRITICAL_SECTION CSUpdateList;
	CList<CUpdateItem *, CUpdateItem *> UpdateList;
	CRITICAL_SECTION CSDownloader;
	CDownloader *Downloader;
	int ErrorCount;
	BOOL Terminate;
	int TotalItems;
	int NewItemsCount;

	void UpdateThread();
	void UpdateFeeds();

	void DownloadHtmlPage(CDownloadItem *di);
	void DownloadFile(CDownloadItem *di);
	void DownloadItems();
	void UpdateProgressText();

	CDownloadQueue DownloadQueue;

	enum {
		UPDATE_STATE_RSS,
		UPDATE_STATE_CACHING,
		UPDATE_STATE_AUTHENTICATING,
		UPDATE_STATE_SYNCING
	} State;
	CString SiteName;

    //{{AFX_MSG(CUpdateBar)
	afx_msg void OnPaint();
	afx_msg void OnSize(UINT nType, int cx, int cy);
	afx_msg void OnStop();
	afx_msg void OnTextClicked();
	//}}AFX_MSG

	DECLARE_MESSAGE_MAP()

	friend class CMainFrame;
	friend DWORD WINAPI UpdateStubProc(LPVOID lpParameter);
};

/////////////////////////////////////////////////////////////////////////////

void FeedDiff(CFeed *first, CFeed *second, CArray<CFeedItem *, CFeedItem *> *diff);

void ClearImages(CArray<CFeedItem *, CFeedItem *> &items);
void ClearHtmlPages(CArray<CFeedItem *, CFeedItem *> &items);
void ClearEnclosures(CArray<CFeedItem *, CFeedItem *> &items);

//{{AFX_INSERT_LOCATION}}
// Microsoft eMbedded Visual C++ will insert additional declarations immediately before the previous line.

#endif // !defined(AFX_UPDATEBAR__INCLUDED_)