Extract module + include

Author: BlakeWilliams

lib/view_component/action_view_compatibility.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module ViewComponent
+  module ActionViewCompatibility
+    attr_accessor :top_level_parent
+
+    def children
+      @_children ||= []
+    end
+
+    def set_output_buffer(buf)
+      @output_buffer = buf
+    end
+
+    def output_buffer=(buf)
+      if top_level_parent
+        top_level_parent.set_output_buffer_for_children(buf)
+      else
+        set_output_buffer_for_children(buf)
+      end
+    end
+
+    def set_output_buffer_for_children(buf)
+      set_output_buffer(buf)
+
+      children.each do |child|
+        child.set_output_buffer(buf)
+      end
+    end
+  end
+end

lib/view_component/base.rb

@@ -10,53 +10,6 @@
 require "view_component/slotable_v2"
 require "view_component/with_content_helper"
 
-class ActionView::Base
-  attr_reader :children
-  attr_accessor :parent
-
-  def _prepare_context
-    @children = []
-    @parent = nil
-
-    super
-  end
-
-  def set_output_buffer(buf)
-    @output_buffer = buf
-  end
-
-  def output_buffer=(buf)
-    if parent.nil?
-      set_output_buffer_for_children(buf)
-    else
-      top_level_parent.set_output_buffer_for_children(buf)
-    end
-  end
-
-  def top_level_parent
-    return if parent.nil?
-    return @_top_level_parent if defined?(@_top_level_parent)
-
-    @_top_level_parent = begin
-      top_level = parent
-
-      while top_level.parent
-        top_level = top_level.parent
-      end
-
-      top_level
-    end
-  end
-
-  def set_output_buffer_for_children(buf)
-    set_output_buffer(buf)
-
-    children.each do |child|
-      child.set_output_buffer(buf)
-    end
-  end
-end
-
 module ViewComponent
   class Base < ActionView::Base
     include ActiveSupport::Configurable

lib/view_component/engine.rb

@@ -81,6 +81,13 @@ class Engine < Rails::Engine # :nodoc:
       end
     end
 
+    initializer "view_component.include_render_compatibility" do |app|
+      ActiveSupport.on_load(:action_view) do
+        require "view_component/action_view_compatibility"
+        ActionView::Base.prepend ViewComponent::ActionViewCompatibility
+      end
+    end
+
     initializer "view_component.include_render_component" do |app|
       next if Rails.version.to_f >= 6.1