favtreectrl.cpp

用VC写的浏览器

// FavTreeCtrl.cpp : implementation file
//

#include "stdafx.h"
#include "GreenBrowser.h"
#include "FavTreeCtrl.h"
#include "MainFrm.h"
#include "setscdlg.h"
#include "inputalias.h"
#include "FavTree.h"
#include "menuorder.h"

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

//#pragma optimize( "s", on )

/////////////////////////////////////////////////////////////////////////////
// CFavTreeCtrl

CFavTreeCtrl::CFavTreeCtrl()
{
	m_hStartFolder = NULL;
	pImageList = NULL;
	m_nExpandTimer = 0;
	m_nScrollTimer =0;
	m_bDragging = FALSE;
	m_hItem = NULL;
	m_nTimerTicks = 0;
	m_hOldDropItem = NULL;
}

CFavTreeCtrl::~CFavTreeCtrl()
{
	try{
	m_astrFav.RemoveAll();
	if(pImageList)
		delete pImageList;
	}catch(...){}
}

BEGIN_MESSAGE_MAP(CFavTreeCtrl, CTreeCtrl)
	//{{AFX_MSG_MAP(CFavTreeCtrl)
	ON_NOTIFY_REFLECT(NM_CLICK, OnClick)
	ON_NOTIFY_REFLECT(NM_RCLICK, OnRclick)
	ON_COMMAND(ID_RFAV_COPY, OnRfavCopy)
	ON_COMMAND(ID_RFAV_DEL, OnRfavDel)
	ON_COMMAND(ID_RFAV_OPENALL, OnRfavOpenall)
	ON_COMMAND(ID_RFAV_START_FOLDER, OnRfavStartFolder)
	ON_COMMAND(ID_RFAV_START, OnRfavStart)
	ON_COMMAND(ID_RFAV_OPEN_NEW, OnRfavOpenNew)
	ON_COMMAND(ID_RFAV_ADDSHORTCUT, OnRfavAddshortcut)
	ON_COMMAND(ID_RFAV_ALIAS, OnRfavAlias)
	ON_COMMAND(ID_RFAVFOLDER_FIND, OnRfavfolderFind)
	ON_COMMAND(ID_RFAV_PROPERTY, OnRfavProperty)
	ON_NOTIFY_REFLECT(TVN_BEGINDRAG, OnBegindrag)
	ON_WM_MOUSEMOVE()
	ON_WM_LBUTTONUP()
	ON_WM_TIMER()
	ON_WM_DESTROY()
	ON_COMMAND(ID_RFAVFOLDER_RESET, OnRfavfolderReset)
	ON_COMMAND(ID_RFAV_RENAME, OnRfavRename)
	ON_NOTIFY_REFLECT(TVN_ENDLABELEDIT, OnEndlabeledit)
	ON_COMMAND(ID_RFAVFOLDER_ONLY, OnRfavfolderOnly)
	ON_NOTIFY_REFLECT(TVN_BEGINLABELEDIT, OnBeginlabeledit)
	//}}AFX_MSG_MAP
	ON_COMMAND_RANGE(0x6101, 0x61ff, OnAddtoGroup)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CFavTreeCtrl message handlers

void CFavTreeCtrl::OnClick(NMHDR* pNMHDR, LRESULT* pResult) 
{
	// TODO: Add your control notification handler code here
	HTREEITEM hItem;

	CPoint pt;
	GetCursorPos(&pt);
	ScreenToClient(&pt);
	hItem = HitTest(pt);

	if ((hItem != NULL) && !ItemHasChildren(hItem))
	{
		int img, selimg;
		GetItemImage(hItem, img, selimg);
		if(img == 0 || img == 5)
		{
			ExpandNode(hItem);
			Expand(hItem,TVE_TOGGLE);
		}
		else
		{
			CGreenBrowserView* pvw = GetCurrentView(1, g_bFavInNewWindow);
			int d = GetItemData(hItem);
			if(pvw!=NULL && d>=0)
			{
				CString url;
				POSITION pos = m_astrFav.FindIndex(d);
				if(pos!=NULL)
					GetURL(m_astrFav.GetAt(pos), url);
				pvw->ToNavigate(url,0,NULL);
			}
		}
	}
	else
		Expand(hItem,TVE_TOGGLE);

	*pResult = 0;	
}

