searching (please review i dont trust myself)

Author: kam

src/DevTools.hpp

@@ -52,6 +52,9 @@ class DevTools {
     Ref<CCNode> m_selectedNode;
     std::vector<std::pair<CCNode*, HighlightMode>> m_toHighlight;
     std::vector<std::function<void(CCNode*)>> m_customCallbacks;
+    std::string m_searchQuery;
+    std::string m_prevQuery;
+    std::unordered_map<CCNode*, bool> m_nodeOpen;
 
     void setupFonts();
     void setupPlatform();
@@ -87,6 +90,8 @@ class DevTools {
     void renderDrawData(ImDrawData*);
     void renderDrawDataFallback(ImDrawData*);
 
+    bool searchBranch(CCNode* node);
+
     bool hasExtension(const std::string& ext) const;
 
     DevTools() { loadSettings(); }

src/pages/Tree.cpp

@@ -2,13 +2,18 @@
 #include <Geode/utils/cocos.hpp>
 #include "../DevTools.hpp"
 #include "../platform/utils.hpp"
+#include <misc/cpp/imgui_stdlib.h>
 #ifndef GEODE_IS_WINDOWS
 #include <cxxabi.h>
 #endif
 
 using namespace geode::prelude;
 
 void DevTools::drawTreeBranch(CCNode* node, size_t index) {
+    if (!searchBranch(node)) {
+        return;
+    }
+
     auto selected = DevTools::get()->getSelectedNode() == node;
 
     ImGuiTreeNodeFlags flags = ImGuiTreeNodeFlags_None;
@@ -23,6 +28,14 @@ void DevTools::drawTreeBranch(CCNode* node, size_t index) {
     {
         flags |= ImGuiTreeNodeFlags_OpenOnArrow;
     }
+
+    if (m_searchQuery.empty()) {
+        ImGui::SetNextItemOpen(m_nodeOpen.contains(node) ? m_nodeOpen[node] : false);
+    }
+    else if (m_searchQuery != m_prevQuery) {
+        ImGui::SetNextItemOpen(true);
+    }
+
     std::stringstream name;
     name << "[" << index << "] " << getNodeName(node) << " ";
     if (node->getTag() != -1) {
@@ -36,13 +49,31 @@ void DevTools::drawTreeBranch(CCNode* node, size_t index) {
     }
     // The order here is unusual due to imgui weirdness; see the second-to-last paragraph in https://kahwei.dev/2022/06/20/imgui-tree-node/
     bool expanded = ImGui::TreeNodeEx(node, flags, "%s", name.str().c_str());
-    if (ImGui::IsItemClicked()) { 
+    if (ImGui::IsItemClicked()) {
         DevTools::get()->selectNode(node);
         selected = true;
+
+        if (!m_searchQuery.empty()) {
+            CCNode* parent = node->getParent();
+            while (parent) {
+                m_nodeOpen[parent] = true;
+                parent = parent->getParent();
+            }
+        }
     }
     if (ImGui::IsItemHovered() && (m_settings.alwaysHighlight || ImGui::IsKeyDown(ImGuiMod_Shift))) {
         DevTools::get()->highlightNode(node, HighlightMode::Hovered);
     }
+    if (ImGui::IsItemToggledOpen() && (m_searchQuery.empty() || m_searchQuery == m_prevQuery)) {
+        if (!m_searchQuery.empty() && expanded) {
+            CCNode* parent = node->getParent();
+            while (parent) {
+                m_nodeOpen[parent] = true;
+                parent = parent->getParent();
+            }
+        }
+        m_nodeOpen[node] = expanded;
+    }
     if (expanded) {
         if (m_settings.attributesInTree) {
             this->drawNodeAttributes(node);
@@ -59,6 +90,30 @@ void DevTools::drawTree() {
 #ifdef GEODE_IS_MOBILE
     ImGui::Dummy({0.f, 60.f});
 #endif
+    m_prevQuery = m_searchQuery;
+    ImGui::InputText("Search Tree", &m_searchQuery, 256);
 
     this->drawTreeBranch(CCDirector::get()->getRunningScene(), 0);
 }
+
+bool DevTools::searchBranch(CCNode* node) {
+    if (m_searchQuery.empty()) return true;
+
+    std::string name = getNodeName(node);
+    std::string id = node->getID();
+    std::string query = m_searchQuery;
+
+    std::ranges::transform(name, name.begin(), ::tolower);
+    std::ranges::transform(id.begin(), id.end(), id.begin(), ::tolower);
+    std::ranges::transform(query.begin(), query.end(), query.begin(), ::tolower);
+
+    if (name.find(query) != std::string::npos || id.find(query) != std::string::npos) {
+        return true;
+    }
+    for (auto child : node->getChildrenExt<CCNode>()) {
+        if (searchBranch(child)) {
+            return true;
+        }
+    }
+    return false;
+}