Filter output helpers with Nokogiri

lib/phlex/rails/helper_macros.rb

@@ -7,18 +7,23 @@ def register_output_helper(method_name)
 			# frozen_string_literal: true
 
 			def #{method_name}(*args, **kwargs, &block)
-  			context = @_context
-  			return if context.fragments && !context.in_target_fragment
 
 				output = if block
 					helpers.#{method_name}(*args, **kwargs) { capture(&block) }
 				else
 					helpers.#{method_name}(*args, **kwargs)
 				end
 
+  			context = @_context
+
+				fragments = context.fragments
+				if fragments && !context.in_target_fragment
+					output = Phlex::Rails::FragmentFinder.extract(output, fragments).html_safe
+				end
+
 				case output
 				when ActiveSupport::SafeBuffer
-					@_context.target << output
+					context.target << output
 				end
 
 				nil
@@ -47,8 +52,6 @@ def register_builder_yielding_helper(method_name, builder)
 			# frozen_string_literal: true
 
 			def #{method_name}(*args, **kwargs)
-  			context = @_context
-  			return if context.fragments && !context.in_target_fragment
 				output = if block_given?
 					helpers.#{method_name}(*args, **kwargs) { |form|
 						capture do
@@ -63,9 +66,16 @@ def #{method_name}(*args, **kwargs)
 					helpers.#{method_name}(*args, **kwargs)
 				end
 
+				context = @_context
+
+				fragments = context.fragments
+				if fragments && !context.in_target_fragment
+					output = Phlex::Rails::FragmentFinder.extract(output, fragments).html_safe
+				end
+
 				case output
 				when ActiveSupport::SafeBuffer
-					@_context.target << output
+					context.target << output
 				end
 
 				nil

spec/fragment_filter_spec.rb → spec/fragment_finder_spec.rb

@@ -2,6 +2,16 @@
 
 describe Phlex::Rails::FragmentFinder do
 	include	Phlex::Rails::FragmentFinder
+	it "find nothing" do
+		expect(
+			extract(<<~HTML, ["d"])
+				<div id="a">A</div>
+				<div id="b">B</div>
+				<div id="c">c</div>
+			HTML
+		).to be == ""
+	end
+
 	it "find one" do
 		expect(
 			extract(<<~HTML, ["b"])