Merge pull request #74 from GoodSign2017/fix-exit

Watfaq · Dec 18, 2023 · a00de79 · a00de79
2 parents 3ae8af8 + ed8c4a0
commit a00de79
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/SoftU2FDaemon/Program.cs b/SoftU2FDaemon/Program.cs
@@ -103,6 +103,11 @@ private void SetupApplication()
             }
 
             _cancellation = new CancellationTokenSource();
+
+            _cancellation.Token.Register(() => {
+                _cancellation.Dispose();
+                if (_exitRequested) { Environment.Exit(0); }
+            });
         }
 
 
@@ -142,6 +147,8 @@ private void ConfigureServices(IServiceCollection service)
 
         #region System Tray Icon
 
+        private bool _exitRequested = false;
+
         private void InitializeTrayIcon()
         {
             _trayMenu = new ContextMenu();
@@ -152,7 +159,10 @@ private void InitializeTrayIcon()
 
             _trayMenu.Items.Add("Reset", null, OnResetClickedOnClick);
             _trayMenu.Items.Add("-");
-            _trayMenu.Items.Add("Exit", null, (sender, args) => Application.Exit());
+            _trayMenu.Items.Add("Exit", null, (sender, args) => {
+                _exitRequested = true;
+                Application.Exit();
+            });
 
             _trayIcon = new NotifyIcon
             {
@@ -176,6 +186,13 @@ private void InitializeTrayIcon()
                 lastActiveWin = GetForegroundWindow();
             };
             _trayIcon.BalloonTipClosed += (sender, args) => _notificationOpen = false;
+
+            this.FormClosing += (sender, e) =>
+            {
+                // Hide and dispose the icon
+                _trayIcon.Visible = false;
+                _trayIcon.Dispose();
+            };
         }
 
         private void OnAutoStartClick(object sender, EventArgs e)