void CFavTreeCtrl::OnRclick(NMHDR* pNMHDR, LRESULT* pResult) 
{
	// TODO: Add your control notification handler code here
	HTREEITEM hItem;

	CPoint pt;
	GetCursorPos(&pt);
	ScreenToClient(&pt);
	hItem = HitTest(pt);
	m_hItem = hItem;

	if ((hItem != NULL) && GetItemData(hItem)!=-1)
	{
		CMenu menu;
		CMenu* pPopup, *pGroups = NULL;

		// the font popup is stored in a resource
		menu.LoadMenu(IDR_RFAV_URL);
		pPopup = menu.GetSubMenu(0);
		LOADMENU(pPopup,"MenuFavUrl");
		int nImage, nSImage;
		GetItemImage(hItem, nImage, nSImage);
		if(m_hStartFolder == NULL)
			pPopup->EnableMenuItem(ID_RFAV_START, MF_GRAYED|MF_DISABLED);

		//load group list
		pGroups = pPopup->GetSubMenu(0);
		pmf->MenuLoadGroupList(pGroups, 0);

		ClientToScreen(&pt);
		pPopup->TrackPopupMenu(TPM_LEFTALIGN | TPM_LEFTBUTTON | TPM_RIGHTBUTTON, pt.x, pt.y + 1, this);
	}
	else if ((hItem != NULL))
	{
		CMenu menu;
		CMenu* pPopup;

		// the font popup is stored in a resource
		menu.LoadMenu(IDR_RFAV_FOLDER);
		pPopup = menu.GetSubMenu(0);
		LOADMENU(pPopup,"MenuFavFolder");
		if(hItem == m_hStartFolder)
		{
			CString menustr;
			LOADSTR(menustr ,IDS_UNSET_START_FOLDER);
			pPopup->ModifyMenu(ID_RFAV_START_FOLDER,MF_BYCOMMAND,ID_RFAV_START_FOLDER, menustr);
		}
		if (((CFavTree*)m_pFavTree)->m_wndFav.m_bFolderOnly)
			pPopup->CheckMenuItem(ID_RFAVFOLDER_ONLY,MF_CHECKED);
		else
			pPopup->CheckMenuItem(ID_RFAVFOLDER_ONLY,MF_UNCHECKED);
		ClientToScreen(&pt);
		pPopup->TrackPopupMenu(TPM_LEFTALIGN | TPM_LEFTBUTTON | TPM_RIGHTBUTTON, pt.x, pt.y + 1, this);
	}
	
	*pResult = 0;
}

void CFavTreeCtrl::OnRfavCopy() 
{
	// TODO: Add your command handler code here
	HTREEITEM hItem;
	hItem = m_hItem;
	if (hItem != NULL )
	{
		int npos = GetItemData(hItem);
		POSITION pos = m_astrFav.FindIndex(npos);
		CString url;
		if(pos!=NULL)
			url = m_astrFav.GetAt(pos);
		_SetClipString(url);
	}
}

void CFavTreeCtrl::OnRfavDel() 
{
	// TODO: Add your command handler code here
	CString strName;
	strName = GetItemText(m_hItem);
	CString path,szPath;
	if(!pmf->GetFavoriteFolder(szPath))
		return ;

	GetItemPath(m_hItem, path);
	path = "\\"+path;
	path = szPath+ path;
	if(DeleteFile(path+strName+".url"))
	{
		DeleteItem(m_hItem);
	}
	else if(DeleteFile(path+strName+".lnk"))
	{
		DeleteItem(m_hItem);
	}
	
	//update menu
	pmf->m_bFavDirty = 2;
}

void CFavTreeCtrl::OnRfavOpenall() 
{
	// TODO: Add your command handler code here
	ExpandNode(m_hItem);
	HTREEITEM hit;
	hit = GetChildItem(m_hItem);
	int nWins = 0;

	while(hit!=NULL && (nWins<g_nMaxWins || g_nMaxWins==0))
	{
		if(GetItemData(hit)!=-1)
		{
			if(nWins!=0)
			{
				CString url;
				int d = GetItemData(hit);
				POSITION pos = m_astrFav.FindIndex(d);
				if(pos!=NULL)
					GetURL(m_astrFav.GetAt(pos), url);
				pmf->m_astrOpenAll.AddTail(url);
				nWins++;
			}
			else
			{
				CGreenBrowserView* pvw = CreateNewView();
				int d = GetItemData(hit);
				if(pvw!=NULL)
				{
					CString url;
					POSITION pos = m_astrFav.FindIndex(d);
					if(pos!=NULL)
						GetURL(m_astrFav.GetAt(pos), url);
					pvw->ToNavigate(url,0,NULL);
					nWins ++;
				}
			}
		}
		hit = GetNextSiblingItem(hit);
	}
	if(nWins >1 && pmf->m_nInterval)
		pmf->m_nTimerID = ::SetTimer(pmf->GetSafeHwnd(), 1000, pmf->m_nInterval*1000,NULL);
	else if(nWins>1)
		pmf->m_nTimerID = ::SetTimer(pmf->GetSafeHwnd(), 1000, 20,NULL);
	//enable toolbar button
	pmf->m_wndToolBar.GetToolBarCtrl().EnableButton(ID_FILE_STOP_OPENALL);
}

