Click here to Skip to main content
13,250,182 members (36,815 online)
Click here to Skip to main content

Stats

272.9K views
8.3K downloads
90 bookmarked
Posted 26 Jun 2005

Shell Tray Info - Arrange your system tray icons

, 26 Jun 2005
A tool with full source code that enumerates tray icons and allows you to reposition them as well as send mouse messages.
/*
Copyright(C) Nishant Sivakumar. All rights reserved.
URLs - http://blog.voidnish.com and http://www.voidnish.com
Contact : nish@voidnish.com
*/

// ShellTrayInfoView.cpp : implementation of the CShellTrayInfoView class
//

#include "stdafx.h"
#include "ShellTrayInfo.h"

#include "ProcessData.h"

#include "ShellTrayInfoView.h"
#include ".\shelltrayinfoview.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// CShellTrayInfoView

IMPLEMENT_DYNCREATE(CShellTrayInfoView, CListView)

BEGIN_MESSAGE_MAP(CShellTrayInfoView, CListView)
	ON_COMMAND(ID_EDIT_COPY, OnEditCopy)
	ON_COMMAND(ID_DBLCLK_ICON, OnDoubleClick)
	ON_COMMAND(ID_LCLK_ICON, OnLeftClick)
	ON_COMMAND(ID_RCLK_ICON, OnRightClick)
	ON_COMMAND(ID_ICON_UP, OnMoveLeft)
	ON_COMMAND(ID_ICON_DOWN, OnMoveRight)
	ON_COMMAND(ID_REFRESH_LIST, OnRefresh)
	ON_NOTIFY_REFLECT(NM_CUSTOMDRAW, OnNMCustomdraw)
	ON_NOTIFY_REFLECT(NM_DBLCLK, OnNMDblclk)
END_MESSAGE_MAP()

// CShellTrayInfoView construction/destruction

CShellTrayInfoView::CShellTrayInfoView() : 
	m_hHiddenFont(NULL),
	m_hRegularFont(NULL)
{
	// TODO: add construction code here

}

CShellTrayInfoView::~CShellTrayInfoView()
{
	DeleteObject(m_hHiddenFont);	
	DeleteObject(m_hRegularFont);	
}

BOOL CShellTrayInfoView::PreCreateWindow(CREATESTRUCT& cs)
{
	// TODO: Modify the Window class or styles here by modifying
	//  the CREATESTRUCT cs

	return CListView::PreCreateWindow(cs);
}

void CShellTrayInfoView::OnInitialUpdate()
{
	CListView::OnInitialUpdate();

	LOGFONT lf = {0};
	GetFont()->GetLogFont(&lf);
	m_hRegularFont = CreateFontIndirect(&lf);
	lf.lfItalic = TRUE;
	m_hHiddenFont = CreateFontIndirect(&lf);

	m_hTrayWnd = FindTrayToolbarWindow();

	m_Image16List.Create(16,16,ILC_COLOR24,16,16);
	GetListCtrl().SetImageList(&m_Image16List, LVSIL_SMALL);
	
	m_Image16List.Add(LoadIcon(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDR_MAINFRAME)));

	GetListCtrl().InsertColumn(0, _T("Icon"), LVCFMT_LEFT, 50);
	GetListCtrl().InsertColumn(1, _T("Associated ToolTip"), LVCFMT_LEFT,400);
	GetListCtrl().InsertColumn(2, _T("Process Path"), LVCFMT_LEFT, 500);	

	GetListCtrl().ModifyStyle(0,LVS_REPORT|LVS_SINGLESEL);
	GetListCtrl().SetExtendedStyle(LVS_EX_FULLROWSELECT);	

	ListTrayIcons();
}


// CShellTrayInfoView diagnostics

#ifdef _DEBUG
void CShellTrayInfoView::AssertValid() const
{
	CListView::AssertValid();
}

void CShellTrayInfoView::Dump(CDumpContext& dc) const
{
	CListView::Dump(dc);
}
#endif //_DEBUG


// CShellTrayInfoView message handlers

