WndBrowseHost.cpp

Go to the documentation of this file.

//
// WndBrowseHost.cpp
//
// Copyright (c) Shareaza Development Team, 2002-2005.
// 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 "GProfile.h"
#include "MatchObjects.h"
#include "Schema.h"
#include "SchemaCache.h"
#include "HostBrowser.h"
#include "RichDocument.h"
#include "WndBrowseHost.h"
#include "DlgHitColumns.h"
#include "ChatWindows.h"
#include "Skin.h"

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

IMPLEMENT_DYNCREATE(CBrowseHostWnd, CBaseMatchWnd)

BEGIN_MESSAGE_MAP(CBrowseHostWnd, CBaseMatchWnd)
        ON_WM_CREATE()
        ON_WM_DESTROY()
        ON_WM_CONTEXTMENU()
        ON_WM_NCLBUTTONUP()
        ON_WM_SIZE()
        ON_LBN_SELCHANGE(IDC_MATCHES, OnSelChangeMatches)
        ON_UPDATE_COMMAND_UI(ID_BROWSE_STOP, OnUpdateBrowseHostStop)
        ON_COMMAND(ID_BROWSE_STOP, OnBrowseHostStop)
        ON_COMMAND(ID_BROWSE_REFRESH, OnBrowseHostRefresh)
        ON_UPDATE_COMMAND_UI(ID_BROWSE_PROFILE, OnUpdateBrowseProfile)
        ON_COMMAND(ID_BROWSE_PROFILE, OnBrowseProfile)
        ON_UPDATE_COMMAND_UI(ID_BROWSE_FILES, OnUpdateBrowseFiles)
        ON_COMMAND(ID_BROWSE_FILES, OnBrowseFiles)
        ON_UPDATE_COMMAND_UI(ID_SEARCH_CHAT, OnUpdateSearchChat)
        ON_COMMAND(ID_SEARCH_CHAT, OnSearchChat)
END_MESSAGE_MAP()


// CBrowseHostWnd construction

CBrowseHostWnd::CBrowseHostWnd(SOCKADDR_IN* pAddress, GGUID* pClientID)
{
        m_pBrowser = new CHostBrowser( this, &pAddress->sin_addr, htons( pAddress->sin_port ), FALSE, pClientID );
        Create( IDR_BROWSEHOSTFRAME );
}

CBrowseHostWnd::CBrowseHostWnd(IN_ADDR* pAddress, WORD nPort, BOOL bMustPush, GGUID* pClientID)
{
        m_pBrowser = new CHostBrowser( this, pAddress, nPort, bMustPush, pClientID );
        Create( IDR_BROWSEHOSTFRAME );
}

CBrowseHostWnd::~CBrowseHostWnd()
{
        if ( m_pBrowser ) delete m_pBrowser;
}

// CBrowseHostWnd message handlers

int CBrowseHostWnd::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
        if ( CBaseMatchWnd::OnCreate( lpCreateStruct ) == -1 ) return -1;

        if ( CSchema* pSchema = SchemaCache.Get( Settings.Search.BlankSchemaURI ) )
        {
                CPtrList pColumns;
                CSchemaColumnsDlg::LoadColumns( pSchema, &pColumns );
                m_wndList.SelectSchema( pSchema, &pColumns );
        }

        m_wndHeader.Create( this );
        m_wndProfile.Create( this );
        m_wndFrame.Create( this, &m_wndList );

        LoadState( _T("CBrowseHostWnd"), TRUE );

        m_pBrowser->Browse();

        OnSkinChange();

        m_bOnFiles = FALSE;

        return 0;
}

void CBrowseHostWnd::OnDestroy()
{
        m_pBrowser->Stop();

        delete m_pBrowser;
        m_pBrowser = NULL;

        if ( m_wndList.m_pSchema != NULL )
                Settings.Search.BlankSchemaURI = m_wndList.m_pSchema->m_sURI;
        else
                Settings.Search.BlankSchemaURI.Empty();

        SaveState( _T("CBrowseHostWnd") );

        CBaseMatchWnd::OnDestroy();
}

void CBrowseHostWnd::OnSkinChange()
{
        CBaseMatchWnd::OnSkinChange();

        Skin.CreateToolBar( _T("CBrowseHostWnd"), &m_wndToolBar );
        m_wndFrame.OnSkinChange();
        m_wndHeader.OnSkinChange();
        m_wndProfile.OnSkinChange();

        UpdateMessages();
}

