Implement .active_method

Author: hoppergee

lib/active_method.rb

@@ -4,4 +4,21 @@
 
 module ActiveMethod
   autoload :Base, "active_method/base"
+  autoload :Util, "active_method/util"
+
+  def self.included(base)
+    base.extend ClassMethods
+  end
+
+  module ClassMethods
+    def active_method(name, method_class = nil)
+      method_class ||= Util.constantize self, Util.camel_case(name)
+
+      define_method name do |*args|
+        method = method_class.new(*args)
+        method.__set_owner(self)
+        method.call
+      end
+    end
+  end
 end

lib/active_method/base.rb

@@ -99,6 +99,15 @@ def initialize(*args)
     def call
     end
 
+    def __set_owner(owner)
+      @__method_owner = owner
+
+      instance_name = Util.snake_case(owner.class.name.split("::").last)
+      self.class.define_method instance_name do
+        @__method_owner
+      end
+    end
+
     private
 
     def arguments

test/test_active_method.rb

@@ -8,4 +8,51 @@ def test_that_it_has_a_version_number
     refute_nil ::ActiveMethod::VERSION
   end
 
+  ################
+  # .active_method
+  ################
+
+  class Buzz < ActiveMethod::Base
+    argument :a
+    argument :b
+    keyword_argument :c
+    keyword_argument :d
+
+    def call
+      [a, b, c, d, foo_bar.x, foo_bar.y, foo_bar.z]
+    end
+  end
+
+  class BAZ < ActiveMethod::Base
+    argument :a
+    argument :b
+    keyword_argument :c
+    keyword_argument :d
+
+    def call
+      [a, b, c, d, foo_bar.x, foo_bar.y, foo_bar.z]
+    end
+  end
+
+  module Example
+    class FooBar
+      include ActiveMethod
+
+      active_method :buzz
+      active_method :baz, BAZ
+
+      def x; 'x'; end
+      def y; 'y'; end
+      def z; 'z'; end
+    end
+  end
+
+  it ".active_method - infer method class base on name" do
+    assert_equal ['a', 'b', 'c', 'd', 'x', 'y', 'z'], Example::FooBar.new.buzz('a', 'b', c: 'c', d: 'd')
+  end
+
+  it ".active_method - direct set method class" do
+    assert_equal ['a', 'b', 'c', 'd', 'x', 'y', 'z'], Example::FooBar.new.baz('a', 'b', c: 'c', d: 'd')
+  end
+
 end