From 879ec9ebfc09838949b3e91756e00081f49e427d Mon Sep 17 00:00:00 2001
From: Chris Cranford <chris@hibernate.org>
Date: Sun, 30 Jun 2024 15:09:11 -0400
Subject: [PATCH] GH-463 Adapt comment nodes to use `GraphFrame`

---
 src/editor/graph/graph_edit.cpp               | 115 ++++++++++++++++--
 src/editor/graph/graph_edit.h                 |  21 +++-
 src/editor/graph/graph_node.cpp               |  15 +++
 src/editor/graph/graph_node.h                 |   1 +
 src/editor/graph/nodes/graph_node_comment.cpp |  76 +++++++++++-
 src/editor/graph/nodes/graph_node_comment.h   |  38 ++++++
 src/editor/graph/nodes/graph_node_factory.cpp |  29 -----
 src/editor/graph/nodes/graph_node_factory.h   |  38 ------
 src/editor/register_editor_types.cpp          |   3 +
 src/editor/theme/theme_cache.cpp              |  29 +++++
 src/script/nodes/utilities/comment.cpp        |  46 +++++++
 src/script/nodes/utilities/comment.h          |  26 ++++
 12 files changed, 355 insertions(+), 82 deletions(-)
 delete mode 100644 src/editor/graph/nodes/graph_node_factory.cpp
 delete mode 100644 src/editor/graph/nodes/graph_node_factory.h

diff --git a/src/editor/graph/graph_edit.cpp b/src/editor/graph/graph_edit.cpp
index c2fec447..2bfa2268 100644
--- a/src/editor/graph/graph_edit.cpp
+++ b/src/editor/graph/graph_edit.cpp
@@ -29,8 +29,8 @@
 #include "editor/graph/graph_node_pin.h"
 #include "editor/graph/graph_node_spawner.h"
 #include "editor/graph/nodes/graph_node_comment.h"
+#include "editor/graph/nodes/graph_node_default.h"
 #include "editor/plugins/orchestrator_editor_plugin.h"
-#include "nodes/graph_node_factory.h"
 #include "script/language.h"
 #include "script/nodes/script_nodes.h"
 #include "script/script.h"
@@ -249,6 +249,9 @@ void OrchestratorGraphEdit::_notification(int p_what)
         connect("copy_nodes_request", callable_mp(this, &OrchestratorGraphEdit::_on_copy_nodes_request));
         connect("duplicate_nodes_request", callable_mp(this, &OrchestratorGraphEdit::_on_duplicate_nodes_request));
         connect("paste_nodes_request", callable_mp(this, &OrchestratorGraphEdit::_on_paste_nodes_request));
+        #if GODOT_VERSION >= 0x040300
+        connect("graph_elements_linked_to_frame_request", callable_mp(this, &OrchestratorGraphEdit::_on_attach_to_frame));
+        #endif
 
         _context_menu->connect("id_pressed", callable_mp(this, &OrchestratorGraphEdit::_on_context_menu_selection));
 
@@ -359,11 +362,7 @@ void OrchestratorGraphEdit::goto_class_help(const String& p_class_name)
 
 void OrchestratorGraphEdit::for_each_graph_node(std::function<void(OrchestratorGraphNode*)> p_func)
 {
-    for (int i = 0; i < get_child_count(); i++)
-    {
-        if (OrchestratorGraphNode* node = Object::cast_to<OrchestratorGraphNode>(get_child(i)))
-            p_func(node);
-    }
+    for_each_child_type<OrchestratorGraphNode>(p_func);
 }
 
 void OrchestratorGraphEdit::execute_action(const String& p_action_name)
@@ -1072,7 +1071,7 @@ void OrchestratorGraphEdit::_synchronize_graph_knots()
     }
 }
 