BOOL CBrowseHostWnd::OnCmdMsg(UINT nID, int nCode, void* pExtra, AFX_CMDHANDLERINFO* pHandlerInfo)
{
        if ( m_wndHeader.m_hWnd != NULL )
        {
                if ( m_wndHeader.OnCmdMsg( nID, nCode, pExtra, pHandlerInfo ) ) return TRUE;
        }

        if ( m_wndProfile.m_hWnd != NULL )
        {
                if ( m_wndProfile.OnCmdMsg( nID, nCode, pExtra, pHandlerInfo ) ) return TRUE;
        }

        if ( m_wndFrame.m_hWnd != NULL )
        {
                if ( m_wndFrame.OnCmdMsg( nID, nCode, pExtra, pHandlerInfo ) ) return TRUE;
        }

        return CBaseMatchWnd::OnCmdMsg( nID, nCode, pExtra, pHandlerInfo );
}

void CBrowseHostWnd::OnSize(UINT nType, int cx, int cy)
{
        CPanelWnd::OnSize( nType, cx, cy );

        CRect rc;
        GetClientRect( &rc );

        rc.top += 64;
        rc.bottom -= 28;

        m_wndHeader.SetWindowPos( NULL, rc.left, 0, rc.Width(), rc.top, SWP_NOZORDER );
        m_wndToolBar.SetWindowPos( NULL, rc.left, rc.bottom, rc.Width(), 28, SWP_NOZORDER );
        m_wndProfile.SetWindowPos( NULL, rc.left, rc.top, rc.Width(), rc.Height(), SWP_NOZORDER );
        m_wndFrame.SetWindowPos( NULL, rc.left, rc.top, rc.Width(), rc.Height(), SWP_NOZORDER );
}

void CBrowseHostWnd::OnContextMenu(CWnd* pWnd, CPoint point)
{
        if ( m_bContextMenu )
        {
                TrackPopupMenu( _T("CBrowseHostWnd"), point, ID_SEARCH_DOWNLOAD );
        }
        else
        {
                CBaseMatchWnd::OnContextMenu( pWnd, point );
        }
}

void CBrowseHostWnd::OnUpdateBrowseProfile(CCmdUI* pCmdUI)
{
        pCmdUI->SetCheck( m_wndProfile.IsWindowVisible() );
}

void CBrowseHostWnd::OnBrowseProfile()
{
        m_wndProfile.SetWindowPos( &wndTop, 0, 0, 0, 0, SWP_NOMOVE|SWP_NOSIZE|SWP_SHOWWINDOW );
        m_wndFrame.ShowWindow( SW_HIDE );
        m_wndProfile.Update( m_pBrowser );
}

void CBrowseHostWnd::OnUpdateBrowseFiles(CCmdUI* pCmdUI)
{
        pCmdUI->SetCheck( m_wndList.IsWindowVisible() );
}

void CBrowseHostWnd::OnBrowseFiles()
{
        m_wndFrame.SetWindowPos( &wndTop, 0, 0, 0, 0, SWP_NOMOVE|SWP_NOSIZE|SWP_SHOWWINDOW );
        m_wndProfile.ShowWindow( SW_HIDE );
        m_bOnFiles = TRUE;
}

void CBrowseHostWnd::OnUpdateBrowseHostStop(CCmdUI* pCmdUI)
{
        pCmdUI->Enable( m_pBrowser->IsBrowsing() );
}

void CBrowseHostWnd::OnBrowseHostStop()
{
        m_pBrowser->Stop();
}

void CBrowseHostWnd::OnBrowseHostRefresh()
{
        m_pBrowser->Stop();
        m_pBrowser->Browse();

        m_wndList.DestructiveUpdate();
        m_pMatches->Clear();

        m_bUpdate = TRUE;
        PostMessage( WM_TIMER, 2 );
}

void CBrowseHostWnd::OnUpdateSearchChat(CCmdUI* pCmdUI)
{
        pCmdUI->Enable( m_pBrowser && m_pBrowser->m_bCanChat );
}

void CBrowseHostWnd::OnSearchChat()
{
        if ( m_pBrowser != NULL )
        {
                ChatWindows.OpenPrivate( &m_pBrowser->m_pClientID,
                        &m_pBrowser->m_pAddress, m_pBrowser->m_nPort, m_pBrowser->m_bMustPush, m_pBrowser->m_nProtocol );
        }
}

void CBrowseHostWnd::OnSelChangeMatches()
{
        m_wndFrame.OnSelChangeMatches();
}

// CBrowseHostWnd event handlers

