开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 界面编程 > 对话框与窗口 > 查看源码
HistoryTree.cpp
资源名称：Xtreme.ToolkitPro.v12.0.0.rar　[点击查看]
上传用户：szled88
上传日期：2015-04-09
资源大小：43957k
文件大小：13k
源码类别：

对话框与窗口

开发平台：

Visual C++

HistoryTree.cpp：源码内容
  1. // HistoryTree.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "Resource.h"
  6. #include "HistoryTree.h"
  7. #include <Wininet.h>
  9. #ifdef _DEBUG
  10. #define new DEBUG_NEW
  11. #undef THIS_FILE
  12. static char THIS_FILE[] = __FILE__;
  13. #endif
  15. /////////////////////////////////////////////////////////////////////////////
  16. // CHistoryTree
  18. CHistoryTree::CHistoryTree()
  19. {
  20. m_pMalloc = NULL;
  21. SHGetMalloc(&m_pMalloc);
  22. m_bIE4 = FALSE;
  23. }
  25. CHistoryTree::~CHistoryTree()
  26. {
  27. // m_wndHisTree.DeleteAllItems();
  28. m_wndHisTree.m_astrHist.RemoveAll();
  29.     m_pMalloc->Release();
  30. }
  31. void CHistoryTree::Destroy()
  32. {
  33. }
  37. /////////////////////////////////////////////////////////////////////////////
  38. // CHistoryTree message handlers
  40. LPCTSTR CHistoryTree::GetTitle()
  41. {
  42. strTitle.LoadString(IDS_TITLE_HIS);
  43. return strTitle;
  44. }
  46. BOOL CHistoryTree::Initialize()
  47. {
  48. // TODO: Add your command handler code here
  50. m_wndHisTree.DeleteAllItems();
  51. m_wndHisTree.m_astrHist.RemoveAll();
  53. LPITEMIDLIST pidlHistory = NULL;
  55.    HRESULT hr = SHGetSpecialFolderLocation(NULL, CSIDL_HISTORY, &pidlHistory);
  56.    if (NOERROR != hr)
  57.    {
  58.    // AfxMessageBox("Can't get history folder.");
  59.    m_bIE4=TRUE;
  60.    //for ie4 or win95
  61.    CWaitCursor wc;
  62.    
  63.    IUrlHistoryStg2* pUrlHistoryStg2 = NULL;
  64.    HRESULT hr = CoCreateInstance(CLSID_CUrlHistory,
  65.    NULL, CLSCTX_INPROC, IID_IUrlHistoryStg2,
  66.    (void**)&pUrlHistoryStg2);
  67.    if (SUCCEEDED(hr))
  68.    {     
  69.    IEnumSTATURL* pEnum;
  70.    hr = pUrlHistoryStg2->EnumUrls(&pEnum);
  71.    ULONG size;
  72.    STATURL staturl;
  73.    int ind=1;
  74.    hr = pEnum->Next(1, &staturl, &size);
  75.    if(SUCCEEDED(hr))
  76.    {
  77.    m_wndHisTree.DeleteAllItems();
  78.    m_wndHisTree.m_astrHist.RemoveAll();
  79.    }
  80.    CTime ttime = CTime::GetCurrentTime();
  81.    today = ttime.GetDayOfWeek();
  82.    int t;
  83.    if(today==1)
  84.    t=8;
  85.    else
  86.    t=today;
  87.    CTimeSpan tt(t-2,0,0,0);
  88.    ttime =ttime - tt;
  89.    
  90.    CTime t2(ttime.GetYear(), ttime.GetMonth(), ttime.GetDay(),0,0,0);
  91.    montime = t2;
  92.    
  93.    while(size>0)
  94.    {
  95.    if(insertUrl(&staturl,ind))
  96.    {
  97.    m_wndHisTree.m_astrHist.Add(CString(staturl.pwcsUrl));
  98.    ind++;
  99.    }
  100.    pEnum->Next(1, &staturl, &size);
  101.    }
  102.    //open today
  103.    
  104.    
  105.    pEnum->Release();
  106.    pUrlHistoryStg2->Release();
  107.    }
  108.    
  109.    return FALSE;
  110.    }
  111.    
  112.    IShellFolder* psfDesktop;
  113.    IShellFolder* psfHistory;
  114.    
  115.    hr = SHGetDesktopFolder(&psfDesktop);
  116.    if (NOERROR != hr)
  117.    {
  118.    AfxMessageBox(_T("Can't get desktop."));
  119.    return FALSE;
  120.    }
  121.    
  122.    hr = psfDesktop->BindToObject(pidlHistory, NULL, IID_IShellFolder,
  123.    (LPVOID*)&psfHistory);
  124.    psfDesktop->Release();
  125.    
  126.    if (NOERROR != hr)
  127.    {
  128.    AfxMessageBox(_T("Can't bind history folder."));
  129.    return FALSE;
  130.    }
  131.    
  132.    AddHistory(psfHistory, NULL);
  133.    
  134.    m_pMalloc->Free(pidlHistory);
  135.    
  136.    psfHistory->Release();
  137.    
  138.    return TRUE;
  139. }
  141. BOOL CHistoryTree::Create(CWnd* pParent)
  142. {
  143. if (!m_wndHisTree.Create(WS_BORDER|WS_CHILD|WS_VISIBLE|TVS_SHOWSELALWAYS|TVS_LINESATROOT|
  144. TVS_FULLROWSELECT |TVS_TRACKSELECT ,
  145. CRect(0,0,0,0), pParent, 100))
  146. {
  147. TRACE0("Failed to create instant bar childn");
  148. return 0; // fail to create
  149. }
  151. CWnd* pTT = m_wndHisTree.FromHandle((HWND) m_wndHisTree.SendMessage(TVM_GETTOOLTIPS));
  152.     if (pTT != NULL)
  153. pTT->SetWindowPos(&CWnd::wndTopMost, 0, 0, 0, 0,
  154. SWP_NOMOVE|SWP_NOSIZE|SWP_NOACTIVATE);
  156. CImageList img;
  158. img.Create(16, 16, ILC_COLORDDB|ILC_MASK, 4, 1);
  159. HBITMAP hbmp = ::LoadBitmap(AfxGetInstanceHandle(), MAKEINTRESOURCE(IDB_BITMAP_TREE));
  160. ImageList_AddMasked(img.GetSafeHandle(), hbmp, RGB(255,0,255));
  162. m_wndHisTree.SetImageList(&img, TVSIL_NORMAL);
  163. img.Detach();
  165. m_wndHisTree.m_pHisTree = this;
  167. return TRUE;
  168. }
  170. CWnd* CHistoryTree::GetWindow()
  171. {
  172. return (CWnd*)&m_wndHisTree;
  173. }
  176. //for ie4 and win95
  177. BOOL CHistoryTree::insertUrl(LPSTATURL pSUrl, int ind)
  178. {
  179. //get host
  180. CString url(pSUrl->pwcsUrl);
  181. CString host;
  182. CString l = url.Left(5);
  183. int i;
  184. if(l == _T("https"))
  185. {
  186. host = url.Right(url.GetLength()-8);
  187. i=host.Find('/');
  188. if(i>0)
  189. host=host.Left(i);
  190. }
  191. if(l == _T("http:"))
  192. {
  193. host = url.Right(url.GetLength()-7);
  194. i=host.Find('/');
  195. if(i>0)
  196. host=host.Left(i);
  197. }
  198. else if(l == _T("ftp:/"))
  199. {
  200. host = url.Right(url.GetLength()-6);
  201. i=host.Find('/');
  202. if(i>0)
  203. host=host.Left(i);
  204. }
  205. else if(l == _T("file:"))
  206. {
  207. host.LoadString(IDS_MY_COMPUTER);
  208. }
  209. else
  210. return FALSE;
  212. //get date
  213. CTime ttime(pSUrl->ftLastVisited);
  214. CTime ttime2(pSUrl->ftLastUpdated);
  215. if( ttime < ttime2)
  216. ttime=ttime2;
  218. CTime filetime(ttime.GetYear(), ttime.GetMonth(), ttime.GetDay(),0,0,0);
  220. int d;
  221. int week = 0, data;
  222. CString root;
  223. CTimeSpan ts = montime-filetime;
  224. if(filetime < montime)
  225. {
  226. week = int(ts.GetDays() + 6)/7;
  227. }
  229. if(week>1)
  230. {
  231. CString tw;
  232. tw.Format(_T("%i"), week);
  233. root.LoadString(IDS_WEEKS_AGO);
  234. root = tw + root;
  235. data = -week-7;
  236. }
  237. else if(week==1)
  238. {
  239. root.LoadString(IDS_LAST_WEEK);
  240. data = -8;
  241. }
  242. else
  243. {
  244. d=filetime.GetDayOfWeek();
  245. if(d==today)
  246. root.LoadString(IDS_TODAY);
  247. else
  248. root.LoadString(IDS_SUNDAY+d-1);
  249. data = -9+d;
  250. if(d==1)
  251. data=-1;
  252. }
  254. HTREEITEM hitem;
  255. hitem = FindAndInsert(NULL, root, 0,0, data);
  257. hitem = FindAndInsert(hitem, host, 1,3);
  259. CString temp;
  260. temp = pSUrl->pwcsTitle;
  261. if(temp.IsEmpty())
  262. {
  263. temp=pSUrl->pwcsUrl;
  264. i=temp.ReverseFind('/');
  265. temp=temp.Right(temp.GetLength()-i-1);
  266. }
  267. HTREEITEM hnode = m_wndHisTree.InsertItem(temp, 2, 2, hitem);
  268. m_wndHisTree.SetItemData(hnode, ind);
  270. return TRUE;
  271. }
  273. HTREEITEM CHistoryTree::FindAndInsert(HTREEITEM  hRoot, LPCTSTR text, int nImg, int nOImg,int data)
  274. {
  275. HTREEITEM hit, hnode = NULL, pre=TVI_FIRST;
  276. BOOL found=FALSE;
  277. hit = m_wndHisTree.GetChildItem(hRoot);
  278. if(data>=0)
  279. while(hit!=NULL && !found)
  280. {
  281. if(m_wndHisTree.GetItemText(hit).CompareNoCase(text)==1)
  282. {
  283. hnode = m_wndHisTree.InsertItem(text, nImg, nOImg, hRoot,TVI_SORT);
  284. found = TRUE;
  285. }
  286. else if(m_wndHisTree.GetItemText(hit).CompareNoCase(text)==0)
  287. {
  288. found = TRUE;
  289. hnode = hit;
  290. }
  291. else
  292. hit = m_wndHisTree.GetNextSiblingItem(hit);
  293. }
  294. else //compare with data
  295. while(hit!=NULL && !found)
  296. {
  297. if((signed)m_wndHisTree.GetItemData(hit)>data)
  298. {
  299. hnode = m_wndHisTree.InsertItem(text, nImg, nOImg, hRoot,pre);
  300. found = TRUE;
  301. }
  302. else if((signed)m_wndHisTree.GetItemData(hit)==data)
  303. {
  304. found = TRUE;
  305. hnode = hit;
  306. }
  307. else
  308. {
  309. pre=hit;
  310. hit = m_wndHisTree.GetNextSiblingItem(hit);
  311. }
  312. }
  314. if(!found)
  315. hnode = m_wndHisTree.InsertItem(text, nImg, nOImg, hRoot,TVI_LAST);
  317. m_wndHisTree.SetItemData(hnode, data);
  319. return hnode;
  320. }
  321. //end
  323. BOOL CHistoryTree::Update(int nLevel)
  324. {
  325. if(m_bIE4)
  326. {
  327. HTREEITEM hit, pre=NULL;
  328. BOOL found=FALSE;
  329. int d;
  330. CString text;
  331. text.LoadString(IDS_TODAY);
  332. //now only update today.
  333. if (nLevel == 0)
  334. {
  335. //get today item
  336. hit = m_wndHisTree.GetChildItem(NULL);
  337. while(hit!=NULL && !found)
  338. {
  339. if(m_wndHisTree.GetItemText(hit).CompareNoCase(text)==0)
  340. {
  341. found = TRUE;
  342. }
  343. else
  344. {
  345. pre = hit;
  346. hit = m_wndHisTree.GetNextSiblingItem(hit);
  347. }
  348. }
  349. int day;
  350. if(found)
  351. {
  352. d = (int)m_wndHisTree.GetItemData(hit);
  353. day=d+9;
  354. if(d==-1)
  355. day=1;
  357. //delete today
  358. m_wndHisTree.DeleteItem(hit);
  359. m_wndHisTree.Expand(pre, TVE_COLLAPSE);
  360. }
  361. else
  362. day=today;
  364. //insert history after that day
  365. CWaitCursor wc;
  367. IUrlHistoryStg2* pUrlHistoryStg2 = NULL;
  368. HRESULT hr = CoCreateInstance(CLSID_CUrlHistory,
  369. NULL, CLSCTX_INPROC, IID_IUrlHistoryStg2,
  370. (void**)&pUrlHistoryStg2);
  371. if (SUCCEEDED(hr))
  372. {     
  373. IEnumSTATURL* pEnum;
  374. hr = pUrlHistoryStg2->EnumUrls(&pEnum);
  375. ULONG size;
  376. STATURL staturl;
  377. int ind = (int)m_wndHisTree.m_astrHist.GetUpperBound() + 2;
  378. hr = pEnum->Next(1,&staturl, &size);
  379. if(!SUCCEEDED(hr))
  380. {
  381. return FALSE;
  382. }
  383. CTime ttime = CTime::GetCurrentTime();
  384. today = ttime.GetDayOfWeek();
  385. CTime lastday;
  387. if(today == day)
  388. {
  389. CTime t1(ttime.GetYear(), ttime.GetMonth(), ttime.GetDay(),0,0,0);
  390. lastday = t1;
  391. }
  392. else
  393. {
  394. CTimeSpan tsp(1,0,0,0);
  395. ttime = ttime-tsp;
  396. CTime t1(ttime.GetYear(), ttime.GetMonth(), ttime.GetDay(),0,0,0);
  397. lastday = t1;
  398. ttime = ttime +tsp;
  399. }
  400. int t;
  401. if (today == 1)
  402. t=8;
  403. else
  404. t = today;
  405. CTimeSpan tt(t-2,0,0,0);
  406. ttime =ttime - tt;
  408. CTime t2(ttime.GetYear(), ttime.GetMonth(), ttime.GetDay(),0,0,0);
  409. montime = t2;
  411. while(size>0)
  412. {
  413. CTime t3(staturl.ftLastVisited);
  414. if(t3>=lastday)
  415. {
  416. if(insertUrl(&staturl,ind))
  417. {
  418. m_wndHisTree.m_astrHist.Add(CString(staturl.pwcsUrl));
  419. ind++;
  420. }
  421. }
  422. pEnum->Next(1,&staturl, &size);
  423. }
  424. //open today
  427. pEnum->Release();
  428. pUrlHistoryStg2->Release();
  429. }
  430. return TRUE;
  431. }
  432. return TRUE;
  433. }
  435. //following is for ie5 and win98
  437. //remove the children of Today
  438. HTREEITEM hit;
  439. BOOL found=FALSE;
  440. CString text;
  441. text.LoadString(IDS_TODAY);
  442. //get today item
  443. hit = m_wndHisTree.GetChildItem(NULL);
  444. while(hit!=NULL && !found)
  445. {
  446. if(m_wndHisTree.GetItemText(hit).CompareNoCase(text)==0)
  447. found = TRUE;
  448. else
  449. hit = m_wndHisTree.GetNextSiblingItem(hit);
  450. }
  451. if(found)
  452. {
  453. //delete today
  454. m_wndHisTree.DeleteItem(hit);
  455. //m_wndHisTree.Expand(pre, TVE_COLLAPSE);
  456. }
  457. LPITEMIDLIST pidlHistory = NULL;
  459.    HRESULT hr = SHGetSpecialFolderLocation(NULL, CSIDL_HISTORY, &pidlHistory);
  460.    if (NOERROR != hr)
  461.    return FALSE;
  462.    
  463.    IShellFolder* psfDesktop;
  464.    IShellFolder* psfHistory;
  465.    
  466.    hr = SHGetDesktopFolder(&psfDesktop);
  467.    if (NOERROR != hr)
  468.    return FALSE;
  469.    
  470.    hr = psfDesktop->BindToObject(pidlHistory, NULL, IID_IShellFolder,
  471.    (LPVOID*)&psfHistory);
  472.    psfDesktop->Release();
  473.    
  474.    if (NOERROR != hr)
  475.    return FALSE;
  476.    
  477.    AddHistory(psfHistory, NULL,0,0,TRUE);
  478.    
  479.    m_pMalloc->Free(pidlHistory);
  480.    
  481.    psfHistory->Release();
  482.    
  483.    return TRUE;
  484. }
  486. void CHistoryTree::AddHistory(IShellFolder *pFolder, HTREEITEM root, int nImg, int nOImg, BOOL bNeedCmp)
  487. {
  488. USES_CONVERSION;
  490. if(pFolder == NULL)
  491. return;
  493. //enum history
  494. IEnumIDList* pItems = NULL;
  495. LPITEMIDLIST pidlNext = NULL;
  496. CString lpszName;
  497. //TCHAR lpszFileName[MAX_PATH];
  498. LPTSTR lpszURL = NULL;
  500. DWORD dwEntrySize=0;
  502. // Enumerate all object in the given folder
  503. HRESULT hr = pFolder->EnumObjects(NULL, SHCONTF_FOLDERS|SHCONTF_NONFOLDERS|SHCONTF_INCLUDEHIDDEN , &pItems);
  504. while (NOERROR == hr)
  505. {
  506. hr = pItems->Next(1, &pidlNext, NULL);
  507. if (NOERROR == hr)
  508. {
  509. GetName(pFolder, pidlNext, SHGDN_NORMAL, lpszName);
  511. //add folder
  512. BOOL found = FALSE;
  513. if(bNeedCmp)
  514. {
  515. HTREEITEM hit;
  516. //get today item
  517. hit = m_wndHisTree.GetChildItem(root);
  518. while(hit!=NULL && !found)
  519. {
  520. if(m_wndHisTree.GetItemText(hit).CompareNoCase(lpszName)==0)
  521. found = TRUE;
  522. else
  523. hit = m_wndHisTree.GetNextSiblingItem(hit);
  524. }
  525. }
  526. if(!found)
  527. {
  528. HTREEITEM hItem;
  529. if(nImg==0)
  530. hItem = m_wndHisTree.InsertItem(lpszName, nImg, nOImg, root ,TVI_LAST);
  531. else
  532. hItem = m_wndHisTree.InsertItem(lpszName, nImg, nOImg, root ,TVI_SORT);
  534. if(nImg==2)
  535. ResolveHistory(pFolder, pidlNext, &lpszURL);
  536. if (lpszURL)
  537. {
  538. // Add the URL to the array and free lpszURL
  539. // since it was created with IMalloc::Alloc
  540. m_wndHisTree.m_astrHist.Add(lpszURL);
  541. m_wndHisTree.SetItemData(hItem, m_wndHisTree.m_astrHist.GetSize());
  543. //is the url cached?
  544. dwEntrySize = 0;
  545. if(_tcsncmp(lpszURL, _T("file:"), 5) != 0)
  546. {
  547. #if _MSC_VER < 1300
  548. if (!GetUrlCacheEntryInfoExA(T2A(lpszURL), NULL, &dwEntrySize, NULL, NULL, NULL, 0))
  549. #else
  550. if (!GetUrlCacheEntryInfoEx(lpszURL, NULL, &dwEntrySize, NULL, NULL, NULL, 0))
  551. #endif
  552. {
  553. if (GetLastError() == ERROR_FILE_NOT_FOUND)
  554. {
  555. m_wndHisTree.SetItemState(hItem, TVIS_CUT, TVIS_CUT);
  556. }
  557. }
  558. }
  560. m_pMalloc->Free(lpszURL);
  561. lpszURL = NULL;
  562. }
  563. }
  565. if (pidlNext)
  566. m_pMalloc->Free(pidlNext);
  567. }
  568. }
  569. if (pItems)
  570. pItems->Release();
  571. }
  573. BOOL CHistoryTree::ResolveHistory(IShellFolder* pFolder, LPCITEMIDLIST pidl, LPTSTR* lpszURL)
  574. {
  575. IShellLink* pShellLink;
  577. *lpszURL = NULL;  // Assume failure
  579. // Get a pointer to the IShellLink interface from the given folder
  580. HRESULT hr = pFolder->GetUIObjectOf(NULL, 1, &pidl, IID_IShellLink, NULL, (LPVOID*)&pShellLink);
  581. if (SUCCEEDED(hr))
  582. {
  583. WIN32_FIND_DATA wfd;      
  585. hr = pShellLink->Resolve(AfxGetApp()->m_pMainWnd->m_hWnd, SLR_NO_UI); 
  586. if (NOERROR == hr)
  587. {
  588. // Get the path to the link target. 
  589. *lpszURL = (LPTSTR)m_pMalloc->Alloc(MAX_PATH);  // Must remember to Free later
  591. hr = pShellLink->GetPath(*lpszURL, MAX_PATH - 1, (WIN32_FIND_DATA*)&wfd, SLGP_UNCPRIORITY);
  592. }
  593. pShellLink->Release();
  594. }
  596. if(SUCCEEDED(hr))
  597. return TRUE;
  598. else
  599. return FALSE;
  600. }