void CShellTrayInfoView::ListTrayIcons(int defindex /*= 0*/)
{
	m_TifoVec.erase(m_TifoVec.begin(), m_TifoVec.end());

	GetListCtrl().DeleteAllItems();
	for(int i=1; i<m_Image16List.GetImageCount(); i++) 
	{		
		m_Image16List.Remove(i);
	}

	if(m_hTrayWnd == NULL)	
		return;

	DWORD dwTrayPid;
	GetWindowThreadProcessId(m_hTrayWnd, &dwTrayPid);

	int count = (int)::SendMessage(m_hTrayWnd, TB_BUTTONCOUNT, 0, 0);

	CProcessData<TBBUTTON> data(dwTrayPid);
	TBBUTTON tb = {0};
	TRAYDATA tray = {0};
	TrayItemInfo tifo = {0};

	for(int i=0; i<count; i++)
	{		
		::SendMessage(m_hTrayWnd, TB_GETBUTTON, i, (LPARAM)data.GetData());		
		data.ReadData(&tb);			
		data.ReadData<TRAYDATA>(&tray,(LPCVOID)tb.dwData);

		DWORD dwProcessId = 0;
		GetWindowThreadProcessId(tray.hwnd,&dwProcessId);

		tifo.sProcessPath = GetFilenameFromPid(dwProcessId);		

		wchar_t TipChar;
		wchar_t sTip[1024] = {0};
		wchar_t* pTip = (wchar_t*)tb.iString;		

		if(!(tb.fsState&TBSTATE_HIDDEN))
		{			
			int x = 0;
			do 
			{	
				if(x == 1023)
				{
					wcscpy(sTip,L"[ToolTip was either too long or not set]");	
					break;
				}
				data.ReadData<wchar_t>(&TipChar, (LPCVOID)pTip++);																			
			}while(sTip[x++] = TipChar);
		}
		else
			wcscpy(sTip,L"[Hidden Icon]");				

		USES_CONVERSION;
		tifo.sTip = W2T(sTip);

		tifo.hwnd = tray.hwnd;
		tifo.uCallbackMessage = tray.uCallbackMessage;
		tifo.uID = tray.uID;

		tifo.bVisible = !(tb.fsState & TBSTATE_HIDDEN);

		int iconindex = 0;
		ICONINFO  iinfo;
		if(GetIconInfo(tray.hIcon,&iinfo) != 0)
		{			
			iconindex = m_Image16List.Add(tray.hIcon);
		}

		LVITEM lv = {0};
		lv.iItem = GetListCtrl().GetItemCount();
		lv.iSubItem = 0;
		lv.mask = LVIF_IMAGE;
		lv.iImage = iconindex;

		m_TifoVec.push_back(tifo);
		int index = GetListCtrl().InsertItem(&lv);
		GetListCtrl().SetItemText(index,1,tifo.sTip);
		GetListCtrl().SetItemText(index,2,tifo.sProcessPath);
	}	

	if( (count>0) && (defindex>=0) && (defindex<count) )
		GetListCtrl().SetItemState(defindex, 
			LVIS_FOCUSED|LVIS_SELECTED,
			LVIS_FOCUSED|LVIS_SELECTED);
}

void CShellTrayInfoView::OnEditCopy()
{
	POSITION pos = GetListCtrl().GetFirstSelectedItemPosition();
	if(pos)
	{
		int index = GetListCtrl().GetNextSelectedItem(pos);	
		OpenClipboard();
		EmptyClipboard();
		HGLOBAL hText = GlobalAlloc(GMEM_MOVEABLE, sizeof TCHAR * 512);
		LPTSTR pStr = (LPTSTR)GlobalLock(hText);		
		_stprintf(pStr, _T("Tray Tip : %s\r\nOwner : %s"), 
			m_TifoVec[index].sTip, m_TifoVec[index].sProcessPath);
		GlobalUnlock(hText);	
		SetClipboardData(CF_TEXT, hText);	
		CloseClipboard();	
	}
}

void CShellTrayInfoView::OnDoubleClick()
{
	PostMessageToTrayIcon(WM_LBUTTONDBLCLK);
}

void CShellTrayInfoView::OnLeftClick()
{
	PostMessageToTrayIcon(WM_LBUTTONDOWN);
	PostMessageToTrayIcon(WM_LBUTTONUP);
}

void CShellTrayInfoView::OnRightClick()
{
	PostMessageToTrayIcon(WM_RBUTTONDOWN);
	PostMessageToTrayIcon(WM_RBUTTONUP);
}

