WizardSharePage.cpp

Go to the documentation of this file.

//
// WizardSharePage.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 "WizardSheet.h"
#include "WizardSharePage.h"
#include "Library.h"
#include "LibraryFolders.h"
#include "SharedFolder.h"
#include "DlgFolderScan.h"
#include "ShellIcons.h"
#include "Skin.h"
#include "DlgHelp.h"

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

IMPLEMENT_DYNCREATE(CWizardSharePage, CWizardPage)

BEGIN_MESSAGE_MAP(CWizardSharePage, CWizardPage)
        //{{AFX_MSG_MAP(CWizardSharePage)
        ON_NOTIFY(LVN_ITEMCHANGED, IDC_SHARE_FOLDERS, OnItemChangedShareFolders)
        ON_BN_CLICKED(IDC_SHARE_ADD, OnShareAdd)
        ON_BN_CLICKED(IDC_SHARE_REMOVE, OnShareRemove)
        //}}AFX_MSG_MAP
END_MESSAGE_MAP()


// CWizardSharePage property page

CWizardSharePage::CWizardSharePage() : CWizardPage(CWizardSharePage::IDD)
{
        //{{AFX_DATA_INIT(CWizardSharePage)
        //}}AFX_DATA_INIT
}

CWizardSharePage::~CWizardSharePage()
{
}

void CWizardSharePage::DoDataExchange(CDataExchange* pDX)
{
        CPropertyPage::DoDataExchange(pDX);
        //{{AFX_DATA_MAP(CWizardSharePage)
        DDX_Control(pDX, IDC_SHARE_FOLDERS, m_wndList);
        DDX_Control(pDX, IDC_SHARE_REMOVE, m_wndRemove);
        //}}AFX_DATA_MAP
}

// CWizardSharePage message handlers

BOOL CWizardSharePage::OnInitDialog()
{
        CPropertyPage::OnInitDialog();

        Skin.Apply( _T("CWizardSharePage"), this );

        CBitmap bmBase;
        bmBase.LoadBitmap( IDB_FOLDERS );

        CRect rc;
        m_wndList.GetClientRect( &rc );
        m_wndList.InsertColumn( 0, _T("Folder"), LVCFMT_LEFT, rc.Width() - GetSystemMetrics( SM_CXVSCROLL ) );
        m_wndList.SetImageList( ShellIcons.GetObject( 16 ), LVSIL_SMALL );
        m_wndList.EnableToolTips( TRUE );

        {
                CQuickLock oLock( Library.m_pSection );

                for ( POSITION pos = LibraryFolders.GetFolderIterator() ; pos ; )
                {
                        CLibraryFolder* pFolder = LibraryFolders.GetNextFolder( pos );

                        m_wndList.InsertItem( LVIF_TEXT|LVIF_IMAGE, m_wndList.GetItemCount(),
                                pFolder->m_sPath, 0, 0, SHI_FOLDER_OPEN, 0 );
                }

                CreateDirectory( Settings.Downloads.CompletePath, NULL );
                AddPhysicalFolder( Settings.Downloads.CompletePath );

                CreateDirectory( Settings.Downloads.CollectionPath, NULL );
                AddPhysicalFolder( Settings.Downloads.CollectionPath );

                AddPhysicalFolder( _T("C:\\Program Files\\Piolet\\My Shared Folder") );
                AddPhysicalFolder( _T("C:\\Program Files\\eMule\\Incoming") );
                AddPhysicalFolder( _T("C:\\Program Files\\eDonkey2000\\incoming") );
                AddPhysicalFolder( _T("C:\\Program Files\\Ares\\My Shared Folder") );
                AddPhysicalFolder( _T("C:\\Program Files\\morpheus\\My Shared Folder") );


                AddRegistryFolder( HKEY_CURRENT_USER, _T("Software\\Kazaa\\Transfer"), _T("DlDir0") );
                AddRegistryFolder( HKEY_CURRENT_USER, _T("Software\\Xolox"), _T("completedir") );
                AddRegistryFolder( HKEY_CURRENT_USER, _T("Software\\Xolox"), _T("sharedir") );
        }

        return TRUE;
}

BOOL CWizardSharePage::OnSetActive()
{
        SetWizardButtons( PSWIZB_BACK | PSWIZB_NEXT );
        return CWizardPage::OnSetActive();
}

