private SafeNotifyIconData GetTrayItemIconData(TrayItem trayItem)
{
SafeNotifyIconData nid = new SafeNotifyIconData();
nid.hWnd = trayItem.hWnd;
nid.uID = trayItem.uID;
nid.uCallbackMessage = trayItem.uCallbackMessage;
nid.szTip = trayItem.szIconText;
nid.hIcon = trayItem.hIcon;
nid.uVersion = trayItem.uVersion;
nid.guidItem = trayItem.guidItem;
nid.dwState = 0;
nid.uFlags = NIF.GUID | NIF.MESSAGE | NIF.TIP;
if (nid.hIcon != IntPtr.Zero)
{
nid.uFlags |= NIF.ICON;
}
else
{
CairoLogger.Instance.Warning($"ExplorerTrayService: Unable to use {trayItem.szIconText} icon handle for NOTIFYICONDATA struct");
}
return(nid);
}
private void GetTrayItems()
{
IntPtr toolbarHwnd = FindExplorerTrayToolbarHwnd();
if (toolbarHwnd == IntPtr.Zero)
{
return;
}
int count = GetNumTrayIcons(toolbarHwnd);
if (count < 1)
{
return;
}
GetWindowThreadProcessId(toolbarHwnd, out var processId);
IntPtr hProcess = OpenProcess(ProcessAccessFlags.All, false, (int)processId);
IntPtr hBuffer = VirtualAllocEx(hProcess, IntPtr.Zero, (uint)Marshal.SizeOf(new TBBUTTON()), AllocationType.Commit,
MemoryProtection.ReadWrite);
for (int i = 0; i < count; i++)
{
TrayItem trayItem = GetTrayItem(i, hBuffer, hProcess, toolbarHwnd);
if (trayItem.hWnd == IntPtr.Zero || !IsWindow(trayItem.hWnd))
{
CairoLogger.Instance.Debug($"ExplorerTrayService: Ignored notify icon {trayItem.szIconText} due to invalid handle");
continue;
}
SafeNotifyIconData nid = GetTrayItemIconData(trayItem);
if (trayDelegate != null)
{
if (!trayDelegate((uint)NIM.NIM_ADD, nid))
{
CairoLogger.Instance.Debug("ExplorerTrayService: Ignored notify icon message");
}
}
else
{
CairoLogger.Instance.Debug("ExplorerTrayService: trayDelegate is null");
}
}
VirtualFreeEx(hProcess, hBuffer, 0, AllocationType.Release);
CloseHandle((int)hProcess);
}
private bool SysTrayCallback(uint message, SafeNotifyIconData nicData)
{
if (nicData.hWnd == IntPtr.Zero)
{
return(false);
}
NotifyIcon trayIcon = new NotifyIcon(nicData.hWnd);
trayIcon.UID = nicData.uID;
lock (_lockObject)
{
if ((NIM)message == NIM.NIM_ADD || (NIM)message == NIM.NIM_MODIFY)
{
try
{
bool exists = false;
// hide icons while we are shell which require UWP support & we have a separate implementation for
if (nicData.guidItem == new Guid(VOLUME_GUID) && ((Shell.IsCairoRunningAsShell && Shell.IsWindows10OrBetter) || GroupPolicyManager.Instance.HideScaVolume))
{
return(false);
}
foreach (NotifyIcon ti in TrayIcons)
{
if (ti.Equals(nicData))
{
exists = true;
trayIcon = ti;
break;
}
}
if ((NIF.STATE & nicData.uFlags) != 0)
{
trayIcon.IsHidden = nicData.dwState == 1;
}
if ((NIF.TIP & nicData.uFlags) != 0 && !string.IsNullOrEmpty(nicData.szTip))
{
trayIcon.Title = nicData.szTip;
}
if ((NIF.ICON & nicData.uFlags) != 0)
{
if (nicData.hIcon != IntPtr.Zero)
{
try
{
System.Windows.Media.Imaging.BitmapSource bs =
System.Windows.Interop.Imaging.CreateBitmapSourceFromHIcon(
nicData.hIcon, Int32Rect.Empty,
System.Windows.Media.Imaging.BitmapSizeOptions.FromEmptyOptions());
DestroyIcon(nicData.hIcon);
if (bs != null)
{
bs.Freeze();
trayIcon.Icon = bs;
}
}
catch
{
if (trayIcon.Icon == null)
{
trayIcon.Icon = Common.IconImageConverter.GetDefaultIcon();
}
}
}
else
{
trayIcon.Icon = null;
}
}
trayIcon.HWnd = nicData.hWnd;
trayIcon.UID = nicData.uID;
if ((NIF.GUID & nicData.uFlags) != 0)
{
trayIcon.GUID = nicData.guidItem;
}
// guess version in case we are receiving icons that aren't sending NIM_SETVERSION to new explorers
if ((NIF.VISTA_MASK & nicData.uFlags) != 0)
{
trayIcon.Version = 4;
}
else if ((NIF.XP_MASK & nicData.uFlags) != 0)
{
trayIcon.Version = 3;
}
if (nicData.uVersion > 0 && nicData.uVersion <= 4)
{
trayIcon.Version = nicData.uVersion;
}
if ((NIF.MESSAGE & nicData.uFlags) != 0)
{
trayIcon.CallbackMessage = nicData.uCallbackMessage;
}
if (!exists)
{
// default placement to a menu bar-like rect
trayIcon.Placement = defaultPlacement;
// set properties used for pinning
trayIcon.Path = Shell.GetPathForHandle(trayIcon.HWnd);
trayIcon.SetPinValues();
if (trayIcon.Icon == null)
{
trayIcon.Icon = Common.IconImageConverter.GetDefaultIcon();
}
TrayIcons.Add(trayIcon);
CairoLogger.Instance.Debug($"NotificationArea: Added: {trayIcon.Title} Path: {trayIcon.Path} Hidden: {trayIcon.IsHidden} GUID: {trayIcon.GUID} UID: {trayIcon.UID} Version: {trayIcon.Version}");
if ((NIM)message == NIM.NIM_MODIFY)
{
// return an error to the notifyicon as we received a modify for an icon we did not yet have
return(false);
}
}
else
{
CairoLogger.Instance.Debug($"NotificationArea: Modified: {trayIcon.Title}");
}
}
catch (Exception ex)
{
CairoLogger.Instance.Error("NotificationArea: Unable to modify the icon in the collection.", ex);
}
}
else if ((NIM)message == NIM.NIM_DELETE)
{
try
{
if (!TrayIcons.Contains(trayIcon))
{
// Nothing to remove.
return(false);
}
TrayIcons.Remove(trayIcon);
CairoLogger.Instance.Debug($"NotificationArea: Removed: {nicData.szTip}");
}
catch (Exception ex)
{
CairoLogger.Instance.Error("NotificationArea: Unable to remove the icon from the collection.", ex);
}
}
else if ((NIM)message == NIM.NIM_SETVERSION)
{
foreach (NotifyIcon ti in TrayIcons)
{
if (ti.Equals(nicData))
{
ti.Version = nicData.uVersion;
CairoLogger.Instance.Debug($"NotificationArea: Modified version to {ti.Version} on: {ti.Title}");
break;
}
}
}
}
return(true);
}
public bool Equals(SafeNotifyIconData other)
{
return((HWnd.Equals(other.hWnd) && UID.Equals(other.uID)) || (other.guidItem != Guid.Empty && GUID.Equals(other.guidItem)));
}
