QuickLook/QuickLook/TrayIconManager.cs

// Copyright © 2017-2025 QL-Win Contributors
//
// This file is part of QuickLook program.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

using QuickLook.Common.Helpers;
using QuickLook.Helpers;
using QuickLook.Properties;
using System;
using System.ComponentModel;
using System.Diagnostics;
using System.Drawing;
using System.IO;
using System.Windows.Forms;

namespace QuickLook;

internal class TrayIconManager : IDisposable
{
    private static TrayIconManager _instance;

    private readonly NotifyIcon _icon;

    private readonly MenuItem _itemAutorun =
        new(TranslationHelper.Get("Icon_RunAtStartup"),
            (sender, e) =>
            {
                if (AutoStartupHelper.IsAutorun())
                    AutoStartupHelper.RemoveAutorunShortcut();
                else
                    AutoStartupHelper.CreateAutorunShortcut();
            })
        { Enabled = !App.IsUWP };

    private readonly string _attr = App.IsUWP ? $" (UWP{(App.IsSandBox ? string.Empty : "-SandBox")})" : string.Empty;

    private TrayIconManager()
    {
        _icon = new NotifyIcon
        {
            Text = string.Format(TranslationHelper.Get("Icon_ToolTip"),
                Application.ProductVersion),
            Icon = GetTrayIconByDPI(),
            ContextMenu = new ContextMenu(
            [
                new MenuItem($"v{Application.ProductVersion}{_attr}") {Enabled = false},
                new MenuItem("-"),
                new MenuItem(TranslationHelper.Get("Icon_CheckUpdate"), (_, _) => Updater.CheckForUpdates()),
                new MenuItem(TranslationHelper.Get("Icon_GetPlugin"),
                    (_, _) => Process.Start("https://github.com/QL-Win/QuickLook/wiki/Available-Plugins")),
                new MenuItem(TranslationHelper.Get("Icon_OpenDataFolder"), (_, _) => Process.Start("explorer.exe", SettingHelper.LocalDataPath)),
                _itemAutorun,
                new MenuItem(TranslationHelper.Get("Icon_Restart"), (_, _) => Restart(forced: true)),
                new MenuItem(TranslationHelper.Get("Icon_Quit"),
                    (_, _) => System.Windows.Application.Current.Shutdown())
            ]),
            Visible = SettingHelper.Get("ShowTrayIcon", true)
        };

        _icon.ContextMenu.Popup += (sender, e) => { _itemAutorun.Checked = AutoStartupHelper.IsAutorun(); };

        // Readjust the display position of ContextMenu
        if (SettingHelper.Get("ModernTrayIcon", true, "QuickLook"))
        {
            _icon.MouseDown += (_, e) =>
            {
                if (e.Button == MouseButtons.Right)
                {
                    // Call ShowContextMenu here will be later than the native call,
                    // so here can readjust the ContextMenu position.
                    // You can check the source code to determine the behavior.
                    _icon.ShowContextMenu();
                }
            };
        }
    }

    public void Dispose()
    {
        _icon.Visible = false;
    }

    public void Restart(string fileName = null, string dir = null, string args = null, int? exitCode = null, bool forced = false)
    {
        _ = args; // Currently there is no cli supported by QL

        try
        {
            using Process process = new()
            {
                StartInfo = new ProcessStartInfo()
                {
                    FileName = fileName ?? Path.Combine(dir ?? AppDomain.CurrentDomain.BaseDirectory, AppDomain.CurrentDomain.FriendlyName),
                    WorkingDirectory = dir ?? Environment.CurrentDirectory,
                    UseShellExecute = true,
                },
            };
            process.Start();
        }
        catch (Win32Exception)
        {
            return;
        }
        if (forced)
        {
            Process.GetCurrentProcess().Kill();
        }
        Environment.Exit(exitCode ?? 'r' + 'e' + 's' + 't' + 'a' + 'r' + 't');
    }

    private Icon GetTrayIconByDPI()
    {
        var scale = DisplayDeviceHelper.GetCurrentScaleFactor().Vertical;

        if (!App.IsWin10)
            return scale > 1 ? Resources.app : Resources.app_16;

        return OSThemeHelper.SystemUsesDarkTheme()
            ? (scale > 1 ? Resources.app_white : Resources.app_white_16)
            : (scale > 1 ? Resources.app_black : Resources.app_black_16);
    }

    public static void ShowNotification(string title, string content, bool isError = false, int timeout = 5000,
        Action clickEvent = null,
        Action closeEvent = null)
    {
        var icon = GetInstance()._icon;
        icon.ShowBalloonTip(timeout, title, content, isError ? ToolTipIcon.Error : ToolTipIcon.Info);
        icon.BalloonTipClicked += OnIconOnBalloonTipClicked;
        icon.BalloonTipClosed += OnIconOnBalloonTipClosed;

        void OnIconOnBalloonTipClicked(object sender, EventArgs e)
        {
            clickEvent?.Invoke();
            icon.BalloonTipClicked -= OnIconOnBalloonTipClicked;
        }

        void OnIconOnBalloonTipClosed(object sender, EventArgs e)
        {
            closeEvent?.Invoke();
            icon.BalloonTipClosed -= OnIconOnBalloonTipClosed;
        }
    }

    public static TrayIconManager GetInstance()
    {
        return _instance ??= new TrayIconManager();
    }
}