From 31156433977453372890f5f98f76c74239d35dff Mon Sep 17 00:00:00 2001
From: Wojtek Figat <wojtek@figat.pl>
Date: Tue, 10 Aug 2021 16:32:00 +0200
Subject: [PATCH] Add option to Scene context menu to unload it

---
 Source/Editor/SceneGraph/Actors/SceneNode.cs  | 29 ++++++++++++++-----
 .../Windows/SceneTreeWindow.ContextMenu.cs    | 11 ++++---
 2 files changed, 27 insertions(+), 13 deletions(-)

diff --git a/Source/Editor/SceneGraph/Actors/SceneNode.cs b/Source/Editor/SceneGraph/Actors/SceneNode.cs
index 35c71bf02..2846715fc 100644
--- a/Source/Editor/SceneGraph/Actors/SceneNode.cs
+++ b/Source/Editor/SceneGraph/Actors/SceneNode.cs
@@ -1,5 +1,7 @@
 // Copyright (c) 2012-2021 Wojciech Figat. All rights reserved.
 
+using System.Net;
+using FlaxEditor.GUI.ContextMenu;
 using FlaxEditor.SceneGraph.GUI;
 using FlaxEngine;
 
@@ -17,9 +19,6 @@ namespace FlaxEditor.SceneGraph.Actors
         /// <summary>
         /// Gets or sets a value indicating whether this scene is edited.
         /// </summary>
-        /// <value>
-        ///   <c>true</c> if this scene is edited; otherwise, <c>false</c>.
-        /// </value>
         public bool IsEdited
         {
             get => _isEdited;
@@ -28,7 +27,6 @@ namespace FlaxEditor.SceneGraph.Actors
                 if (_isEdited != value)
                 {
                     _isEdited = value;
-
                     _treeNode.UpdateText();
                 }
             }
@@ -37,9 +35,6 @@ namespace FlaxEditor.SceneGraph.Actors
         /// <summary>
         /// Gets the scene.
         /// </summary>
-        /// <value>
-        /// The scene.
-        /// </value>
         public Scene Scene => _actor as Scene;
 
         /// <summary>
@@ -68,5 +63,25 @@ namespace FlaxEditor.SceneGraph.Actors
 
         /// <inheritdoc />
         public override SceneNode ParentScene => this;
+
+        /// <inheritdoc />
+        public override void OnContextMenu(ContextMenu contextMenu)
+        {
+            contextMenu.AddSeparator();
+            contextMenu.AddButton("Save scene", OnSave).LinkTooltip("Saves this scene.").Enabled = IsEdited && !Editor.IsPlayMode;
+            contextMenu.AddButton("Unload scene", OnUnload).LinkTooltip("Unloads this scene.").Enabled = Editor.Instance.StateMachine.CurrentState.CanChangeScene;
+
+            base.OnContextMenu(contextMenu);
+        }
+
+        private void OnSave()
+        {
+            Editor.Instance.Scene.SaveScene(this);
+        }
+
+        private void OnUnload()
+        {
+            Editor.Instance.Scene.CloseScene(Scene);
+        }
     }
 }
diff --git a/Source/Editor/Windows/SceneTreeWindow.ContextMenu.cs b/Source/Editor/Windows/SceneTreeWindow.ContextMenu.cs
index 50187ed3f..fe88a13a1 100644
--- a/Source/Editor/Windows/SceneTreeWindow.ContextMenu.cs
+++ b/Source/Editor/Windows/SceneTreeWindow.ContextMenu.cs
@@ -104,12 +104,11 @@ namespace FlaxEditor.Windows
                         Editor.Windows.ContentWin.CurrentViewFolder.CanHaveAssets;
 
             bool hasPrefabLink = canEditScene && isSingleActorSelected && (Editor.SceneEditing.Selection[0] as ActorNode).HasPrefabLink;
-
-            b = contextMenu.AddButton("Select Prefab", Editor.Prefabs.SelectPrefab);
-            b.Enabled = hasPrefabLink;
-
-            b = contextMenu.AddButton("Break Prefab Link", Editor.Prefabs.BreakLinks);
-            b.Enabled = hasPrefabLink;
+            if (hasPrefabLink)
+            {
+                contextMenu.AddButton("Select Prefab", Editor.Prefabs.SelectPrefab);
+                contextMenu.AddButton("Break Prefab Link", Editor.Prefabs.BreakLinks);
+            }
 
             // Spawning actors options