using UnityEngine;
using UnityEditor;
using UnityEngine.UI;
using System.Collections.Generic;
using System.Linq;

[InitializeOnLoad]
public class QuickSelectorHud
{
    public static GameObject HoveredObject;

    static QuickSelectorHud()
    {
        SceneView.duringSceneGui += OnSceneGUI;
    }

    private static void OnSceneGUI(SceneView sceneView)
    {
        if (HoveredObject != null)
        {
            DrawHighlight(HoveredObject);
            sceneView.Repaint();
        }

        Event e = Event.current;
        if (e.type == EventType.MouseDown && e.button == 1 && e.shift)
        {
            var entries = CollectData(sceneView, e.mousePosition);
            QuickSelectorPopup popup = new QuickSelectorPopup(entries);
            PopupWindow.Show(new Rect(e.mousePosition.x, e.mousePosition.y, 0, 0), popup);
            e.Use();
        }
    }

    private static void DrawHighlight(GameObject go)
    {
        Handles.color = new Color(0f, 0.7f, 1f, 1f); // 亮蓝色
        RectTransform rt = go.GetComponent<RectTransform>();
        if (rt != null)
        {
            Vector3[] corners = new Vector3[4];
            rt.GetWorldCorners(corners);
            Handles.DrawPolyLine(corners[0], corners[1], corners[2], corners[3], corners[0]);
            Handles.color = new Color(0f, 0.7f, 1f, 0.1f);
            Handles.DrawAAConvexPolygon(corners);
        }
        else
        {
            Renderer r = go.GetComponent<Renderer>();
            if (r != null) Handles.DrawWireCube(r.bounds.center, r.bounds.size);
        }
    }

    private static List<SelectionEntry> CollectData(SceneView sceneView, Vector2 mousePos)
    {
        bool enableUI = EditorPrefs.GetBool("QS_UI", true);
        bool enable3D = EditorPrefs.GetBool("QS_3D", true);
        Vector2 guiPos = mousePos;
        guiPos.y = sceneView.camera.pixelHeight - guiPos.y;
        Ray ray = HandleUtility.GUIPointToWorldRay(mousePos);

        HashSet<GameObject> processed = new HashSet<GameObject>();
        List<SelectionEntry> list = new List<SelectionEntry>();

        GameObject smart = HandleUtility.PickGameObject(mousePos, false);
        if (smart != null) AddEntry(list, processed, smart, "PICK");

        if (enableUI)
        {
            var rects = GameObject.FindObjectsByType<RectTransform>(FindObjectsSortMode.None);
            foreach (var rect in rects)
            {
                if (rect.gameObject.activeInHierarchy && RectTransformUtility.RectangleContainsScreenPoint(rect, guiPos, sceneView.camera))
                    AddEntry(list, processed, rect.gameObject, "UI");
            }
        }

        if (enable3D)
        {
            RaycastHit[] hits = Physics.RaycastAll(ray, float.MaxValue);
            foreach (var hit in hits) AddEntry(list, processed, hit.collider.gameObject, "3D");
            foreach (var r in GameObject.FindObjectsByType<Renderer>(FindObjectsSortMode.None))
                if (!processed.Contains(r.gameObject) && r.bounds.IntersectRay(ray, out _))
                    AddEntry(list, processed, r.gameObject, "Mesh");
        }
        return list;
    }

    private static void AddEntry(List<SelectionEntry> list, HashSet<GameObject> set, GameObject go, string src)
    {
        if (go == null || !set.Add(go)) return;
        string maj = (go.GetComponent<RectTransform>() != null) ? "UI" : (go.GetComponent<Collider>() != null ? "3D" : "Mesh");
        string min = (maj == "UI") ? GetUIType(go) : (go.GetComponent<Renderer>()?.GetType().Name ?? "Object");
        list.Add(new SelectionEntry { go = go, root = GetRoot(go.transform), depth = GetDepth(go.transform), sibling = go.transform.GetSiblingIndex(), major = maj, minor = min });
    }

    private static string GetUIType(GameObject go)
    {
        if (go.GetComponent("TextMeshProUGUI")) return "TMPro";
        if (go.GetComponent<Text>()) return "Text";
        if (go.GetComponent<Image>()) return "Image";
        return "Rect";
    }
    private static GameObject GetRoot(Transform t) { while (t.parent != null) t = t.parent; return t.gameObject; }
    private static int GetDepth(Transform t) { int d = 0; while (t.parent != null) { d++; t = t.parent; } return d; }
}

public class SelectionEntry { public GameObject go, root; public int depth, sibling; public string major, minor; }

public class QuickSelectorPopup : PopupWindowContent
{
    private List<SelectionEntry> _entries;
    private List<DisplayItem> _displayItems;
    private Vector2 _scroll;
    private GUIStyle _hoverStyle;
    private GUIStyle _richLabelStyle; // 修正点：手动创建支持富文本的 Style

