#if UNITY_2019_4_OR_NEWER
using System;
using System.Collections.Generic;
using UnityEditor;
using UnityEngine;
using UnityEngine.UIElements;
using UnityEditor.UIElements;
using UnityEditor.Networking.PlayerConnection;
using UnityEngine.Networking.PlayerConnection;

namespace YooAsset.Editor
{
    public class AssetBundleDebuggerWindow : EditorWindow
    {
        [MenuItem("YooAsset/AssetBundle Debugger", false, 104)]
        public static void OpenWindow()
        {
            AssetBundleDebuggerWindow wnd = GetWindow<AssetBundleDebuggerWindow>("AssetBundle Debugger", true, WindowsDefine.DockedWindowTypes);
            wnd.minSize = new Vector2(800, 600);
        }

        /// <summary>
        /// 视图模式
        /// </summary>
        private enum EViewMode
        {
            /// <summary>
            /// 内存视图
            /// </summary>
            MemoryView,

            /// <summary>
            /// 资源对象视图
            /// </summary>
            AssetView,

            /// <summary>
            /// 资源包视图
            /// </summary>
            BundleView,

            /// <summary>
            /// 帧对比视图
            /// </summary>
            CompareView,
        }


        private readonly Dictionary<int, RemotePlayerSession> _playerSessions = new Dictionary<int, RemotePlayerSession>();

        private Label _playerName;
        private ToolbarMenu _viewModeMenu;
        private SliderInt _frameSlider;
        private DebuggerAssetListViewer _assetListViewer;
        private DebuggerBundleListViewer _bundleListViewer;
        private DebuggerCompareViewer _compareViewer;

        private EViewMode _viewMode;
        private string _searchKeyWord;
        private DebugReport _currentReport;
        private RemotePlayerSession _currentPlayerSession;
        private int _rangeIndex = 0;

        // 帧对比相关
        private SliderInt _compareFrameSlider1;
        private SliderInt _compareFrameSlider2;
        private VisualElement _compareToolbar;
        private VisualElement _frameToolbar;