void CBrowseHostWnd::UpdateMessages(BOOL bActive)
{
        CString strCaption, strOld;

        m_wndHeader.Update( m_pBrowser );

        LoadString( strCaption, IDR_BROWSEHOSTFRAME );
        if ( theApp.m_bRTL ) strCaption = _T("\x200F") + strCaption + _T("\x202E");
        strCaption += _T(" : ");
        if ( theApp.m_bRTL ) strCaption += _T("\x202A\x200E");

        if ( m_pBrowser->m_pProfile != NULL && m_pBrowser->m_pProfile->IsValid() )
        {
                strOld.Format( _T("%s (%s:%lu)"),
                        (LPCTSTR)m_pBrowser->m_pProfile->GetNick(),
                        (LPCTSTR)CString( inet_ntoa( m_pBrowser->m_pAddress ) ),
                        m_pBrowser->m_nPort );
                strCaption += strOld;
        }
        else
        {
                strOld.Format( _T("%s:%lu"),
                        (LPCTSTR)CString( inet_ntoa( m_pBrowser->m_pAddress ) ),
                        m_pBrowser->m_nPort );
                strCaption += strOld;
        }

        if ( m_pMatches->m_nFilteredFiles || m_pMatches->m_nFilteredHits )
        {
                strOld.Format( _T(" [%lu/%lu]"),
                        m_pMatches->m_nFilteredFiles, m_pMatches->m_nFilteredHits );
                if ( theApp.m_bRTL ) strOld = _T("\x202D") + strOld + _T(" ");
                strCaption += strOld;
        }

        GetWindowText( strOld );

        if ( strOld != strCaption ) SetWindowText( strCaption );

        if ( m_pMatches->m_nFilteredFiles == 0 )
        {
                if ( m_pMatches->m_nFiles > 0 )
                {
                        m_wndList.SetMessage( IDS_SEARCH_FILTERED, FALSE );
                        return;
                }

                switch ( m_pBrowser->m_nState )
                {
                case CHostBrowser::hbsNull:
                        m_wndList.SetMessage( m_pBrowser->m_bConnect ? IDS_BROWSE_NOT_SUPPORTED : IDS_BROWSE_CANT_CONNECT );
                        break;
                case CHostBrowser::hbsConnecting:
                        m_wndList.SetMessage( m_pBrowser->m_tPushed ? IDS_BROWSE_PUSHED : IDS_BROWSE_CONNECTING );
                        break;
                case CHostBrowser::hbsRequesting:
                case CHostBrowser::hbsHeaders:
                        m_wndList.SetMessage( IDS_BROWSE_REQUESTING );
                        break;
                case CHostBrowser::hbsContent:
                        LoadString( strOld, IDS_BROWSE_DOWNLOADING );
                        strCaption.Format( strOld, m_pBrowser->GetProgress() * 100.0f );
                        m_wndList.SetMessage( strCaption );
                        break;
                }
        }
}

void CBrowseHostWnd::OnProfileReceived()
{
        if ( ! m_bPaused && m_hWnd != NULL && m_wndProfile.m_hWnd != NULL)
        {
                if ( m_bOnFiles == FALSE || m_wndProfile.IsWindowVisible() )
                {
                        PostMessage( WM_COMMAND, ID_BROWSE_PROFILE );
                }

                m_bUpdate = TRUE;
        }
}

void CBrowseHostWnd::OnBrowseHits(CQueryHit* pHits)
{
        if ( m_bPaused || m_hWnd == NULL ) return;

        CSingleLock pLock( &m_pMatches->m_pSection );

        if ( pLock.Lock( 100 ) && ! m_bPaused )
        {
                m_pMatches->AddHits( pHits );
                m_bUpdate = TRUE;
        }
}

void CBrowseHostWnd::OnHeadPacket(CG2Packet* pPacket)
{
        if ( m_bPaused || m_hWnd == NULL ) return;

        m_wndProfile.OnHeadPacket( pPacket );
}

void CBrowseHostWnd::OnPhysicalTree(CG2Packet* pPacket)
{
        if ( m_bPaused || m_hWnd == NULL ) return;

        m_wndFrame.OnPhysicalTree( pPacket );
}

void CBrowseHostWnd::OnVirtualTree(CG2Packet* pPacket)
{
        if ( m_bPaused || m_hWnd == NULL ) return;

        m_wndFrame.OnVirtualTree( pPacket );
}

BOOL CBrowseHostWnd::OnPush(GGUID* pClientID, CConnection* pConnection)
{
        return m_pBrowser != NULL && m_pBrowser->OnPush( pClientID, pConnection );
}

---