- better detection of method visibility changes

lib/action_cable/connection/base.rb

@@ -162,8 +162,15 @@ def send_welcome_message
   end
 end
 
-ActiveSupport.run_load_hooks(:action_cable_connection, ActionCable::Connection::Base)
-
 # Hack to make sure legacy extensions do not changes the visibility of API methods
 # (see https://github.com/anycable/actioncable-next/issues/7)
-ActionCable::Connection::Base.send :public, :handle_open, :handle_close, :handle_channel_command, :transmit, :close
+# We call it from the engine after application initialization.
+class << ActionCable::Connection::Base
+  def __restore_visibility__
+    %i[handle_open handle_close handle_channel_command transmit close].each do |method|
+      instance_method(method).owner.send(:public, method)
+    end
+  end
+end
+
+ActiveSupport.run_load_hooks(:action_cable_connection, ActionCable::Connection::Base)

lib/action_cable/engine.rb

@@ -94,5 +94,11 @@ class Engine < Rails::Engine # :nodoc:
         end
       end
     end
+
+    config.after_initialize do
+      ActiveSupport.on_load(:action_cable_connection) do
+        ActionCable::Connection::Base.__restore_visibility__
+      end
+    end
   end
 end