CollectAllNetworkPrefabsEditor.cs

namespace Editor.Networking
{
    using System.Collections.Generic;
    using Core.Helper;
    using global::Networking;
    using Unity.Netcode;
    using UnityEditor;
    using UnityEngine;

    [CustomEditor(typeof(CollectAllNetworkPrefabs))]
    public sealed class CollectAllNetworkPrefabsEditor : Editor
    {
        public override void OnInspectorGUI()
        {
            base.OnInspectorGUI();

            NetworkManager networkManager = (target as CollectAllNetworkPrefabs).GetComponent<NetworkManager>();

            if (GUILayout.Button("Update GlobalObjectIds"))
            {
                foreach (GameObject go in UnityHelpers.EnumeratePrefabs())
                {
                    if (go.GetComponentsInChildren<NetworkObject>(true).Length <= 0)
                    {
                        continue;
                    }

                    UnityHelpers.ModifyAndSavePrefab(
                        go, networkGameObject =>
                        {
                            foreach (NetworkObject networkObject in networkGameObject
                                         .GetComponentsInChildren<NetworkObject>(true))
                            {
                                networkObject.GenerateGlobalObjectIdHash();
                            }
                        });
                }
            }

            if (GUILayout.Button("Update All Network Prefabs"))
            {
                UpdateNetworkManager(networkManager);
            }
        }

        public static void UpdateNetworkManager(NetworkManager networkManager)
        {
            UnityHelpers.ModifyAndSavePrefab(
                networkManager.gameObject, networkManagerGameObject =>
                {
                    networkManager = networkManagerGameObject.GetComponent<NetworkManager>();
                    List<NetworkPrefab> networkPrefabs = networkManager.NetworkConfig.NetworkPrefabs;
                    networkPrefabs.Clear();

                    foreach (GameObject go in UnityHelpers.EnumeratePrefabs())
                    {
                        Buffers<NetworkObject>.List.Clear();
                        go.GetComponentsInChildren(true, Buffers<NetworkObject>.List);
                        foreach (NetworkObject networkObject in Buffers<NetworkObject>.List)
                        {
                            NetworkPrefab networkPrefab = new()
                            {
                                Prefab = networkObject.gameObject,
                            };

                            networkPrefabs.Add(networkPrefab);
                        }
                    }

                    networkPrefabs.Sort(
                        (lhs, rhs) =>
                        {
                            int comparison = string.CompareOrdinal(lhs.Prefab.name, rhs.Prefab.name);
                            if (comparison != 0)
                            {
                                return comparison;
                            }

                            string lhsAssetPath = AssetDatabase.GetAssetPath(lhs.Prefab);
                            string rhsAssetPath = AssetDatabase.GetAssetPath(rhs.Prefab);
                            return string.CompareOrdinal(lhsAssetPath, rhsAssetPath);
                        });
                });
        }
    }
}