// // /*===============================================================================
// // Copyright (C) 2025 PhantomsXR Ltd. All Rights Reserved.
// //
// // This file is part of the Phantom.XRMOD.PackageTools.Editor.
// //
// // The XR-MOD cannot be copied, distributed, or made available to
// // third-parties for commercial purposes without written permission of PhantomsXR Ltd.
// //
// // Contact nswell@phantomsxr.com for licensing requests.
// // ===============================================================================*/

using System;
using System.Collections.Generic;
using System.IO;
using System.Reflection;
using UnityEditor;
using UnityEngine;
using UnityEngine.UIElements;
using UnityEngine.Video;
using Object = UnityEngine.Object;

namespace Phantom.XRMOD.XRMODPackageTools.Editor
{
    public class ExperienceInspectorWindow : EditorWindow
    {
        static ExperienceInspectorWindow _WINDOW;
        private const string _CONST_PACKAGE_PATH = "Packages/com.phantomsxr.xrmodpackagetools";
        private static string _UXML_AND_USS_PATH = $"{_CONST_PACKAGE_PATH}/Editor/ExperienceInspector";
        private const string _CONST_FILE_NAME = "ExperienceInspectorWindow";
        private ScrollView _contentScrollView;
        private Button _resetButton;
        private ListView _assetListView;
        private List<string> _assetPathList = new();
        private ScrollView _assetPreviewView;
        private VisualElement _splitView;
        private PreviewRenderUtility _previewUtility;
        private Label _previewHintLabel;

        private GameObject _previewInstance;
        private Vector2 _previewRotation = new Vector2(120, -20);
        private Vector2 _previewDragStart;
        private bool _isDraggingPreview;
        private bool isPlaying = false;

        private List<Object> experienceObjects = new();
        private static AssetBundle _currentBundle;

        [MenuItem("Tools/XR-MOD/Tools/Experience Inspector Window")]
        private static void GetWindow()
        {
            if (_WINDOW == null)
            {
                _WINDOW = (ExperienceInspectorWindow) EditorWindow.GetWindow(typeof(ExperienceInspectorWindow));
            }

            _WINDOW.Show();
        }

        public void CreateGUI()
        {
            // 加载 UXML
            var visualTree =
                AssetDatabase.LoadAssetAtPath<VisualTreeAsset>($"{_UXML_AND_USS_PATH}/{_CONST_FILE_NAME}.uxml");
            var styleSheet =
                AssetDatabase.LoadAssetAtPath<StyleSheet>($"{_UXML_AND_USS_PATH}/{_CONST_FILE_NAME}.uss");


            VisualElement root = visualTree.CloneTree();
            root.style.flexGrow = 1;
            root.styleSheets.Add(styleSheet);
            rootVisualElement.Add(root);

            _assetListView = root.Q<ListView>("AssetListView");
            _assetListView.makeItem = () =>
            {
                var label = new Label();
                label.style.unityTextAlign = TextAnchor.MiddleLeft;
                label.style.marginBottom = 2;
                return label;
            };
            _assetListView.bindItem = (element, i) => { (element as Label).text = _assetPathList[i]; };
            _assetListView.selectionType = SelectionType.Single;
            _assetListView.onSelectionChange += objects =>
            {
                foreach (var obj in objects) // 只取第一个
                {
                    if (obj is string path)
                    {
                        ShowAssetPreview(_currentBundle, path);
                    }
                }
            };

            _assetPreviewView = root.Q<ScrollView>("AssetPreviewView");
            _splitView = root.Q<TwoPaneSplitView>("SplitView");
            _splitView.style.display = DisplayStyle.None;

            _previewHintLabel = new Label("Select an object to preview")
            {
                style =
                {
                    unityTextAlign = TextAnchor.MiddleCenter,
                    marginTop = 40,
                    marginBottom = 20,
                    fontSize = 14,
                    color = Color.gray,
                    alignSelf = Align.Center
                }
            };
            _assetPreviewView.Add(_previewHintLabel);

            _resetButton = root.Q<Button>("ResetButton");
            _resetButton.clicked += () =>
            {
                AssetBundle.UnloadAllAssetBundles(true);
                _previewHintLabel.style.display = DisplayStyle.Flex;
                _assetListView.itemsSource = null;
                _assetPathList.Clear();
                _resetButton.style.display = DisplayStyle.None;
                _splitView.style.display = DisplayStyle.None;

                var dropArea = root.Q<VisualElement>("DropArea");
                dropArea.style.display = DisplayStyle.Flex;
            };


            // 设置拖拽回调
            var dropArea = root.Q<VisualElement>("DropArea");
            _contentScrollView = root.Q<ScrollView>("ContentView");
            dropArea.pickingMode = PickingMode.Position;
            dropArea.RegisterCallback<DragUpdatedEvent>(evt =>
            {
                DragAndDrop.visualMode = DragAndDropVisualMode.Copy;
                evt.StopPropagation();
            });
            dropArea.RegisterCallback<DragPerformEvent>(evt =>
            {
                DragAndDrop.AcceptDrag();
                foreach (var path in DragAndDrop.paths)
                {
                    if (!string.IsNullOrEmpty(path) && File.Exists(path) &&
                        Path.GetFileName(path).ToLower().EndsWith("arexperience"))
                    {
                        LoadAssetBundle(path);
                    }
                }

                evt.StopPropagation();
            });
        }

