Unity 编辑器工具 - 资源引用查找器

news2024/11/18 13:48:15

在Unity项目开发过程中，管理和维护资源之间的引用关系是至关重要的。当然我们项目也是需要这个功能毕竟项目大了之后查找资源引用还是交给 资源引用查找器 比较好。

功能概述

资源引用查找器允许开发者选择一个目标资源，并在整个项目中查找引用了该资源的其他资源。其主要功能包括：

在Unity编辑器菜单栏下添加一个名为"Resource Reference Finder"的菜单项。
提供一个可调整大小的窗口，用于选择目标资源和显示引用结果。
通过点击"Search"按钮来触发搜索过程，查找所有引用了目标资源的Prefab文件，并将结果展示在窗口中。

完整代码

using UnityEditor;
using UnityEngine;
using System.Collections.Generic;
using System.IO;

public class ResourceReferenceFinder : EditorWindow
{
    [MenuItem("Assets/Resource Reference Finder")]
    static void SearchReference()
    {
        GetWindow<ResourceReferenceFinder>("Resource Reference Finder");
    }

    private static Object targetResource;
    private List<Object> referencingAssets = new List<Object>();
    private Vector2 scrollPosition;

    private void OnGUI()
    {
        // 显示资源选择字段和搜索按钮
        EditorGUILayout.BeginHorizontal();
        GUILayout.Label("Select Target Resource:", GUILayout.Width(150));
        targetResource = EditorGUILayout.ObjectField(targetResource, typeof(Object), true, GUILayout.Width(200));
        if (GUILayout.Button("Search", GUILayout.Width(100)))
        {
            ReferenceFinder();
        }
        EditorGUILayout.EndHorizontal();

        // 滚动视图开始
        scrollPosition = EditorGUILayout.BeginScrollView(scrollPosition);
        EditorGUILayout.BeginVertical();

        // 显示搜索结果
        for (int i = 0; i < referencingAssets.Count; i++)
        {
            EditorGUILayout.ObjectField(referencingAssets[i], typeof(Object), true, GUILayout.Width(300));
        }

        EditorGUILayout.EndVertical();
        EditorGUILayout.EndScrollView();
        // 滚动视图结束
    }

    // 查找引用
    private void ReferenceFinder()
    {
        referencingAssets.Clear();

        // 检查是否选择了资源
        if (targetResource == null)
        {
            Debug.LogWarning("Please select a resource to search for references.");
            return;
        }

        // 获取选择资源的 GUID
        string assetPath = AssetDatabase.GetAssetPath(targetResource);
        string assetGuid = AssetDatabase.AssetPathToGUID(assetPath);

        // 遍历项目中所有 Prefab 文件
        string[] guids = AssetDatabase.FindAssets("t:Prefab", new[] { "Assets" });
        int length = guids.Length;
        for (int i = 0; i < length; i++)
        {
            string filePath = AssetDatabase.GUIDToAssetPath(guids[i]);
            EditorUtility.DisplayCancelableProgressBar("Checking", filePath, (float)i / length);

            // 读取 Prefab 文件内容，检查是否包含选择资源的 GUID
            string content = File.ReadAllText(filePath);
            if (content.Contains(assetGuid))
            {
                // 如果包含，将该 Prefab 添加到结果列表中
                Object referencingAsset = AssetDatabase.LoadAssetAtPath(filePath, typeof(Object));
                referencingAssets.Add(referencingAsset);
            }
        }

        // 清除进度条
        EditorUtility.ClearProgressBar();
    }
}