Allow broadcasting actions without rendering, and passing stream tag …

…attributes

In order to allow custom Turbo actions that don't require a template and work only via attributes passed to the turbo_stream_tag.

app/channels/turbo/streams/broadcasts.rb

@@ -33,9 +33,10 @@ def broadcast_prepend_to(*streamables, **opts)
     broadcast_action_to(*streamables, action: :prepend, **opts)
   end
 
-  def broadcast_action_to(*streamables, action:, target: nil, targets: nil, **rendering)
+  def broadcast_action_to(*streamables, action:, target: nil, targets: nil, attributes: {}, **rendering)
     broadcast_stream_to(*streamables, content: turbo_stream_action_tag(action, target: target, targets: targets, template:
-      rendering.delete(:content) || rendering.delete(:html) || (rendering.any? ? render_format(:html, **rendering) : nil)
+      rendering.delete(:content) || rendering.delete(:html) || (rendering[:render] != false && rendering.any? ? render_format(:html, **rendering) : nil),
+      **attributes
     ))
   end
 
@@ -63,9 +64,9 @@ def broadcast_prepend_later_to(*streamables, **opts)
     broadcast_action_later_to(*streamables, action: :prepend, **opts)
   end
 
-  def broadcast_action_later_to(*streamables, action:, target: nil, targets: nil, **rendering)
+  def broadcast_action_later_to(*streamables, action:, target: nil, targets: nil, attributes: {}, **rendering)
     Turbo::Streams::ActionBroadcastJob.perform_later \
-      stream_name_from(streamables), action: action, target: target, targets: targets, **rendering
+      stream_name_from(streamables), action: action, target: target, targets: targets, attributes: attributes, **rendering
   end
 
   def broadcast_render_to(*streamables, **rendering)

app/jobs/turbo/streams/action_broadcast_job.rb

@@ -2,7 +2,7 @@
 class Turbo::Streams::ActionBroadcastJob < ActiveJob::Base
   discard_on ActiveJob::DeserializationError
 
-  def perform(stream, action:, target:, **rendering)
-    Turbo::StreamsChannel.broadcast_action_to stream, action: action, target: target, **rendering
+  def perform(stream, action:, target:, attributes: {}, **rendering)
+    Turbo::StreamsChannel.broadcast_action_to stream, action: action, target: target, attributes: attributes, **rendering
   end
 end

app/models/concerns/turbo/broadcastable.rb

@@ -236,13 +236,13 @@ def broadcast_prepend(target: broadcast_target_default, **rendering)
   #   # Sends <turbo-stream action="prepend" target="clearances"><template><div id="clearance_5">My Clearance</div></template></turbo-stream>
   #   # to the stream named "identity:2:clearances"
   #   clearance.broadcast_action_to examiner.identity, :clearances, action: :prepend, target: "clearances"
-  def broadcast_action_to(*streamables, action:, target: broadcast_target_default, **rendering)
-    Turbo::StreamsChannel.broadcast_action_to(*streamables, action: action, target: target, **broadcast_rendering_with_defaults(rendering))
+  def broadcast_action_to(*streamables, action:, target: broadcast_target_default, attributes: {}, **rendering)
+    Turbo::StreamsChannel.broadcast_action_to(*streamables, action: action, target: target, attributes: attributes, **broadcast_rendering_with_defaults(rendering))
   end
 
   # Same as <tt>#broadcast_action_to</tt>, but the designated stream is automatically set to the current model.
-  def broadcast_action(action, target: broadcast_target_default, **rendering)
-    broadcast_action_to self, action: action, target: target, **rendering
+  def broadcast_action(action, target: broadcast_target_default, attributes: {}, **rendering)
+    broadcast_action_to self, action: action, target: target, attributes: attributes, **rendering
   end
 
 
@@ -287,13 +287,13 @@ def broadcast_prepend_later(target: broadcast_target_default, **rendering)
   end
 
   # Same as <tt>broadcast_action_to</tt> but run asynchronously via a <tt>Turbo::Streams::BroadcastJob</tt>.