void CFavTreeCtrl::OnRfavStartFolder() 
{
	// TODO: Add your command handler code here
	try{
	HTREEITEM hItem ;
	hItem = m_hItem;
	if (hItem != NULL )
	{
		//get the path
		CString	szPath;
		// find out from the registry where the favorites are located.
		if(!pmf->GetFavoriteFolder(szPath))
			return ;
		CString path;
		while(hItem !=NULL)
		{
			path = GetItemText(hItem)+"\\"+path;
			hItem = GetParentItem(hItem);
		}
		if(m_hItem != GetRootItem())
		{
			path = "\\"+path;
			path = szPath+ path;
		}
		else
			path = szPath;

		CString oldpath = pmf->m_strStartFavFolder;
		if(oldpath == path)
		{
			//clean the startup folder
			pmf->m_strStartFavFolder = "";
			SetItemState(m_hStartFolder,0 , TVIS_BOLD);
			m_hStartFolder = NULL;
			return;
		}

		//update icon
		SetItemState(m_hItem,TVIS_BOLD,TVIS_BOLD);
		SetItemState(m_hStartFolder,0 , TVIS_BOLD);

		//save the path
		pmf->m_strStartFavFolder = path;
		m_hStartFolder = m_hItem;
	}
	}catch(...){}
}

void CFavTreeCtrl::OnRfavStart() 
{
	// TODO: Add your command handler code here
	try{

	if(GetParentItem(m_hItem)==m_hStartFolder || m_hStartFolder == NULL)
		return;

	CString strName;
	strName = GetItemText(m_hItem);
	int id = GetItemData(m_hItem);
	CString path,szPath;
	if(!pmf->GetFavoriteFolder(szPath))
		return ;

	GetItemPath(m_hItem, path);
	HTREEITEM hItem;
	path = "\\"+path;
	path = szPath+ path;

	if(!CopyFile(path+strName+".url", pmf->m_strStartFavFolder+strName+".url",TRUE))
	{
		if(CopyFile(path+strName+".lnk", pmf->m_strStartFavFolder+strName+".lnk",TRUE))
		{
			if(ItemHasChildren(m_hStartFolder))
			{
				hItem = InsertItem(strName, 1,1,m_hStartFolder, TVI_SORT);
				SetItemData(hItem, id);
			}
		}
	}
	else
	{
		if(ItemHasChildren(m_hStartFolder))
		{
			hItem = InsertItem(strName, 1,1,m_hStartFolder, TVI_SORT);
			SetItemData(hItem, id);
		}
	}

	//update menu
	pmf->m_bFavDirty = 2;
	}catch(...){}
}

void CFavTreeCtrl::OnRfavOpenNew() 
{
	// TODO: Add your command handler code here
	CGreenBrowserView* pvw = CreateNewView();
	int d = GetItemData(m_hItem);
	if(pvw!=NULL && d>=0)
	{
		CString url;
		POSITION pos = m_astrFav.FindIndex(d);
		if(pos!=NULL)
			GetURL(m_astrFav.GetAt(pos), url);
		pvw->ToNavigate(url,0,NULL);
	}
}

BOOL CFavTreeCtrl::GetURL(CString &path, CString &url)
{
	url = path;
	if(url.Right(4) == _T(".lnk"))
	{
		TCHAR buf[2048];
		if(pmf->ResolveLink(url, buf))
		{
			url = buf;
			path = url;
		}
		else
			url = "about:blank";
	}
	return TRUE;
}

void CFavTreeCtrl::OnRfavAddshortcut() 
{
	// TODO: Add your command handler code here
	int d = GetItemData(m_hItem);
	if(d>=0)
	{
		CString url;
		POSITION pos = m_astrFav.FindIndex(d);
		if(pos!=NULL)
			GetURL(m_astrFav.GetAt(pos), url);
		CSetSCDlg dlg;
		dlg.m_strEdit = url;
		dlg.DoModal();
	}	
}

void CFavTreeCtrl::OnRfavAlias() 
{
	// TODO: Add your command handler code here
	int d = GetItemData(m_hItem);
	if(d>=0)
	{
		CString url, alias;
		POSITION pos = m_astrFav.FindIndex(d);
		if(pos!=NULL)
			GetURL(m_astrFav.GetAt(pos), url);
		CInputAlias dlg;
		dlg.m_url = url;
		dlg.m_alias = GetItemText(m_hItem);
		if(dlg.DoModal()==IDOK)
		{
			alias = dlg.m_alias; url = dlg.m_url;
			alias.TrimLeft();
			url.TrimLeft();
			if(alias !="" && !alias.IsEmpty() && url!="" && !url.IsEmpty())
				pmf->m_mapAlias.SetAt(alias, url);
		}
	}	
}

int CFavTreeCtrl::ExpandNode(HTREEITEM hItem)
{
	if ((hItem != NULL) && !ItemHasChildren(hItem))
	{
		int img, selimg;
		GetItemImage(hItem, img, selimg);
		if(img == 0 || img == 5)
		{
			int nEndPos = m_astrFav.GetCount();
			
			CString strName;
			strName = GetItemText(hItem);
			CString path,szPath;
			if(!pmf->GetFavoriteFolder(szPath))
				return 0;

			GetItemPath(hItem, path);
			path = "\\"+path;
			path = szPath+ path + strName;

			return ((CFavTree*)m_pFavTree)->BuildFavorites(path, nEndPos, hItem);
		}
	}
	return 0;
}

void CFavTreeCtrl::OnRfavfolderFind() 
{
	// TODO: Add your command handler code here
	CString path,szPath;
	if(!pmf->GetFavoriteFolder(szPath))
		return ;