From 6c276c75761e1e05a70e11c08a293ec37e1df757 Mon Sep 17 00:00:00 2001
From: DubiousDuck <118256740+DubiousDuck@users.noreply.github.com>
Date: Wed, 15 Jan 2025 12:25:25 -0800
Subject: [PATCH] fix project dialog removing home dir (+style fix)

---
 editor/project_manager/project_dialog.cpp | 37 ++++++++++++++++-------
 editor/project_manager/project_dialog.h   |  2 ++
 2 files changed, 28 insertions(+), 11 deletions(-)

diff --git a/editor/project_manager/project_dialog.cpp b/editor/project_manager/project_dialog.cpp
index 6f2e9da3e0c7..8926fbb7e538 100644
--- a/editor/project_manager/project_dialog.cpp
+++ b/editor/project_manager/project_dialog.cpp
@@ -186,14 +186,7 @@ void ProjectDialog::_validate_path() {
 		return;
 	}
 
-	// TODO: The following 5 lines could be simplified if OS.get_user_home_dir() or SYSTEM_DIR_HOME is implemented. See: https://github.com/godotengine/godot-proposals/issues/4851.
-#ifdef WINDOWS_ENABLED
-	String home_dir = OS::get_singleton()->get_environment("USERPROFILE");
-#else
-	String home_dir = OS::get_singleton()->get_environment("HOME");
-#endif
-	String documents_dir = OS::get_singleton()->get_system_dir(OS::SYSTEM_DIR_DOCUMENTS);
-	if (target_path == home_dir || target_path == documents_dir) {
+	if (_is_path_home_or_doc_dir(target_path)) {
 		_set_message(TTR("You cannot save a project at the selected path. Please create a subfolder or choose a new path."), MESSAGE_ERROR, target_path_input_type);
 		return;
 	}
@@ -324,13 +317,18 @@ void ProjectDialog::_create_dir_toggled(bool p_pressed) {
 	} else {
 		// Strip any trailing slash.
 		target_path = target_path.rstrip("/\\");
+
 		// Save and remove target dir name.
-		if (target_path.get_file() == auto_dir) {
+		if (_is_path_home_or_doc_dir(target_path)) {
 			last_custom_target_dir = "";
 		} else {
-			last_custom_target_dir = target_path.get_file();
+			if (target_path.get_file() == auto_dir) {
+				last_custom_target_dir = "";
+			} else {
+				last_custom_target_dir = target_path.get_file();
+			}
+			target_path = target_path.get_base_dir();
 		}
-		target_path = target_path.get_base_dir();
 	}
 
 	_set_target_path(target_path);
@@ -731,6 +729,23 @@ void ProjectDialog::ok_pressed() {
 	}
 }
 
+bool ProjectDialog::_is_path_home_or_doc_dir(const String &p_path) {
+	String target_path = p_path;
+
+	// TODO: The following 5 lines could be simplified if OS.get_user_home_dir() or SYSTEM_DIR_HOME is implemented. See: https://github.com/godotengine/godot-proposals/issues/4851.
+#ifdef WINDOWS_ENABLED
+	String home_dir = OS::get_singleton()->get_environment("USERPROFILE");
+#else
+	String home_dir = OS::get_singleton()->get_environment("HOME");
+#endif
+	String documents_dir = OS::get_singleton()->get_system_dir(OS::SYSTEM_DIR_DOCUMENTS);
+
+	if (target_path == home_dir || target_path == documents_dir) {
+		return true;
+	}
+	return false;
+}
+
 void ProjectDialog::set_zip_path(const String &p_path) {
 	zip_path = p_path;
 }
diff --git a/editor/project_manager/project_dialog.h b/editor/project_manager/project_dialog.h
index f0e07a05dc65..2ec01d3c2d0e 100644
--- a/editor/project_manager/project_dialog.h
+++ b/editor/project_manager/project_dialog.h
@@ -134,6 +134,8 @@ class ProjectDialog : public ConfirmationDialog {
 
 	void ok_pressed() override;
 
+	bool _is_path_home_or_doc_dir(const String &p_path);
+
 protected:
 	static void _bind_methods();
 	void _notification(int p_what);