favtreectrl.cpp

myie的源程序是用VC++写自已可以扩搌使用

// FavTreeCtrl.cpp : implementation file
//

#include "stdafx.h"
#include "MyIE.h"
#include "FavTreeCtrl.h"
#include "MainFrm.h"
#include "setscdlg.h"
#include "inputalias.h"
#include "FavTree.h"

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


//MyIE url extension
//[myie]
//MostFav = 1 if it is the most favorite url

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

CFavTreeCtrl::CFavTreeCtrl()
{
	m_hStartFolder = NULL;
	pImageList = NULL;
	m_idTimer = 0;
	m_nTimerID =0;
	m_bDragging = FALSE;
	m_hItem = NULL;
	m_timerticks = 0;
	m_hOldDropItem = NULL;
}

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


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_MOST_FAV, OnRfavMostFav)
	ON_COMMAND(ID_RFAVFOLDER_OPENMOSTFAVORITES, OnRfavfolderOpenmostfavorites)
	ON_COMMAND(ID_RFAV_ADDSHORTCUT, OnRfavAddshortcut)
	ON_COMMAND(ID_RFAV_ALIAS, OnRfavAlias)
	ON_COMMAND(ID_RFAV_MOST_FOLDER, OnRfavMostFolder)
	ON_NOTIFY_REFLECT(NM_DBLCLK, OnDblclk)
	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()
	//}}AFX_MSG_MAP
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
		{
			CMyIEView* pvw = GetCurrentView(1);
			int d = GetItemData(hItem);
			if(pvw!=NULL && d>=0)
			{
				CString url;
				GetURL(m_astrFav[d], 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;

		// the font popup is stored in a resource
		menu.LoadMenu(IDR_RFAV_URL);
		pPopup = menu.GetSubMenu(0);
		int nImage, nSImage;
		GetItemImage(hItem, nImage, nSImage);
		if(nImage == 4)
		{
			CString menustr;
			menustr.LoadString(IDS_UNSET_MOST_FAV);
			pPopup->ModifyMenu(ID_RFAV_MOST_FAV,MF_BYCOMMAND,ID_RFAV_MOST_FAV, menustr);
		}
		if(m_hStartFolder == NULL)
			pPopup->EnableMenuItem(ID_RFAV_START, MF_GRAYED|MF_DISABLED);
		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);
		if(hItem == m_hStartFolder)
		{
			CString menustr;
			menustr.LoadString(IDS_UNSET_START_FOLDER);
			pPopup->ModifyMenu(ID_RFAV_START_FOLDER,MF_BYCOMMAND,ID_RFAV_START_FOLDER, menustr);
		}
		int nImage, nSImage;
		GetItemImage(hItem, nImage, nSImage);
		if(nImage == 5)
		{
			CString menustr;
			menustr.LoadString(IDS_UNSET_MOST_FOLDER);
			pPopup->ModifyMenu(ID_RFAV_MOST_FOLDER,MF_BYCOMMAND,ID_RFAV_MOST_FOLDER, menustr);
		}
		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);
		int l = m_astrFav[npos].GetLength();

		if ( !OpenClipboard() )
		  {
			AfxMessageBox( "Cannot open the Clipboard" );
			return;
		  }
		  // Remove the current Clipboard contents
		  if( !EmptyClipboard() )
		  {
			AfxMessageBox( "Cannot empty the Clipboard" );
			return;
		  }

		HGLOBAL hData = GlobalAlloc(GMEM_MOVEABLE|GMEM_DDESHARE, l+1);
		CopyMemory(GlobalLock(hData), m_astrFav[npos].GetBuffer(1), l+1);
		GlobalUnlock(hData);

		  // For the appropriate data formats...
		  if ( ::SetClipboardData( CF_TEXT, hData ) == NULL )
		  {
			AfxMessageBox( "Unable to set Clipboard data" );
			CloseClipboard();
			GlobalFree(hData);
			return;
		  }
		  // ...
		  CloseClipboard();
		  GlobalFree(hData);
	}
}

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

	GetItemPath(m_hItem, path);
