Client/GameClient/Assets/YooAsset/Editor/AssetBundleDebugger/AssetBundleDebuggerWindow.cs

#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>("资源包调试工具", true, WindowsDefine.DockedWindowTypes);
			wnd.minSize = new Vector2(800, 600);
		}

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

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

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


		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 EViewMode _viewMode;
		private string _searchKeyWord;
		private DebugReport _currentReport;
		private RemotePlayerSession _currentPlayerSession;
		private int _rangeIndex = 0;


		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.text = EViewMode.AssetView.ToString();

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

				// 帧数相关
				{
					_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;
				}

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

				// 加载视图
				_bundleListViewer = new DebuggerBundleListViewer();
				_bundleListViewer.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;
			UpdateFrameView(_currentPlayerSession);
		}
		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;
				_currentPlayerSession.ClearDebugReport();
				_assetListViewer.ClearView();
				_bundleListViewer.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);
			}
		}
		private void OnViewModeMenuChange(DropdownMenuAction action)
		{
			var viewMode = (EViewMode)action.userData;
			if (_viewMode != viewMode)
			{
				_viewMode = viewMode;
				VisualElement root = this.rootVisualElement;
				_viewModeMenu.text = viewMode.ToString();

				if (viewMode == EViewMode.AssetView)
				{
					_assetListViewer.AttachParent(root);
					_bundleListViewer.DetachParent();
				}
				else if (viewMode == EViewMode.BundleView)
				{
					_assetListViewer.DetachParent();
					_bundleListViewer.AttachParent(root);
				}
				else
				{
					throw new NotImplementedException(viewMode.ToString());
				}
			}
		}
		private DropdownMenuAction.Status OnViewModeMenuStatusUpdate(DropdownMenuAction action)
		{
			var viewMode = (EViewMode)action.userData;
			if (_viewMode == viewMode)
				return DropdownMenuAction.Status.Checked;
			else
				return DropdownMenuAction.Status.Normal;
		}
	}
}
#endif