Added options to disable stripping folders from prefabs in play mode or build

Author: SolidAlloy

Editor/Settings/SettingsDrawer.cs

@@ -15,21 +15,21 @@ internal static class SettingsDrawer
         {
             { nameof(StripSettings.PlayMode), "Play Mode Stripping Type" },
             { nameof(StripSettings.Build), "Build Stripping Type" },
-            { nameof(StripSettings.CapitalizeName), "Capitalize Folder Names" }
+            { nameof(StripSettings.CapitalizeName), "Capitalize Folder Names" },
+            { nameof(StripSettings.StripFoldersFromPrefabsInPlayMode), "Strip folders from prefabs in Play Mode" },
+            { nameof(StripSettings.StripFoldersFromPrefabsInBuild), "Strip folders from prefabs in build" },
         };
 
         private static readonly GUIContent _buildStrippingName = new GUIContent(_fieldNames[nameof(StripSettings.Build)]);
 
         [SettingsProvider]
         public static SettingsProvider CreateSettingsProvider()
         {
-            var provider = new SettingsProvider("Preferences/Hierarchy Folders", SettingsScope.User)
+            return new SettingsProvider("Preferences/Hierarchy Folders", SettingsScope.User)
             {
                 guiHandler = OnGUI,
                 keywords = GetKeywords()
             };
-
-            return provider;
         }
 
         private static void OnGUI(string searchContext)
@@ -45,6 +45,22 @@ private static void OnGUI(string searchContext)
 
             StripSettings.Build = (StrippingMode) EditorGUILayout.EnumPopup(
                 _buildStrippingName, StripSettings.Build, TypeCanBeInBuild, true);
+
+            EditorGUILayout.Space(EditorGUIUtility.singleLineHeight);
+
+            EditorGUILayout.HelpBox(
+                "If you notice that entering play mode takes too long, you can try disabling this option. " +
+                "Folders will not be stripped from prefabs that are instantiated at runtime, but if performance in " +
+                "Play Mode does not matter, you will be fine.", MessageType.Warning);
+
+            using (new TemporaryLabelWidth(230f))
+            {
+                StripSettings.StripFoldersFromPrefabsInPlayMode =
+                    EditorGUILayout.Toggle(_fieldNames[nameof(StripSettings.StripFoldersFromPrefabsInPlayMode)], StripSettings.StripFoldersFromPrefabsInPlayMode);
+
+                StripSettings.StripFoldersFromPrefabsInBuild =
+                    EditorGUILayout.Toggle(_fieldNames[nameof(StripSettings.StripFoldersFromPrefabsInBuild)], StripSettings.StripFoldersFromPrefabsInBuild);
+            }
         }
 
         private static HashSet<string> GetKeywords()
@@ -72,5 +88,24 @@ private static bool TypeCanBeInBuild(Enum enumValue)
             var mode = (StrippingMode) enumValue;
             return mode == StrippingMode.PrependWithFolderName || mode == StrippingMode.Delete;
         }
+
+        /// <summary>
+        /// Temporarily sets <see cref="EditorGUIUtility.labelWidth"/> to a certain value, than reverts it.
+        /// </summary>
+        private readonly struct TemporaryLabelWidth : IDisposable
+        {
+            private readonly float _oldWidth;
+
+            public TemporaryLabelWidth(float width)
+            {
+                _oldWidth = EditorGUIUtility.labelWidth;
+                EditorGUIUtility.labelWidth = width;
+            }
+
+            public void Dispose()
+            {
+                EditorGUIUtility.labelWidth = _oldWidth;
+            }
+        }
     }
 }

Editor/Settings/StripSettings.cs

@@ -12,16 +12,14 @@ internal static class StripSettings
         private static UserSetting<StrippingMode> _playModeSetting;
         private static UserSetting<StrippingMode> _buildSetting;
         private static UserSetting<bool> _capitalizeName;
+        private static UserSetting<bool> _stripFoldersFromPrefabsInPlayMode;
+        private static UserSetting<bool> _stripFoldersFromPrefabsInBuild;
 
         public static StrippingMode PlayMode
         {
             get
             {
-                if (_playModeSetting == null)
-                {
-                    Initialize();
-                }
-
+                InitializeIfNeeded();
                 return _playModeSetting.value;
             }
 
@@ -32,11 +30,7 @@ public static StrippingMode Build
         {
             get
             {
-                if (_buildSetting == null)
-                {
-                    Initialize();
-                }
-
+                InitializeIfNeeded();
                 return _buildSetting.value;
             }
 
@@ -47,19 +41,40 @@ public static bool CapitalizeName
         {
             get
             {
-                if (_capitalizeName == null)
-                {
-                    Initialize();
-                }
-
+                InitializeIfNeeded();
                 return _capitalizeName.value;
             }
 
             set => _capitalizeName.value = value;
         }
 
-        private static void Initialize()
+        public static bool StripFoldersFromPrefabsInPlayMode
+        {
+            get
+            {
+                InitializeIfNeeded();
+                return _stripFoldersFromPrefabsInPlayMode.value;
+            }
+
+            set => _stripFoldersFromPrefabsInPlayMode.value = value;
+        }
+
+        public static bool StripFoldersFromPrefabsInBuild
         {
+            get
+            {
+                InitializeIfNeeded();
+                return _stripFoldersFromPrefabsInBuild.value;
+            }
+
+            set => _stripFoldersFromPrefabsInBuild.value = value;
+        }
+
+        private static void InitializeIfNeeded()
+        {
+            if (_instance != null)
+                return;
+
             _instance = new Settings(PackageName);
 
             _playModeSetting = new UserSetting<StrippingMode>(_instance, nameof(_playModeSetting),
@@ -69,6 +84,12 @@ private static void Initialize()
                 StrippingMode.PrependWithFolderName, SettingsScope.User);
 
             _capitalizeName = new UserSetting<bool>(_instance, nameof(_capitalizeName), true, SettingsScope.User);
+
+            _stripFoldersFromPrefabsInPlayMode = new UserSetting<bool>(_instance,
+                nameof(_stripFoldersFromPrefabsInPlayMode), true, SettingsScope.User);
+
+            _stripFoldersFromPrefabsInBuild = new UserSetting<bool>(_instance,
+                nameof(_stripFoldersFromPrefabsInPlayMode), true, SettingsScope.User);
         }
     }
 }