From cff4626b2ddb97f5ea77a1a9a01b52fa1039957a Mon Sep 17 00:00:00 2001
From: Serhii Snitsaruk <limbonaut@pm.me>
Date: Sat, 26 Aug 2023 10:08:01 +0200
Subject: [PATCH] Extract TaskPanel code => TaskPalette

---
 editor/limbo_ai_editor_plugin.cpp | 348 +----------------------------
 editor/limbo_ai_editor_plugin.h   |  73 +-----
 editor/task_palette.cpp           | 354 ++++++++++++++++++++++++++++++
 editor/task_palette.h             |  91 ++++++++
 4 files changed, 454 insertions(+), 412 deletions(-)
 create mode 100644 editor/task_palette.cpp
 create mode 100644 editor/task_palette.h

diff --git a/editor/limbo_ai_editor_plugin.cpp b/editor/limbo_ai_editor_plugin.cpp
index accccdb..c77a0c5 100644
--- a/editor/limbo_ai_editor_plugin.cpp
+++ b/editor/limbo_ai_editor_plugin.cpp
@@ -357,340 +357,6 @@ TaskTree::~TaskTree() {
 
 //**** TaskTree ^
 
-//**** TaskButton
-
-Control *TaskButton::make_custom_tooltip(const String &p_text) const {
-	EditorHelpBit *help_bit = memnew(EditorHelpBit);
-	help_bit->get_rich_text()->set_custom_minimum_size(Size2(360 * EDSCALE, 1));
-
-	String help_text;
-	if (!p_text.is_empty()) {
-		help_text = p_text;
-	} else {
-		help_text = "[i]" + TTR("No description.") + "[/i]";
-	}
-
-	help_bit->set_text(help_text);
-
-	return help_bit;
-}
-
-//**** TaskButton ^
-
-//**** TaskSection
-
-void TaskSection::_on_task_button_pressed(const String &p_task) {
-	emit_signal(SNAME("task_button_pressed"), p_task);
-}
-
-void TaskSection::_on_task_button_gui_input(const Ref<InputEvent> &p_event, const String &p_task) {
-	if (!p_event->is_pressed()) {
-		return;
-	}
-
-	Ref<InputEventMouseButton> mb = p_event;
-	if (mb.is_valid() && mb->get_button_index() == MouseButton::RIGHT) {
-		emit_signal(SNAME("task_button_rmb"), p_task);
-	}
-}
-
-void TaskSection::_on_header_pressed() {
-	set_collapsed(!is_collapsed());
-}
-
-void TaskSection::set_filter(String p_filter_text) {
-	int num_hidden = 0;
-	if (p_filter_text.is_empty()) {
-		for (int i = 0; i < tasks_container->get_child_count(); i++) {
-			Object::cast_to<Button>(tasks_container->get_child(i))->show();
-		}
-		set_visible(tasks_container->get_child_count() > 0);
-	} else {
-		for (int i = 0; i < tasks_container->get_child_count(); i++) {
-			Button *btn = Object::cast_to<Button>(tasks_container->get_child(i));
-			btn->set_visible(btn->get_text().findn(p_filter_text) != -1);
-			num_hidden += !btn->is_visible();
-		}
-		set_visible(num_hidden < tasks_container->get_child_count());
-	}
-}
-
-void TaskSection::add_task_button(const String &p_name, const Ref<Texture> &icon, const String &p_tooltip, Variant p_meta) {
-	TaskButton *btn = memnew(TaskButton);
-	btn->set_text(p_name);
-	btn->set_icon(icon);
-	btn->set_tooltip_text(p_tooltip);
-	btn->add_theme_constant_override(SNAME("icon_max_width"), 16 * EDSCALE); // Force user icons to  be of the proper size.
-	btn->connect(SNAME("pressed"), callable_mp(this, &TaskSection::_on_task_button_pressed).bind(p_meta));
-	btn->connect(SNAME("gui_input"), callable_mp(this, &TaskSection::_on_task_button_gui_input).bind(p_meta));
-	tasks_container->add_child(btn);
-}
-
-void TaskSection::set_collapsed(bool p_collapsed) {
-	tasks_container->set_visible(!p_collapsed);
-	section_header->set_icon(p_collapsed ? get_theme_icon(SNAME("GuiTreeArrowRight"), SNAME("EditorIcons")) : get_theme_icon(SNAME("GuiTreeArrowDown"), SNAME("EditorIcons")));
-}
-
-bool TaskSection::is_collapsed() const {
-	return !tasks_container->is_visible();
-}
-
-void TaskSection::_notification(int p_what) {
-	if (p_what == NOTIFICATION_THEME_CHANGED) {
-		section_header->set_icon(is_collapsed() ? get_theme_icon(SNAME("GuiTreeArrowRight"), SNAME("EditorIcons")) : get_theme_icon(SNAME("GuiTreeArrowDown"), SNAME("EditorIcons")));
-		section_header->add_theme_font_override(SNAME("font"), get_theme_font(SNAME("bold"), SNAME("EditorFonts")));
-	}
-}
-
-void TaskSection::_bind_methods() {
-	ADD_SIGNAL(MethodInfo("task_button_pressed"));
-	ADD_SIGNAL(MethodInfo("task_button_rmb"));
-}
-
-TaskSection::TaskSection(String p_category_name) {
-	section_header = memnew(Button);
-	add_child(section_header);
-	section_header->set_text(p_category_name);
-	section_header->set_focus_mode(FOCUS_NONE);
-	section_header->connect("pressed", callable_mp(this, &TaskSection::_on_header_pressed));
-
-	tasks_container = memnew(HFlowContainer);
-	add_child(tasks_container);
-}
-
-TaskSection::~TaskSection() {
-}
-
-//**** TaskSection ^
-
-//**** TaskPanel
-
-void TaskPanel::_menu_action_selected(int p_id) {
-	ERR_FAIL_COND(context_task.is_empty());
-	switch (p_id) {
-		case MENU_OPEN_DOC: {
-			String help_class;
-			if (context_task.begins_with("res://")) {
-				Ref<Script> s = ResourceLoader::load(context_task, "Script");
-				help_class = s->get_language()->get_global_class_name(context_task);
-			}
-			if (help_class.is_empty()) {
-				// Assuming context task is core class.
-				help_class = context_task;
-			}
-			ScriptEditor::get_singleton()->goto_help("class_name:" + help_class);
-			EditorNode::get_singleton()->set_visible_editor(EditorNode::EDITOR_SCRIPT);
-		} break;
-		case MENU_EDIT_SCRIPT: {
-			ERR_FAIL_COND(!context_task.begins_with("res://"));
-			ScriptEditor::get_singleton()->open_file(context_task);
-		} break;
-		case MENU_FAVORITE: {
-			PackedStringArray favorite_tasks = GLOBAL_GET("limbo_ai/behavior_tree/favorite_tasks");
-			if (favorite_tasks.has(context_task)) {
-				favorite_tasks.erase(context_task);
-			} else {
-				favorite_tasks.append(context_task);
-			}
-			ProjectSettings::get_singleton()->set_setting("limbo_ai/behavior_tree/favorite_tasks", favorite_tasks);
-			ProjectSettings::get_singleton()->save();
-			emit_signal(SNAME("favorite_tasks_changed"));
-		} break;
-	}
-}
-
-void TaskPanel::_on_task_button_pressed(const String &p_task) {
-	emit_signal(SNAME("task_selected"), p_task);
-}
-
-void TaskPanel::_on_task_button_rmb(const String &p_task) {
-	ERR_FAIL_COND(p_task.is_empty());
-
-	context_task = p_task;
-	menu->clear();
-
-	menu->add_icon_item(get_theme_icon(SNAME("Script"), SNAME("EditorIcons")), TTR("Edit Script"), MENU_EDIT_SCRIPT);
-	menu->set_item_disabled(MENU_EDIT_SCRIPT, !context_task.begins_with("res://"));
-	menu->add_icon_item(get_theme_icon(SNAME("Help"), SNAME("EditorIcons")), TTR("Open Documentation"), MENU_OPEN_DOC);
-
-	menu->add_separator();
-	Array favorite_tasks = GLOBAL_GET("limbo_ai/behavior_tree/favorite_tasks");
-	if (favorite_tasks.has(context_task)) {
-		menu->add_icon_item(get_theme_icon(SNAME("NonFavorite"), SNAME("EditorIcons")), TTR("Remove from Favorites"), MENU_FAVORITE);
-	} else {
-		menu->add_icon_item(get_theme_icon(SNAME("Favorites"), SNAME("EditorIcons")), TTR("Add to Favorites"), MENU_FAVORITE);
-	}
-
-	menu->reset_size();
-	menu->set_position(get_screen_position() + get_local_mouse_position());
-	menu->popup();
-}
-
-void TaskPanel::_apply_filter(const String &p_text) {
-	for (int i = 0; i < sections->get_child_count(); i++) {
-		TaskSection *sec = Object::cast_to<TaskSection>(sections->get_child(i));
-		ERR_FAIL_COND(sec == nullptr);
-		sec->set_filter(p_text);
-	}
-}
-
-void TaskPanel::refresh() {
-	filter_edit->set_right_icon(get_theme_icon(SNAME("Search"), SNAME("EditorIcons")));
-
-	HashSet<String> collapsed_sections;
-	if (sections->get_child_count() == 0) {
-		// Restore collapsed state from config.
-		ConfigFile conf;
-		String conf_path = EditorPaths::get_singleton()->get_project_settings_dir().path_join("limbo_ai.cfg");
-		if (conf.load(conf_path) == OK) {
-			Variant value = conf.get_value("bt_editor", "collapsed_sections", Array());
-			if (value.is_array()) {
-				Array arr = value;
-				for (int i = 0; i < arr.size(); i++) {
-					if (arr[i].get_type() == Variant::STRING) {
-						collapsed_sections.insert(arr[i]);
-					}
-				}
-			}
-		}
-	} else {
-		for (int i = 0; i < sections->get_child_count(); i++) {
-			TaskSection *sec = Object::cast_to<TaskSection>(sections->get_child(i));
-			if (sec->is_collapsed()) {
-				collapsed_sections.insert(sec->get_category_name());
-			}
-			sections->get_child(i)->queue_free();
-		}
-	}
-
-	LimboTaskDB::scan_user_tasks();
-	List<String> categories = LimboTaskDB::get_categories();
-	categories.sort();
-
-	for (String cat : categories) {
-		List<String> tasks = LimboTaskDB::get_tasks_in_category(cat);
-
-		if (tasks.size() == 0) {
-			continue;
-		}
-
-		TaskSection *sec = memnew(TaskSection(cat));
-		for (String task_meta : tasks) {
-			Ref<Texture2D> icon = LimboUtility::get_singleton()->get_task_icon(task_meta);
-
-			String tname;
-			DocTools *dd = EditorHelp::get_doc_data();
-			HashMap<String, DocData::ClassDoc>::Iterator E;
-			if (task_meta.begins_with("res:")) {
-				tname = task_meta.get_file().get_basename().trim_prefix("BT").to_pascal_case();
-				E = dd->class_list.find(vformat("\"%s\"", task_meta.trim_prefix("res://")));
-				if (!E) {
-					E = dd->class_list.find(tname);
-				}
-			} else {
-				tname = task_meta.trim_prefix("BT");
-				E = dd->class_list.find(task_meta);
-			}
-
-			String descr;
-			if (E) {
-				if (E->value.description.is_empty() || E->value.description.length() > 1000) {
-					descr = DTR(E->value.brief_description);
-				} else {
-					descr = DTR(E->value.description);
-				}
-			}
-
-			sec->add_task_button(tname, icon, descr, task_meta);
-		}
-		sec->set_filter("");
-		sec->connect(SNAME("task_button_pressed"), callable_mp(this, &TaskPanel::_on_task_button_pressed));
-		sec->connect(SNAME("task_button_rmb"), callable_mp(this, &TaskPanel::_on_task_button_rmb));
-		sections->add_child(sec);
-		sec->set_collapsed(collapsed_sections.has(cat));
-	}
-
-	if (!filter_edit->get_text().is_empty()) {
-		_apply_filter(filter_edit->get_text());
-	}
-}
-
-void TaskPanel::_notification(int p_what) {
-	switch (p_what) {
-		case NOTIFICATION_EXIT_TREE: {
-			if (sections->get_child_count() == 0) {
-				return;
-			}
-			Array collapsed_sections;
-			for (int i = 0; i < sections->get_child_count(); i++) {
-				TaskSection *sec = Object::cast_to<TaskSection>(sections->get_child(i));
-				if (sec->is_collapsed()) {
-					collapsed_sections.push_back(sec->get_category_name());
-				}
-			}
-			ConfigFile conf;
-			String conf_path = EditorPaths::get_singleton()->get_project_settings_dir().path_join("limbo_ai.cfg");
-			conf.load(conf_path);
-			conf.set_value("bt_editor", "collapsed_sections", collapsed_sections);
-			conf.save(conf_path);
-		} break;
-		case NOTIFICATION_THEME_CHANGED: {
-			refresh_btn->set_icon(get_theme_icon(SNAME("Reload"), SNAME("EditorIcons")));
-			if (is_visible_in_tree()) {
-				refresh();
-			}
-		} break;
-	}
-}
-
-void TaskPanel::_bind_methods() {
-	ClassDB::bind_method(D_METHOD("refresh"), &TaskPanel::refresh);
-
-	ADD_SIGNAL(MethodInfo("task_selected"));
-	ADD_SIGNAL(MethodInfo("favorite_tasks_changed"));
-}
-
-TaskPanel::TaskPanel() {
-	VBoxContainer *vb = memnew(VBoxContainer);
-	add_child(vb);
-
-	HBoxContainer *hb = memnew(HBoxContainer);
-	vb->add_child(hb);
-
-	filter_edit = memnew(LineEdit);
-	filter_edit->set_clear_button_enabled(true);
-	filter_edit->set_placeholder(TTR("Filter tasks"));
-	filter_edit->connect("text_changed", callable_mp(this, &TaskPanel::_apply_filter));
-	filter_edit->set_h_size_flags(SIZE_EXPAND_FILL);
-	hb->add_child(filter_edit);
-
-	refresh_btn = memnew(Button);
-	refresh_btn->set_tooltip_text(TTR("Refresh tasks"));
-	refresh_btn->set_flat(true);
-	refresh_btn->set_focus_mode(FocusMode::FOCUS_NONE);
-	refresh_btn->connect("pressed", callable_mp(this, &TaskPanel::refresh));
-	hb->add_child(refresh_btn);
-
-	ScrollContainer *sc = memnew(ScrollContainer);
-	sc->set_h_size_flags(SIZE_EXPAND_FILL);
-	sc->set_v_size_flags(SIZE_EXPAND_FILL);
-	vb->add_child(sc);
-
-	sections = memnew(VBoxContainer);
-	sections->set_h_size_flags(SIZE_EXPAND_FILL);
-	sections->set_v_size_flags(SIZE_EXPAND_FILL);
-	sc->add_child(sections);
-
-	menu = memnew(PopupMenu);
-	add_child(menu);
-	menu->connect("id_pressed", callable_mp(this, &TaskPanel::_menu_action_selected));
-}
-
-TaskPanel::~TaskPanel() {
-}
-
-//**** TaskPanel ^
-
 //**** LimboAIEditor
 
 _FORCE_INLINE_ String _get_script_template_path() {
@@ -835,7 +501,7 @@ void LimboAIEditor::edit_bt(Ref<BehaviorTree> p_behavior_tree, bool p_force_refr
 
 	usage_hint->hide();
 	task_tree->show();
-	task_panel->show();
+	task_palette->show();
 
 	_update_history_buttons();
 	_update_header();
@@ -1162,7 +828,7 @@ void LimboAIEditor::_on_visibility_changed() {
 			EditorNode::get_singleton()->edit_resource(task_tree->get_bt());
 		}
 
-		task_panel->refresh();
+		task_palette->refresh();
 	}
 	_update_favorite_tasks();
 }
@@ -1592,12 +1258,12 @@ LimboAIEditor::LimboAIEditor() {
 	usage_label->set_text(TTR("Create a new or load an existing behavior tree."));
 	usage_hint->add_child(usage_label);
 
-	task_panel = memnew(TaskPanel());
+	task_palette = memnew(TaskPalette());
 	hsc->set_split_offset(-300);
-	task_panel->connect("task_selected", callable_mp(this, &LimboAIEditor::_add_task_by_class_or_path));
-	task_panel->connect("favorite_tasks_changed", callable_mp(this, &LimboAIEditor::_update_favorite_tasks));
-	task_panel->hide();
-	hsc->add_child(task_panel);
+	task_palette->connect("task_selected", callable_mp(this, &LimboAIEditor::_add_task_by_class_or_path));
+	task_palette->connect("favorite_tasks_changed", callable_mp(this, &LimboAIEditor::_update_favorite_tasks));
+	task_palette->hide();
+	hsc->add_child(task_palette);
 
 	banners = memnew(VBoxContainer);
 	vbox->add_child(banners);
diff --git a/editor/limbo_ai_editor_plugin.h b/editor/limbo_ai_editor_plugin.h
index 8c23427..8933434 100644
--- a/editor/limbo_ai_editor_plugin.h
+++ b/editor/limbo_ai_editor_plugin.h
@@ -15,6 +15,7 @@
 
 #include "modules/limboai/bt/behavior_tree.h"
 #include "modules/limboai/bt/tasks/bt_task.h"
+#include "task_palette.h"
 
 #include "core/object/class_db.h"
 #include "core/object/object.h"
@@ -76,76 +77,6 @@ public:
 	~TaskTree();
 };
 
-class TaskButton : public Button {
-	GDCLASS(TaskButton, Button);
-
-public:
-	virtual Control *make_custom_tooltip(const String &p_text) const override;
-};
-
-class TaskSection : public VBoxContainer {
-	GDCLASS(TaskSection, VBoxContainer);
-
-private:
-	FlowContainer *tasks_container;
-	Button *section_header;
-
-	void _on_task_button_pressed(const String &p_task);
-	void _on_task_button_gui_input(const Ref<InputEvent> &p_event, const String &p_task);
-	void _on_header_pressed();
-
-protected:
-	static void _bind_methods();
-
-	void _notification(int p_what);
-
-public:
-	void set_filter(String p_filter);
-	void add_task_button(const String &p_name, const Ref<Texture> &icon, const String &p_tooltip, Variant p_meta);
-
-	void set_collapsed(bool p_collapsed);
-	bool is_collapsed() const;
-
-	String get_category_name() const { return section_header->get_text(); }
-
-	TaskSection(String p_category_name);
-	~TaskSection();
-};
-
-class TaskPanel : public PanelContainer {
-	GDCLASS(TaskPanel, PanelContainer)
-
-private:
-	enum MenuAction {
-		MENU_EDIT_SCRIPT,
-		MENU_OPEN_DOC,
-		MENU_FAVORITE,
-	};
-
-	LineEdit *filter_edit;
-	VBoxContainer *sections;
-	PopupMenu *menu;
-	Button *refresh_btn;
-
-	String context_task;
-
-	void _menu_action_selected(int p_id);
-	void _on_task_button_pressed(const String &p_task);
-	void _on_task_button_rmb(const String &p_task);
-	void _apply_filter(const String &p_text);
-
-protected:
-	static void _bind_methods();
-
-	void _notification(int p_what);
-
-public:
-	void refresh();
-
-	TaskPanel();
-	~TaskPanel();
-};
-
 class LimboAIEditor : public Control {
 	GDCLASS(LimboAIEditor, Control);
 
@@ -182,7 +113,7 @@ private:
 	FileDialog *load_dialog;
 	Button *history_back;
 	Button *history_forward;
-	TaskPanel *task_panel;
+	TaskPalette *task_palette;
 	HBoxContainer *fav_tasks_hbox;
 
 	Button *comment_btn;
diff --git a/editor/task_palette.cpp b/editor/task_palette.cpp
new file mode 100644
index 0000000..387a857
--- /dev/null
+++ b/editor/task_palette.cpp
@@ -0,0 +1,354 @@
+/**
+ * task_palette.cpp
+ * =============================================================================
+ * Copyright 2021-2023 Serhii Snitsaruk
+ *
+ * Use of this source code is governed by an MIT-style
+ * license that can be found in the LICENSE file or at
+ * https://opensource.org/licenses/MIT.
+ * =============================================================================
+ */
+
+#include "task_palette.h"
+
+#include "modules/limboai/util/limbo_task_db.h"
+#include "modules/limboai/util/limbo_utility.h"
+
+#include "core/config/project_settings.h"
+#include "editor/editor_help.h"
+#include "editor/editor_node.h"
+#include "editor/editor_paths.h"
+#include "editor/editor_scale.h"
+#include "editor/plugins/script_editor_plugin.h"
+
+//**** TaskButton
+
+Control *TaskButton::make_custom_tooltip(const String &p_text) const {
+	EditorHelpBit *help_bit = memnew(EditorHelpBit);
+	help_bit->get_rich_text()->set_custom_minimum_size(Size2(360 * EDSCALE, 1));
+
+	String help_text;
+	if (!p_text.is_empty()) {
+		help_text = p_text;
+	} else {
+		help_text = "[i]" + TTR("No description.") + "[/i]";
+	}
+
+	help_bit->set_text(help_text);
+
+	return help_bit;
+}
+
+//**** TaskButton ^
+
+//**** TaskPaletteSection
+
+void TaskPaletteSection::_on_task_button_pressed(const String &p_task) {
+	emit_signal(SNAME("task_button_pressed"), p_task);
+}
+
+void TaskPaletteSection::_on_task_button_gui_input(const Ref<InputEvent> &p_event, const String &p_task) {
+	if (!p_event->is_pressed()) {
+		return;
+	}
+
+	Ref<InputEventMouseButton> mb = p_event;
+	if (mb.is_valid() && mb->get_button_index() == MouseButton::RIGHT) {
+		emit_signal(SNAME("task_button_rmb"), p_task);
+	}
+}
+
+void TaskPaletteSection::_on_header_pressed() {
+	set_collapsed(!is_collapsed());
+}
+
+void TaskPaletteSection::set_filter(String p_filter_text) {
+	int num_hidden = 0;
+	if (p_filter_text.is_empty()) {
+		for (int i = 0; i < tasks_container->get_child_count(); i++) {
+			Object::cast_to<Button>(tasks_container->get_child(i))->show();
+		}
+		set_visible(tasks_container->get_child_count() > 0);
+	} else {
+		for (int i = 0; i < tasks_container->get_child_count(); i++) {
+			Button *btn = Object::cast_to<Button>(tasks_container->get_child(i));
+			btn->set_visible(btn->get_text().findn(p_filter_text) != -1);
+			num_hidden += !btn->is_visible();
+		}
+		set_visible(num_hidden < tasks_container->get_child_count());
+	}
+}
+
+void TaskPaletteSection::add_task_button(const String &p_name, const Ref<Texture> &icon, const String &p_tooltip, Variant p_meta) {
+	TaskButton *btn = memnew(TaskButton);
+	btn->set_text(p_name);
+	btn->set_icon(icon);
+	btn->set_tooltip_text(p_tooltip);
+	btn->add_theme_constant_override(SNAME("icon_max_width"), 16 * EDSCALE); // Force user icons to  be of the proper size.
+	btn->connect(SNAME("pressed"), callable_mp(this, &TaskPaletteSection::_on_task_button_pressed).bind(p_meta));
+	btn->connect(SNAME("gui_input"), callable_mp(this, &TaskPaletteSection::_on_task_button_gui_input).bind(p_meta));
+	tasks_container->add_child(btn);
+}
+
+void TaskPaletteSection::set_collapsed(bool p_collapsed) {
+	tasks_container->set_visible(!p_collapsed);
+	section_header->set_icon(p_collapsed ? get_theme_icon(SNAME("GuiTreeArrowRight"), SNAME("EditorIcons")) : get_theme_icon(SNAME("GuiTreeArrowDown"), SNAME("EditorIcons")));
+}
+
+bool TaskPaletteSection::is_collapsed() const {
+	return !tasks_container->is_visible();
+}
+
+void TaskPaletteSection::_notification(int p_what) {
+	if (p_what == NOTIFICATION_THEME_CHANGED) {
+		section_header->set_icon(is_collapsed() ? get_theme_icon(SNAME("GuiTreeArrowRight"), SNAME("EditorIcons")) : get_theme_icon(SNAME("GuiTreeArrowDown"), SNAME("EditorIcons")));
+		section_header->add_theme_font_override(SNAME("font"), get_theme_font(SNAME("bold"), SNAME("EditorFonts")));
+	}
+}
+
+void TaskPaletteSection::_bind_methods() {
+	ADD_SIGNAL(MethodInfo("task_button_pressed"));
+	ADD_SIGNAL(MethodInfo("task_button_rmb"));
+}
+
+TaskPaletteSection::TaskPaletteSection(String p_category_name) {
+	section_header = memnew(Button);
+	add_child(section_header);
+	section_header->set_text(p_category_name);
+	section_header->set_focus_mode(FOCUS_NONE);
+	section_header->connect("pressed", callable_mp(this, &TaskPaletteSection::_on_header_pressed));
+
+	tasks_container = memnew(HFlowContainer);
+	add_child(tasks_container);
+}
+
+TaskPaletteSection::~TaskPaletteSection() {
+}
+
+//**** TaskPaletteSection ^
+
+//**** TaskPalette
+
+void TaskPalette::_menu_action_selected(int p_id) {
+	ERR_FAIL_COND(context_task.is_empty());
+	switch (p_id) {
+		case MENU_OPEN_DOC: {
+			String help_class;
+			if (context_task.begins_with("res://")) {
+				Ref<Script> s = ResourceLoader::load(context_task, "Script");
+				help_class = s->get_language()->get_global_class_name(context_task);
+			}
+			if (help_class.is_empty()) {
+				// Assuming context task is core class.
+				help_class = context_task;
+			}
+			ScriptEditor::get_singleton()->goto_help("class_name:" + help_class);
+			EditorNode::get_singleton()->set_visible_editor(EditorNode::EDITOR_SCRIPT);
+		} break;
+		case MENU_EDIT_SCRIPT: {
+			ERR_FAIL_COND(!context_task.begins_with("res://"));
+			ScriptEditor::get_singleton()->open_file(context_task);
+		} break;
+		case MENU_FAVORITE: {
+			PackedStringArray favorite_tasks = GLOBAL_GET("limbo_ai/behavior_tree/favorite_tasks");
+			if (favorite_tasks.has(context_task)) {
+				favorite_tasks.erase(context_task);
+			} else {
+				favorite_tasks.append(context_task);
+			}
+			ProjectSettings::get_singleton()->set_setting("limbo_ai/behavior_tree/favorite_tasks", favorite_tasks);
+			ProjectSettings::get_singleton()->save();
+			emit_signal(SNAME("favorite_tasks_changed"));
+		} break;
+	}
+}
+
+void TaskPalette::_on_task_button_pressed(const String &p_task) {
+	emit_signal(SNAME("task_selected"), p_task);
+}
+
+void TaskPalette::_on_task_button_rmb(const String &p_task) {
+	ERR_FAIL_COND(p_task.is_empty());
+
+	context_task = p_task;
+	menu->clear();
+
+	menu->add_icon_item(get_theme_icon(SNAME("Script"), SNAME("EditorIcons")), TTR("Edit Script"), MENU_EDIT_SCRIPT);
+	menu->set_item_disabled(MENU_EDIT_SCRIPT, !context_task.begins_with("res://"));
+	menu->add_icon_item(get_theme_icon(SNAME("Help"), SNAME("EditorIcons")), TTR("Open Documentation"), MENU_OPEN_DOC);
+
+	menu->add_separator();
+	Array favorite_tasks = GLOBAL_GET("limbo_ai/behavior_tree/favorite_tasks");
+	if (favorite_tasks.has(context_task)) {
+		menu->add_icon_item(get_theme_icon(SNAME("NonFavorite"), SNAME("EditorIcons")), TTR("Remove from Favorites"), MENU_FAVORITE);
+	} else {
+		menu->add_icon_item(get_theme_icon(SNAME("Favorites"), SNAME("EditorIcons")), TTR("Add to Favorites"), MENU_FAVORITE);
+	}
+
+	menu->reset_size();
+	menu->set_position(get_screen_position() + get_local_mouse_position());
+	menu->popup();
+}
+
+void TaskPalette::_apply_filter(const String &p_text) {
+	for (int i = 0; i < sections->get_child_count(); i++) {
+		TaskPaletteSection *sec = Object::cast_to<TaskPaletteSection>(sections->get_child(i));
+		ERR_FAIL_COND(sec == nullptr);
+		sec->set_filter(p_text);
+	}
+}
+
+void TaskPalette::refresh() {
+	filter_edit->set_right_icon(get_theme_icon(SNAME("Search"), SNAME("EditorIcons")));
+
+	HashSet<String> collapsed_sections;
+	if (sections->get_child_count() == 0) {
+		// Restore collapsed state from config.
+		ConfigFile conf;
+		String conf_path = EditorPaths::get_singleton()->get_project_settings_dir().path_join("limbo_ai.cfg");
+		if (conf.load(conf_path) == OK) {
+			Variant value = conf.get_value("bt_editor", "collapsed_sections", Array());
+			if (value.is_array()) {
+				Array arr = value;
+				for (int i = 0; i < arr.size(); i++) {
+					if (arr[i].get_type() == Variant::STRING) {
+						collapsed_sections.insert(arr[i]);
+					}
+				}
+			}
+		}
+	} else {
+		for (int i = 0; i < sections->get_child_count(); i++) {
+			TaskPaletteSection *sec = Object::cast_to<TaskPaletteSection>(sections->get_child(i));
+			if (sec->is_collapsed()) {
+				collapsed_sections.insert(sec->get_category_name());
+			}
+			sections->get_child(i)->queue_free();
+		}
+	}
+
+	LimboTaskDB::scan_user_tasks();
+	List<String> categories = LimboTaskDB::get_categories();
+	categories.sort();
+
+	for (String cat : categories) {
+		List<String> tasks = LimboTaskDB::get_tasks_in_category(cat);
+
+		if (tasks.size() == 0) {
+			continue;
+		}
+
+		TaskPaletteSection *sec = memnew(TaskPaletteSection(cat));
+		for (String task_meta : tasks) {
+			Ref<Texture2D> icon = LimboUtility::get_singleton()->get_task_icon(task_meta);
+
+			String tname;
+			DocTools *dd = EditorHelp::get_doc_data();
+			HashMap<String, DocData::ClassDoc>::Iterator E;
+			if (task_meta.begins_with("res:")) {
+				tname = task_meta.get_file().get_basename().trim_prefix("BT").to_pascal_case();
+				E = dd->class_list.find(vformat("\"%s\"", task_meta.trim_prefix("res://")));
+				if (!E) {
+					E = dd->class_list.find(tname);
+				}
+			} else {
+				tname = task_meta.trim_prefix("BT");
+				E = dd->class_list.find(task_meta);
+			}
+
+			String descr;
+			if (E) {
+				if (E->value.description.is_empty() || E->value.description.length() > 1000) {
+					descr = DTR(E->value.brief_description);
+				} else {
+					descr = DTR(E->value.description);
+				}
+			}
+
+			sec->add_task_button(tname, icon, descr, task_meta);
+		}
+		sec->set_filter("");
+		sec->connect(SNAME("task_button_pressed"), callable_mp(this, &TaskPalette::_on_task_button_pressed));
+		sec->connect(SNAME("task_button_rmb"), callable_mp(this, &TaskPalette::_on_task_button_rmb));
+		sections->add_child(sec);
+		sec->set_collapsed(collapsed_sections.has(cat));
+	}
+
+	if (!filter_edit->get_text().is_empty()) {
+		_apply_filter(filter_edit->get_text());
+	}
+}
+
+void TaskPalette::_notification(int p_what) {
+	switch (p_what) {
+		case NOTIFICATION_EXIT_TREE: {
+			if (sections->get_child_count() == 0) {
+				return;
+			}
+			Array collapsed_sections;
+			for (int i = 0; i < sections->get_child_count(); i++) {
+				TaskPaletteSection *sec = Object::cast_to<TaskPaletteSection>(sections->get_child(i));
+				if (sec->is_collapsed()) {
+					collapsed_sections.push_back(sec->get_category_name());
+				}
+			}
+			ConfigFile conf;
+			String conf_path = EditorPaths::get_singleton()->get_project_settings_dir().path_join("limbo_ai.cfg");
+			conf.load(conf_path);
+			conf.set_value("bt_editor", "collapsed_sections", collapsed_sections);
+			conf.save(conf_path);
+		} break;
+		case NOTIFICATION_THEME_CHANGED: {
+			refresh_btn->set_icon(get_theme_icon(SNAME("Reload"), SNAME("EditorIcons")));
+			if (is_visible_in_tree()) {
+				refresh();
+			}
+		} break;
+	}
+}
+
+void TaskPalette::_bind_methods() {
+	ClassDB::bind_method(D_METHOD("refresh"), &TaskPalette::refresh);
+
+	ADD_SIGNAL(MethodInfo("task_selected"));
+	ADD_SIGNAL(MethodInfo("favorite_tasks_changed"));
+}
+
+TaskPalette::TaskPalette() {
+	VBoxContainer *vb = memnew(VBoxContainer);
+	add_child(vb);
+
+	HBoxContainer *hb = memnew(HBoxContainer);
+	vb->add_child(hb);
+
+	filter_edit = memnew(LineEdit);
+	filter_edit->set_clear_button_enabled(true);
+	filter_edit->set_placeholder(TTR("Filter tasks"));
+	filter_edit->connect("text_changed", callable_mp(this, &TaskPalette::_apply_filter));
+	filter_edit->set_h_size_flags(SIZE_EXPAND_FILL);
+	hb->add_child(filter_edit);
+
+	refresh_btn = memnew(Button);
+	refresh_btn->set_tooltip_text(TTR("Refresh tasks"));
+	refresh_btn->set_flat(true);
+	refresh_btn->set_focus_mode(FocusMode::FOCUS_NONE);
+	refresh_btn->connect("pressed", callable_mp(this, &TaskPalette::refresh));
+	hb->add_child(refresh_btn);
+
+	ScrollContainer *sc = memnew(ScrollContainer);
+	sc->set_h_size_flags(SIZE_EXPAND_FILL);
+	sc->set_v_size_flags(SIZE_EXPAND_FILL);
+	vb->add_child(sc);
+
+	sections = memnew(VBoxContainer);
+	sections->set_h_size_flags(SIZE_EXPAND_FILL);
+	sections->set_v_size_flags(SIZE_EXPAND_FILL);
+	sc->add_child(sections);
+
+	menu = memnew(PopupMenu);
+	add_child(menu);
+	menu->connect("id_pressed", callable_mp(this, &TaskPalette::_menu_action_selected));
+}
+
+TaskPalette::~TaskPalette() {
+}
\ No newline at end of file
diff --git a/editor/task_palette.h b/editor/task_palette.h
new file mode 100644
index 0000000..d058035
--- /dev/null
+++ b/editor/task_palette.h
@@ -0,0 +1,91 @@
+/**
+ * task_palette.h
+ * =============================================================================
+ * Copyright 2021-2023 Serhii Snitsaruk
+ *
+ * Use of this source code is governed by an MIT-style
+ * license that can be found in the LICENSE file or at
+ * https://opensource.org/licenses/MIT.
+ * =============================================================================
+ */
+#ifndef TASK_PALETTE_H
+#define TASK_PALETTE_H
+
+#include "scene/gui/panel_container.h"
+
+#include "scene/gui/box_container.h"
+#include "scene/gui/button.h"
+#include "scene/gui/flow_container.h"
+#include "scene/gui/line_edit.h"
+
+class TaskButton : public Button {
+	GDCLASS(TaskButton, Button);
+
+public:
+	virtual Control *make_custom_tooltip(const String &p_text) const override;
+};
+
+class TaskPaletteSection : public VBoxContainer {
+	GDCLASS(TaskPaletteSection, VBoxContainer);
+
+private:
+	FlowContainer *tasks_container;
+	Button *section_header;
+
+	void _on_task_button_pressed(const String &p_task);
+	void _on_task_button_gui_input(const Ref<InputEvent> &p_event, const String &p_task);
+	void _on_header_pressed();
+
+protected:
+	static void _bind_methods();
+
+	void _notification(int p_what);
+
+public:
+	void set_filter(String p_filter);
+	void add_task_button(const String &p_name, const Ref<Texture> &icon, const String &p_tooltip, Variant p_meta);
+
+	void set_collapsed(bool p_collapsed);
+	bool is_collapsed() const;
+
+	String get_category_name() const { return section_header->get_text(); }
+
+	TaskPaletteSection(String p_category_name);
+	~TaskPaletteSection();
+};
+
+class TaskPalette : public PanelContainer {
+	GDCLASS(TaskPalette, PanelContainer)
+
+private:
+	enum MenuAction {
+		MENU_EDIT_SCRIPT,
+		MENU_OPEN_DOC,
+		MENU_FAVORITE,
+	};
+
+	LineEdit *filter_edit;
+	VBoxContainer *sections;
+	PopupMenu *menu;
+	Button *refresh_btn;
+
+	String context_task;
+
+	void _menu_action_selected(int p_id);
+	void _on_task_button_pressed(const String &p_task);
+	void _on_task_button_rmb(const String &p_task);
+	void _apply_filter(const String &p_text);
+
+protected:
+	static void _bind_methods();
+
+	void _notification(int p_what);
+
+public:
+	void refresh();
+
+	TaskPalette();
+	~TaskPalette();
+};
+
+#endif // TASK_PALETTE
\ No newline at end of file