www.pudn.com > downloadingNet.rar > document.h

// Document.h : interface of the CSnaggerDoc class 
// 
///////////////////////////////////////////////////////////////////////////// 
 
#if !defined(__Document_h__) 
#define __Document_h__ 
 
#if _MSC_VER >= 1000 
#pragma once 
#endif // _MSC_VER >= 1000 
 
 
#include "parser.h" 
#include "snagpars.h" 
#include "inet.h" 
#include "tree.h" 
#include "options.h" 
 
typedef struct 
{ 
   CStringArray arrLinks; 
   CStringArray arrMedia; 
   CByteArray arrOffsite; 
   int nIndex; 
} LINKS; 
 
 
class CProgress; 
 
class CSnaggerDoc : public CDocument 
{ 
protected: 
	CProgress *m_pProgress; 
	CSnaggerHtmlParser m_Parser; 
	CInet m_Inet; 
	MAP_FILES_ARRAY m_arrPagesDone; 
	MAP_FILES_ARRAY m_arrMediaDone; 
	LINKS m_aLinks[MAX_LEVELS]; 
	int m_nLevel; 
	int m_nTotalTicks; 
	int m_nTotalBytes; 
	CString m_strDirectory; 
	int m_nGottenPageCount; 
	int m_nGottenFileCount; 
	int m_nQueuedPageCount; 
	BYTE m_byBuffer[MAX_INET_BUFFER]; 
	CSnagOptions m_Options; 
	CFile m_fileLog; 
	CFile m_fileContents; 
	BOOL m_bProjectLoaded; 
	CString m_strPageTitle; 
	CString m_strStartPage; 
	BOOL m_bAutoMode; 
 
protected: // create from serialization only 
	CSnaggerDoc(); 
	DECLARE_DYNCREATE(CSnaggerDoc) 
 
protected: 
	BOOL ShouldGetPage(CString& strPage, MAP_FILES*& pMapEntry); 
	BOOL ShouldGetMedia(CString& strPage, MAP_FILES*& pMapEntry); 
	void ClearCacheMaps(); 
	void SetPageCacheEntry(LPCTSTR lpszPage, LPCTSTR lpszFileName, int nLevel); 
	void SetMediaCacheEntry(LPCTSTR lpszMedia, LPCTSTR lpszFileName); 
	BOOL GetPage(CString& strPage, CString& strFileName, LINKS& linkEntry); 
	BOOL GetMedia(CString& strMedia, CString& strFileName); 
	void UpdateStatus(); 
	void FixupLinks(); 
	static UINT DownloadThread(LPVOID lpvData); 
	void ResetLink(int nLevel); 
	BOOL ShouldQueuePage(CString& strNewPage, BOOL bOffsite); 
 
// Attributes 
public: 
	void SetDirectory(CString &strDirectory) { m_strDirectory = strDirectory; } 
	CString& GetDirectory() { return m_strDirectory; } 
	BOOL ProjectLoaded() { return m_bProjectLoaded; } 
	CSnagOptions *GetOptions() { return &m_Options; } 
	void GetOptions(CSnagOptions& Options); 
	void SetOptions(CSnagOptions& Options); 
	CString GetStartPage() { return m_strStartPage; } 
	void SetStartPage(CString& strStartPage) { m_strStartPage = strStartPage; } 
	void SetAutoMode(BOOL bAutoMode) { m_bAutoMode = bAutoMode; } 
 
// Operations 
public: 
	void RecursiveDownload(LPCTSTR lpszURL); 
	void Reset(LPCTSTR lpszProjName=""); 
 
// Overrides 
	// ClassWizard generated virtual function overrides 
	//{{AFX_VIRTUAL(CSnaggerDoc) 
	public: 
	virtual BOOL OnNewDocument(); 
	virtual void Serialize(CArchive& ar); 
	virtual BOOL OnOpenDocument(LPCTSTR lpszPathName); 
	virtual BOOL SaveModified(); 
	virtual BOOL CanCloseFrame(CFrameWnd* pFrame); 
	virtual void OnCloseDocument(); 
	//}}AFX_VIRTUAL 
 
// Implementation 
public: 
	virtual ~CSnaggerDoc(); 
#ifdef _DEBUG 
	virtual void AssertValid() const; 
	virtual void Dump(CDumpContext& dc) const; 
#endif 
 
protected: 
 
// Generated message map functions 
protected: 
	//{{AFX_MSG(CSnaggerDoc) 
	//}}AFX_MSG 
	DECLARE_MESSAGE_MAP() 
}; 
 
///////////////////////////////////////////////////////////////////////////// 
 
//{{AFX_INSERT_LOCATION}} 
// Microsoft Developer Studio will insert additional declarations immediately before the previous line. 
 
#endif // !defined(__Document_h__)