Awake Updates - NEWKONA_11272024

src/modules/awake/Awake/Core/Manager.cs

@@ -14,7 +14,6 @@
 using System.Text;
 using System.Text.Json;
 using System.Threading;
-
 using Awake.Core.Models;
 using Awake.Core.Native;
 using Awake.Properties;

src/modules/awake/Awake/Core/Threading/SingleThreadSynchronizationContext.cs

@@ -5,6 +5,7 @@
 using System;
 using System.Collections.Generic;
 using System.Threading;
+using ManagedCommon;
 
 using ManagedCommon;
 

src/modules/awake/Awake/Core/TrayHelper.cs

@@ -193,12 +193,24 @@ internal static void SetShellIcon(IntPtr hWnd, string text, Icon? icon, TrayIcon
                     };
                 }
 
-                if (!Bridge.Shell_NotifyIcon(message, ref _notifyIconData))
+                for (int attempt = 1; attempt <= 3; attempt++)
                 {
-                    int errorCode = Marshal.GetLastWin32Error();
-                    Logger.LogInfo($"Could not set the shell icon. Action: {action} and error code: {errorCode}. HIcon handle is {icon?.Handle} and HWnd is {hWnd}");
+                    if (Bridge.Shell_NotifyIcon(message, ref _notifyIconData))
+                    {
+                        break;
+                    }
+                    else
+                    {
+                        int errorCode = Marshal.GetLastWin32Error();
+                        Logger.LogInfo($"Could not set the shell icon. Action: {action}, error code: {errorCode}. HIcon handle is {icon?.Handle} and HWnd is {hWnd}");
+
+                        if (attempt == 3)
+                        {
+                            throw new Win32Exception(errorCode, $"Failed to change tray icon after 3 attempts. Action: {action} and error code: {errorCode}");
+                        }
 
-                    throw new Win32Exception(errorCode, $"Failed to change tray icon. Action: {action} and error code: {errorCode}");
+                        Thread.Sleep(100);
+                    }
                 }
 
                 if (action == TrayIconAction.Delete)

src/modules/awake/Awake/Program.cs

@@ -15,7 +15,6 @@
 using System.Text.Json;
 using System.Threading;
 using System.Threading.Tasks;
-
 using Awake.Core;
 using Awake.Core.Models;
 using Awake.Core.Native;