-void OrchestratorGraphEdit::_synchronize_graph_node(Ref<OScriptNode> p_node)
+void OrchestratorGraphEdit::_synchronize_graph_node(const Ref<OScriptNode>& p_node)
 {
     if (!p_node.is_valid())
         return;
@@ -1082,11 +1081,32 @@ void OrchestratorGraphEdit::_synchronize_graph_node(Ref<OScriptNode> p_node)
     {
         const Vector2 node_size = p_node->get_size();
 
-        OrchestratorGraphNode* graph_node = OrchestratorGraphNodeFactory::create_node(this, p_node);
-        graph_node->set_title(p_node->get_node_title());
-        graph_node->set_position_offset(p_node->get_position());
-        graph_node->set_size(node_size.is_zero_approx() ? graph_node->get_size() : node_size);
-        add_child(graph_node);
+        Ref<OScriptNodeComment> comment = p_node;
+        if (comment.is_valid())
+        {
+            #if GODOT_VERSION >= 0x040300
+            OrchestratorGraphFrameComment* frame = memnew(OrchestratorGraphFrameComment(this, p_node));
+            frame->set_title(p_node->get_node_title());
+            frame->set_position_offset(p_node->get_position());
+            frame->set_size(node_size.is_zero_approx() ? frame->get_size() : node_size);
+            frame->set_name(vformat("%s", node_id));
+            add_child(frame);
+            #else
+            OrchestratorGraphNodeComment* node = memnew(OrchestratorGraphNodeComment(this, p_node));
+            node->set_title(p_node->get_node_title());
+            node->set_position_offset(p_node->get_position());
+            node->set_size(node_size.is_zero_approx() ? node->get_size() : node_size);
+            add_child(node);
+            #endif
+        }
+        else
+        {
+            OrchestratorGraphNode* graph_node = memnew(OrchestratorGraphNodeDefault(this, p_node));
+            graph_node->set_title(p_node->get_node_title());
+            graph_node->set_position_offset(p_node->get_position());
+            graph_node->set_size(node_size.is_zero_approx() ? graph_node->get_size() : node_size);
+            add_child(graph_node);
+        }
     }
     else
     {
@@ -1105,6 +1125,51 @@ void OrchestratorGraphEdit::_synchronize_child_order()
             comment->call_deferred("raise_request_node_reorder");
         }
     });
+
+    #if GODOT_VERSION >= 0x040300
+    // Performs a small update synchronization step for comment nodes to track their frame attachments
+    bool any_upgraded = false;
+    for_each_child_type<OrchestratorGraphFrameComment>([&, this](OrchestratorGraphFrameComment* frame) {
+        Ref<OScriptNodeComment> comment = frame->get_comment_node();
+        if (comment.is_valid())
+        {
+            if (comment->get_state() == OScriptNodeComment::State_Initial)
+            {
+                // Collect a list of nodes that overlap but are not attached to the frame
+                Vector<StringName> overlaps;
+                for_each_child_type<OrchestratorGraphNode>([&](OrchestratorGraphNode* node) {
+                    if (frame->get_rect().intersects(node->get_rect()))
+                        overlaps.push_back(node->get_name());
+                });
+
+                // Compare existing attachments with overlaps
+                PackedInt64Array attachment_node_ids;
+                for (const StringName& overlap : overlaps)
+                {
+                    if (OrchestratorGraphNode* node = _get_by_name<OrchestratorGraphNode>(overlap))
+                    {
+                        attach_graph_element_to_frame(overlap, frame->get_name());
+                        attachment_node_ids.push_back(node->get_script_node_id());
+                    }
+                }
+
+                // Update attachment list
+                comment->set_attachments(attachment_node_ids);
+                any_upgraded = true;
+            }
+            else if (comment->get_state() >= OScriptNodeComment::State_Tracks_Attachments)
+            {
+                const PackedInt64Array attachments = comment->get_attachments();
+                for (int i = 0; i < attachments.size(); i++)
+                    attach_graph_element_to_frame(itos(attachments[i]), frame->get_name());
+            }
+        }
+    });
+
+    if (any_upgraded)
+        UtilityFunctions::print(get_orchestration()->get_self()->get_path(), ": Graph '", _script_graph->get_graph_name(), "': Comment nodes have been upgraded to GraphFrame(s).");
+
+    #endif
 }
 
 void OrchestratorGraphEdit::_complete_spawn(const Ref<OScriptNode>& p_spawned, const Callable& p_callback)
