From 3fefbb078a4382e833018cc09107f3c455579b6b Mon Sep 17 00:00:00 2001
From: aXu-AP <1621768+aXu-AP@users.noreply.github.com>
Date: Wed, 11 Sep 2024 18:57:45 +0300
Subject: [PATCH] Add class icons to script list

Draws icons representing the script's or doc page's class on the right side of the script editor. Supports custom icons and icons inherited from parent class.
---
 editor/plugins/script_editor_plugin.cpp | 57 +++++++++++++++++++++++++
 editor/plugins/script_editor_plugin.h   |  1 +
 2 files changed, 58 insertions(+)

diff --git a/editor/plugins/script_editor_plugin.cpp b/editor/plugins/script_editor_plugin.cpp
index 499696497680..acb660b48c75 100644
--- a/editor/plugins/script_editor_plugin.cpp
+++ b/editor/plugins/script_editor_plugin.cpp
@@ -3340,6 +3340,62 @@ void ScriptEditor::_script_list_clicked(int p_item, Vector2 p_local_mouse_pos, M
 	}
 }
 
+void ScriptEditor::_draw_script_list() {
+	// Draw class icons on right side.
+	Size2 max_icon_size = EditorNode::get_singleton()->get_class_icon("Node")->get_size();
+
+	VScrollBar *scroll = script_list->get_v_scroll_bar();
+	real_t offset = -scroll->get_value();
+
+	real_t margin_right = script_list->get_theme_stylebox("panel")->get_margin(SIDE_RIGHT);
+	margin_right += script_list->get_theme_constant("h_separation") / 2;
+	if (scroll->is_visible()) {
+		margin_right += scroll->get_size().x;
+	}
+
+	for (int i = 0; i < script_list->get_item_count(); i++) {
+		Control *c = tab_container->get_tab_control(i);
+		if (!c) {
+			continue;
+		}
+
+		Ref<Texture2D> icon;
+		if (Object::cast_to<ScriptEditorBase>(c)) {
+			ScriptEditorBase *se = Object::cast_to<ScriptEditorBase>(c);
+			Ref<Script> scr = se->get_edited_resource();
+
+			icon = EditorNode::get_editor_data().get_script_icon(scr);
+			if (icon.is_null()) {
+				icon = EditorNode::get_singleton()->get_class_icon(scr->get_instance_base_type());
+			}
+		}
+		if (Object::cast_to<EditorHelp>(c)) {
+			EditorHelp *help = Object::cast_to<EditorHelp>(c);
+			icon = EditorNode::get_singleton()->get_class_icon(help->get_class());
+		}
+
+		if (icon.is_valid()) {
+			Size2 icon_size = icon->get_size();
+			if (icon_size.x > max_icon_size.x) {
+				icon_size.y = icon_size.y * (max_icon_size.x / icon_size.x);
+				icon_size.x = max_icon_size.x;
+			}
+			if (icon_size.y > max_icon_size.y) {
+				icon_size.x = icon_size.x * (max_icon_size.y / icon_size.y);
+				icon_size.y = max_icon_size.y;
+			}
+			Rect2 item_rect = script_list->get_item_rect(i);
+			real_t margin_top = (item_rect.size.y - icon_size.y) / 2;
+			real_t pos_from_right = max_icon_size.x - (max_icon_size.x - icon_size.x) / 2;
+			Rect2 icon_rect = Rect2(
+					item_rect.get_end().x - pos_from_right - margin_right,
+					item_rect.position.y + margin_top + offset,
+					icon_size.x, icon_size.y);
+			script_list->draw_texture_rect(icon, icon_rect);
+		}
+	}
+}
+
 void ScriptEditor::_make_script_list_context_menu() {
 	context_menu->clear();
 
@@ -4111,6 +4167,7 @@ ScriptEditor::ScriptEditor(WindowWrapper *p_wrapper) {
 	script_split->set_split_offset(70 * EDSCALE);
 	_sort_list_on_update = true;
 	script_list->connect("item_clicked", callable_mp(this, &ScriptEditor::_script_list_clicked), CONNECT_DEFERRED);
+	script_list->connect("draw", callable_mp(this, &ScriptEditor::_draw_script_list));
 	script_list->set_allow_rmb_select(true);
 	SET_DRAG_FORWARDING_GCD(script_list, ScriptEditor);
 
diff --git a/editor/plugins/script_editor_plugin.h b/editor/plugins/script_editor_plugin.h
index e0fac5d0c6e7..32a3ffaf94c9 100644
--- a/editor/plugins/script_editor_plugin.h
+++ b/editor/plugins/script_editor_plugin.h
@@ -469,6 +469,7 @@ class ScriptEditor : public PanelContainer {
 	virtual void shortcut_input(const Ref<InputEvent> &p_event) override;
 
 	void _script_list_clicked(int p_item, Vector2 p_local_mouse_pos, MouseButton p_mouse_button_index);
+	void _draw_script_list();
 	void _make_script_list_context_menu();
 
 	void _help_search(const String &p_text);