#if UNITY_EDITOR_WIN && UNITY_EDITOR

using System;
using System.Diagnostics;
using System.Runtime.InteropServices;
using System.Text.RegularExpressions;
using UnityEditor;

[InitializeOnLoad]
public static class UnityEditorPerformanceBoost
{
    /// <summary>
    /// Is it currently in high-performance mode?
    /// </summary>
    private static bool isHighPerformance = false;

    /// <summary>
    /// Power-Saving Power Plan
    /// </summary>
    private static readonly string powerSchemeSaving = "a1841308-3541-4fab-bc81-f71556f20b4a";

    /// <summary>
    /// Balanced Power Plan
    /// </summary>
    private static readonly string powerSchemeBalance = "381b4222-f694-41f0-9685-ff5bb260df2e";

    /// <summary>
    /// High-Performance Power Plan
    /// </summary>
    private static readonly string PowerSchemeHighPerformance = "8c5e7fda-e8bf-4a96-9a85-a6e23a8c635c";

    /// <summary>
    /// Timer
    /// </summary>
    private static double lastTime;

    /// <summary>
    /// Previous power plan
    /// </summary>
    private static string originalPowerScheme;

    /// <summary>
    /// GUID Regular Expressions
    /// </summary>
    private static readonly Regex GuidRegex = new(@"\b[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}\b", RegexOptions.Compiled);

    static UnityEditorPerformanceBoost()
    {
        EditorApplication.update += OnUpdate;
    }

    /// <summary>
    /// Scheduled Timer
    /// </summary>
    private static void OnUpdate()
    {
        if (EditorApplication.timeSinceStartup - lastTime > 2)
        {
            lastTime = EditorApplication.timeSinceStartup;
            CheckFocused();
        }
    }

    /// <summary>
    /// Focus Check
    /// </summary>
    private static void CheckFocused()
    {
        if (IsWindowFocused())
        {
            EnableHighPerformance();
        }
        else
        {
            RestoreOriginalPowerScheme();
        }
    }

    /// <summary>
    /// Enable High Performance Mode
    /// </summary>
    private static void EnableHighPerformance()
    {
        try
        {
            if (isHighPerformance) return;
            isHighPerformance = true;

            // Save the current power plan

            var str = CommandPowerCfg($"/getactivescheme");
            var match = GuidRegex.Match(str);
            originalPowerScheme = match.Value;

            // Raise the priority of your process

            Process.GetCurrentProcess().PriorityClass = ProcessPriorityClass.High;

            // Set the Windows power plan to High Performance

            CommandPowerCfg($"/setactive {PowerSchemeHighPerformance}");

            UnityEngine.Debug.Log("High-Performance Power Plan Enabled：" + PowerSchemeHighPerformance);
        }
        catch (Exception e)
        {
            UnityEngine.Debug.LogError(e);
        }
    }

    /// <summary>
    /// 元の電源プランに戻す
    /// </summary>
    private static void RestoreOriginalPowerScheme()
    {
        try
        {
            if (!isHighPerformance) return;
            isHighPerformance = false;

            // Set the process priority to default

            Process.GetCurrentProcess().PriorityClass = ProcessPriorityClass.Normal;

            // Restore Windows power plans

            CommandPowerCfg($"/setactive {originalPowerScheme}");

            UnityEngine.Debug.Log("I switched back to the original plan：" + originalPowerScheme);
        }
        catch (Exception e)
        {
            UnityEngine.Debug.LogError(e);
        }
    }

    [DllImport("user32.dll")]
    private static extern IntPtr GetForegroundWindow();

    [DllImport("user32.dll")]
    private static extern uint GetWindowThreadProcessId(IntPtr hWnd, out uint processId);

    /// <summary>
    /// Is the window in the foreground?
    /// </summary>
    /// <returns></returns>
    private static bool IsWindowFocused()
    {
        IntPtr hwnd = GetForegroundWindow();
        if (hwnd == IntPtr.Zero) return false;

        GetWindowThreadProcessId(hwnd, out uint processId);
        return (processId == (uint)Process.GetCurrentProcess().Id);
    }

    /// <summary>
    /// Power Plan Commands
    /// </summary>
    /// <param name="command"></param>
    /// <returns></returns>
    private static string CommandPowerCfg(string command)
    {
        var processStartInfo = new ProcessStartInfo("powercfg", command)
        {
            RedirectStandardOutput = true,
            UseShellExecute = false,
            CreateNoWindow = true
        };

        using (var process = Process.Start(processStartInfo))
        {
            return process.StandardOutput.ReadToEnd();
        }
    }
}

#endif