-  def broadcast_action_later_to(*streamables, action:, target: broadcast_target_default, **rendering)
-    Turbo::StreamsChannel.broadcast_action_later_to(*streamables, action: action, target: target, **broadcast_rendering_with_defaults(rendering))
+  def broadcast_action_later_to(*streamables, action:, target: broadcast_target_default, attributes: {}, **rendering)
+    Turbo::StreamsChannel.broadcast_action_later_to(*streamables, action: action, target: target, attributes: attributes, **broadcast_rendering_with_defaults(rendering))
   end
 
   # Same as <tt>#broadcast_action_later_to</tt>, but the designated stream is automatically set to the current model.
-  def broadcast_action_later(action:, target: broadcast_target_default, **rendering)
-    broadcast_action_later_to self, action: action, target: target, **rendering
+  def broadcast_action_later(action:, target: broadcast_target_default, attributes: {}, **rendering)
+    broadcast_action_later_to self, action: action, target: target, attributes: attributes, **rendering
   end
 
   # Render a turbo stream template with this broadcastable model passed as the local variable. Example:
@@ -354,6 +354,8 @@ def broadcast_rendering_with_defaults(options)
           return o
         elsif o[:template]
           o[:layout] = false
+        elsif o[:render] == false
+          return o
         else
           # if none of these options are passed in, it will set a partial from #to_partial_path
           o[:partial] ||= to_partial_path

test/streams/broadcastable_test.rb

@@ -108,6 +108,65 @@ class Turbo::BroadcastableTest < ActionCable::Channel::TestCase
     end
   end
 
+  test "broadcasting action with attributes" do
+    assert_broadcast_on @message.to_gid_param, turbo_stream_action_tag("prepend", target: "messages", template: render(@message), "data-foo" => "bar") do
+      @message.broadcast_action "prepend", target: "messages", attributes: { "data-foo" => "bar" }
+    end
+  end
+
+  test "broadcasting action with no rendering" do
+    assert_broadcast_on @message.to_gid_param, turbo_stream_action_tag("prepend", target: "messages", template: nil) do
+      @message.broadcast_action "prepend", target: "messages", render: false
+    end
+  end
+
+  test "broadcasting action to with attributes" do
+    assert_broadcast_on "stream", turbo_stream_action_tag("prepend", target: "messages", template: render(@message), "data-foo" => "bar") do
+      @message.broadcast_action_to "stream", action: "prepend", attributes: { "data-foo" => "bar" }
+    end
+  end
+
+  test "broadcasting action to with no rendering" do
+    assert_broadcast_on "stream", turbo_stream_action_tag("prepend", target: "messages", template: nil) do
+      @message.broadcast_action_to "stream", action: "prepend", render: false
+    end
+  end
+
+  # The below commented tests pass if you change `Turbo::Streams::Broadcasts#broadcast_action_later_to`'s
+  # job call from `perform_later` to `perform_now`, but for some reason `perform_enqueued_jobs` is not running them.
+
+  # test "broadcasting action later to with attributes" do
+  #   assert_broadcast_on @message.to_gid_param, turbo_stream_action_tag("prepend", target: "messages", template: render(@message), "data-foo" => "bar") do
+  #     perform_enqueued_jobs do
+  #       @message.broadcast_action_later_to @message, action: "prepend", target: "messages", attributes: { "data-foo" => "bar" }
+  #     end
+  #   end
+  # end
+
+  # test "broadcasting action later to with no rendering" do
+  #   assert_broadcast_on @message.to_gid_param, turbo_stream_action_tag("prepend", target: "messages", template: nil) do
+  #     perform_enqueued_jobs do
+  #       @message.broadcast_action_later_to @message, action: "prepend", target: "messages", render: false
+  #     end
+  #   end
+  # end
+
+  # test "broadcasting action later with attributes" do
+  #   assert_broadcast_on @message.to_gid_param, turbo_stream_action_tag("prepend", target: "messages", template: render(@message), "data-foo" => "bar") do
+  #     perform_enqueued_jobs do
+  #       @message.broadcast_action_later action: "prepend", target: "messages", attributes: { "data-foo" => "bar" }
+  #     end
+  #   end
+  # end
+
+  # test "broadcasting action later with no rendering" do
+  #   assert_broadcast_on @message.to_gid_param, turbo_stream_action_tag("prepend", target: "messages", template: nil) do
+  #     perform_enqueued_jobs do
+  #       @message.broadcast_action_later action: "prepend", target: "messages", render: false
+  #     end
+  #   end
+  # end
+
   test "render correct local name in partial for namespaced models" do
     @profile = Users::Profile.new(id: 1, name: "Ryan")
     assert_broadcast_on @profile.to_param, turbo_stream_action_tag("replace", target: "users_profile_1", template: "<p>Ryan</p>\n") do

