Main.cs

using BaseMacro.Macro;
using HarmonyLib;
using SA.GoogleDoc;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using System.Reflection;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading.Tasks;
using UnityEngine;
using UnityModManagerNet;
using static UnityModManagerNet.UnityModManager;

#nullable enable

namespace BaseMacro
{
#if DEBUG
    [EnableReloading]
#endif
    public static class Main
    {
        public static UnityModManager.ModEntry? Mod { get; private set; }
        public static Harmony? Harmony { get; private set; }
        public static Settings Settings { get; private set; } = null!;
        private static GameObject? _uiObject;
        public static bool Load(UnityModManager.ModEntry modEntry)
        {
            Mod = modEntry;
            Settings = Settings.Load(modEntry);

            // 手动初始化 InputSystem
            if (InputSystem.Initialize())
            {
                modEntry.Logger.Log("[InputSystem] 初始化成功");
            }
            else
            {
                modEntry.Logger.Log("[InputSystem] 初始化失败");
            }

            if (InputSystem.IsUsingNtFunctions())
            {
                modEntry.Logger.Log("[InputSystem] 当前使用 NT 内核函数");
            }
            else
            {
                modEntry.Logger.Log("[InputSystem] 当前使用传统输入模拟");
            }

            modEntry.OnToggle = OnToggle;
            modEntry.OnGUI = Settings.OnGUI;
            modEntry.OnSaveGUI = Settings.OnSaveGUI;
            modEntry.OnUnload = Unload;

            Harmony = new Harmony(modEntry.Info.Id);
            return true;
        }

        public static bool Unload(UnityModManager.ModEntry modEntry)
        {
            var harmony = new Harmony(modEntry.Info.Id);
            harmony.UnpatchAll(modEntry.Info.Id);

            // If you have created any dependent objects, they also need to be deleted.

            return true;
        }

        public static bool IsDebugAssembly()
        {
            Assembly assembly = Assembly.GetExecutingAssembly();
            var attribute = assembly.GetCustomAttribute<DebuggableAttribute>();
            if (attribute == null)
                return false;
            return attribute.IsJITOptimizerDisabled;
        }

        private static bool OnToggle(UnityModManager.ModEntry modEntry, bool value)
        {
            if (value)
            {
#if DEBUG
                // None

#else
                if (modEntry.Info.Version != "1.1.1" || modEntry.Info.Id != "BaseMacro" || modEntry.Info.DisplayName != "Base Macro" || modEntry.Info.Author != "HitMargin" || modEntry.Info.AssemblyName != "BaseMacro.dll" || modEntry.Info.EntryMethod != "BaseMacro.Main.Load")
                {
                    Mod?.Logger.Error("Modifying the Info.json file is NOT allowed!");
                    Application.Quit();
                }

#endif
                Mod?.Info.IsCheat = true;
                if (Mod!.Info.IsCheat)
                {
                    Mod.Info.DisplayName += " (Cheat)";
                }
                if (IsDebugAssembly())
                    Mod.Info.DisplayName += " <color=grey>(Debug)</color>";
                if (Settings.IsBeta)
                    Mod.Info.Version += $"\nBeta{Settings.BetaVersion}";

                IsEnabled = true;
                Harmony?.PatchAll(Assembly.GetExecutingAssembly());
                if (_uiObject == null)
                {
                    _uiObject = new GameObject("MacroText");
                    _uiObject.AddComponent<ShowText>();
                    UnityEngine.Object.DontDestroyOnLoad(_uiObject);
                    TrySetWindowTitle($"{GetClean(modEntry.Info.DisplayName)}, {GetClean(modEntry.Info.Version)}, {modEntry.Info.Author}");
                }
            }
            else
            {
                IsEnabled = false;
                Harmony?.UnpatchAll();
                TrySetWindowTitle(null);
                InputSystem.EmergencyStop();
            }
            return true;
        }

        public static string GetClean(string version)
        {
            if (version.Contains("<color="))
            {
                int startIndex = version.IndexOf('>') + 1;
                int endIndex = version.LastIndexOf('<');
                if (startIndex > 0 && endIndex > startIndex)
                {
                    return version.Substring(startIndex, endIndex - startIndex);
                }
            }
            return version;
        }

        [DllImport("user32.dll")]
        private static extern bool SetWindowText(IntPtr hWnd, string lpString);

        private static void TrySetWindowTitle(string? title)
        {
            try
            {
                // 获取当前进程
                Process currentProcess = Process.GetCurrentProcess();

                // 等待主窗口句柄可用
                for (int i = 0; i < 10 && currentProcess.MainWindowHandle == IntPtr.Zero; i++)
                {
                    currentProcess.Refresh();
                    System.Threading.Thread.Sleep(100);
                }

                IntPtr hwnd = currentProcess.MainWindowHandle;

                if (hwnd != IntPtr.Zero)
                {
                    string newTitle = $"A Dance of Fire and Ice";
                    if (title != null)
                    {
                        // 修复：将换行符替换为空格
                        string cleanTitle = title.Replace('\n', ' ').Replace('\r', ' ');
                        newTitle = $"A Dance of Fire and Ice - {cleanTitle}";
                    }

                    if (SetWindowText(hwnd, newTitle))
                    {
                        Mod?.Logger.Log($"成功设置窗口标题: {newTitle}");
                    }
                    else
                    {
                        Mod?.Logger.Log("设置窗口标题失败");
                    }
                }
                else
                {
                    Mod?.Logger.Log("未获取到主窗口句柄");
                }
            }
            catch (Exception ex)
            {
                Mod?.Logger.Log($"设置窗口标题异常: {ex.Message}");
            }
        }
        public static bool IsEnabled { get; internal set; }
    }
}