@@ -1402,15 +1467,21 @@ void OrchestratorGraphEdit::_on_node_selected(Node* p_node)
         return;
 
     OrchestratorGraphNode* graph_node = Object::cast_to<OrchestratorGraphNode>(p_node);
+    #if GODOT_VERSION >= 0x040300
+    OrchestratorGraphFrameComment* frame_node = Object::cast_to<OrchestratorGraphFrameComment>(p_node);
+    if (!graph_node && !frame_node)
+        return;
+    #else
     if (!graph_node)
         return;
+    #endif
 
     Ref<OScriptNode> node = p_node->get_meta("__script_node");
     if (node.is_null())
         return;
 
     OrchestratorSettings* os = OrchestratorSettings::get_singleton();
-    if (os->get_setting("ui/nodes/highlight_selected_connections", false))
+    if (os->get_setting("ui/nodes/highlight_selected_connections", false) && !frame_node)
     {
         // Get list of all selected nodes
         List<Ref<OScriptNode>> selected_nodes;
@@ -1830,4 +1901,22 @@ void OrchestratorGraphEdit::_on_grid_style_selected(int p_index)
     const GridPattern pattern = static_cast<GridPattern>(int(_grid_pattern->get_item_metadata(p_index)));
     set_grid_pattern(pattern);
 }
+
+void OrchestratorGraphEdit::_on_attach_to_frame(const TypedArray<StringName>& p_elements, const StringName& p_frame_name)
+{
+    for (int i = 0; i < p_elements.size(); i++)
+    {
+        OrchestratorGraphFrameComment* frame = _get_by_name<OrchestratorGraphFrameComment>(p_frame_name);
+        OrchestratorGraphNode* node = _get_by_name<OrchestratorGraphNode>(p_elements[i]);
+        if (frame && node)
+        {
+            PackedInt64Array attachments = frame->get_comment_node()->get_attachments();
+            attachments.push_back(node->get_script_node_id());
+
+            attach_graph_element_to_frame(p_elements[i], p_frame_name);
+            frame->get_comment_node()->set_attachments(attachments);
+        }
+    }
+}
+
 #endif
\ No newline at end of file
diff --git a/src/editor/graph/graph_edit.h b/src/editor/graph/graph_edit.h
index 60445b0c..4bfe7a73 100644
--- a/src/editor/graph/graph_edit.h
+++ b/src/editor/graph/graph_edit.h
@@ -204,8 +204,22 @@ class OrchestratorGraphEdit : public GraphEdit
 
     /// Perform an action for each graph node
     /// @param p_func the lambda to be applied
+    /// @deprecated use for_each_child_type
     void for_each_graph_node(std::function<void(OrchestratorGraphNode*)> p_func);
 
+    /// Perform an action for each child type
+    /// @tparam T the child class type to only operate on
+    /// @param p_func the lambda to be applied
+    template <typename T>
+    void for_each_child_type(std::function<void(T*)> p_func)
+    {
+        for (int i = 0; i < get_child_count(); i++)
+        {
+            if (T* child = Object::cast_to<T>(get_child(i)))
+                p_func(child);
+        }
+    }
+
     /// Execute the specified action
     /// @param p_action_name the action to execute
     void execute_action(const String& p_action_name);
@@ -315,7 +329,7 @@ class OrchestratorGraphEdit : public GraphEdit
 
     /// Updates only the specific graph node
     /// @param p_node the node to update.
-    void _synchronize_graph_node(Ref<OScriptNode> p_node);
+    void _synchronize_graph_node(const Ref<OScriptNode>& p_node);
 
     /// Synchronizes the child order
     void _synchronize_child_order();
