sitedownload.cpp

《Visual C++网络通信编程实用案例精选》的配套源代码。

#include "stdafx.h"

#include <direct.h>
#include <io.h>

#include "SiteDownload.h"

#include "MyFrame.h"
#include "MyDocument.h"
#include "MyView.h"
#include "InternetDownload.h"

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

/////////////////////////////////////////////////////////////////////////////
// CSiteDownloadApp

BEGIN_MESSAGE_MAP(CSiteDownloadApp, CWinApp)
	//{{AFX_MSG_MAP(CSiteDownloadApp)
	ON_COMMAND(ID_APP_ABOUT, OnAppAbout)
	//}}AFX_MSG_MAP
	// Standard file based document commands
	ON_COMMAND(ID_FILE_NEW, CWinApp::OnFileNew)
	ON_COMMAND(ID_FILE_OPEN, CWinApp::OnFileOpen)
END_MESSAGE_MAP()

// Determine if we are running the new shell (i.e. Windows 95 or Windows NT 4.0
// or later)
BOOL CSiteDownloadApp::HasNewShell()
{
	// Get the Windows Version
	DWORD dwVersion = GetVersion();

	// Windows NT?
	if(dwVersion < 0x80000000) 
	{
		// Yes, Windows NT 4.0 or greater?
		if(LOBYTE(LOWORD(dwVersion)) >= 4) 
			return TRUE; 
	} // Windows 95?
	else if(LOBYTE(LOWORD(dwVersion)) >= 4) 
			return TRUE;

   return FALSE;  // Win16, 32s or NT3.x.
}

// Determine if SiteSnager is already running by using a Mutex object -- returns
// TRUE if already running, FALSE otherwise.  Note that a mutex is created
// with the name of the application
BOOL CSiteDownloadApp::AlreadyRunning()
{
	BOOL bFound = FALSE;

	// Try to create a mutex with the app's name
	HANDLE hMutexOneInstance = ::CreateMutex(NULL,TRUE,_T(AfxGetAppName()));

	// Already there...means that we are already running an instance
	if(::GetLastError() == ERROR_ALREADY_EXISTS)
		bFound = TRUE;

	// Release the mutex
	if(hMutexOneInstance)
		::ReleaseMutex(hMutexOneInstance);

	return(bFound);
}

/////////////////////////////////////////////////////////////////////////////
// CSiteDownloadApp construction

CSiteDownloadApp::CSiteDownloadApp()
{
}

/////////////////////////////////////////////////////////////////////////////
// The one and only CSiteDownloadApp object

CSiteDownloadApp theApp;

/////////////////////////////////////////////////////////////////////////////
// CSiteDownloadApp initialization

// Initialize the app -- check to see if is already running, make sure we're
// running the right OS
BOOL CSiteDownloadApp::InitInstance()
{
	// Only allow one instance of our application to execute
	if(AlreadyRunning())
	{
		AfxMessageBox(IDS_ALREADY_RUNNING,MB_OK|MB_ICONWARNING);
		return(FALSE);
	}

	// Only run with the new shell --- we need a system tray
	if(!HasNewShell())
	{
		AfxMessageBox(IDS_WRONG_SHELL,MB_OK|MB_ICONSTOP);
		return(FALSE);
	}

	AfxEnableControlContainer();

	// Standard initialization
	// If you are not using these features and wish to reduce the size
	//  of your final executable, you should remove from the following
	//  the specific initialization routines you do not need.

#ifdef _AFXDLL
	Enable3dControls();			// Call this when using MFC in a shared DLL
#else
	Enable3dControlsStatic();	// Call this when linking to MFC statically
#endif

	int nSaveCmdShow = m_nCmdShow;
	m_nCmdShow &= ~SW_SHOW;

	// Setup the registry key, we only use this to store the MRU list
	SetRegistryKey(_T("PC Magazine\\SiteSnagger"));

	LoadStdProfileSettings();  // Load standard INI file options (including MRU)

	// Register the application's document templates.  Document templates
	//  serve as the connection between documents, frame windows and views.

	CSingleDocTemplate* pDocTemplate;
	pDocTemplate = new CSingleDocTemplate(
		IDR_MAINFRAME,
		RUNTIME_CLASS(CMyDoc),
		RUNTIME_CLASS(CMyFrame),       // main SDI frame window
		RUNTIME_CLASS(CMyView));
	AddDocTemplate(pDocTemplate);

	// Parse the command line 
	if(!ProcessCommandLine())
		return FALSE;

	// The one and only window has been initialized, so show and update it.
	CRect rect;
	this->GetMainWnd()->GetWindowRect(&rect);
	m_pMainWnd->SetWindowPos(NULL,-1,-1,rect.right-rect.left,rect.bottom-rect.top,SWP_NOMOVE);
	m_pMainWnd->CenterWindow();
	m_pMainWnd->ShowWindow(SW_SHOW);
	m_pMainWnd->UpdateWindow();

	return TRUE;
}

