[Awake]PROMETHEAN_09082024 - tray icon fixes (#34717)

* Update with bug fixes for tray icon and support for parent process

* Process information enum

* Update the docs

* Fix spelling

* Make sure that PID is used in PT config flow

* Logic for checks based on #34148

* Update with link to PR

* Small cleanup

* Proper task segmentation in a function

* Cleanup the code

* Fix synchronization context issue

* Update planning doc

* Test disabling caching to see if that manages to pass CI

.pipelines/v2/ci.yml

@@ -32,7 +32,7 @@ parameters:
   - name: enableMsBuildCaching
     type: boolean
     displayName: "Enable MSBuild Caching"
-    default: true
+    default: false
   - name: runTests
     type: boolean
     displayName: "Run Tests"

doc/planning/awake.md

@@ -10,13 +10,22 @@ The build ID can be found in `Core\Constants.cs` in the `BuildId` variable - it
 
 The build ID moniker is made up of two components - a reference to a [Halo](https://en.wikipedia.org/wiki/Halo_(franchise)) character, and the date when the work on the specific build started in the format of `MMDDYYYY`.
 
-| Build ID                                                           | Build Date      |
+| Build ID                                                           | Build Date         |
-|:-------------------------------------------------------------------|:----------------|
+|:-------------------------------------------------------------------|:------------------|
-| [`VISEGRADRELAY_08152024`](#VISEGRADRELAY_08152024-august-15-2024) | August 15, 2024 |
+| [`PROMETHEAN_09082024`](#PROMETHEAN_09082024-september-8-2024)     | September 8, 2024 |
-| [`DAISY023_04102024`](#DAISY023_04102024-april-10-2024)            | April 10, 2024  |
+| [`VISEGRADRELAY_08152024`](#VISEGRADRELAY_08152024-august-15-2024) | August 15, 2024   |
-| [`ATRIOX_04132023`](#ATRIOX_04132023-april-13-2023)                | April 13, 2023  |
+| [`DAISY023_04102024`](#DAISY023_04102024-april-10-2024)            | April 10, 2024    |
-| [`LIBRARIAN_03202022`](#librarian_03202022-march-20-2022)          | March 20, 2022  |
+| [`ATRIOX_04132023`](#ATRIOX_04132023-april-13-2023)                | April 13, 2023    |
-| `ARBITER_01312022`                                                 | January 31, 2022 |
+| [`LIBRARIAN_03202022`](#librarian_03202022-march-20-2022)          | March 20, 2022    |
+| `ARBITER_01312022`                                                 | January 31, 2022  |
+
+### `PROMETHEAN_09082024` (September 8, 2024)
+
+>[!NOTE]
+>See pull request: [Awake - `PROMETHEAN_09082024`](https://github.com/microsoft/PowerToys/pull/34717)
+
+- Updating the initialization logic to make sure that settings are respected for proper group policy and single-instance detection.
+- [#34148] Fixed a bug from the previous release that incorrectly synchronized threads for shell icon creation and initialized parent PID when it was not parented.
 
 ### `VISEGRADRELAY_08152024` (August 15, 2024)
 

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

@@ -17,6 +17,6 @@ namespace Awake.Core
         // Format of the build ID is: CODENAME_MMDDYYYY, where MMDDYYYY
         // is representative of the date when the last change was made before
         // the pull request is issued.
-        internal const string BuildId = "VISEGRADRELAY_08152024";
+        internal const string BuildId = "PROMETHEAN_09082024";
     }
 }

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

@@ -6,17 +6,16 @@ using System;
 using System.Collections.Generic;
 using System.Threading;
 
-namespace Awake.Core.Models
+namespace Awake.Core.Threading
 {
     internal sealed class SingleThreadSynchronizationContext : SynchronizationContext
     {
-        private readonly Queue<Tuple<SendOrPostCallback, object>> queue =
+        private readonly Queue<Tuple<SendOrPostCallback, object?>?> queue = new();
-            new();
 
-#pragma warning disable CS8765 // Nullability of type of parameter doesn't match overridden member (possibly because of nullability attributes).
+        public override void Post(SendOrPostCallback d, object? state)
-        public override void Post(SendOrPostCallback d, object state)
-#pragma warning restore CS8765 // Nullability of type of parameter doesn't match overridden member (possibly because of nullability attributes).
         {
+            ArgumentNullException.ThrowIfNull(d);
+
             lock (queue)
             {
                 queue.Enqueue(Tuple.Create(d, state));
@@ -28,7 +27,7 @@ namespace Awake.Core.Models
         {
             while (true)
             {
-                Tuple<SendOrPostCallback, object> work;
+                Tuple<SendOrPostCallback, object?>? work;
                 lock (queue)
                 {
                     while (queue.Count == 0)
@@ -44,7 +43,14 @@ namespace Awake.Core.Models
                     break;
                 }
 
-                work.Item1(work.Item2);
+                try
+                {
+                    work.Item1(work.Item2);
+                }
+                catch (Exception e)
+                {
+                    Console.WriteLine("Error during execution: " + e.Message);
+                }
             }
         }
 
@@ -52,9 +58,7 @@ namespace Awake.Core.Models
         {
             lock (queue)
             {
-#pragma warning disable CS8625 // Cannot convert null literal to non-nullable reference type.
                 queue.Enqueue(null);  // Signal the end of the message loop
-#pragma warning restore CS8625 // Cannot convert null literal to non-nullable reference type.
                 Monitor.Pulse(queue);
             }
         }

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

@@ -9,10 +9,9 @@ using System.Drawing;
 using System.Linq;
 using System.Runtime.InteropServices;
 using System.Threading;
-using System.Threading.Tasks;
-
 using Awake.Core.Models;
 using Awake.Core.Native;
+using Awake.Core.Threading;
 using Awake.Properties;
 using ManagedCommon;
 using Microsoft.PowerToys.Settings.UI.Library;
@@ -29,13 +28,13 @@ namespace Awake.Core
     internal static class TrayHelper
     {
         private static NotifyIconData _notifyIconData;
-
         private static IntPtr _trayMenu;
+        private static IntPtr _hiddenWindowHandle;
+        private static SingleThreadSynchronizationContext? _syncContext;
+        private static Thread? _mainThread;
 
         private static IntPtr TrayMenu { get => _trayMenu; set => _trayMenu = value; }
 
-        private static IntPtr _hiddenWindowHandle;
-
         internal static IntPtr HiddenWindowHandle { get => _hiddenWindowHandle; private set => _hiddenWindowHandle = value; }
 
         static TrayHelper()
@@ -46,42 +45,35 @@ namespace Awake.Core
 
         private static void ShowContextMenu(IntPtr hWnd)
         {
-            if (TrayMenu != IntPtr.Zero)
+            if (TrayMenu == IntPtr.Zero)
             {
-                Bridge.SetForegroundWindow(hWnd);
-
-                // Get the handle to the context menu associated with the tray icon
-                IntPtr hMenu = TrayMenu;
-
-                // Get the current cursor position
-                Bridge.GetCursorPos(out Models.Point cursorPos);
-
-                Bridge.ScreenToClient(hWnd, ref cursorPos);
-
-                MenuInfo menuInfo = new()
-                {
-                    CbSize = (uint)Marshal.SizeOf(typeof(MenuInfo)),
-                    FMask = Native.Constants.MIM_STYLE,
-                    DwStyle = Native.Constants.MNS_AUTO_DISMISS,
-                };
-                Bridge.SetMenuInfo(hMenu, ref menuInfo);
-
-                // Display the context menu at the cursor position
-                Bridge.TrackPopupMenuEx(
-                      hMenu,
-                      Native.Constants.TPM_LEFT_ALIGN | Native.Constants.TPM_BOTTOMALIGN | Native.Constants.TPM_LEFT_BUTTON,
-                      cursorPos.X,
-                      cursorPos.Y,
-                      hWnd,
-                      IntPtr.Zero);
-            }
-            else
-            {
-                // Tray menu was not initialized. Log the issue.
-                // This is normal when operating in "standalone mode" - that is, detached
-                // from the PowerToys configuration file.
                 Logger.LogError("Tried to create a context menu while the TrayMenu object is a null pointer. Normal when used in standalone mode.");
+                return;
             }
+
+            Bridge.SetForegroundWindow(hWnd);
+
+            // Get cursor position and convert it to client coordinates
+            Bridge.GetCursorPos(out Models.Point cursorPos);
+            Bridge.ScreenToClient(hWnd, ref cursorPos);
+
+            // Set menu information
+            var menuInfo = new MenuInfo
+            {
+                CbSize = (uint)Marshal.SizeOf<MenuInfo>(),
+                FMask = Native.Constants.MIM_STYLE,
+                DwStyle = Native.Constants.MNS_AUTO_DISMISS,
+            };
+            Bridge.SetMenuInfo(TrayMenu, ref menuInfo);
+
+            // Display the context menu at the cursor position
+            Bridge.TrackPopupMenuEx(
+                  TrayMenu,
+                  Native.Constants.TPM_LEFT_ALIGN | Native.Constants.TPM_BOTTOMALIGN | Native.Constants.TPM_LEFT_BUTTON,
+                  cursorPos.X,
+                  cursorPos.Y,
+                  hWnd,
+                  IntPtr.Zero);
         }
 
         public static void InitializeTray(Icon icon, string text)
@@ -89,8 +81,11 @@ namespace Awake.Core
             IntPtr hWnd = IntPtr.Zero;
 
             // Start the message loop asynchronously
-            Task.Run(() =>
+            _mainThread = new Thread(() =>
             {
+                _syncContext = new SingleThreadSynchronizationContext();
+                SynchronizationContext.SetSynchronizationContext(_syncContext);
+
                 RunOnMainThread(() =>
                 {
                     WndClassEx wcex = new()
@@ -137,74 +132,82 @@ namespace Awake.Core
 
                     Bridge.ShowWindow(hWnd, 0); // SW_HIDE
                     Bridge.UpdateWindow(hWnd);
+                    Logger.LogInfo($"Created HWND for the window: {hWnd}");
 
                     SetShellIcon(hWnd, text, icon);
                 });
-            }).Wait();
 
-            Task.Run(() =>
-            {
                 RunOnMainThread(() =>
                 {
                     RunMessageLoop();
                 });
+
+                _syncContext!.BeginMessageLoop();
             });
+
+            _mainThread.IsBackground = true;
+            _mainThread.Start();
         }
 
         internal static void SetShellIcon(IntPtr hWnd, string text, Icon? icon, TrayIconAction action = TrayIconAction.Add)
         {
-            Logger.LogInfo($"Setting the shell icon.\nText: {text}\nAction: {action}");
+            if (hWnd != IntPtr.Zero && icon != null)
-
-            int message = Native.Constants.NIM_ADD;
-
-            switch (action)
             {
-                case TrayIconAction.Update:
+                int message = Native.Constants.NIM_ADD;
-                    message = Native.Constants.NIM_MODIFY;
-                    break;
-                case TrayIconAction.Delete:
-                    message = Native.Constants.NIM_DELETE;
-                    break;
-                case TrayIconAction.Add:
-                default:
-                    break;
-            }
 
-            if (action == TrayIconAction.Add || action == TrayIconAction.Update)
+                switch (action)
-            {
-                Logger.LogInfo($"Adding or updating tray icon. HIcon handle is {icon?.Handle}\nHWnd: {hWnd}");
-
-                _notifyIconData = new NotifyIconData
                 {
-                    CbSize = Marshal.SizeOf(typeof(NotifyIconData)),
+                    case TrayIconAction.Update:
-                    HWnd = hWnd,
+                        message = Native.Constants.NIM_MODIFY;
-                    UId = 1000,
+                        break;
-                    UFlags = Native.Constants.NIF_ICON | Native.Constants.NIF_TIP | Native.Constants.NIF_MESSAGE,
+                    case TrayIconAction.Delete:
-                    UCallbackMessage = (int)Native.Constants.WM_USER,
+                        message = Native.Constants.NIM_DELETE;
-                    HIcon = icon?.Handle ?? IntPtr.Zero,
+                        break;
-                    SzTip = text,
+                    case TrayIconAction.Add:
-                };
+                    default:
-            }
+                        break;
-            else if (action == TrayIconAction.Delete)
+                }
-            {
+
-                _notifyIconData = new NotifyIconData
+                if (action == TrayIconAction.Add || action == TrayIconAction.Update)
                 {
-                    CbSize = Marshal.SizeOf(typeof(NotifyIconData)),
+                    _notifyIconData = new NotifyIconData
-                    HWnd = hWnd,
+                    {
-                    UId = 1000,
+                        CbSize = Marshal.SizeOf(typeof(NotifyIconData)),
-                    UFlags = 0,
+                        HWnd = hWnd,
-                };
+                        UId = 1000,
-            }
+                        UFlags = Native.Constants.NIF_ICON | Native.Constants.NIF_TIP | Native.Constants.NIF_MESSAGE,
+                        UCallbackMessage = (int)Native.Constants.WM_USER,
+                        HIcon = icon?.Handle ?? IntPtr.Zero,
+                        SzTip = text,
+                    };
+                }
+                else if (action == TrayIconAction.Delete)
+                {
+                    _notifyIconData = new NotifyIconData
+                    {
+                        CbSize = Marshal.SizeOf(typeof(NotifyIconData)),
+                        HWnd = hWnd,
+                        UId = 1000,
+                        UFlags = 0,
+                    };
+                }
 
-            if (!Bridge.Shell_NotifyIcon(message, ref _notifyIconData))
+                if (!Bridge.Shell_NotifyIcon(message, ref _notifyIconData))
-            {
+                {
-                int errorCode = Marshal.GetLastWin32Error();
+                    int errorCode = Marshal.GetLastWin32Error();
-                throw new Win32Exception(errorCode, $"Failed to change tray icon. Action: {action} and error code: {errorCode}");
+                    Logger.LogInfo($"Could not set the shell icon. Action: {action} and error code: {errorCode}. HIcon handle is {icon?.Handle} and HWnd is {hWnd}");
-            }
 
-            if (action == TrayIconAction.Delete)
+                    throw new Win32Exception(errorCode, $"Failed to change tray icon. Action: {action} and error code: {errorCode}");
+                }
+
+                if (action == TrayIconAction.Delete)
+                {
+                    _notifyIconData = default;
+                }
+            }
+            else
             {
-                _notifyIconData = default;
+                Logger.LogInfo($"Cannot set the shell icon - parent window handle is zero or icon is not available. Text: {text} Action: {action}");
             }
         }
 
@@ -215,6 +218,8 @@ namespace Awake.Core
                 Bridge.TranslateMessage(ref msg);
                 Bridge.DispatchMessage(ref msg);
             }
+
+            Logger.LogInfo("Message loop terminated.");
         }
 
         private static int WndProc(IntPtr hWnd, uint message, IntPtr wParam, IntPtr lParam)
@@ -295,30 +300,20 @@ namespace Awake.Core
 
         internal static void RunOnMainThread(Action action)
         {
-            var syncContext = new SingleThreadSynchronizationContext();
+            _syncContext!.Post(
-            SynchronizationContext.SetSynchronizationContext(syncContext);
+                _ =>
-
-#pragma warning disable CS8625 // Cannot convert null literal to non-nullable reference type.
-            syncContext.Post(
-            _ =>
-            {
-                try
                 {
-                    action();
+                    try
-                }
+                    {
-                catch (Exception e)
+                        Logger.LogInfo($"Thread execution is on: {Environment.CurrentManagedThreadId}");
-                {
+                        action();
-                    Console.WriteLine("Error: " + e.Message);
+                    }
-                }
+                    catch (Exception e)
-                finally
+                    {
-                {
+                        Console.WriteLine("Error: " + e.Message);
-                    syncContext.EndMessageLoop();
+                    }
-                }
+                },
-            },
+                null);
-            null);
-#pragma warning restore CS8625 // Cannot convert null literal to non-nullable reference type.
-
-            syncContext.BeginMessageLoop();
         }
 
         internal static void SetTray(AwakeSettings settings, bool startedFromPowerToys)
@@ -353,6 +348,7 @@ namespace Awake.Core
         private static void CreateNewTrayMenu(bool startedFromPowerToys, bool keepDisplayOn, AwakeMode mode)
         {
             TrayMenu = Bridge.CreatePopupMenu();
+
             if (TrayMenu == IntPtr.Zero)
             {
                 return;

src/modules/awake/Awake/Program.cs

@@ -53,7 +53,6 @@ namespace Awake
         {
             _settingsUtils = new SettingsUtils();
             LockMutex = new Mutex(true, Core.Constants.AppName, out bool instantiated);
-
             Logger.InitializeLogger(Path.Combine("\\", Core.Constants.AppName, "Logs"));
 
             try
@@ -71,86 +70,93 @@ namespace Awake
 
             AppDomain.CurrentDomain.UnhandledException += AwakeUnhandledExceptionCatcher;
 
-            if (PowerToys.GPOWrapper.GPOWrapper.GetConfiguredAwakeEnabledValue() == PowerToys.GPOWrapper.GpoRuleConfigured.Disabled)
-            {
-                Exit("PowerToys.Awake tried to start with a group policy setting that disables the tool. Please contact your system administrator.", 1);
-                return 0;
-            }
-
             if (!instantiated)
             {
+                // Awake is already running - there is no need for us to process
+                // anything further
                 Exit(Core.Constants.AppName + " is already running! Exiting the application.", 1);
+                return 1;
             }
-
+            else
-            Logger.LogInfo($"Launching {Core.Constants.AppName}...");
-            Logger.LogInfo(FileVersionInfo.GetVersionInfo(Assembly.GetExecutingAssembly().Location).FileVersion);
-            Logger.LogInfo($"Build: {Core.Constants.BuildId}");
-            Logger.LogInfo($"OS: {Environment.OSVersion}");
-            Logger.LogInfo($"OS Build: {Manager.GetOperatingSystemBuild()}");
-
-            TaskScheduler.UnobservedTaskException += (sender, args) =>
             {
-                Trace.WriteLine($"Task scheduler error: {args.Exception.Message}"); // somebody forgot to check!
+                if (PowerToys.GPOWrapper.GPOWrapper.GetConfiguredAwakeEnabledValue() == PowerToys.GPOWrapper.GpoRuleConfigured.Disabled)
-                args.SetObserved();
+                {
-            };
+                    Exit("PowerToys.Awake tried to start with a group policy setting that disables the tool. Please contact your system administrator.", 1);
+                    return 1;
+                }
+                else
+                {
+                    Logger.LogInfo($"Launching {Core.Constants.AppName}...");
+                    Logger.LogInfo(FileVersionInfo.GetVersionInfo(Assembly.GetExecutingAssembly().Location).FileVersion);
+                    Logger.LogInfo($"Build: {Core.Constants.BuildId}");
+                    Logger.LogInfo($"OS: {Environment.OSVersion}");
+                    Logger.LogInfo($"OS Build: {Manager.GetOperatingSystemBuild()}");
 
-            // To make it easier to diagnose future issues, let's get the
+                    TaskScheduler.UnobservedTaskException += (sender, args) =>
-            // system power capabilities and aggregate them in the log.
+                    {
-            Bridge.GetPwrCapabilities(out _powerCapabilities);
+                        Trace.WriteLine($"Task scheduler error: {args.Exception.Message}"); // somebody forgot to check!
-            Logger.LogInfo(JsonSerializer.Serialize(_powerCapabilities));
+                        args.SetObserved();
+                    };
 
-            Logger.LogInfo("Parsing parameters...");
+                    // To make it easier to diagnose future issues, let's get the
+                    // system power capabilities and aggregate them in the log.
+                    Bridge.GetPwrCapabilities(out _powerCapabilities);
+                    Logger.LogInfo(JsonSerializer.Serialize(_powerCapabilities));
 
-            var configOption = new Option<bool>(AliasesConfigOption, () => false, Resources.AWAKE_CMD_HELP_CONFIG_OPTION)
+                    Logger.LogInfo("Parsing parameters...");
-            {
-                Arity = ArgumentArity.ZeroOrOne,
-                IsRequired = false,
-            };
 
-            var displayOption = new Option<bool>(AliasesDisplayOption, () => true, Resources.AWAKE_CMD_HELP_DISPLAY_OPTION)
+                    var configOption = new Option<bool>(AliasesConfigOption, () => false, Resources.AWAKE_CMD_HELP_CONFIG_OPTION)
-            {
+                    {
-                Arity = ArgumentArity.ZeroOrOne,
+                        Arity = ArgumentArity.ZeroOrOne,
-                IsRequired = false,
+                        IsRequired = false,
-            };
+                    };
 
-            var timeOption = new Option<uint>(AliasesTimeOption, () => 0, Resources.AWAKE_CMD_HELP_TIME_OPTION)
+                    var displayOption = new Option<bool>(AliasesDisplayOption, () => true, Resources.AWAKE_CMD_HELP_DISPLAY_OPTION)
-            {
+                    {
-                Arity = ArgumentArity.ExactlyOne,
+                        Arity = ArgumentArity.ZeroOrOne,
-                IsRequired = false,
+                        IsRequired = false,
-            };
+                    };
 
-            var pidOption = new Option<int>(AliasesPidOption, () => 0, Resources.AWAKE_CMD_HELP_PID_OPTION)
+                    var timeOption = new Option<uint>(AliasesTimeOption, () => 0, Resources.AWAKE_CMD_HELP_TIME_OPTION)
-            {
+                    {
-                Arity = ArgumentArity.ZeroOrOne,
+                        Arity = ArgumentArity.ExactlyOne,
-                IsRequired = false,
+                        IsRequired = false,
-            };
+                    };
 
-            var expireAtOption = new Option<string>(AliasesExpireAtOption, () => string.Empty, Resources.AWAKE_CMD_HELP_EXPIRE_AT_OPTION)
+                    var pidOption = new Option<int>(AliasesPidOption, () => 0, Resources.AWAKE_CMD_HELP_PID_OPTION)
-            {
+                    {
-                Arity = ArgumentArity.ZeroOrOne,
+                        Arity = ArgumentArity.ZeroOrOne,
-                IsRequired = false,
+                        IsRequired = false,
-            };
+                    };
 
-            var parentPidOption = new Option<bool>(AliasesParentPidOption, () => true, Resources.AWAKE_CMD_PARENT_PID_OPTION)
+                    var expireAtOption = new Option<string>(AliasesExpireAtOption, () => string.Empty, Resources.AWAKE_CMD_HELP_EXPIRE_AT_OPTION)
-            {
+                    {
-                Arity = ArgumentArity.ZeroOrOne,
+                        Arity = ArgumentArity.ZeroOrOne,
-                IsRequired = false,
+                        IsRequired = false,
-            };
+                    };
 
-            RootCommand? rootCommand =
+                    var parentPidOption = new Option<bool>(AliasesParentPidOption, () => false, Resources.AWAKE_CMD_PARENT_PID_OPTION)
-            [
+                    {
-                configOption,
+                        Arity = ArgumentArity.ZeroOrOne,
-                displayOption,
+                        IsRequired = false,
-                timeOption,
+                    };
-                pidOption,
-                expireAtOption,
-                parentPidOption,
-            ];
 
-            rootCommand.Description = Core.Constants.AppName;
+                    RootCommand? rootCommand =
-            rootCommand.SetHandler(HandleCommandLineArguments, configOption, displayOption, timeOption, pidOption, expireAtOption, parentPidOption);
+                    [
+                        configOption,
+                        displayOption,
+                        timeOption,
+                        pidOption,
+                        expireAtOption,
+                        parentPidOption,
+                    ];
 
-            return rootCommand.InvokeAsync(args).Result;
+                    rootCommand.Description = Core.Constants.AppName;
+                    rootCommand.SetHandler(HandleCommandLineArguments, configOption, displayOption, timeOption, pidOption, expireAtOption, parentPidOption);
+
+                    return rootCommand.InvokeAsync(args).Result;
+                }
+            }
         }
 
         private static void AwakeUnhandledExceptionCatcher(object sender, UnhandledExceptionEventArgs e)
@@ -180,15 +186,11 @@ namespace Awake
             if (pid == 0 && !useParentPid)
             {
                 Logger.LogInfo("No PID specified. Allocating console...");
-                Manager.AllocateConsole();
+                AllocateLocalConsole();
-
-                _handler += new ConsoleEventHandler(ExitHandler);
-                Manager.SetConsoleControlHandler(_handler, true);
-
-                Trace.Listeners.Add(new ConsoleTraceListener());
             }
             else
             {
+                Logger.LogInfo("Starting with PID binding.");
                 _startedFromPowerToys = true;
             }
 
@@ -246,7 +248,7 @@ namespace Awake
                 }
                 catch (Exception ex)
                 {
-                    Logger.LogError($"There was a problem with the configuration file. Make sure it exists.\n{ex.Message}");
+                    Logger.LogError($"There was a problem with the configuration file. Make sure it exists. {ex.Message}");
                 }
             }
             else if (pid != 0 || useParentPid)
@@ -307,38 +309,22 @@ namespace Awake
             }
         }
 
+        private static void AllocateLocalConsole()
+        {
+            Manager.AllocateConsole();
+
+            _handler += new ConsoleEventHandler(ExitHandler);
+            Manager.SetConsoleControlHandler(_handler, true);
+
+            Trace.Listeners.Add(new ConsoleTraceListener());
+        }
+
         private static void ScaffoldConfiguration(string settingsPath)
         {
             try
             {
-                var directory = Path.GetDirectoryName(settingsPath)!;
+                SetupFileSystemWatcher(settingsPath);
-                var fileName = Path.GetFileName(settingsPath);
+                InitializeSettings();
-
-                _watcher = new FileSystemWatcher
-                {
-                    Path = directory,
-                    EnableRaisingEvents = true,
-                    NotifyFilter = NotifyFilters.LastWrite | NotifyFilters.CreationTime,
-                    Filter = fileName,
-                };
-
-                var mergedObservable = Observable.Merge(
-                    Observable.FromEventPattern<FileSystemEventHandler, FileSystemEventArgs>(
-                        h => _watcher.Changed += h,
-                        h => _watcher.Changed -= h),
-                    Observable.FromEventPattern<FileSystemEventHandler, FileSystemEventArgs>(
-                        h => _watcher.Created += h,
-                        h => _watcher.Created -= h));
-
-                mergedObservable
-                    .Throttle(TimeSpan.FromMilliseconds(25))
-                    .SubscribeOn(TaskPoolScheduler.Default)
-                    .Select(e => e.EventArgs)
-                    .Subscribe(HandleAwakeConfigChange);
-
-                var settings = Manager.ModuleSettings!.GetSettings<AwakeSettings>(Core.Constants.AppName) ?? new AwakeSettings();
-                TrayHelper.SetTray(settings, _startedFromPowerToys);
-
                 ProcessSettings();
             }
             catch (Exception ex)
@@ -347,6 +333,38 @@ namespace Awake
             }
         }
 
+        private static void SetupFileSystemWatcher(string settingsPath)
+        {
+            var directory = Path.GetDirectoryName(settingsPath)!;
+            var fileName = Path.GetFileName(settingsPath);
+
+            _watcher = new FileSystemWatcher
+            {
+                Path = directory,
+                EnableRaisingEvents = true,
+                NotifyFilter = NotifyFilters.LastWrite | NotifyFilters.CreationTime,
+                Filter = fileName,
+            };
+
+            Observable.Merge(
+                Observable.FromEventPattern<FileSystemEventHandler, FileSystemEventArgs>(
+                    h => _watcher.Changed += h,
+                    h => _watcher.Changed -= h),
+                Observable.FromEventPattern<FileSystemEventHandler, FileSystemEventArgs>(
+                    h => _watcher.Created += h,
+                    h => _watcher.Created -= h))
+            .Throttle(TimeSpan.FromMilliseconds(25))
+            .SubscribeOn(TaskPoolScheduler.Default)
+            .Select(e => e.EventArgs)
+            .Subscribe(HandleAwakeConfigChange);
+        }
+
+        private static void InitializeSettings()
+        {
+            var settings = Manager.ModuleSettings?.GetSettings<AwakeSettings>(Core.Constants.AppName) ?? new AwakeSettings();
+            TrayHelper.SetTray(settings, _startedFromPowerToys);
+        }
+
         private static void HandleAwakeConfigChange(FileSystemEventArgs fileEvent)
         {
             try
@@ -364,7 +382,9 @@ namespace Awake
         {
             try
             {
-                var settings = _settingsUtils!.GetSettings<AwakeSettings>(Core.Constants.AppName) ?? throw new InvalidOperationException("Settings are null.");
+                var settings = _settingsUtils!.GetSettings<AwakeSettings>(Core.Constants.AppName)
+                    ?? throw new InvalidOperationException("Settings are null.");
+
                 Logger.LogInfo($"Identified custom time shortcuts for the tray: {settings.Properties.CustomTrayTimes.Count}");
 
                 switch (settings.Properties.Mode)
@@ -378,7 +398,7 @@ namespace Awake
                         break;
 
                     case AwakeMode.TIMED:
-                        uint computedTime = (settings.Properties.IntervalHours * 60 * 60) + (settings.Properties.IntervalMinutes * 60);
+                        uint computedTime = (settings.Properties.IntervalHours * 3600) + (settings.Properties.IntervalMinutes * 60);
                         Manager.SetTimedKeepAwake(computedTime, settings.Properties.KeepDisplayOn);
                         break;