void CShellTrayInfoView::PostMessageToTrayIcon(LPARAM lParam)
{
	POSITION pos = GetListCtrl().GetFirstSelectedItemPosition();
	if(pos)
	{
		int index = GetListCtrl().GetNextSelectedItem(pos);	

		::PostMessage(m_TifoVec[index].hwnd,
			m_TifoVec[index].uCallbackMessage,
			m_TifoVec[index].uID,
			lParam);		
	}
}

void CShellTrayInfoView::OnMoveLeft()
{
	POSITION pos = GetListCtrl().GetFirstSelectedItemPosition();
	if(pos)
	{
		int index = GetListCtrl().GetNextSelectedItem(pos);	
		if(index > 0)
		{
			::SendMessage(m_hTrayWnd, TB_MOVEBUTTON, index, index-1);		
			ListTrayIcons(index-1);
		}
	}
}

void CShellTrayInfoView::OnMoveRight()
{
	POSITION pos = GetListCtrl().GetFirstSelectedItemPosition();
	if(pos)
	{
		int index = GetListCtrl().GetNextSelectedItem(pos);	
		if(index < (GetListCtrl().GetItemCount()-1))
		{
			::SendMessage(m_hTrayWnd, TB_MOVEBUTTON, index, index+1);		
			ListTrayIcons(index+1);
		}
	}
}

void CShellTrayInfoView::OnNMCustomdraw(NMHDR *pNMHDR, LRESULT *pResult)
{
	LPNMCUSTOMDRAW pNMCD = reinterpret_cast<LPNMCUSTOMDRAW>(pNMHDR);
	*pResult = CDRF_DODEFAULT;

	if(pNMCD->dwDrawStage == CDDS_PREPAINT)
	{		
		*pResult = CDRF_NOTIFYITEMDRAW;		
	}
	else if(pNMCD->dwDrawStage == CDDS_ITEMPREPAINT)
	{		
		NMLVCUSTOMDRAW* pLVCD = reinterpret_cast<NMLVCUSTOMDRAW*>( pNMHDR );				
		if(!m_TifoVec[pNMCD->dwItemSpec].bVisible)
		{
			pLVCD->clrText = RGB(120,120,120);				
			SelectObject(pNMCD->hdc,m_hHiddenFont);				
		}
		else
		{			
			pLVCD->clrText = RGB(0,0,130);
			SelectObject(pNMCD->hdc,m_hRegularFont);				
		}				
		*pResult = CDRF_NEWFONT;	
	}	
}

void CShellTrayInfoView::OnNMDblclk(NMHDR *pNMHDR, LRESULT *pResult)
{	
	LPNMITEMACTIVATE pNMItem  = (LPNMITEMACTIVATE)pNMHDR;
	if(pNMItem->iItem != -1)
		PostMessageToTrayIcon(WM_LBUTTONDBLCLK);
	*pResult = 0;
}

void CShellTrayInfoView::OnRefresh()
{
	ListTrayIcons();
}

By viewing downloads associated with this article you agree to the Terms of Service and the article's licence.

If a file you wish to view isn't highlighted, and is a text file (not binary), please let us know and we'll add colourisation support for it.

License

This article, along with any associated source code and files, is licensed under The Code Project Open License (CPOL)

Share

About the Author

Nish Nishant
United States United States
Nish Nishant is the Principal Software Architect/Consultant for Ganymede Software Solutions LLC, and is based out of Columbus, Ohio. He has over 17 years of software industry experience in various roles including Lead Software Architect, Principal Software Engineer, and Product Manager. Nish was a Microsoft Visual C++ MVP between 2002 and 2015.

Nish is an industry acknowledged expert in the Microsoft technology stack. He authored C++/CLI in Action for Manning Publications in 2005, and had previously co-authored Extending MFC Applications with the .NET Framework for Addison Wesley in 2003. In addition, he has over 140 published technology articles on CodeProject.com and another 250+ blog articles on his WordPress blog. Nish is vastly experienced in team management, mentoring teams, and directing all stages of software development.

Contact Nish : If you are interested in hiring Nish as a consultant, you can reach him via his google email id voidnish.

Company Website : www.ganymedesoftwaresolutions.com

You may also be interested in...

Permalink | Advertise | Privacy | Terms of Use | Mobile
Web01 | 2.8.171114.1 | Last Updated 27 Jun 2005
Article Copyright 2005 by Nish Nishant
Everything else Copyright © CodeProject, 1999-2017
Layout: fixed | fluid