Remotely/Desktop.Win/Services/SessionIndicatorWin.cs

using Microsoft.Extensions.DependencyInjection;
using Remotely.Desktop.Core;
using Remotely.Desktop.Core.Interfaces;
using Remotely.Desktop.Core.Services;
using Remotely.Desktop.Core.Utilities;
using Remotely.Shared.Utilities;
using System;
using System.ComponentModel;
using System.Drawing;
using System.IO;
using System.Linq;
using System.Windows.Forms;
using System.Windows.Media.Imaging;

namespace Remotely.Desktop.Win.Services
{
    public class SessionIndicatorWin : ISessionIndicator
    {
        private readonly Form _backgroundForm;
        private readonly IDeviceInitService _deviceInitService;
        private Container _container;
        private ContextMenuStrip _contextMenuStrip;
        private NotifyIcon _notifyIcon;

        public SessionIndicatorWin(Form backgroundForm, IDeviceInitService deviceInitService)
        {
            _backgroundForm = backgroundForm;
            _deviceInitService = deviceInitService;
        }
        public void Show()
        {
            try
            {
                if (_notifyIcon != null)
                {
                    return;
                }

                App.Current.Dispatcher.Invoke(() =>
                {
                    App.Current.Exit += App_Exit;
                });

                _backgroundForm.Invoke(new Action(() =>
                {
                    _container = new Container();
                    _contextMenuStrip = new ContextMenuStrip(_container);
                    _contextMenuStrip.Items.Add("Exit", null, ExitMenuItem_Click);

                    Icon icon;

                    if (_deviceInitService.BrandingInfo?.Icon?.Any() == true)
                    {
                        using var ms = new MemoryStream(_deviceInitService.BrandingInfo.Icon);
                        using var bitmap = new Bitmap(ms);
                        icon = Icon.FromHandle(bitmap.GetHicon());
                    }
                    else
                    {
                        icon = Icon.ExtractAssociatedIcon(Path.Combine(AppDomain.CurrentDomain.BaseDirectory, EnvironmentHelper.DesktopExecutableFileName));
                    }

                    _notifyIcon = new NotifyIcon(_container)
                    {
                        Icon = icon,
                        Text = "Remote Control Session",
                        BalloonTipIcon = ToolTipIcon.Info,
                        BalloonTipText = "A remote control session has started.",
                        BalloonTipTitle = "Remote Control Started",
                        ContextMenuStrip = _contextMenuStrip
                    };

                    _notifyIcon.Visible = true;
                    _notifyIcon.ShowBalloonTip(3000);
                }));
            }
            catch (Exception ex)
            {
                Logger.Write(ex);
            }
        }

        private void App_Exit(object sender, EventArgs e)
        {
            if (_notifyIcon != null)
            {
                _notifyIcon.Visible = false;
                _notifyIcon?.Dispose();
                _notifyIcon?.Icon?.Dispose();
            }
        }
        private async void ExitMenuItem_Click(object sender, EventArgs e)
        {
            var casterSocket = ServiceContainer.Instance.GetRequiredService<ICasterSocket>();
            await casterSocket.DisconnectAllViewers();
        }
    }
}