@@ -458,6 +472,11 @@ class OrchestratorGraphEdit : public GraphEdit
     /// Dispatched when a grid style option is selected
     /// @param p_index the selected item index
     void _on_grid_style_selected(int p_index);
+
+    /// Dispatched when dragging nodes to be attached to a frame
+    /// @param p_elements graph elements to be attached
+    /// @param p_frame_name the frame name to be linked
+    void _on_attach_to_frame(const TypedArray<StringName>& p_elements, const StringName& p_frame_name);
     #endif
 };
 
diff --git a/src/editor/graph/graph_node.cpp b/src/editor/graph/graph_node.cpp
index 6adb7807..cdd7cca7 100644
--- a/src/editor/graph/graph_node.cpp
+++ b/src/editor/graph/graph_node.cpp
@@ -26,6 +26,7 @@
 #include "script/nodes/editable_pin_node.h"
 #include "script/nodes/functions/call_function.h"
 #include "script/nodes/functions/call_script_function.h"
+#include "script/nodes/utilities/comment.h"
 #include "script/script.h"
 
 #include <godot_cpp/classes/button.hpp>
@@ -470,7 +471,16 @@ void OrchestratorGraphNode::_show_context_menu(const Vector2& p_position)
     if (!call_script_function.is_valid())
         _context_menu->set_item_disabled(_context_menu->get_item_index(CM_EXPAND_NODE), true);
 
+
     #if GODOT_VERSION >= 0x040300
+    Ref<OScriptNodeComment> comment = _node;
+    if (comment.is_null())
+    {
+        // Anything but comments
+        if (GraphFrame* frame = get_graph()->get_element_frame(get_name()))
+            _context_menu->add_item("Detach from comment frame", CM_DETACH_FRAME);
+    }
+
     _context_menu->add_separator("Breakpoints");
     _context_menu->add_item("Toggle Breakpoint", CM_TOGGLE_BREAKPOINT, KEY_F9);
     if (_node->has_breakpoint())
@@ -774,6 +784,11 @@ void OrchestratorGraphNode::_on_context_menu_selection(int p_id)
                 _set_breakpoint_state(OScriptNode::BreakpointFlags::BREAKPOINT_DISABLED);
                 break;
             }
+            case CM_DETACH_FRAME:
+            {
+                get_graph()->detach_graph_element_from_frame(get_name());
+                break;
+            }
             #endif
             #ifdef _DEBUG
             case CM_SHOW_DETAILS:
diff --git a/src/editor/graph/graph_node.h b/src/editor/graph/graph_node.h
index d74c5a94..2c63505b 100644
--- a/src/editor/graph/graph_node.h
+++ b/src/editor/graph/graph_node.h
@@ -65,6 +65,7 @@ class OrchestratorGraphNode : public GraphNode
         CM_COLLAPSE_FUNCTION,
         CM_EXPAND_NODE,
         CM_RESIZABLE,
+        CM_DETACH_FRAME,
         #ifdef _DEBUG
         CM_SHOW_DETAILS = 999,
         #endif
diff --git a/src/editor/graph/nodes/graph_node_comment.cpp b/src/editor/graph/nodes/graph_node_comment.cpp
index 11bc3888..de3bb145 100644
--- a/src/editor/graph/nodes/graph_node_comment.cpp
+++ b/src/editor/graph/nodes/graph_node_comment.cpp
@@ -17,6 +17,8 @@
 #include "graph_node_comment.h"
 
 #include "editor/graph/graph_edit.h"
+#include "editor/plugins/orchestrator_editor_plugin.h"
+#include "editor/theme/theme_cache.h"
 
 #include <godot_cpp/classes/input_event_mouse_button.hpp>
 #include <godot_cpp/classes/margin_container.hpp>
@@ -151,4 +153,76 @@ void OrchestratorGraphNodeComment::raise_request_node_reorder()
     List<OrchestratorGraphNode*> intersections = get_nodes_within_global_rect();
     for (OrchestratorGraphNode* node : intersections)
         get_parent()->move_child(node, -1);
