dlgnewsearch.cpp

著名的下载软件核心Shareaza

//
// DlgNewSearch.cpp
//
// Copyright (c) Shareaza Development Team, 2002-2004.
// This file is part of SHAREAZA (www.shareaza.com)
//
// Shareaza 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.
//
// Shareaza 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 Shareaza; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
//

#include "StdAfx.h"
#include "Shareaza.h"
#include "Settings.h"
#include "QuerySearch.h"
#include "Schema.h"
#include "XML.h"
#include "Skin.h"
#include "DlgNewSearch.h"

#include "SHA.h"
#include "ED2K.h"
#include "TigerTree.h"

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

BEGIN_MESSAGE_MAP(CNewSearchDlg, CSkinDialog)
	//{{AFX_MSG_MAP(CNewSearchDlg)
	ON_WM_SIZE()
	ON_WM_GETMINMAXINFO()
	ON_CBN_SELCHANGE(IDC_SCHEMAS, OnSelChangeSchemas)
	ON_CBN_CLOSEUP(IDC_SCHEMAS, OnCloseUpSchemas)
	ON_EN_CHANGE(IDC_SEARCH, OnChangeSearch)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()


/////////////////////////////////////////////////////////////////////////////
// CNewSearchDlg dialog

CNewSearchDlg::CNewSearchDlg(CWnd* pParent, CQuerySearch* pSearch, BOOL bLocal, BOOL bAgain) : CSkinDialog( CNewSearchDlg::IDD, pParent )
{
	//{{AFX_DATA_INIT(CNewSearchDlg)
	//}}AFX_DATA_INIT
	m_pSearch	= pSearch;
	m_bLocal	= bLocal;
	m_bAgain	= bAgain;
}

CNewSearchDlg::~CNewSearchDlg()
{
	if ( m_pSearch ) delete m_pSearch;
}

void CNewSearchDlg::DoDataExchange(CDataExchange* pDX)
{
	CSkinDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CNewSearchDlg)
	DDX_Control(pDX, IDCANCEL, m_wndCancel);
	DDX_Control(pDX, IDOK, m_wndOK);
	DDX_Control(pDX, IDC_SCHEMAS, m_wndSchemas);
	DDX_Control(pDX, IDC_SEARCH, m_wndSearch);
	//}}AFX_DATA_MAP
}

/////////////////////////////////////////////////////////////////////////////
// CNewSearchDlg message handlers

BOOL CNewSearchDlg::OnInitDialog() 
{
	CSkinDialog::OnInitDialog();

	SkinMe( _T("CNewSearchDlg"), IDR_SEARCHFRAME );

	SelectCaption( this, m_bLocal ? 2 : ( m_bAgain ? 1 : 0 ) );

	CRect rc;
	m_wndSchema.Create( WS_CHILD|WS_VISIBLE|WS_BORDER|WS_TABSTOP, rc, this, IDC_METADATA );
	
	m_wndSchemas.m_sNoSchemaText = _T("Plain Text Search");
	m_wndSchemas.Load( Settings.Search.LastSchemaURI );
	
	if ( m_pSearch != NULL )
	{
		m_wndSchemas.Select( m_pSearch->m_pSchema );
	}
	else
	{
		m_pSearch = new CQuerySearch();
	}
	
	OnSelChangeSchemas();
	
	if ( m_pSearch->m_pXML )
	{
		m_wndSchema.UpdateData( m_pSearch->m_pXML->GetFirstElement(), FALSE );
	}
	
	Settings.LoadWindow( _T("NewSearch"), this );
	
	OnCloseUpSchemas();
	
	if ( m_pSearch->m_bSHA1 )
	{
		m_wndSearch.SetWindowText( CSHA::HashToString( &m_pSearch->m_pSHA1, TRUE ) );
		m_wndSchema.ShowWindow( SW_HIDE );
	}
	else
	{
		m_wndSearch.SetWindowText( m_pSearch->m_sSearch );
	}
	
	if ( m_wndSchemas.GetCurSel() > 0 ) m_wndSchemas.SetFocus();
	
	return FALSE;
}

void CNewSearchDlg::OnGetMinMaxInfo(MINMAXINFO FAR* lpMMI) 
{
	CSkinDialog::OnGetMinMaxInfo( lpMMI );
	lpMMI->ptMinTrackSize.x = 256;
	lpMMI->ptMinTrackSize.y = 128;
}

