Merge pull request #3 from hoppergee/support_class_method

Support class method

Author: hoppergee

CHANGELOG.md

@@ -1,5 +1,9 @@
 ## [Unreleased]
 
+## [1.2.0] - 2022-11-21
+
+- Support `class_method`
+
 ## [1.1.1] - 2022-10-08
 
 - Change owner method define from `self.calss.define_method` to `self.define_singleton_method`

README.md

@@ -108,6 +108,27 @@ Say.hi('John')
 # => "Hi, John"
 ```
 
+### Work as a class method
+
+```ruby
+class BuildARobot < ActiveMethod::Base
+  argument :name
+
+  def call
+    "#{robot_factory} build a robot called #{name}"
+  end
+end
+
+class RobotFactory
+  include ActiveMethod
+
+  active_method :build_a_robot, class_method: true
+end
+
+RobotFactory.build_a_robot
+# => "RobotFactory build a robot called B-2"
+```
+
 ## Development
 
 ```bash

lib/active_method.rb

@@ -15,13 +15,21 @@ module ClassMethods
     def active_method(name, method_class = nil, **options)
       method_class ||= Util.constantize self, Util.camel_case(name)
 
-      define_method name do |*args|
-        method_class[self].call(*args)
-      end
+      if options[:class_method]
+        define_singleton_method name do |*args|
+          method_class[self].call(*args)
+        end
+
+      else
+        define_method name do |*args|
+          method_class[self].call(*args)
+        end
 
-      if options[:module_function]
-        module_function name
+        if options[:module_function]
+          module_function name
+        end
       end
+
     end
   end
 end

lib/active_method/base.rb

@@ -104,7 +104,8 @@ def initialize(*args)
     def __set_owner(owner)
       @__method_owner = owner
 
-      instance_name = Util.snake_case(owner.class.name.split("::").last)
+      klass = owner.is_a?(Class) ? owner : owner.class
+      instance_name = Util.snake_case(klass.name.split("::").last)
       self.define_singleton_method instance_name do
         @__method_owner
       end

lib/active_method/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ActiveMethod
-  VERSION = "1.1.1"
+  VERSION = "1.2.0"
 end

test/test_active_method.rb

@@ -105,4 +105,33 @@ class Dog
     assert_includes error.message, "undefined method `module_function'"
   end
 
+  ################
+  # .active_method for class method
+  ################
+
+  class BuildARobot < ActiveMethod::Base
+    argument :name
+
+    def call
+      "#{robot_factory} build a robot called #{name}"
+    end
+  end
+
+  class RobotFactory
+    include ActiveMethod
+
+    active_method :build_a_robot, class_method: true
+  end
+
+  it ".acive_method - works as a class method" do
+    assert_equal "TestActiveMethod::RobotFactory build a robot called B-2", RobotFactory.build_a_robot("B-2")
+  end
+
+  it ".acive_method - won't works as a instance method" do
+    error = assert_raises NoMethodError do
+      RobotFactory.new.build_a_robot("B-2")
+    end
+    assert_includes error.message, "undefined method `build_a_robot'"
+  end
+
 end