/*	HTREEITEM hItem = GetParentItem(m_hItem);
	HTREEITEM hParent = GetParentItem(hItem);
	HTREEITEM hRoot = GetRootItem();
	while(hParent !=NULL)
	{
		path = GetItemText(hItem)+"\\"+path;
		hItem = hParent;
		hParent = GetParentItem(hParent);
	}*/
	path = "\\"+path;
	path = szPath+ path;

	if(DeleteFile(path+strName+".url"))
	{
		DeleteItem(m_hItem);
	}
	else if(DeleteFile(path+strName+".lnk"))
	{
		DeleteItem(m_hItem);
	}
	
	//update menu
	((CMainFrame*)pMainFrame)->m_bFavDirty = TRUE;

}

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<nMaxWins || nMaxWins==0))
	{
		if(GetItemData(hit)!=-1)
		{
			if(nWins!=0)
			{
				CString url;
				int d = GetItemData(hit);
				GetURL(m_astrFav[d], url);
				((CMainFrame*)pMainFrame)->m_astrOpenAll.AddTail(url);
				nWins++;
			}
			else
			{
				CMyIEView* pvw = CreateNewView();
				int d = GetItemData(hit);
				if(pvw!=NULL)
				{
					CString url;
					GetURL(m_astrFav[d], url);
					pvw->ToNavigate(url,0,NULL);
					nWins ++;
				}
			}
		}
		hit = GetNextSiblingItem(hit);
	}
	if(nWins >1 && ((CMainFrame*)pMainFrame)->m_nInterval)
		((CMainFrame*)pMainFrame)->m_nTimerID = ::SetTimer(((CMainFrame*)pMainFrame)->GetSafeHwnd(), 1000, ((CMainFrame*)pMainFrame)->m_nInterval*1000,NULL);
	else if(nWins>1)
		((CMainFrame*)pMainFrame)->m_nTimerID = ::SetTimer(((CMainFrame*)pMainFrame)->GetSafeHwnd(), 1000, 20,NULL);
	//enable toolbar button
	((CMainFrame*)pMainFrame)->m_wndToolBar.GetToolBarCtrl().EnableButton(ID_FILE_STOP_OPENALL);
}