-}
\ No newline at end of file
+}
+
+#if GODOT_VERSION >= 0x040300
+void OrchestratorGraphFrameComment::_node_moved(Vector2 p_old_pos, Vector2 p_new_pos)
+{
+    _node->set_position(p_new_pos);
+}
+
+void OrchestratorGraphFrameComment::_node_resized()
+{
+    _node->set_size(get_size());
+}
+
+void OrchestratorGraphFrameComment::_script_node_changed()
+{
+    set_tint_color(_node->get_background_color());
+    set_title(_node->get_node_title());
+
+    Label* title = Object::cast_to<Label>(get_titlebar_hbox()->get_child(0));
+    if (title)
+    {
+        title->set_horizontal_alignment(HORIZONTAL_ALIGNMENT_LEFT);
+        if (_node->is_title_center_aligned())
+            title->set_horizontal_alignment(HORIZONTAL_ALIGNMENT_CENTER);
+    }
+
+    const int font_size = _node->get_font_size();
+    _text->add_theme_font_size_override("font_size", font_size != 0 ? font_size : 14);
+    _text->set_text(_node->get("comments"));
+    _text->add_theme_color_override("font_color", _node->get_text_color());
+}
+
+void OrchestratorGraphFrameComment::_notification(int p_what)
+{
+    if (p_what == NOTIFICATION_READY)
+    {
+        // Used to replicate size/position state to underlying node resource
+        connect("dragged", callable_mp(this, &OrchestratorGraphFrameComment::_node_moved));
+        connect("resized", callable_mp(this, &OrchestratorGraphFrameComment::_node_resized));
+
+        _node->connect("changed", callable_mp(this, &OrchestratorGraphFrameComment::_script_node_changed));
+
+        Ref<OrchestratorThemeCache> cache = OrchestratorPlugin::get_singleton()->get_theme_cache();
+        if (cache.is_valid())
+        {
+            begin_bulk_theme_override();
+            add_theme_stylebox_override("panel", cache->get_theme_stylebox("panel", "GraphFrame"));
+            add_theme_stylebox_override("panel_selected", cache->get_theme_stylebox("panel_selected", "GraphFrame"));
+            end_bulk_theme_override();
+        }
+    }
+}
+
+void OrchestratorGraphFrameComment::_bind_methods()
+{
+}
+
+OrchestratorGraphFrameComment::OrchestratorGraphFrameComment(OrchestratorGraphEdit* p_graph, const Ref<OScriptNodeComment>& p_node)
+    : _graph(p_graph)
+    , _node(p_node)
+{
+    set_meta("__script_node", p_node);
+    set_tint_color_enabled(true);
+
+    _text = memnew(Label);
+    _text->set_vertical_alignment(VERTICAL_ALIGNMENT_TOP);
+    _text->set_v_size_flags(SIZE_SHRINK_BEGIN);
+
+    add_child(_text);
+
+    _script_node_changed();
+}
+#endif
\ No newline at end of file
diff --git a/src/editor/graph/nodes/graph_node_comment.h b/src/editor/graph/nodes/graph_node_comment.h
index 5ea428aa..eaaae01b 100644
--- a/src/editor/graph/nodes/graph_node_comment.h
+++ b/src/editor/graph/nodes/graph_node_comment.h
@@ -24,6 +24,10 @@
 #include <godot_cpp/classes/h_box_container.hpp>
 #include <godot_cpp/classes/label.hpp>
 
+#if GODOT_VERSION >= 0x040300
+#include <godot_cpp/classes/graph_frame.hpp>
+#endif
+
 /// A simple OrchestratorGraphNode implementation to render comment nodes.
 class OrchestratorGraphNodeComment : public OrchestratorGraphNode
 {
@@ -69,4 +73,38 @@ class OrchestratorGraphNodeComment : public OrchestratorGraphNode
     //~ End Control Interface
 };
 
