Adds "Set as Main Scene" option to EditorSceneTabs context menu

Adds the option to quickly set main scene by right-clicking a scene tab. If the scene has no root, then the option is greyed out. If the scene is unsaved, the file is saved and then set as main scene.

Co-Authored-By: Alex Tam <[email protected]>

Author: vaner-org

editor/editor_node.cpp

@@ -2127,6 +2127,12 @@ void EditorNode::save_before_run() {
 	file->set_title(TTR("Save scene before running..."));
 }
 
+void EditorNode::save_then_set_main_scene() {
+	current_menu_option = SAVE_AND_SET_MAIN_SCENE;
+	_menu_option_confirm(SCENE_SAVE_AS_SCENE, true);
+	file->set_title(TTR("Save new main scene..."));
+}
+
 void EditorNode::try_autosave() {
 	if (!bool(EDITOR_GET("run/auto_save/save_before_running"))) {
 		return;
@@ -2295,6 +2301,17 @@ void EditorNode::_dialog_action(String p_file) {
 			}
 		} break;
 
+		case SAVE_AND_SET_MAIN_SCENE: {
+			if (file->get_file_mode() == EditorFileDialog::FILE_MODE_SAVE_FILE) {
+				save_default_environment();
+				_save_scene(p_file);
+			}
+
+			ProjectSettings::get_singleton()->set("application/run/main_scene", ResourceUID::path_to_uid(p_file));
+			ProjectSettings::get_singleton()->save();
+			FileSystemDock::get_singleton()->update_all();
+		} break;
+
 		case FILE_EXPORT_MESH_LIBRARY: {
 			const Dictionary &fd_options = file_export_lib->get_selected_options();
 			bool merge_with_existing_library = fd_options.get(TTR("Merge With Existing"), true);
@@ -3028,6 +3045,17 @@ void EditorNode::_menu_option_confirm(int p_option, bool p_confirmed) {
 
 		} break;
 
+		case SCENE_SET_MAIN_SCENE: {
+			String scene_path = editor_data.get_scene_path(editor_data.get_edited_scene());
+			if (scene_path.is_empty()) {
+				save_then_set_main_scene();
+			} else {
+				ProjectSettings::get_singleton()->set("application/run/main_scene", ResourceUID::path_to_uid(scene_path));
+				ProjectSettings::get_singleton()->save();
+				FileSystemDock::get_singleton()->update_all();
+			}
+		} break;
+
 		case SCENE_SAVE_ALL_SCENES: {
 			_save_all_scenes();
 		} break;

editor/editor_node.h

@@ -146,6 +146,7 @@ class EditorNode : public Node {
 		SCENE_SAVE_SCENE,
 		SCENE_SAVE_AS_SCENE,
 		SCENE_SAVE_ALL_SCENES,
+		SCENE_SET_MAIN_SCENE,
 		SCENE_MULTI_SAVE_AS_SCENE,
 		SCENE_QUICK_OPEN,
 		SCENE_QUICK_OPEN_SCENE,
@@ -213,6 +214,7 @@ class EditorNode : public Node {
 		SCENE_TAB_CLOSE,
 		SAVE_AND_RUN,
 		SAVE_AND_RUN_MAIN_SCENE,
+		SAVE_AND_SET_MAIN_SCENE,
 		RESOURCE_SAVE,
 		RESOURCE_SAVE_AS,
 		SETTINGS_PICK_MAIN_SCENE,
@@ -973,6 +975,7 @@ class EditorNode : public Node {
 	void save_scene_if_open(const String &p_scene_path);
 	void save_scene_list(const HashSet<String> &p_scene_paths);
 	void save_before_run();
+	void save_then_set_main_scene();
 	void try_autosave();
 	void restart_editor(bool p_goto_project_manager = false);
 	void unload_editor_addons();

editor/gui/editor_scene_tabs.cpp

@@ -192,11 +192,15 @@ void EditorSceneTabs::_update_context_menu() {
 	DISABLE_LAST_OPTION_IF(!can_save_all_scenes);
 
 	if (tab_id >= 0) {
+		String scene_path = EditorNode::get_editor_data().get_scene_path(tab_id);
+
 		scene_tabs_context_menu->add_separator();
 		scene_tabs_context_menu->add_item(TTR("Show in FileSystem"), SCENE_SHOW_IN_FILESYSTEM);
-		DISABLE_LAST_OPTION_IF(!ResourceLoader::exists(EditorNode::get_editor_data().get_scene_path(tab_id)));
+		DISABLE_LAST_OPTION_IF(!ResourceLoader::exists(scene_path));
 		scene_tabs_context_menu->add_item(TTR("Play This Scene"), SCENE_RUN);
 		DISABLE_LAST_OPTION_IF(no_root_node);
+		scene_tabs_context_menu->add_item(TTR("Set as Main Scene"), EditorNode::SCENE_SET_MAIN_SCENE);
+		DISABLE_LAST_OPTION_IF(no_root_node || GLOBAL_GET("application/run/main_scene") == ResourceUID::path_to_uid(scene_path));
 
 		scene_tabs_context_menu->add_separator();
 		scene_tabs_context_menu->add_shortcut(ED_GET_SHORTCUT("editor/close_scene"), EditorNode::SCENE_CLOSE);