void CWizardSharePage::AddPhysicalFolder(LPCTSTR pszFolder)
{
        DWORD nFlags = GetFileAttributes( pszFolder );

        if ( nFlags == 0xFFFFFFFF ) return;
        if ( ( nFlags & FILE_ATTRIBUTE_DIRECTORY ) == 0 ) return;

        for ( int nItem = 0 ; nItem < m_wndList.GetItemCount() ; nItem++ )
        {
                if ( m_wndList.GetItemText( nItem, 0 ).CompareNoCase( pszFolder ) == 0 ) return;
        }

        m_wndList.InsertItem( LVIF_TEXT|LVIF_IMAGE, m_wndList.GetItemCount(),
                pszFolder, 0, 0, SHI_FOLDER_OPEN, 0 );
}

void CWizardSharePage::AddRegistryFolder(HKEY hRoot, LPCTSTR pszKey, LPCTSTR pszValue)
{
        TCHAR szFolder[MAX_PATH];
        DWORD dwType, dwFolder;
        HKEY hKey = NULL;

        if ( RegOpenKeyEx( hRoot, pszKey, 0, KEY_READ, &hKey ) != ERROR_SUCCESS ) return;

        dwType = REG_SZ;
        dwFolder = MAX_PATH - 1;

        if ( RegQueryValueEx( hKey, pszValue, NULL, &dwType, (LPBYTE)szFolder, &dwFolder )
                 != ERROR_SUCCESS || dwType != REG_SZ )
        {
                RegCloseKey( hKey );
                return;
        }

        RegCloseKey( hKey );

        szFolder[ dwFolder ] = 0;

        AddPhysicalFolder( szFolder );
}

void CWizardSharePage::OnItemChangedShareFolders(NMHDR* pNMHDR, LRESULT* pResult)
{
        NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
        m_wndRemove.EnableWindow( m_wndList.GetSelectedCount() > 0 );
        *pResult = 0;
}

void CWizardSharePage::OnShareAdd()
{
        //Let user select path to share
        TCHAR szPath[MAX_PATH];
        LPITEMIDLIST pPath;
        LPMALLOC pMalloc;
        BROWSEINFO pBI;

        ZeroMemory( &pBI, sizeof(pBI) );
        pBI.hwndOwner           = AfxGetMainWnd()->GetSafeHwnd();
        pBI.pszDisplayName      = szPath;
        pBI.lpszTitle           = _T("Select folder to share:");
        pBI.ulFlags                     = BIF_RETURNONLYFSDIRS | BIF_NEWDIALOGSTYLE;

        pPath = SHBrowseForFolder( &pBI );

        if ( pPath == NULL ) return;

        SHGetPathFromIDList( pPath, szPath );
        SHGetMalloc( &pMalloc );
        pMalloc->Free( pPath );

        CString strPathLC( szPath );
        CharLower( strPathLC.GetBuffer() );
        strPathLC.ReleaseBuffer();


        //Get system paths (to compare)
        CString strWindowsLC, strProgramsLC;
        PTSTR pszWindowsPath, pszProgramsPath;

        pszWindowsPath = strWindowsLC.GetBuffer( MAX_PATH + 1 );
        pszProgramsPath = strProgramsLC.GetBuffer( MAX_PATH + 1 );

        if ( HINSTANCE hShell = LoadLibrary( _T("shfolder.dll") ) )
        {
                HRESULT (WINAPI *pfnSHGetFolderPath)(HWND, int, HANDLE, DWORD, LPWSTR);
                (FARPROC&)pfnSHGetFolderPath = GetProcAddress( hShell, "SHGetFolderPathW" );
                if ( pfnSHGetFolderPath != NULL )
                {
                        (*pfnSHGetFolderPath)(NULL, CSIDL_WINDOWS, NULL, NULL, pszWindowsPath);
                        (*pfnSHGetFolderPath)(NULL, CSIDL_PROGRAM_FILES, NULL, NULL, pszProgramsPath);
                }
                FreeLibrary( hShell );
        }
        CharLower( pszWindowsPath );
        CharLower( pszProgramsPath );

        strWindowsLC.ReleaseBuffer();
        strProgramsLC.ReleaseBuffer();

        if ( strWindowsLC.IsEmpty() ) strWindowsLC = _T("c:\\windows");
        if ( strProgramsLC.IsEmpty() ) strProgramsLC = _T("c:\\program files");


        //Get various shareaza paths (to compare)
        CString strIncompletePathLC = Settings.Downloads.IncompletePath;
        CharLower( strIncompletePathLC.GetBuffer() );
        strIncompletePathLC.ReleaseBuffer();

        CString strGeneralPathLC = Settings.General.Path;
        CharLower( strGeneralPathLC.GetBuffer() );
        strGeneralPathLC.ReleaseBuffer();

        CString strUserPathLC = Settings.General.UserPath;
        CharLower( strUserPathLC.GetBuffer() );
        strUserPathLC.ReleaseBuffer();


        //Check shared path isn't invalid
        if ( strPathLC == _T( "" ) ||
                 strPathLC == strWindowsLC.Left( 3 ) ||
                 strPathLC == strProgramsLC ||
                 strPathLC == strWindowsLC ||
                 strPathLC == strGeneralPathLC ||
                 strPathLC == strGeneralPathLC + _T("\\data") ||
                 strPathLC == strUserPathLC ||
                 strPathLC == strUserPathLC + _T("\\data") ||
                 strPathLC == strIncompletePathLC )
        {
                CHelpDlg::Show( _T("ShareHelp.BadShare") );
                return;
        }

        BOOL bForceAdd = FALSE;
        //Check shared path isn't already shared
        for ( int nItem = 0 ; nItem < m_wndList.GetItemCount() ; nItem++ )
        {
                BOOL bSubFolder = FALSE;
                CString strOldLC( m_wndList.GetItemText( nItem, 0 ) );
                CharLower( strOldLC.GetBuffer() );
                strOldLC.ReleaseBuffer();

                if ( strPathLC == strOldLC )
                {
                        // Matches exactly
                }
                else if ( strPathLC.GetLength() > strOldLC.GetLength() )
                {
                        if ( strPathLC.Left( strOldLC.GetLength() + 1 ) != strOldLC + '\\' )
                                continue;
                }
                else if ( strPathLC.GetLength() < strOldLC.GetLength() )
                {
                        bSubFolder = TRUE;
                        if ( strOldLC.Left( strPathLC.GetLength() + 1 ) != strPathLC + '\\' )
                                continue;
                }
                else
                {
                        continue;
                }

                if ( bSubFolder )
                {
                        CString strFormat, strMessage;
                        LoadString( strFormat, IDS_LIBRARY_SUBFOLDER_IN_LIBRARY );
                        strMessage.Format( strFormat, (LPCTSTR)szPath );

                        if ( bForceAdd || AfxMessageBox( strMessage, MB_ICONQUESTION|MB_YESNO ) == IDYES )
                        {
                                // Don't bother asking again- remove all sub-folders
                                bForceAdd = TRUE;
                                // Remove the sub-folder
                                m_wndList.DeleteItem( nItem );
                                nItem--;
                        }
                        else
                        {
                                return;
                        }
                }
                else
                {
                        CString strFormat, strMessage;
                        Skin.LoadString( strFormat, IDS_WIZARD_SHARE_ALREADY );
                        strMessage.Format( strFormat, (LPCTSTR)strOldLC );
                        AfxMessageBox( strMessage, MB_ICONINFORMATION );
                        //CHelpDlg::Show(  _T( "ShareHelp.AlreadyShared" ) );
                        return;
                }
        }

        //Add path to shared list
        m_wndList.InsertItem( LVIF_TEXT|LVIF_IMAGE, m_wndList.GetItemCount(),
                szPath, 0, 0, SHI_FOLDER_OPEN, 0 );
}

