Mapping tool second iteration, pending code review

Author: danigfedev

Editor/MappingTool/EventSystemAuditor.cs

@@ -9,136 +9,161 @@ public class EventSystemAuditor : EditorWindow
 {
     private Vector2 _scrollPos;
     private List<ScriptableObject> _allProjectEvents = new List<ScriptableObject>();
-    private ScriptableObject _selectedEvent;
-    
-    // Diccionario para guardar: Ruta del Asset -> Lista de detalles (GameObject + Componente)
-    private Dictionary<string, List<UsageDetail>> _usageResults = new Dictionary<string, List<UsageDetail>>();
+    private Dictionary<ScriptableObject, bool> _expansionStates = new Dictionary<ScriptableObject, bool>();
+    private Dictionary<ScriptableObject, Dictionary<string, List<UsageDetail>>> _masterResults = new Dictionary<ScriptableObject, Dictionary<string, List<UsageDetail>>>();
 
     private struct UsageDetail
     {
         public string GameObjectName;
         public string ComponentTypeName;
-        public Object Context; // Para hacer ping al componente exacto
+        public Object Context;
     }
-    
-    [MenuItem("EspidiGames/SO Events/SO Event System Auditor")]
+
+    [MenuItem("Tools/SO Event System Auditor")]
     public static void ShowWindow() => GetWindow<EventSystemAuditor>("Event Auditor");
 
-    private void OnEnable() => RefreshEventList();
+    private void OnEnable() => RefreshAndScanAll();
 
     private void OnGUI()
     {
-        RenderEventSelector();
-        EditorGUILayout.Space(10);
-        RenderUsageResults();
-    }
+        // --- TOOLBAR SUPERIOR ---
+        EditorGUILayout.BeginHorizontal(EditorStyles.toolbar);
+        if (GUILayout.Button("Refrescar y Escanear Proyecto", EditorStyles.toolbarButton)) RefreshAndScanAll();
+        GUILayout.FlexibleSpace();
+        
+        if (GUILayout.Button("Expandir Todo", EditorStyles.toolbarButton)) SetAllExpansion(true);
+        if (GUILayout.Button("Colapsar Todo", EditorStyles.toolbarButton)) SetAllExpansion(false);
+        EditorGUILayout.EndHorizontal();
 
-    private void RenderEventSelector()
-    {
-        GUILayout.Label("1. Selección de Evento (ISOEventBase)", EditorStyles.boldLabel);
-        if (GUILayout.Button("Refrescar Lista de Proyecto")) RefreshEventList();
+        _scrollPos = EditorGUILayout.BeginScrollView(_scrollPos);
+
+        if (_allProjectEvents.Count == 0)
+        {
+            EditorGUILayout.HelpBox("No se encontraron eventos que implementen ISOEventBase.", MessageType.Info);
+        }
 
-        _scrollPos = EditorGUILayout.BeginScrollView(_scrollPos, GUILayout.Height(150));
         foreach (var ev in _allProjectEvents)
         {
-            bool isSelected = (_selectedEvent == ev);
-            GUI.color = isSelected ? Color.cyan : Color.white;
-            if (GUILayout.Button($"{ev.name} ({ev.GetType().Name})", EditorStyles.miniButton))
-            {
-                _selectedEvent = ev;
-                PerformDeepScan(ev);
-            }
+            RenderEventGroup(ev);
         }
-        GUI.color = Color.white;
+
         EditorGUILayout.EndScrollView();
     }
 
-    private void RenderUsageResults()
+    private void RenderEventGroup(ScriptableObject ev)
     {
-        GUILayout.Label("2. Detalle de Referencias Encontradas", EditorStyles.boldLabel);
-        if (_selectedEvent == null) return;
+        if (!_expansionStates.ContainsKey(ev)) _expansionStates[ev] = true;
+        bool expanded = _expansionStates[ev];
 
-        if (_usageResults.Count == 0)
+        // Definimos un estilo que cambie visualmente si está expandido
+        GUIStyle headerStyle = new GUIStyle(EditorStyles.miniButtonMid);
+        headerStyle.alignment = TextAnchor.MiddleLeft;
+        headerStyle.fontStyle = FontStyle.Bold;
+        headerStyle.fontSize = 11;
+        headerStyle.fixedHeight = 25;
+
+        // Feedback visual: Si está expandido, resaltamos el botón
+        if (expanded) GUI.backgroundColor = new Color(0.8f, 0.9f, 1f); 
+
+        EditorGUILayout.BeginVertical(EditorStyles.helpBox);
+        
+        // El icono de flecha (foldout) ayuda a entender que es colapsable
+        string arrow = expanded ? "▼" : "▶";
+        if (GUILayout.Button($" {arrow}  {ev.name.ToUpper()} [{ev.GetType().Name}]", headerStyle))
         {
-            EditorGUILayout.HelpBox("No se han encontrado referencias directas en componentes.", MessageType.Info);
-            return;
+            _expansionStates[ev] = !expanded;
         }
+        GUI.backgroundColor = Color.white;
 
-        foreach (var entry in _usageResults)
+        if (expanded)
         {
-            EditorGUILayout.BeginVertical("helpbox");
-            
-            // Título: Nombre del Prefab o Escena
-            GUILayout.Label(System.IO.Path.GetFileName(entry.Key), EditorStyles.whiteLargeLabel);
-            
-            foreach (var detail in entry.Value)
+            EditorGUILayout.Space(2);
+            if (_masterResults.ContainsKey(ev) && _masterResults[ev].Count > 0)
             {
-                EditorGUILayout.BeginHorizontal();
-                GUILayout.Space(20);
-                if (GUILayout.Button($"GO: {detail.GameObjectName} | Comp: {detail.ComponentTypeName}", EditorStyles.label))
+                foreach (var assetEntry in _masterResults[ev])
                 {
-                    EditorGUIUtility.PingObject(detail.Context);
+                    // Contenedor para cada Prefab/Escena
+                    EditorGUILayout.BeginVertical(EditorStyles.textArea);
+                    GUILayout.Label($"📂 {System.IO.Path.GetFileName(assetEntry.Key)}", EditorStyles.boldLabel);
+                    
+                    foreach (var detail in assetEntry.Value)
+                    {
+                        EditorGUILayout.BeginHorizontal();
+                        GUILayout.Space(15);
+                        // El botón de cada componente para hacer ping
+                        if (GUILayout.Button($"   # GO: {detail.GameObjectName} ({detail.ComponentTypeName})", EditorStyles.label))
+                        {
+                            EditorGUIUtility.PingObject(detail.Context);
+                        }
+                        EditorGUILayout.EndHorizontal();
+                    }
+                    EditorGUILayout.EndVertical();
+                    EditorGUILayout.Space(1);
                 }
-                EditorGUILayout.EndHorizontal();
             }
-            EditorGUILayout.EndVertical();
-            EditorGUILayout.Space(2);
+            else
+            {
+                EditorGUILayout.LabelField("   No se detectaron usos en el proyecto.", EditorStyles.centeredGreyMiniLabel);
+            }
         }
+
+        EditorGUILayout.EndVertical();
+        EditorGUILayout.Space(2);
     }
 
-    private void RefreshEventList()
+    private void SetAllExpansion(bool state)
+    {
+        var keys = _expansionStates.Keys.ToList();
+        foreach (var key in keys) _expansionStates[key] = state;
+    }
+
+    // --- LÓGICA DE ESCANEO (Sin cambios significativos para mantener la funcionalidad) ---
+    private void RefreshAndScanAll()
     {
         _allProjectEvents.Clear();
+        _masterResults.Clear();
+
         string[] guids = AssetDatabase.FindAssets("t:ScriptableObject");
         foreach (var guid in guids)
         {
             string path = AssetDatabase.GUIDToAssetPath(guid);
             var so = AssetDatabase.LoadAssetAtPath<ScriptableObject>(path);
-            if (so is ISOEventBase) _allProjectEvents.Add(so);
+            if (so is ISOEventBase) //
+            {
+                _allProjectEvents.Add(so);
+                _masterResults[so] = new Dictionary<string, List<UsageDetail>>();
+                if (!_expansionStates.ContainsKey(so)) _expansionStates[so] = true; // Expandido por defecto
+            }
         }
-    }
 
-    private void PerformDeepScan(ScriptableObject targetEvent)
-    {
-        _usageResults.Clear();
-        string targetPath = AssetDatabase.GetAssetPath(targetEvent);
         string[] potentialAssets = AssetDatabase.FindAssets("t:Prefab t:Scene");
-
         foreach (var guid in potentialAssets)
         {
-            string path = AssetDatabase.GUIDToAssetPath(guid);
-            string[] deps = AssetDatabase.GetDependencies(path);
-            if (!deps.Contains(targetPath)) continue;
+            string assetPath = AssetDatabase.GUIDToAssetPath(guid);
+            string[] deps = AssetDatabase.GetDependencies(assetPath);
 
-            if (path.EndsWith(".prefab"))
-            {
-                ScanPrefab(path, targetEvent);
-            }
-            else if (path.EndsWith(".unity"))
+            foreach (var ev in _allProjectEvents)
             {
-                ScanScene(path, targetEvent);
+                if (deps.Contains(AssetDatabase.GetAssetPath(ev)))
+                {
+                    if (assetPath.EndsWith(".prefab")) ScanPrefab(assetPath, ev);
+                    else if (assetPath.EndsWith(".unity")) ScanScene(assetPath, ev);
+                }
             }
         }
     }
 
     private void ScanPrefab(string path, ScriptableObject target)
     {
         GameObject root = AssetDatabase.LoadAssetAtPath<GameObject>(path);
-        var components = root.GetComponentsInChildren<Component>(true);
-        CheckComponents(path, components, target);
+        CheckComponents(path, root.GetComponentsInChildren<Component>(true), target);
     }
 
     private void ScanScene(string path, ScriptableObject target)
     {
-        // Nota: Para escenas no abiertas, hay que cargarlas en el editor de forma temporal y silenciosa
-        SceneAsset sceneAsset = AssetDatabase.LoadAssetAtPath<SceneAsset>(path);
         var tempScene = EditorSceneManager.OpenScene(path, OpenSceneMode.Additive);
-        
-        var allComponents = Resources.FindObjectsOfTypeAll<Component>()
-            .Where(c => c.gameObject.scene == tempScene);
-        
+        var allComponents = Resources.FindObjectsOfTypeAll<Component>().Where(c => c.gameObject.scene == tempScene);
         CheckComponents(path, allComponents, target);
-        
         EditorSceneManager.CloseScene(tempScene, true);
     }
 
@@ -147,24 +172,22 @@ private void CheckComponents(string assetPath, IEnumerable<Component> components
         foreach (var comp in components)
         {
             if (comp == null) continue;
-
-            // Usamos SerializedObject para iterar por todas las propiedades del componente
-            // Esto detecta el evento incluso si está en un script personalizado que no sea un EventListener
             SerializedObject so = new SerializedObject(comp);
             SerializedProperty prop = so.GetIterator();
 
             while (prop.NextVisible(true))
             {
                 if (prop.propertyType == SerializedPropertyType.ObjectReference && prop.objectReferenceValue == target)
                 {
-                    if (!_usageResults.ContainsKey(assetPath)) _usageResults[assetPath] = new List<UsageDetail>();
+                    if (!_masterResults[target].ContainsKey(assetPath)) 
+                        _masterResults[target][assetPath] = new List<UsageDetail>();
 
-                    _usageResults[assetPath].Add(new UsageDetail {
+                    _masterResults[target][assetPath].Add(new UsageDetail {
                         GameObjectName = comp.gameObject.name,
                         ComponentTypeName = comp.GetType().Name,
                         Context = comp
                     });
-                    break; 
+                    break;
                 }
             }
         }