void CFavTreeCtrl::OnRfavStartFolder() 
{
	// TODO: Add your command handler code here
	HTREEITEM hItem, hParent, hRoot;
	hItem = m_hItem;

	if (hItem != NULL )
	{
		//get the path
		CString           szPath;

		// find out from the registry where the favorites are located.
		if(!((CMainFrame*)pMainFrame)->GetFavoriteFolder(szPath))
			return ;


		CString path;

		hParent = GetParentItem(hItem);
		hRoot = GetRootItem();
		while(hParent !=NULL)
		{
			path = GetItemText(hItem)+"\\"+path;
			hItem = hParent;
			hParent = GetParentItem(hParent);
		}
		path = "\\"+path;
		path = szPath+ path;

		CString oldpath = ((CMainFrame*)pMainFrame)->m_strStartFavFolder;
		if(oldpath == path)
		{
			//clean the startup folder
			((CMainFrame*)pMainFrame)->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
		((CMainFrame*)pMainFrame)->m_strStartFavFolder = path;
		m_hStartFolder = m_hItem;
	}
}

void CFavTreeCtrl::OnRfavStart() 
{
	// TODO: Add your command handler code here
	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(!((CMainFrame*)pMainFrame)->GetFavoriteFolder(szPath))
		return ;

	GetItemPath(m_hItem, path);
	HTREEITEM hItem;
	/*
	HTREEITEM hItem = GetParentItem(m_hItem);
	HTREEITEM hParent = GetParentItem(hItem);
	HTREEITEM hRoot = GetRootItem();
	while(hParent !=NULL)
	{
		path = GetItemText(hItem)+"\\"+path;
		hItem = hParent;
		hParent = GetParentItem(hParent);
	}*/
	path = "\\"+path;
	path = szPath+ path;

	if(!CopyFile(path+strName+".url", ((CMainFrame*)pMainFrame)->m_strStartFavFolder+strName+".url",TRUE))
	{
		if(CopyFile(path+strName+".lnk", ((CMainFrame*)pMainFrame)->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
	((CMainFrame*)pMainFrame)->m_bFavDirty = TRUE;

}

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

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

	GetItemPath(m_hItem, path);
	/*
	HTREEITEM hItem = GetParentItem(m_hItem);
	HTREEITEM hParent = GetParentItem(hItem);
	HTREEITEM hRoot = GetRootItem();
	while(hParent !=NULL)
	{
		path = GetItemText(hItem)+"\\"+path;
		hItem = hParent;
		hParent = GetParentItem(hParent);
	}*/
	path = "\\"+path;
	path = szPath+ path;

	OFSTRUCT ofs;
	if(OpenFile(path+strName+".url", &ofs, OF_EXIST)!=HFILE_ERROR)
	{
		int nImage, nSImage;
		GetItemImage(m_hItem, nImage, nSImage);
		if(nImage == 1)
		{
			SetItemImage(m_hItem, 4,4);
			WritePrivateProfileString("MyIE", "MostFav", "1", path+strName+".url");
		}
		else if (nImage == 4)
		{
			SetItemImage(m_hItem, 1,1);
			WritePrivateProfileString("MyIE", "MostFav", "0", path+strName+".url");
		}
		//update menu
		((CMainFrame*)pMainFrame)->m_bFavDirty = TRUE;
	}
	else 
	{
		AfxMessageBox(IDS_NOT_URL);
	}
	
	
}

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

	while(hit!=NULL && (nWins<nMaxWins || nMaxWins==0))
	{
		if(GetItemData(hit)!=-1)
		{
			GetItemImage(hit, nImage, nSImage);
			if(nImage == 4)
			{
				if(nWins!=0)
				{
					CString url;
					int d = GetItemData(hit);
					GetURL(m_astrFav[d], url);
					((CMainFrame*)pMainFrame)->m_astrOpenAll.AddTail(url);
					nWins ++;
				}
				else
				{
					CMyIEView* pvw = CreateNewView();
					int d = GetItemData(hit);
					if(pvw!=NULL)
					{
						CString url;
						GetURL(m_astrFav[d], url);
						pvw->ToNavigate(url,0,NULL);
						nWins ++;
					}
				}
			}
		}
		hit = GetNextSiblingItem(hit);
	}
	if(nWins >1 && ((CMainFrame*)pMainFrame)->m_nInterval)
		((CMainFrame*)pMainFrame)->m_nTimerID = ::SetTimer(((CMainFrame*)pMainFrame)->GetSafeHwnd(), 1000, ((CMainFrame*)pMainFrame)->m_nInterval*1000,NULL);
	else if(nWins>1)
		((CMainFrame*)pMainFrame)->m_nTimerID = ::SetTimer(((CMainFrame*)pMainFrame)->GetSafeHwnd(), 1000, 20,NULL);
	//enable toolbar button
	((CMainFrame*)pMainFrame)->m_wndToolBar.GetToolBarCtrl().EnableButton(ID_FILE_STOP_OPENALL);
}

BOOL CFavTreeCtrl::GetURL(CString &path, CString &url)
{
	url = path;
	if(url.Right(4) == _T(".lnk"))
	{
		TCHAR buf[2048];
		if(((CMainFrame*)pMainFrame)->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;
		GetURL(m_astrFav[d], 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;
		GetURL(m_astrFav[d], url);
		CInputAlias dlg;
		dlg.m_url = url;
		dlg.m_alias = GetItemText(m_hItem);
		if(dlg.DoModal()==IDOK)
		{