+#if GODOT_VERSION >= 0x040300
+class OrchestratorGraphFrameComment : public GraphFrame
+{
+    GDCLASS(OrchestratorGraphFrameComment, GraphFrame);
+    static void _bind_methods();
+
+protected:
+    OrchestratorGraphEdit* _graph{ nullptr };
+    Ref<OScriptNodeComment> _node;
+    Label* _text{ nullptr };
+
+    //~ Begin Object Interface
+    void _notification(int p_what);
+    //~ End Object Interface
+
+    void _node_moved(Vector2 p_old_pos, Vector2 p_new_pos);
+    void _node_resized();
+    void _script_node_changed();
+
+    /// Default constructor, intentionally protected
+    OrchestratorGraphFrameComment() = default;
+
+public:
+    /// Get a reference to the underlying comment node
+    /// @return the comment node or an invalid reference if its invalid.
+    Ref<OScriptNodeComment> get_comment_node() { return _node; }
+
+    /// Constructor
+    /// @param p_graph the graph edit instance
+    /// @param p_node the comment node
+    OrchestratorGraphFrameComment(OrchestratorGraphEdit* p_graph, const Ref<OScriptNodeComment>& p_node);
+};
+#endif
+
 #endif // ORCHESTRATOR_GRAPH_NODE_COMMENT_H
\ No newline at end of file
diff --git a/src/editor/graph/nodes/graph_node_factory.cpp b/src/editor/graph/nodes/graph_node_factory.cpp
deleted file mode 100644
index 9de12454..00000000
--- a/src/editor/graph/nodes/graph_node_factory.cpp
+++ /dev/null
@@ -1,29 +0,0 @@
-// This file is part of the Godot Orchestrator project.
-//
-// Copyright (c) 2023-present Vahera Studios LLC and its contributors.
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//		http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-//
-#include "graph_node_factory.h"
-
-#include "editor/graph/nodes/graph_node_comment.h"
-#include "editor/graph/nodes/graph_node_default.h"
-
-OrchestratorGraphNode* OrchestratorGraphNodeFactory::create_node(OrchestratorGraphEdit* p_graph, const Ref<OScriptNode>& p_node)
-{
-    Ref<OScriptNodeComment> comment = p_node;
-    if (comment.is_valid())
-        return memnew(OrchestratorGraphNodeComment(p_graph, comment));
-
-    return memnew(OrchestratorGraphNodeDefault(p_graph, p_node));
-}
\ No newline at end of file
diff --git a/src/editor/graph/nodes/graph_node_factory.h b/src/editor/graph/nodes/graph_node_factory.h
deleted file mode 100644
index 8f0ce94a..00000000
--- a/src/editor/graph/nodes/graph_node_factory.h
+++ /dev/null
@@ -1,38 +0,0 @@
-// This file is part of the Godot Orchestrator project.
-//
-// Copyright (c) 2023-present Vahera Studios LLC and its contributors.
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//		http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-//
-#ifndef ORCHESTRATOR_GRAPH_NODE_FACTORY_H
-#define ORCHESTRATOR_GRAPH_NODE_FACTORY_H
-
-#include "editor/graph/graph_node.h"
-
-/// A simple OrchestratorGraphNode factory helper
-class OrchestratorGraphNodeFactory
-{
-    // Intentionally private
-    OrchestratorGraphNodeFactory() = default;
-
-public:
-
-    /// Creates the appropriate OrchestratorGraphNode implementation for the given graph and node.
-    /// @param p_graph the graph that will own the node
-    /// @param p_node the orchestration node reference
-    /// @return the editor graph node instance, never null
-    static OrchestratorGraphNode* create_node(OrchestratorGraphEdit* p_graph, const Ref<OScriptNode>& p_node);
-
-};
-
-#endif  // ORCHESTRATOR_GRAPH_NODE_FACTORY_H
diff --git a/src/editor/register_editor_types.cpp b/src/editor/register_editor_types.cpp
index 47b1c977..edf0820e 100644
--- a/src/editor/register_editor_types.cpp
+++ b/src/editor/register_editor_types.cpp
@@ -121,6 +121,9 @@ void register_editor_types()
     // Graph Node Type
     ORCHESTRATOR_REGISTER_INTERNAL_CLASS(OrchestratorGraphNodeDefault)
     ORCHESTRATOR_REGISTER_INTERNAL_CLASS(OrchestratorGraphNodeComment)
