queryitems.cpp

用Visual C++开发的一个FTP搜索工具。具有站点搜索、文件列表下载、文件搜索等功能

// QueryItems.cpp : implementation file
//

#include "stdafx.h"
#include "IntelligentFtp.h"
#include "QueryItems.h"

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

/////////////////////////////////////////////////////////////////////////////
// CQueryItems dialog


CQueryItems::CQueryItems(CWnd* pParent /*=NULL*/)
	: CDialog(CQueryItems::IDD, pParent)
{
	p_Database = NULL;
	p_Record  = NULL;
	m_IsDbOpen = FALSE;
	m_strFilter = "*";
	//{{AFX_DATA_INIT(CQueryItems)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
}


void CQueryItems::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CQueryItems)
	DDX_Control(pDX, IDC_CONTENTLIST, m_listQuery);
	DDX_Control(pDX, IDC_QUERYITEMS, m_btnQuery);
	DDX_Control(pDX, IDC_KEYWORD, m_editQuery);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CQueryItems, CDialog)
	//{{AFX_MSG_MAP(CQueryItems)
	ON_BN_CLICKED(IDC_QUERYITEMS, OnQueryitems)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CQueryItems message handlers

void CQueryItems::OnOK() 
{
	// TODO: Add extra validation here
	
	// CDialog::OnOK();
}

void CQueryItems::OnCancel() 
{
	// TODO: Add extra cleanup here
	
	// CDialog::OnCancel();
}

void CQueryItems::OnQueryitems() 
{
	CString strQuery;
	CString strLen;
	m_editQuery.GetWindowText(strQuery);

	if(strQuery.GetLength() == 0) {
		return;
	}

	m_strFilter = "FileName Like \'%" + strQuery + "%\'";

	if(!m_IsDbOpen) {
		if( !OpenDb() ) {
			return;
		}
	}

	if(!p_Record->IsBOF())
	{
		m_listQuery.DeleteAllItems();

		while(!p_Record->IsEOF())
		{
			int	i = m_listQuery.InsertItem(0,p_Record->m_FileName);
			strLen.Format("%d",p_Record->m_FileLength);
			m_listQuery.SetItemText(i,1,strLen);
			m_listQuery.SetItemText(i,2,p_Record->m_IP);
			m_listQuery.SetItemText(i,3,p_Record->m_Path);
			p_Record->MoveNext();
		}
	}

	CloseDb();
}

BOOL CQueryItems::OpenDb()
{
	if(m_IsDbOpen) return TRUE;

	p_Database = new CDatabase;

	p_Database->SetLoginTimeout(5);

	BOOL p_Status = FALSE;
	try
	{
		p_Status = p_Database->Open("ftpdb");
		if(p_Status)
			TRACE("\nDB opened successfully\n");
		else
			TRACE("\nOpen DSN failed.\n");
	}
	catch(CMemoryException* pEx)
	{
		pEx->ReportError();
	}
	catch(CDBException*	pDBEx)
	{
		pDBEx->ReportError();
		TRACE("RetCOde:%d strError:[%d] strState: [%d]\n",
			pDBEx->m_nRetCode,
			pDBEx->m_strError,
			pDBEx->m_strStateNativeOrigin);
	}

	if(!p_Status)
	{
		MessageBox("Open DSN Failed.");

		if(p_Database->IsOpen())	p_Database->Close();
		if(p_Database	!=	NULL)	delete	p_Database;

		return	FALSE;
	}

	p_Record	=	new	CFtpContent(p_Database);
	p_Record->m_strFilter = m_strFilter;

	if(!p_Record->Open(AFX_DB_USE_DEFAULT_TYPE,(LPCTSTR)"Content"))
	{
		MessageBox("Open Table SiteData Failed.");

		if(p_Record->IsOpen())		p_Record->Close();
		if(p_Database->IsOpen())	p_Database->Close();

		delete	p_Record;
		delete	p_Database;

		return FALSE;
	}

	m_IsDbOpen = TRUE;

	return TRUE;
}

void CQueryItems::CloseDb()
{
	if(!m_IsDbOpen) return;

	if(p_Record->IsOpen())		p_Record->Close();
	if(p_Database->IsOpen())	p_Database->Close();

	delete	p_Record;
	delete	p_Database;	

	m_IsDbOpen = FALSE;
}

BOOL CQueryItems::OnInitDialog() 
{
	CDialog::OnInitDialog();
	
	// TODO: Add extra initialization here
	
	m_listQuery.InsertColumn(0,"URL",LVCFMT_LEFT,150);
	m_listQuery.InsertColumn(0,"FtpSite",LVCFMT_LEFT,100);
	m_listQuery.InsertColumn(0,"FileLength",LVCFMT_LEFT,100);
	m_listQuery.InsertColumn(0,"FileName",LVCFMT_LEFT,100);
	
	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}