histreectrl.cpp

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

// HisTreeCtrl.cpp : implementation file
//

#include "stdafx.h"
#include "MyIE.h"
#include "HisTreeCtrl.h"
#include "HistoryTree.h"
#include "MainFrm.h"

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

/////////////////////////////////////////////////////////////////////////////
// CHisTreeCtrl

CHisTreeCtrl::CHisTreeCtrl()
{
}

CHisTreeCtrl::~CHisTreeCtrl()
{
	m_astrHist.RemoveAll();
}


BEGIN_MESSAGE_MAP(CHisTreeCtrl, CTreeCtrl)
	//{{AFX_MSG_MAP(CHisTreeCtrl)
	ON_NOTIFY_REFLECT(NM_CLICK, OnClick)
	ON_WM_MOUSEMOVE()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CHisTreeCtrl message handlers


void CHisTreeCtrl::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 nImg, nOImg;
		GetItemImage(hItem, nImg, nOImg);
		if(nImg==2)
		{
			CMyIEView* pvw = GetCurrentView(1);
			if(pvw!=NULL)
				pvw->ToNavigate(m_astrHist[GetItemData(hItem)-1],0,NULL);
		}
		else
		{
			//get folder
			LPITEMIDLIST pidlHistory = NULL;

		   HRESULT hr = SHGetSpecialFolderLocation(NULL, CSIDL_HISTORY, &pidlHistory);
		   if (NOERROR != hr)
			  return;

		   IShellFolder* psfDesktop;
		   IShellFolder* psfHistory;

		   hr = SHGetDesktopFolder(&psfDesktop);
		   if (NOERROR != hr)
			  return;

		   hr = psfDesktop->BindToObject(pidlHistory, NULL, IID_IShellFolder,
										 (LPVOID*)&psfHistory);
		   psfDesktop->Release();

		   if (NOERROR != hr)
			  return;

		   HTREEITEM hDayItem;
		   hDayItem = GetParentItem(hItem);
		   if(hDayItem == NULL)
			   hDayItem = hItem;

			CString folder = GetItemText(hDayItem);
			LPITEMIDLIST pidlSubFolder = FindFolder(folder, psfHistory);
			if(pidlSubFolder!=NULL)
			{
				IShellFolder* psfSubFolder = NULL;
				IShellFolder* psfSubFolder2 = NULL;
			    hr = psfHistory->BindToObject(pidlSubFolder, NULL, IID_IShellFolder,
					 (LPVOID*)&psfSubFolder);

				if( hDayItem != hItem)
				{
					if (pidlSubFolder)
						((CHistoryTree*)m_pHisTree)->m_pMalloc->Free(pidlSubFolder);

					folder = GetItemText(hItem);
					pidlSubFolder = FindFolder(folder, psfSubFolder);
					//psfSubFolder->Release();
					hr = psfSubFolder->BindToObject(pidlSubFolder, NULL, IID_IShellFolder,
						 (LPVOID*)&psfSubFolder2);
				}
				if(nImg ==0)
					((CHistoryTree*)m_pHisTree)->AddHistory(psfSubFolder, hItem, 1,3);
				else
					((CHistoryTree*)m_pHisTree)->AddHistory(psfSubFolder2, hItem, 2,2);

				if(psfSubFolder)
					psfSubFolder ->Release();
				if(psfSubFolder2)
					psfSubFolder2 ->Release();
			}

			((CHistoryTree*)m_pHisTree)->m_pMalloc->Free(pidlSubFolder);
		   ((CHistoryTree*)m_pHisTree)->m_pMalloc->Free(pidlHistory);

		   psfHistory->Release();

			Expand(hItem,TVE_TOGGLE);
		}
	}
	else
		Expand(hItem,TVE_TOGGLE);

	*pResult = 0;
	
}

LPITEMIDLIST CHisTreeCtrl::FindFolder(LPCSTR foldername, IShellFolder *pFolder)
{
	IEnumIDList* pItems = NULL;
	LPITEMIDLIST pidlNext = NULL;
	STRRET StrRetName;
	LPTSTR lpszName = NULL;

	HRESULT hr = pFolder->EnumObjects(NULL, SHCONTF_FOLDERS|SHCONTF_NONFOLDERS, &pItems);
   while (NOERROR == hr)
   {
	  hr = pItems->Next(1, &pidlNext, NULL);
	  if (NOERROR == hr)
	  {
		 pFolder->GetDisplayNameOf(pidlNext, SHGDN_NORMAL, &StrRetName);

         ((CHistoryTree*)m_pHisTree)->StrRetToStr(StrRetName, &lpszName, pidlNext);


		 if(strcmp(foldername, lpszName)==0)
		 {
	 		 if (lpszName)
				((CHistoryTree*)m_pHisTree)->m_pMalloc->Free(lpszName);

	 		 if (pItems)
				  pItems->Release();
			 return pidlNext;
		 }
		 else if (pidlNext)
			((CHistoryTree*)m_pHisTree)->m_pMalloc->Free(pidlNext);

		 if (lpszName)
			((CHistoryTree*)m_pHisTree)->m_pMalloc->Free(lpszName);

	  }
   }

   if (pItems)
	  pItems->Release();

   return NULL;
}


void CHisTreeCtrl::OnMouseMove(UINT nFlags, CPoint point) 
{
	// TODO: Add your message handler code here and/or call default
	HTREEITEM hItem;
	
	CPoint pt;
	GetCursorPos(&pt);
	ScreenToClient(&pt);
	hItem = HitTest(pt);

	if ((hItem != NULL) && !ItemHasChildren(hItem))
	{
		int nImg, nOImg;
		GetItemImage(hItem, nImg, nOImg);
		if(nImg==2)
		{
			pMainFrame->SetMessageText(m_astrHist[GetItemData(hItem)-1]);
		}
	}	
	
	CTreeCtrl::OnMouseMove(nFlags, point);
}