+    #if GODOT_VERSION >= 0x040300
+    ORCHESTRATOR_REGISTER_INTERNAL_CLASS(OrchestratorGraphFrameComment)
+    #endif
     ORCHESTRATOR_REGISTER_INTERNAL_CLASS(OrchestratorGraphKnot)
 
     // Pin Helpers
diff --git a/src/editor/theme/theme_cache.cpp b/src/editor/theme/theme_cache.cpp
index 2455c086..85f0428d 100644
--- a/src/editor/theme/theme_cache.cpp
+++ b/src/editor/theme/theme_cache.cpp
@@ -18,6 +18,7 @@
 
 #include "common/scene_utils.h"
 #include "common/settings.h"
+#include "common/version.h"
 
 #include <godot_cpp/classes/editor_interface.hpp>
 #include <godot_cpp/classes/graph_node.hpp>
@@ -40,6 +41,34 @@ void OrchestratorThemeCache::_settings_changed()
     const Color bkgrnd = settings->get_setting("ui/nodes/background_color", Color::html("#191d23"));
     const float bwidth = settings->get_setting("ui/nodes/border_width", 2);
 
+    #if GODOT_VERSION >= 0x040300
+    Ref<StyleBoxFlat> frame_panel = get_theme_stylebox("panel", "GraphFrame");
+    if (!frame_panel.is_valid())
+    {
+        Ref<StyleBoxFlat> new_frame_panel = _get_editor_theme_stylebox("panel", "GraphFrame")->duplicate(true);
+        if (new_frame_panel.is_valid())
+        {
+            add_theme_stylebox("panel", "GraphFrame", new_frame_panel);
+
+            Ref<StyleBoxFlat> new_frame_panel_selected = new_frame_panel->duplicate(true);
+            if (new_frame_panel_selected.is_valid())
+            {
+                new_frame_panel_selected->set_border_color(select);
+                add_theme_stylebox("panel_selected", "GraphFrame", new_frame_panel_selected);
+            }
+        }
+    }
+    else
+    {
+        Ref<StyleBoxFlat> panel_selected = get_theme_stylebox("panel_selected", "GraphFrame");
+        if (panel_selected.is_valid())
+        {
+            if (panel_selected->get_border_color() != select)
+                panel_selected->set_border_color(select);
+        }
+    }
+    #endif
+
     Ref<StyleBoxFlat> panel = get_theme_stylebox("panel", "GraphNode");
     if (!panel.is_valid())
     {
diff --git a/src/script/nodes/utilities/comment.cpp b/src/script/nodes/utilities/comment.cpp
index 3f3e125d..7222a6c7 100644
--- a/src/script/nodes/utilities/comment.cpp
+++ b/src/script/nodes/utilities/comment.cpp
@@ -26,6 +26,13 @@ void OScriptNodeComment::_get_property_list(List<PropertyInfo>* r_list) const
     r_list->push_back(PropertyInfo(Variant::INT, "font_size", PROPERTY_HINT_RANGE, "0,64"));
     r_list->push_back(PropertyInfo(Variant::COLOR, "text_color", PROPERTY_HINT_COLOR_NO_ALPHA));
     r_list->push_back(PropertyInfo(Variant::STRING, "comments", PROPERTY_HINT_MULTILINE_TEXT));
+
+    #if GODOT_VERSION >= 0x040300
+    // Tracks the internal logic version for this node type
+    r_list->push_back(PropertyInfo(Variant::INT, "state", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_STORAGE));
+    // Tracks the nodes that are attached/bounded to the comment frame
+    r_list->push_back(PropertyInfo(Variant::PACKED_INT64_ARRAY, "attachments", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_STORAGE));
+    #endif
 }
 
 bool OScriptNodeComment::_get(const StringName& p_name, Variant& r_value) const