        public void CreateGUI()
        {
            try
            {
                VisualElement root = rootVisualElement;

                // 加载布局文件
                var visualAsset = UxmlLoader.LoadWindowUXML<AssetBundleDebuggerWindow>();
                if (visualAsset == null)
                    return;

                visualAsset.CloneTree(root);

                // 采样按钮
                var sampleBtn = root.Q<Button>("SampleButton");
                sampleBtn.clicked += SampleBtn_onClick;

                // 导出按钮
                var exportBtn = root.Q<Button>("ExportButton");
                exportBtn.clicked += ExportBtn_clicked;

                // 用户列表菜单
                _playerName = root.Q<Label>("PlayerName");
                _playerName.text = "Editor player";

                // 视口模式菜单
                _viewModeMenu = root.Q<ToolbarMenu>("ViewModeMenu");
                _viewModeMenu.menu.AppendAction(EViewMode.AssetView.ToString(), OnViewModeMenuChange, OnViewModeMenuStatusUpdate, EViewMode.AssetView);
                _viewModeMenu.menu.AppendAction(EViewMode.BundleView.ToString(), OnViewModeMenuChange, OnViewModeMenuStatusUpdate, EViewMode.BundleView);
                _viewModeMenu.menu.AppendAction(EViewMode.CompareView.ToString(), OnViewModeMenuChange, OnViewModeMenuStatusUpdate, EViewMode.CompareView);
                _viewModeMenu.text = EViewMode.AssetView.ToString();

                // 搜索栏
                var searchField = root.Q<ToolbarSearchField>("SearchField");
                searchField.RegisterValueChangedCallback(OnSearchKeyWordChange);

                // 帧数相关
                {
                    _frameToolbar = root.Q<VisualElement>("FrameToolbar");
                    
                    _frameSlider = root.Q<SliderInt>("FrameSlider");
                    _frameSlider.label = "Frame:";
                    _frameSlider.highValue = 0;
                    _frameSlider.lowValue = 0;
                    _frameSlider.value = 0;
                    _frameSlider.RegisterValueChangedCallback(evt =>
                    {
                        OnFrameSliderChange(evt.newValue);
                    });

                    var frameLast = root.Q<ToolbarButton>("FrameLast");
                    frameLast.clicked += OnFrameLast_clicked;

                    var frameNext = root.Q<ToolbarButton>("FrameNext");
                    frameNext.clicked += OnFrameNext_clicked;

                    var frameClear = root.Q<ToolbarButton>("FrameClear");
                    frameClear.clicked += OnFrameClear_clicked;
                }

                // 帧对比工具栏
                {
                    _compareToolbar = new Toolbar();
                    _compareToolbar.style.display = DisplayStyle.None;
                    
                    var compareLabel = new Label("帧对比: ");
                    compareLabel.style.unityTextAlign = TextAnchor.MiddleLeft;
                    compareLabel.style.marginLeft = 5;
                    _compareToolbar.Add(compareLabel);

                    _compareFrameSlider1 = new SliderInt();
                    _compareFrameSlider1.label = "Frame A:";
                    _compareFrameSlider1.highValue = 0;
                    _compareFrameSlider1.lowValue = 0;
                    _compareFrameSlider1.value = 0;
                    _compareFrameSlider1.style.flexGrow = 1;
                    _compareFrameSlider1.RegisterValueChangedCallback(evt => OnCompareFrameSliderChange());
                    _compareToolbar.Add(_compareFrameSlider1);

                    _compareFrameSlider2 = new SliderInt();
                    _compareFrameSlider2.label = "Frame B:";
                    _compareFrameSlider2.highValue = 0;
                    _compareFrameSlider2.lowValue = 0;
                    _compareFrameSlider2.value = 0;
                    _compareFrameSlider2.style.flexGrow = 1;
                    _compareFrameSlider2.RegisterValueChangedCallback(evt => OnCompareFrameSliderChange());
                    _compareToolbar.Add(_compareFrameSlider2);

                    // 插入到 FrameToolbar 之后
                    int frameToolbarIndex = root.IndexOf(_frameToolbar);
                    root.Insert(frameToolbarIndex + 1, _compareToolbar);
                }

                // 加载视图
                _assetListViewer = new DebuggerAssetListViewer();
                _assetListViewer.InitViewer();

                // 加载视图
                _bundleListViewer = new DebuggerBundleListViewer();
                _bundleListViewer.InitViewer();

                // 加载对比视图
                _compareViewer = new DebuggerCompareViewer();
                _compareViewer.InitViewer();

                // 显示视图
                _viewMode = EViewMode.AssetView;
                _assetListViewer.AttachParent(root);

                // 远程调试
                EditorConnection.instance.Initialize();
                EditorConnection.instance.RegisterConnection(OnHandleConnectionEvent);
                EditorConnection.instance.RegisterDisconnection(OnHandleDisconnectionEvent);
                EditorConnection.instance.Register(RemoteDebuggerDefine.kMsgSendPlayerToEditor, OnHandlePlayerMessage);
                RemoteDebuggerInRuntime.EditorHandleDebugReportCallback = OnHandleDebugReport;
            }
            catch (Exception e)
            {
                Debug.LogError(e.ToString());
            }
        }
        public void OnDestroy()
        {
            // 远程调试
            EditorConnection.instance.UnregisterConnection(OnHandleConnectionEvent);
            EditorConnection.instance.UnregisterDisconnection(OnHandleDisconnectionEvent);
            EditorConnection.instance.Unregister(RemoteDebuggerDefine.kMsgSendPlayerToEditor, OnHandlePlayerMessage);
            _playerSessions.Clear();
        }