void CWizardSharePage::OnShareRemove()
{
        for ( int nItem = 0 ; nItem < m_wndList.GetItemCount() ; nItem++ )
        {
                if ( m_wndList.GetItemState( nItem, LVIS_SELECTED ) )
                {
                        m_wndList.DeleteItem( nItem-- );
                }
        }
}

LRESULT CWizardSharePage::OnWizardNext()
{
        CWaitCursor pCursor;

        if ( m_wndList.GetItemCount() == 0 )
        {
                CString strMessage;
                Skin.LoadString( strMessage, IDS_WIZARD_SHARE_CONFIRM );
                if ( AfxMessageBox( strMessage, MB_ICONQUESTION|MB_YESNO ) == IDNO )
                        return -1;
        }

        {
                CQuickLock oLock( Library.m_pSection );

                for ( POSITION pos = LibraryFolders.GetFolderIterator() ; pos ; )
                {
                        CLibraryFolder* pFolder = LibraryFolders.GetNextFolder( pos );

                        int nItem = 0;
                        for ( ; nItem < m_wndList.GetItemCount() ; nItem++ )
                        {
                                CString strFolder = m_wndList.GetItemText( nItem, 0 );
                                if ( strFolder.CompareNoCase( pFolder->m_sPath ) == 0 ) break;
                        }

                        if ( nItem >= m_wndList.GetItemCount() )
                        {
                                LibraryFolders.RemoveFolder( pFolder );
                        }
                }

                for ( int nItem = 0 ; nItem < m_wndList.GetItemCount() ; nItem++ )
                {
                        LibraryFolders.AddFolder( m_wndList.GetItemText( nItem, 0 ) );
                }
        }

        CFolderScanDlg dlgScan;
        dlgScan.DoModal();

        return 0;
}

---