@@ -60,6 +67,18 @@ bool OScriptNodeComment::_get(const StringName& p_name, Variant& r_value) const
         r_value = _title;
         return true;
     }
+    #if GODOT_VERSION >= 0x040300
+    else if (p_name.match("state"))
+    {
+        r_value = _state;
+        return true;
+    }
+    else if (p_name.match("attachments"))
+    {
+        r_value = _attachments;
+        return true;
+    }
+    #endif
     return false;
 }
 
@@ -101,6 +120,20 @@ bool OScriptNodeComment::_set(const StringName& p_name, const Variant& p_value)
         _notify_pins_changed();
         return true;
     }
+    #if GODOT_VERSION >= 0x040300
+    else if (p_name.match("state"))
+    {
+        _state = p_value;
+        _notify_pins_changed();
+        return true;
+    }
+    else if (p_name.match("attachments"))
+    {
+        _attachments = p_value;
+        _notify_pins_changed();
+        return true;
+    }
+    #endif
     return false;
 }
 
@@ -121,3 +154,16 @@ String OScriptNodeComment::get_icon() const
 {
     return "VisualShaderNodeComment";
 }
+
+#if GODOT_VERSION >= 0x040300
+void OScriptNodeComment::set_attachments(const PackedInt64Array& p_attachments)
+{
+    if (_attachments != p_attachments)
+    {
+        _attachments = p_attachments;
+        _state = State_Tracks_Attachments;
+
+        emit_changed();
+    }
+}
+#endif
\ No newline at end of file
diff --git a/src/script/nodes/utilities/comment.h b/src/script/nodes/utilities/comment.h
index b7a1f018..18fd04cc 100644
--- a/src/script/nodes/utilities/comment.h
+++ b/src/script/nodes/utilities/comment.h
@@ -25,6 +25,14 @@ class OScriptNodeComment : public OScriptNode
     ORCHESTRATOR_NODE_CLASS(OScriptNodeComment, OScriptNode);
     static void _bind_methods() { }
 
+public:
+    // State types for the node
+    enum State : int
+    {
+        State_Initial = 1, // Orchestrator 2.1.dev3 or before
+        State_Tracks_Attachments = 2 // Used by Orchestrator 2.1.dev4
+    };
+
 protected:
     String _comments;
     String _title{ "Comment" };
@@ -32,6 +40,10 @@ class OScriptNodeComment : public OScriptNode
     Color _background_color{ 0.6, 0.6, 0.6, 0.05 };
     Color _text_color{ 1.0, 1.0, 1.0, 1.0 };
     int _font_size{ 0 };
+    #if GODOT_VERSION >= 0x040300
+    int _state{ State_Initial };
+    PackedInt64Array _attachments;
+    #endif
 
     //~ Begin Wrapped Interface
     void _get_property_list(List<PropertyInfo> *r_list) const;
@@ -62,6 +74,20 @@ class OScriptNodeComment : public OScriptNode
     /// Get the comment text's font size
     /// @return the font size, 0 means use the default size
     int get_font_size() const { return _font_size; }
+
+    #if GODOT_VERSION >= 0x040300
+    /// Gets the comment node's current data state
+    /// @return the data state
+    int get_state() const { return _state; }
+
+    /// Gets the comment node's attachments
+    /// @return array of attached node ids
+    const PackedInt64Array& get_attachments() const { return _attachments; }
+
+    /// Sets the node's attachments
+    /// @param p_attachments the attachments to set
+    void set_attachments(const PackedInt64Array& p_attachments);
+    #endif
 };
 
 #endif  // ORCHESTRATOR_SCRIPT_NODE_COMMENT_H