        private void OnHandleConnectionEvent(int playerId)
        {
            Debug.Log($"Game player connection : {playerId}");
            _playerName.text = $"Connected player : {playerId}";
        }
        private void OnHandleDisconnectionEvent(int playerId)
        {
            Debug.Log($"Game player disconnection : {playerId}");
            _playerName.text = $"Disconneced player : {playerId}";
        }
        private void OnHandlePlayerMessage(MessageEventArgs args)
        {
            var debugReport = DebugReport.Deserialize(args.data);
            OnHandleDebugReport(args.playerId, debugReport);
        }
        private void OnHandleDebugReport(int playerId, DebugReport debugReport)
        {
            Debug.Log($"Handle player {playerId} debug report !");
            _currentPlayerSession = GetOrCreatePlayerSession(playerId);
            _currentPlayerSession.AddDebugReport(debugReport);
            _frameSlider.highValue = _currentPlayerSession.MaxRangeValue;
            _frameSlider.value = _currentPlayerSession.MaxRangeValue;
            
            // 更新对比帧滑块范围
            _compareFrameSlider1.highValue = _currentPlayerSession.MaxRangeValue;
            _compareFrameSlider2.highValue = _currentPlayerSession.MaxRangeValue;
            
            UpdateFrameView(_currentPlayerSession);
            
            // 如果当前在对比视图，刷新对比数据
            if (_viewMode == EViewMode.CompareView)
            {
                OnCompareFrameSliderChange();
            }
        }
        private void OnFrameSliderChange(int sliderValue)
        {
            if (_currentPlayerSession != null)
            {
                _rangeIndex = _currentPlayerSession.ClampRangeIndex(sliderValue); ;
                UpdateFrameView(_currentPlayerSession, _rangeIndex);
            }
        }
        private void OnFrameLast_clicked()
        {
            if (_currentPlayerSession != null)
            {
                _rangeIndex = _currentPlayerSession.ClampRangeIndex(_rangeIndex - 1);
                _frameSlider.value = _rangeIndex;
                UpdateFrameView(_currentPlayerSession, _rangeIndex);
            }
        }
        private void OnFrameNext_clicked()
        {
            if (_currentPlayerSession != null)
            {
                _rangeIndex = _currentPlayerSession.ClampRangeIndex(_rangeIndex + 1);
                _frameSlider.value = _rangeIndex;
                UpdateFrameView(_currentPlayerSession, _rangeIndex);
            }
        }
        private void OnFrameClear_clicked()
        {
            if (_currentPlayerSession != null)
            {
                _frameSlider.label = $"Frame:";
                _frameSlider.value = 0;
                _frameSlider.lowValue = 0;
                _frameSlider.highValue = 0;
                
                // 重置对比帧滑块
                _compareFrameSlider1.value = 0;
                _compareFrameSlider1.lowValue = 0;
                _compareFrameSlider1.highValue = 0;
                _compareFrameSlider2.value = 0;
                _compareFrameSlider2.lowValue = 0;
                _compareFrameSlider2.highValue = 0;
                
                _currentPlayerSession.ClearDebugReport();
                _assetListViewer.ClearView();
                _bundleListViewer.ClearView();
                _compareViewer.ClearView();
            }
        }

        private RemotePlayerSession GetOrCreatePlayerSession(int playerId)
        {
            if (_playerSessions.TryGetValue(playerId, out RemotePlayerSession session))
            {
                return session;
            }
            else
            {
                RemotePlayerSession newSession = new RemotePlayerSession(playerId);
                _playerSessions.Add(playerId, newSession);
                return newSession;
            }
        }
        private void UpdateFrameView(RemotePlayerSession playerSession)
        {
            if (playerSession != null)
            {
                UpdateFrameView(playerSession, playerSession.MaxRangeValue);
            }
        }
        private void UpdateFrameView(RemotePlayerSession playerSession, int rangeIndex)
        {
            if (playerSession == null)
                return;

            var debugReport = playerSession.GetDebugReport(rangeIndex);
            if (debugReport != null)
            {
                _currentReport = debugReport;
                _frameSlider.label = $"Frame: {debugReport.FrameCount}";
                _assetListViewer.FillViewData(debugReport, _searchKeyWord);
                _bundleListViewer.FillViewData(debugReport, _searchKeyWord);
            }
        }