    public QuickSelectorPopup(List<SelectionEntry> entries)
    {
        _entries = entries;
        RefreshList();
    }

    public override Vector2 GetWindowSize() => new Vector2(280, Mathf.Min((_displayItems.Count * 22) + 30, 450));

    public override void OnGUI(Rect rect)
    {
        // 样式初始化
        if (_hoverStyle == null)
        {
            _hoverStyle = new GUIStyle(EditorStyles.label);
            _hoverStyle.normal.background = MakeTex(2, 2, new Color(0.2f, 0.5f, 1f, 0.4f));
        }
        if (_richLabelStyle == null)
        {
            _richLabelStyle = new GUIStyle(EditorStyles.label);
            _richLabelStyle.richText = true; // 关键修正：开启富文本支持
        }

        _scroll = EditorGUILayout.BeginScrollView(_scroll);
        Event e = Event.current;

        foreach (var item in _displayItems)
        {
            if (item.isSep) { EditorGUILayout.LabelField("", GUI.skin.horizontalSlider); continue; }

            Rect r = EditorGUILayout.GetControlRect(false, 20);
            bool isHover = r.Contains(e.mousePosition);

            if (isHover)
            {
                GUI.Box(r, "", _hoverStyle);
                if (QuickSelectorHud.HoveredObject != item.ent.go)
                {
                    QuickSelectorHud.HoveredObject = item.ent.go;
                    EditorGUIUtility.PingObject(item.ent.go);
                    SceneView.RepaintAll();
                }
                if (e.type == EventType.MouseDown && e.button == 0)
                {
                    Selection.activeGameObject = item.ent.go;
                    this.editorWindow.Close();
                }
            }

            // 使用修正后的富文本样式绘制
            GUI.Label(r, item.label, _richLabelStyle);
        }
        EditorGUILayout.EndScrollView();

        DrawToolbar();

        if (e.type == EventType.MouseMove && !rect.Contains(e.mousePosition))
        {
            QuickSelectorHud.HoveredObject = null;
        }
    }

    private void DrawToolbar()
    {
        GUILayout.BeginHorizontal(EditorStyles.toolbar);
        if (GUILayout.Toggle(EditorPrefs.GetBool("QS_UI", true), "UI", EditorStyles.toolbarButton) != EditorPrefs.GetBool("QS_UI", true))
        {
            EditorPrefs.SetBool("QS_UI", !EditorPrefs.GetBool("QS_UI")); RefreshList();
        }
        if (GUILayout.Toggle(EditorPrefs.GetBool("QS_3D", true), "3D", EditorStyles.toolbarButton) != EditorPrefs.GetBool("QS_3D", true))
        {
            EditorPrefs.SetBool("QS_3D", !EditorPrefs.GetBool("QS_3D")); RefreshList();
        }
        GUILayout.FlexibleSpace();
        if (GUILayout.Toggle(EditorPrefs.GetBool("QS_REV", false), "Reverse", EditorStyles.toolbarButton) != EditorPrefs.GetBool("QS_REV", false))
        {
            EditorPrefs.SetBool("QS_REV", !EditorPrefs.GetBool("QS_REV")); RefreshList();
        }
        GUILayout.EndHorizontal();
    }

    public override void OnClose() { QuickSelectorHud.HoveredObject = null; SceneView.RepaintAll(); }

    private void RefreshList()
    {
        _displayItems = new List<DisplayItem>();
        bool rev = EditorPrefs.GetBool("QS_REV", false);
        var groups = _entries.GroupBy(x => x.root).OrderByDescending(g => g.Any(e => e.major == "UI")).ThenBy(g => g.Key.name);
        bool first = true;
        foreach (var g in groups)
        {
            if (!first) _displayItems.Add(new DisplayItem { isSep = true });
            first = false;
            var sorted = rev ? g.OrderByDescending(x => x.depth).ThenByDescending(x => x.sibling) : g.OrderBy(x => x.depth).ThenBy(x => x.sibling);
            int minD = g.Min(x => x.depth);
            int maxD = g.Max(x => x.depth);
            foreach (var ent in sorted)
            {
                int ind = rev ? (maxD - ent.depth) : (ent.depth - minD);
                string color = (ent.major == "UI" ? "#00E6FF" : "#AAAAAA"); // 调整 UI 标签为更亮的青色
                string label = $"{new string(' ', ind * 4)}<color={color}>[{ent.major}|{ent.minor}]</color> {ent.go.name}";
                _displayItems.Add(new DisplayItem { ent = ent, label = label });
            }
        }
        if (editorWindow != null) editorWindow.Repaint();
    }

    private Texture2D MakeTex(int w, int h, Color col)
    {
        Color[] pix = new Color[w * h]; for (int i = 0; i < pix.Length; i++) pix[i] = col;
        Texture2D t = new Texture2D(w, h); t.SetPixels(pix); t.Apply(); return t;
    }

    private class DisplayItem { public bool isSep; public SelectionEntry ent; public string label; }
}