Stimulus reflex concern (#464)

Gemfile.lock

@@ -91,7 +91,9 @@ GEM
     marcel (0.3.3)
       mimemagic (~> 0.3.2)
     method_source (0.9.2)
-    mimemagic (0.3.5)
+    mimemagic (0.3.10)
+      nokogiri (~> 1)
+      rake
     mini_mime (1.0.2)
     mini_portile2 (2.5.0)
     minitest (5.14.4)
@@ -191,4 +193,4 @@ DEPENDENCIES
   stimulus_reflex!
 
 BUNDLED WITH
-   2.1.4
+   2.2.4

lib/stimulus_reflex.rb

@@ -11,6 +11,7 @@
 require "cable_ready"
 require "stimulus_reflex/version"
 require "stimulus_reflex/cable_ready_channels"
+require "stimulus_reflex/concern_enhancer"
 require "stimulus_reflex/configuration"
 require "stimulus_reflex/callbacks"
 require "stimulus_reflex/reflex"

lib/stimulus_reflex/concern_enhancer.rb

@@ -0,0 +1,37 @@
+module StimulusReflex
+  module ConcernEnhancer
+    extend ActiveSupport::Concern
+
+    class_methods do
+      def method_missing(name, *args)
+        case ancestors
+        when ->(a) { !(a & [StimulusReflex::Reflex]).empty? }
+          if (ActiveRecord::Base.public_methods + ActionController::Base.public_methods).include? name
+            nil
+          else
+            super
+          end
+        when ->(a) { !(a & [ActiveRecord::Base, ActionController::Base]).empty? }
+          if StimulusReflex::Reflex.public_methods.include? name
+            nil
+          else
+            super
+          end
+        else
+          super
+        end
+      end
+
+      def respond_to_missing?(name, include_all = false)
+        case ancestors
+        when ->(a) { !(a & [StimulusReflex::Reflex]).empty? }
+          (ActiveRecord::Base.public_methods + ActionController::Base.public_methods).include?(name) || super
+        when ->(a) { !(a & [ActiveRecord::Base, ActionController::Base]).empty? }
+          StimulusReflex::Reflex.public_methods.include?(name) || super
+        else
+          super
+        end
+      end
+    end
+  end
+end

test/concern_enhancer_test.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require_relative "test_helper"
+
+class StimulusReflex::ConcernTest < ActiveSupport::TestCase
+  module TestConcern
+    extend ActiveSupport::Concern
+    include StimulusReflex::ConcernEnhancer
+  end
+
+  class TestReflex < StimulusReflex::Reflex
+    include TestConcern
+
+    def initialize
+    end
+  end
+
+  class TestController < ActionController::Base
+    include TestConcern
+  end
+
+  class TestModel < ActiveRecord::Base
+    include TestConcern
+  end
+
+  test "included in a reflex it stubs controller and model methods" do
+    assert_nil TestReflex.helper_method
+    assert_nil TestReflex.before_action
+    assert_nil TestReflex.around_action
+    assert_nil TestReflex.after_action
+
+    assert_nil TestReflex.before_save
+    assert_nil TestReflex.around_save
+    assert_nil TestReflex.after_save
+
+    refute_nil TestReflex.before_reflex
+  end
+
+  test "included in a controller it stubs reflex methods" do
+    assert_nil TestController.before_reflex
+    assert_nil TestController.around_reflex
+    assert_nil TestController.after_reflex
+
+    refute_nil TestController.before_action
+  end
+
+  test "included in a model it stubs reflex methods" do
+    assert_nil TestModel.before_reflex
+    assert_nil TestModel.around_reflex
+    assert_nil TestModel.after_reflex
+
+    refute_nil TestModel.after_save
+  end
+end