        private void SampleBtn_onClick()
        {
            // 发送采集数据的命令
            RemoteCommand command = new RemoteCommand();
            command.CommandType = (int)ERemoteCommand.SampleOnce;
            command.CommandParam = string.Empty;
            byte[] data = RemoteCommand.Serialize(command);
            EditorConnection.instance.Send(RemoteDebuggerDefine.kMsgSendEditorToPlayer, data);
            RemoteDebuggerInRuntime.EditorRequestDebugReport();
        }
        private void ExportBtn_clicked()
        {
            if (_currentReport == null)
            {
                Debug.LogWarning("Debug report is null.");
                return;
            }

            string resultPath = EditorTools.OpenFolderPanel("Export JSON", "Assets/");
            if (resultPath != null)
            {
                // 注意：排序保证生成配置的稳定性
                foreach (var packageData in _currentReport.PackageDatas)
                {
                    packageData.ProviderInfos.Sort();
                    foreach (var providerInfo in packageData.ProviderInfos)
                    {
                        providerInfo.DependBundleInfos.Sort();
                    }
                }

                string filePath = $"{resultPath}/{nameof(DebugReport)}_{_currentReport.FrameCount}.json";
                string fileContent = JsonUtility.ToJson(_currentReport, true);
                FileUtility.WriteAllText(filePath, fileContent);
            }
        }
        private void OnSearchKeyWordChange(ChangeEvent<string> e)
        {
            _searchKeyWord = e.newValue;
            if (_currentReport != null)
            {
                _assetListViewer.FillViewData(_currentReport, _searchKeyWord);
                _bundleListViewer.FillViewData(_currentReport, _searchKeyWord);
            }
            _compareViewer.SetSearchKeyWord(_searchKeyWord);
        }
        private void OnViewModeMenuChange(DropdownMenuAction action)
        {
            var viewMode = (EViewMode)action.userData;
            if (_viewMode != viewMode)
            {
                _viewMode = viewMode;
                VisualElement root = this.rootVisualElement;
                _viewModeMenu.text = viewMode.ToString();

                // 先隐藏所有视图
                _assetListViewer.DetachParent();
                _bundleListViewer.DetachParent();
                _compareViewer.DetachParent();

                // 切换工具栏显示
                if (viewMode == EViewMode.CompareView)
                {
                    _frameToolbar.style.display = DisplayStyle.None;
                    _compareToolbar.style.display = DisplayStyle.Flex;
                    _compareViewer.AttachParent(root);
                    
                    // 更新对比帧滑块范围
                    if (_currentPlayerSession != null)
                    {
                        _compareFrameSlider1.highValue = _currentPlayerSession.MaxRangeValue;
                        _compareFrameSlider2.highValue = _currentPlayerSession.MaxRangeValue;
                    }
                }
                else
                {
                    _frameToolbar.style.display = DisplayStyle.Flex;
                    _compareToolbar.style.display = DisplayStyle.None;
                    
                    if (viewMode == EViewMode.AssetView)
                    {
                        _assetListViewer.AttachParent(root);
                    }
                    else if (viewMode == EViewMode.BundleView)
                    {
                        _bundleListViewer.AttachParent(root);
                    }
                }
            }
        }
        private DropdownMenuAction.Status OnViewModeMenuStatusUpdate(DropdownMenuAction action)
        {
            var viewMode = (EViewMode)action.userData;
            if (_viewMode == viewMode)
                return DropdownMenuAction.Status.Checked;
            else
                return DropdownMenuAction.Status.Normal;
        }
        
        /// <summary>
        /// 帧对比滑块变化事件
        /// </summary>
        private void OnCompareFrameSliderChange()
        {
            if (_currentPlayerSession == null)
                return;

            int frameIndex1 = _currentPlayerSession.ClampRangeIndex(_compareFrameSlider1.value);
            int frameIndex2 = _currentPlayerSession.ClampRangeIndex(_compareFrameSlider2.value);

            var report1 = _currentPlayerSession.GetDebugReport(frameIndex1);
            var report2 = _currentPlayerSession.GetDebugReport(frameIndex2);

            _compareViewer.SetCompareFrames(report1, report2);
        }
    }
}
#endif