Click here to Skip to main content
12,622,568 members (31,002 online)
Click here to Skip to main content

Stats

260.1K views
8K downloads
90 bookmarked
Posted

Shell Tray Info - Arrange your system tray icons

, 26 Jun 2005 CPOL
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 a Software Architect/Consultant based out of Columbus, Ohio. He has over 16 years of software industry experience in various roles including Lead Software Architect, Principal Software Engineer, and Product Manager. Nish is a recipient of the annual Microsoft Visual C++ MVP Award since 2002 (14 consecutive awards as of 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 : You can reach Nish on his google email id voidnish.

Website and Blog

You may also be interested in...

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