test/system/broadcasts_test.rb

@@ -1,6 +1,8 @@
 require "application_system_test_case"
 
 class BroadcastsTest < ApplicationSystemTestCase
+  include ActiveJob::TestHelper
+
   test "Message broadcasts Turbo Streams" do
     visit messages_path
     wait_for_stream_to_be_connected
@@ -19,6 +21,67 @@ class BroadcastsTest < ApplicationSystemTestCase
     end
   end
 
+  test "Message broadcasts with extra attributes to turbo stream tag" do
+    visit messages_path
+    subscribe_to_broadcasts
+
+    assert_broadcasts_text "Message 1", to: :messages do |text, target|
+      Message.create(content: text).broadcast_action_to(target, action: :append, attributes: { "data-foo": "bar" })
+    end
+  end
+
+  test "Message broadcasts with correct extra attributes to turbo stream tag" do
+    visit messages_path
+    subscribe_to_broadcasts
+
+    assert_forwards_turbo_stream_tag_attribute attr_key: "data-foo", attr_value: "bar", to: :messages do |attr_key, attr_value, target|
+      Message.create(content: text).broadcast_action_to(target, action: :test, attributes: { attr_key => attr_value })
+    end
+  end
+
+  test "Message broadcasts with no rendering" do
+    visit messages_path
+    subscribe_to_broadcasts
+
+    assert_forwards_turbo_stream_tag_attribute attr_key: "data-foo", attr_value: "bar", to: :messages do |attr_key, attr_value, target|
+      Message.create(content: text).broadcast_action_to(target, action: :test, render: false, partial: "non_existant", attributes: { attr_key => attr_value })
+    end
+  end
+
+  test "Message broadcasts later with extra attributes to turbo stream tag" do
+    visit messages_path
+    subscribe_to_broadcasts
+
+    perform_enqueued_jobs do
+      assert_broadcasts_text "Message 1", to: :messages do |text, target|
+        Message.create(content: text).broadcast_action_later_to(target, action: :append, attributes: { "data-foo": "bar" })
+      end
+    end
+  end
+
+
+  test "Message broadcasts later with correct extra attributes to turbo stream tag" do
+    visit messages_path
+    subscribe_to_broadcasts
+
+    perform_enqueued_jobs do
+      assert_forwards_turbo_stream_tag_attribute attr_key: "data-foo", attr_value: "bar", to: :messages do |attr_key, attr_value, target|
+        Message.create(content: text).broadcast_action_later_to(target, action: :test, attributes: { attr_key => attr_value })
+      end
+    end
+  end
+
+  test "Message broadcasts later with no rendering" do
+    visit messages_path
+    subscribe_to_broadcasts
+
+    perform_enqueued_jobs do
+      assert_forwards_turbo_stream_tag_attribute attr_key: "data-foo", attr_value: "bar", to: :messages do |attr_key, attr_value, target|
+        Message.create(content: text).broadcast_action_to(target, action: :test, render: false, partial: "non_existant", attributes: { attr_key => attr_value })
+      end
+    end
+  end
+
   test "Users::Profile broadcasts Turbo Streams" do
     visit users_profiles_path
     wait_for_stream_to_be_connected
@@ -50,4 +113,20 @@ def assert_broadcasts_text(text, to:, &block)
 
     within(:element, id: to) { assert_text text }
   end
+
+  def assert_forwards_turbo_stream_tag_attribute(attr_key:, attr_value:, to:, &block)
+    execute_script(<<~SCRIPT)
+      Turbo.StreamActions.test = function () {
+        const attribute = this.getAttribute('#{attr_key}')
+
+        document.getElementById('#{to}').innerHTML = attribute
+      }
+    SCRIPT
+
+    within(:element, id: to) { assert_no_text attr_value }
+
+    [attr_key, attr_value, to].yield_self(&block)
+
+    within(:element, id: to) { assert_text attr_value }
+  end
 end