// Allows access to the applications Most Recently Used (MRU) file list
void CSiteDownloadApp::UpdateMRU(CCmdUI *pCmdUI)
{
	CWinApp::OnUpdateRecentFileMenu(pCmdUI);
}

// Process the command line arguments passed in
//
// The format is: sitesnag project url [options] 
//
BOOL CSiteDownloadApp::ProcessCommandLine()
{
	CString strURL;
	CConfigure Options;

	// Create the projects directory (if not in command line mode)
	if(__argc < 2)
		mkdir("Projects");

	// Create an empty document -- this ensures that the tree control will
	// always exist
	OnFileNew();

	// Was a document name specified?
	if(__argc > 1)
	{
		// Make sure that the Projects subdirectory exists, if not, then
		// we're running from the wrong directory or there is no project
		if(access("Projects",0) != 0)
		{
			AfxMessageBox("The working directory must be the same as the SiteSnagger program directory.",
										MB_ICONERROR|MB_OK);
			return(FALSE);
		}

		// Add the "Projects" subdirectory if the user didn't specify a 
		// subdirectory location
		CString strFileName;
		CString strPath = CInternetDownload::SplitFileName(__argv[1],CInternetDownload::PATH);
		if(strPath.IsEmpty())
			strFileName = CString("Projects\\");
		strFileName += __argv[1];

		// Try to open the document that the user specified on the command line
		CMyDoc *pDoc = (CMyDoc *) m_pDocManager->OpenDocumentFile(strFileName);

		// Didn't work, get out
		if(pDoc == NULL)
			return(FALSE);

		// Do we have a URL???
		if(__argc > 2)
		{
			// Yes, this means that it it automatic mode
			strURL = __argv[2];

			// Set some default options (these will override the current 
			// project options)
			Options.nMaxDepth = 2;
			Options.nMaxPages = 0;
			Options.bOffsiteLinks = FALSE;
			Options.bMultimedia = TRUE;
			Options.bFixupLinks = TRUE;
			Options.bContents = TRUE;

			for(int i = 3; i < __argc; i++)
			{
				LPCTSTR lpszParam = __targv[i];
				if(*lpszParam == '-' || *lpszParam == '/')
				{
					// Skip the flag
					++lpszParam;

					switch(tolower(*lpszParam))
					{
						// Levels
						case 'l':
							Options.nMaxDepth = atoi(lpszParam+1)%(MAX_LEVELS+1);
							if(Options.nMaxDepth <= 0)
								Options.nMaxDepth = 0;
							break;

						// Pages
						case 'p':
							Options.nMaxPages = atoi(lpszParam+1);
							if(Options.nMaxPages < 0)
								Options.nMaxPages = 0;
							break;

						// Offsite links						
						case 'o':
							Options.bOffsiteLinks = *(lpszParam+1) == '+';
							break;

						// Multimedia 
						case 'm':
							Options.bMultimedia = *(lpszParam+1) == '+';
							break;

						// Fixup for browsing
						case 'b':
							Options.bFixupLinks = *(lpszParam+1) == '+';
							break;

						// Generate table of contents
						case 'c':
							Options.bContents = *(lpszParam+1) == '+';
							break;
					}
				}
			}

			// Setup the document and view for snagging
			POSITION pos = pDoc->GetFirstViewPosition();
			CMyView* pView = (CMyView *) pDoc->GetNextView(pos);

			pDoc->SetAutoMode(TRUE);
			pDoc->SetOptions(Options);
			CString strPath = CInternetDownload::SplitFileName(pDoc->GetPathName(),
						CInternetDownload::DRIVE|CInternetDownload::PATH|CInternetDownload::FNAME)+"\\";
			pView->ClearProject(strPath);
			pView->InitTree(strURL);
			pView->SetSnagging(TRUE);

			// Start the snagging operation
			pDoc->RecursiveDownload(strURL);
			pDoc->SetModifiedFlag(TRUE);
		}
	}

	return(TRUE);
}

/////////////////////////////////////////////////////////////////////////////
// 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)
		// No message handlers
	//}}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()

// App command to run the dialog
void CSiteDownloadApp::OnAppAbout()
{
	CAboutDlg aboutDlg;
	aboutDlg.DoModal();
}

/////////////////////////////////////////////////////////////////////////////
// CSiteDownloadApp commands