// HisTreeCtrl.cpp
//
// (c)1998-2025 Codejock Software, All Rights Reserved.
//
// THIS SOURCE FILE IS THE PROPERTY OF CODEJOCK SOFTWARE AND IS NOT TO BE
// RE-DISTRIBUTED BY ANY MEANS WHATSOEVER WITHOUT THE EXPRESSED WRITTEN
// CONSENT OF CODEJOCK SOFTWARE.
//
// THIS SOURCE CODE CAN ONLY BE USED UNDER THE TERMS AND CONDITIONS OUTLINED
// IN THE XTREME TOOLKIT PRO LICENSE AGREEMENT. CODEJOCK SOFTWARE GRANTS TO
// YOU (ONE SOFTWARE DEVELOPER) THE LIMITED RIGHT TO USE THIS SOFTWARE ON A
// SINGLE COMPUTER.
//
// CONTACT INFORMATION:
// support@codejock.com
// http://www.codejock.com
//
/////////////////////////////////////////////////////////////////////////////

#include "stdafx.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;
	try
	{
		GetCursorPos(&pt);
		ScreenToClient(&pt);
		hItem = HitTest(pt);

		if ((hItem != NULL) && !ItemHasChildren(hItem))
		{
			int nImg, nOImg;
			GetItemImage(hItem, nImg, nOImg);
			if (nImg == 2)
			{
				OpenWindow(m_astrHist[GetItemData(hItem) - 1]);
			}
			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);
	}
	catch (...)
	{
	}

	*pResult = 0;
}

LPITEMIDLIST CHisTreeCtrl::FindFolder(LPCTSTR foldername, IShellFolder* pFolder)
{
	IEnumIDList* pItems   = NULL;
	LPITEMIDLIST pidlNext = NULL;
	CString lpszName;

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

			if (_tcscmp(foldername, lpszName) == 0)
			{
				if (pItems)
					pItems->Release();
				return pidlNext;
			}
			else if (pidlNext)
			{
				((CHistoryTree*)m_pHisTree)->m_pMalloc->Free(pidlNext);
			}
		}
	}

	if (pItems)
		pItems->Release();

	return NULL;
}

void CHisTreeCtrl::OnMouseMove(UINT nFlags, CPoint point)
{
	CTreeCtrl::OnMouseMove(nFlags, point);
}