From edd8668695a2a27ea85ba3f46b1406d9c8df01d2 Mon Sep 17 00:00:00 2001
From: Saracen <SaracenOne@gmail.com>
Date: Fri, 29 Sep 2023 14:58:13 +0100
Subject: [PATCH] Script path type support in editor: Allow script path type
 hints to be used in drag and drop and scene tree popup

---
 editor/editor_properties.cpp     |  8 +++++
 editor/gui/scene_tree_editor.cpp | 56 ++++++++++++++++++++++++++++++--
 2 files changed, 62 insertions(+), 2 deletions(-)

diff --git a/editor/editor_properties.cpp b/editor/editor_properties.cpp
index a197eb9dced8f0..6ea903d5575c9d 100644
--- a/editor/editor_properties.cpp
+++ b/editor/editor_properties.cpp
@@ -2858,6 +2858,14 @@ bool EditorPropertyNodePath::is_drop_valid(const Dictionary &p_drag_data) const
 		if (dropped_node->is_class(E) ||
 				EditorNode::get_singleton()->is_object_of_custom_type(dropped_node, E)) {
 			return true;
+		} else {
+			Ref<Script> dropped_node_script = dropped_node->get_script();
+			while (dropped_node_script.is_valid()) {
+				if (dropped_node_script->get_path() == E) {
+					return true;
+				}
+				dropped_node_script = dropped_node_script->get_base_script();
+			}
 		}
 	}
 
diff --git a/editor/gui/scene_tree_editor.cpp b/editor/gui/scene_tree_editor.cpp
index fff469044543f9..e77a45ca33ff7f 100644
--- a/editor/gui/scene_tree_editor.cpp
+++ b/editor/gui/scene_tree_editor.cpp
@@ -498,6 +498,18 @@ void SceneTreeEditor::_add_nodes(Node *p_node, TreeItem *p_parent) {
 					EditorNode::get_singleton()->is_object_of_custom_type(p_node, E)) {
 				valid = true;
 				break;
+			} else {
+				Ref<Script> node_script = p_node->get_script();
+				while (node_script.is_valid()) {
+					if (node_script->get_path() == E) {
+						valid = true;
+						break;
+					}
+					node_script = node_script->get_base_script();
+				}
+				if (valid) {
+					break;
+				}
 			}
 		}
 
@@ -656,6 +668,18 @@ bool SceneTreeEditor::_update_filter(TreeItem *p_parent, bool p_scroll_to_select
 					EditorNode::get_singleton()->is_object_of_custom_type(n, E)) {
 				selectable = true;
 				break;
+			} else {
+				Ref<Script> dropped_node_script = n->get_script();
+				while (dropped_node_script.is_valid()) {
+					if (dropped_node_script->get_path() == E) {
+						selectable = true;
+						break;
+					}
+					dropped_node_script = dropped_node_script->get_base_script();
+				}
+				if (selectable) {
+					break;
+				}
 			}
 		}
 	}
@@ -1548,16 +1572,44 @@ void SceneTreeDialog::set_valid_types(const Vector<StringName> &p_valid) {
 		HBoxContainer *hb = memnew(HBoxContainer);
 		hflow->add_child(hb);
 
+		// Attempt to get the correct name and icon for script path types
+		String type_name = type;
+		String type_icon = type;
+		if (ResourceLoader::exists(type, "Script")) {
+			Ref<Script> script = ResourceLoader::load(type);
+			if (script.is_valid()) {
+				if (script->get_global_name() == "") {
+					type_name = type_name.get_file();
+					while (script.is_valid()) {
+						if (script->get_global_name() != "") {
+							type_icon = script->get_global_name();
+							break;
+						}
+
+						Ref<Script> base_script = script->get_base_script();
+
+						if (base_script.is_null()) {
+							type_icon = script->get_instance_base_type();
+						}
+						script = base_script;
+					}
+				} else {
+					type_name = script->get_global_name();
+					type_icon = type_name;
+				}
+			}
+		}
+
 		TextureRect *trect = memnew(TextureRect);
 		hb->add_child(trect);
 		trect->set_expand_mode(TextureRect::EXPAND_IGNORE_SIZE);
 		trect->set_stretch_mode(TextureRect::STRETCH_KEEP_ASPECT_CENTERED);
-		trect->set_meta("type", type);
+		trect->set_meta("type", type_icon);
 		valid_type_icons.push_back(trect);
 
 		Label *label = memnew(Label);
 		hb->add_child(label);
-		label->set_text(type);
+		label->set_text(type_name);
 		label->set_auto_translate(false);
 	}