Toggle objects on build (#25)

Co-authored-by: 1 <[email protected]>
oneVR · Jan 31, 2024 · f657283 · f657283
1 parent 5908149
commit f657283
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/Editor/DisableOnBuildManager.cs b/Editor/DisableOnBuildManager.cs
@@ -3,10 +3,12 @@
 using UnityEditor.SceneManagement;
 using UnityEngine;
 using UnityEngine.SceneManagement;
+using UnityEditor.Build;
+using UnityEditor.Build.Reporting;
 
 namespace VRWorldToolkit.Editor
 {
-    public class DisableOnBuildCallback : IVRCSDKBuildRequestedCallback
+    public class DisableOnBuildCallback : IVRCSDKBuildRequestedCallback, IProcessSceneWithReport
     {
         public int callbackOrder => 1;
 
@@ -17,6 +19,12 @@ public bool OnBuildRequested(VRCSDKRequestedBuildType requestedBuildType)
 
             return true;
         }
+
+        public void OnProcessScene(Scene scene, BuildReport report)
+        {
+            DisableOnBuildManager.ToggleObjectsUsingTag("DisableOnBuild", false, false);
+            DisableOnBuildManager.ToggleObjectsUsingTag("EnableOnBuild", true, false);
+        }
     }
 
     public class DisableOnBuildManager : UnityEditor.Editor