void CNewSearchDlg::OnSize(UINT nType, int cx, int cy) 
{
	CSkinDialog::OnSize( nType, cx, cy );
	
	if ( ! IsWindow( m_wndSchema.m_hWnd ) ) return;

	int nSpacing	= 8;
	int nHeight		= 20;
	int nButtonSize	= 72;
	
	m_wndSearch.SetWindowPos( NULL, nSpacing, nSpacing, cx - nSpacing * 2, nHeight, SWP_NOZORDER );
	m_wndSchemas.SetWindowPos( NULL, nSpacing, nSpacing * 2 + nHeight, cx - nSpacing * 2, nHeight, SWP_NOZORDER );

	if ( cy > nSpacing * 5 + nHeight * 3 + 4 )
	{
		m_wndSchema.SetWindowPos( NULL,
			nSpacing, nSpacing * 3 + nHeight * 2,
			cx - nSpacing * 2, cy - nSpacing * 5 - nHeight * 3 - 4,
			SWP_NOZORDER|SWP_SHOWWINDOW );
	}
	else
	{
		m_wndSchema.ShowWindow( SW_HIDE );
	}

	nHeight += 4;

	m_wndOK.SetWindowPos( NULL, nSpacing - 1, cy - nSpacing - nHeight, nButtonSize, nHeight, SWP_NOZORDER );
	m_wndCancel.SetWindowPos( NULL, nSpacing * 2 + nButtonSize - 1, cy - nSpacing - nHeight, nButtonSize, nHeight, SWP_NOZORDER );
}

void CNewSearchDlg::OnSelChangeSchemas() 
{
	CSchema* pSchema = m_wndSchemas.GetSelected();
	m_wndSchema.SetSchema( pSchema, TRUE );
}

void CNewSearchDlg::OnCloseUpSchemas() 
{
	CSchema* pSchema = m_wndSchemas.GetSelected();

	CRect rcWindow;
	GetWindowRect( &rcWindow );

	if ( pSchema != NULL )
	{
		if ( rcWindow.Height() <= 200 )
		{
			SetWindowPos( NULL, 0, 0, rcWindow.Width(), 264, SWP_NOMOVE|SWP_NOZORDER );
		}
		PostMessage( WM_KEYDOWN, VK_TAB );
	}
	else
	{
		m_wndSearch.SetFocus();
	}
}

void CNewSearchDlg::OnChangeSearch() 
{
	CString strSearch;
	m_wndSearch.GetWindowText( strSearch );
	
	BOOL bHash = FALSE;
	TIGEROOT pTiger;
	SHA1 pSHA1;
	MD4 pED2K;
	
	bHash |= CSHA::HashFromURN( strSearch, &pSHA1 );
	bHash |= CTigerNode::HashFromURN( strSearch, &pTiger );
	bHash |= CED2K::HashFromURN( strSearch, &pED2K );
	
	if ( m_wndSchema.IsWindowVisible() == bHash )
	{
		m_wndSchema.ShowWindow( bHash ? SW_HIDE : SW_SHOW );
	}
}

BOOL CNewSearchDlg::PreTranslateMessage(MSG* pMsg) 
{
	if ( pMsg->message == WM_KEYDOWN && pMsg->wParam == VK_TAB )
	{
		CWnd* pFocus = GetFocus();
		if ( m_wndSchema.OnTab() ) return TRUE;
	}
	
	return CSkinDialog::PreTranslateMessage( pMsg );
}

void CNewSearchDlg::OnOK() 
{
	Settings.SaveWindow( _T("NewSearch"), this );
	
	m_wndSearch.GetWindowText( m_pSearch->m_sSearch );
	
	m_pSearch->m_bSHA1	= CSHA::HashFromURN( m_pSearch->m_sSearch, &m_pSearch->m_pSHA1 );
	m_pSearch->m_bTiger	= CTigerNode::HashFromURN( m_pSearch->m_sSearch, &m_pSearch->m_pTiger );
	m_pSearch->m_bED2K	= CED2K::HashFromURN( m_pSearch->m_sSearch, &m_pSearch->m_pED2K );
	
	if ( m_pSearch->m_bSHA1 || m_pSearch->m_bTiger || m_pSearch->m_bED2K )
	{
		m_pSearch->m_sSearch.Empty();
	}
	
	CSchema* pSchema = m_wndSchemas.GetSelected();
	
	if ( m_pSearch->m_pXML != NULL ) delete m_pSearch->m_pXML;
	
	m_pSearch->m_pSchema	= NULL;
	m_pSearch->m_pXML		= NULL;
	
	if ( pSchema != NULL && ! m_pSearch->m_bSHA1 )
	{
		m_pSearch->m_pSchema	= pSchema;
		m_pSearch->m_pXML		= pSchema->Instantiate();
		
		m_wndSchema.UpdateData( m_pSearch->m_pXML->AddElement( pSchema->m_sSingular ), TRUE );
		
		m_pSearch->GetHashFromXML();
		
		Settings.Search.LastSchemaURI = pSchema->m_sURI;
	}
	else
	{
		Settings.Search.LastSchemaURI.Empty();
	}
	
	m_pSearch->m_sSearch.TrimLeft();
	m_pSearch->m_sSearch.TrimRight();
	
	m_pSearch->BuildWordList();
	
	if ( m_pSearch->m_nWords == 0 && ! m_pSearch->m_bSHA1 &&
		 ! m_pSearch->m_bTiger && ! m_pSearch->m_bED2K )
	{
		m_wndSearch.SetFocus();
		return;
	}
	
	CSkinDialog::OnOK();
}

CQuerySearch* CNewSearchDlg::GetSearch()
{
	CQuerySearch* pSearch = m_pSearch;
	m_pSearch = NULL;
	return pSearch;
}