        protected void OnDisable()
        {
            CleanupPreview();
        }

        private void LoadAssetBundle(string path)
        {
            _assetListView.Clear();
            _assetPreviewView.Clear();
            _previewHintLabel.style.display = DisplayStyle.Flex;

            AssetBundle.UnloadAllAssetBundles(true);

            var dropArea = rootVisualElement.Q<VisualElement>("DropArea");
            dropArea.style.display = DisplayStyle.None;
            _resetButton.style.display = DisplayStyle.Flex;
            _splitView.style.display = DisplayStyle.Flex;

            var bundle = AssetBundle.LoadFromFile(path);
            if (bundle == null)
            {
                Debug.LogError("Load experience package failed!");
                return;
            }

            _assetPathList.Clear();
            _assetPreviewView.Clear();

            var allAssets = bundle.GetAllAssetNames();
            _assetPathList.AddRange(allAssets);
            _assetListView.itemsSource = _assetPathList;
            _assetListView.Rebuild(); // 强制刷新显示

            _currentBundle?.Unload(false);
            _currentBundle = bundle;
        }


        private void ShowAssetPreview(AssetBundle bundle, string assetPath)
        {
            _previewHintLabel.style.display = DisplayStyle.None;
            _assetPreviewView.Clear();
            CleanupPreview();

            var asset = bundle.LoadAsset(assetPath);

            switch (asset)
            {
                case AudioClip audioClip:
                {
                    var btnContainer = new VisualElement
                    {
                        style =
                        {
                            flexDirection = FlexDirection.Row,
                            justifyContent = Justify.Center,
                            marginBottom = 10,
                        }
                    };
                    var playButton = new Button(() =>
                    {
                        // 只能在Editor下播放一次
                        Type audioUtilType = typeof(AudioImporter).Assembly.GetType("UnityEditor.AudioUtil");
                        var _playClipMethod = audioUtilType.GetMethod("PlayClip",
                                                  BindingFlags.Static | BindingFlags.Public, null,
                                                  new Type[] {typeof(AudioClip)}, null)
                                              ?? audioUtilType.GetMethod("PlayPreviewClip",
                                                  BindingFlags.Static | BindingFlags.Public, null,
                                                  new Type[] {typeof(AudioClip), typeof(int), typeof(bool)}, null);
                        if (!isPlaying)
                        {
                            _playClipMethod.Invoke(null, new object[] {audioClip, 0, false});
                            isPlaying = true;
                        }
                    })
                    {
                        text = $"▶ Play Audio: {audioClip.name}"
                    };
                    var stopButton = new Button(() =>
                    {
                        Type audioUtilType = typeof(AudioImporter).Assembly.GetType("UnityEditor.AudioUtil");
                        var stopMethod = audioUtilType.GetMethod("StopPreviewClip",
                                             BindingFlags.Static | BindingFlags.Public, null,
                                             new Type[] {typeof(AudioClip)}, null)
                                         ?? audioUtilType.GetMethod("StopAllPreviewClips",
                                             BindingFlags.Static | BindingFlags.Public);
                        stopMethod?.Invoke(null, null);
                        isPlaying = false;
                    })
                    {
                        text = "■ Stop Audio"
                    };


                    playButton.style.marginBottom = 10;
                    playButton.style.alignSelf = Align.Center;
                    stopButton.style.marginBottom = 10;
                    stopButton.style.alignSelf = Align.Center;
                    btnContainer.Add(playButton);
                    btnContainer.Add(stopButton);
                    _assetPreviewView.Add(btnContainer);
                    break;
                }
                case Material mat:
                {
                    _previewUtility = new PreviewRenderUtility();
                    _previewUtility.cameraFieldOfView = 30f;

                    var sphere = GameObject.CreatePrimitive(PrimitiveType.Sphere);
                    sphere.GetComponent<Renderer>().sharedMaterial = mat;

                    _previewInstance = sphere;
                    SetupPreviewScene(_previewInstance);

                    var previewElement = new IMGUIContainer(() =>
                    {
                        float fallbackWidth = EditorGUIUtility.currentViewWidth * 0.5f;
                        float width = _assetPreviewView.resolvedStyle.width;
                        if (width < 1) width = fallbackWidth;

                        float height = Mathf.Min(400, width * 0.75f);
                        Rect r = GUILayoutUtility.GetRect(width, height, GUILayout.ExpandWidth(true));

                        Event evt = Event.current;
                        if (evt.type == EventType.MouseDrag && r.Contains(evt.mousePosition))
                        {
                            _previewRotation += evt.delta;
                            evt.Use();
                            _assetPreviewView.MarkDirtyRepaint();
                        }

                        _previewUtility.BeginPreview(r, GUIStyle.none);

                        _previewUtility.camera.transform.position = Vector3.zero;
                        _previewUtility.camera.nearClipPlane = 0.001f;
                        _previewUtility.camera.farClipPlane = 1000f;
                        Quaternion rot = Quaternion.Euler(_previewRotation.y, -_previewRotation.x, 0);
                        Vector3 dir = rot * Vector3.forward;
                        _previewUtility.camera.transform.position = -dir * 5f;
                        _previewUtility.camera.transform.LookAt(Vector3.zero);

                        _previewUtility.Render(true);
                        Texture tex = _previewUtility.EndPreview();
                        if (tex != null)
                            GUI.DrawTexture(r, tex, ScaleMode.ScaleToFit, false);
                    });

                    previewElement.style.flexGrow = 1;
                    previewElement.style.minHeight = 200;
                    previewElement.style.marginBottom = 10;
                    _assetPreviewView.Add(previewElement);
                    break;
                }

                case VideoClip videoClip:
                {
                    var label = new Label($"🎞 VideoClip: {videoClip.name}")
                    {
                        style =
                        {
                            color = Color.white,
                            unityTextAlign = TextAnchor.MiddleCenter,
                            marginBottom = 10,
                            alignSelf = Align.Center
                        }
                    };
                    _assetPreviewView.Add(label);
                    break;
                }
                case Texture2D texture:
                {
                    var image = new Image
                    {
                        image = texture,
                        scaleMode = ScaleMode.ScaleToFit
                    };
                    image.style.height = 300;
                    image.style.marginBottom = 10;
                    image.style.unityTextAlign = TextAnchor.MiddleCenter;
                    image.style.alignSelf = Align.Center;
                    _assetPreviewView.Add(image);
                    break;
                }
                case TextAsset textAsset:
                {
                    var textElement = new Label(textAsset.text)
                    {
                        style =
                        {
                            whiteSpace = WhiteSpace.Normal,
                            unityFontStyleAndWeight = FontStyle.Normal,
                            marginBottom = 10
                        }
                    };
                    textElement.style.color = Color.white;
                    _assetPreviewView.Add(textElement);
                    break;
                }
                case GameObject go:
                {
                    _previewUtility = new PreviewRenderUtility();
                    _previewUtility.cameraFieldOfView = 30f;

                    _previewInstance = (GameObject) GameObject.Instantiate(go);
                    SetupPreviewScene(_previewInstance);

                    var previewElement = new IMGUIContainer(() =>
                    {
                        float fallbackWidth = EditorGUIUtility.currentViewWidth * 0.5f;
                        float width = _assetPreviewView.resolvedStyle.width;
                        if (width < 1) width = fallbackWidth;

                        float height = _assetPreviewView.resolvedStyle.height;
                        Rect r = GUILayoutUtility.GetRect(width, height, GUILayout.ExpandWidth(true));


                        Event evt = Event.current;
                        if (evt.type == EventType.MouseDrag && r.Contains(evt.mousePosition))
                        {
                            _previewRotation += evt.delta;
                            evt.Use();
                            _assetPreviewView.MarkDirtyRepaint(); // 强制重绘
                        }

                        _previewUtility.BeginPreview(r, GUIStyle.none);

                        _previewUtility.camera.transform.position = Vector3.zero;
                        _previewUtility.camera.nearClipPlane = 0.001f;
                        _previewUtility.camera.farClipPlane = 1000f;
                        Quaternion rot = Quaternion.Euler(_previewRotation.y, -_previewRotation.x, 0);
                        Vector3 dir = rot * Vector3.forward;
                        _previewUtility.camera.transform.position = -dir * 5f;
                        _previewUtility.camera.transform.LookAt(Vector3.zero);

                        _previewUtility.Render(true);
                        var tex = _previewUtility.EndPreview();
                        GUI.DrawTexture(r, tex, ScaleMode.ScaleToFit, false);
                    });


                    previewElement.style.flexGrow = 1;
                    previewElement.style.minHeight = 200;
                    _assetPreviewView.Add(previewElement);
                    break;
                }
                default:
                {
                    var label = new Label($"[Unknow Type] {asset.name}");
                    label.style.color = Color.gray;
                    label.style.marginBottom = 6;
                    _assetPreviewView.Add(label);
                    break;
                }
            }
        }

        private void SetupPreviewScene(GameObject previewGO)
        {
            _previewUtility.AddSingleGO(previewGO);
            _previewUtility.camera.transform.position = new Vector3(0, 0, -5);
            _previewUtility.camera.transform.LookAt(Vector3.zero);
            _previewUtility.lights[0].intensity = 1.2f;
            _previewUtility.lights[0].transform.rotation = Quaternion.Euler(50f, 50f, 0f);
            _previewUtility.lights[1].intensity = 0.5f;
        }


        private void CleanupPreview()
        {
            if (_previewInstance != null)
            {
                DestroyImmediate(_previewInstance);
                _previewInstance = null;
            }

            if (_previewUtility != null)
            {
                _previewUtility.Cleanup();
                _previewUtility = null;
            }
        }
    }
}