lib/reverse_parameters.rb in reverse_parameters-1.0.0 vs lib/reverse_parameters.rb in reverse_parameters-1.1.0

@@ -1,43 +1,52 @@
 require "reverse_parameters/version"
+require "reverse_parameters/core_ext/refinements"
 
-class ReverseParameters
+module ReverseParameters
 
   # @param [Proc, Array] input
-  def initialize(input)
-    if input.respond_to?(:to_proc)
-      @params = input.to_proc.parameters
-    elsif input.respond_to?(:to_ary)
-      @params = input.to_ary
-    else
-      raise ArgumentError.new("Input must be an Array of parameters or a Proc object.")
-    end
+  def self.new(*args)
+    Base.new(*args)
   end
 
-  # Method parameters are the names listed in the function definition.
-  # @return [ReverseParameters::Parameters]
-  def parameters
-    Parameters.new(params)
-  end
+  class Base
 
-  # Method arguments are the real values passed to (and received by) the function.
-  # @return [ReverseParameters::Arguments]
-  # @param [true, false] blocks_as_values: express block as variable vs a proc passed to the end of a method.
-  # def my_method(&block)
-  # end
-  # ReverseParameters.new(method(:my_method)).arguments(blocks_as_values: true).to_s
-  #   #=> "block"
-  #
-  # # ReverseParameters.new(method(:my_method)).arguments.to_s
-  #   #=> "&block"
-  def arguments(blocks_as_values: false)
-    Arguments.new(params, blocks_as_values: blocks_as_values)
-  end
+    # @param [Proc, Array] input
+    def initialize(input)
+      if input.respond_to?(:to_proc)
+        @params = input.to_proc.parameters
+      elsif input.respond_to?(:to_ary)
+        @params = input.to_ary
+      else
+        raise ArgumentError.new("Input must be an Array of parameters or a Proc object.")
+      end
+    end
 
-  private
-  attr_reader :params
+    # Method parameters are the names listed in the function definition.
+    # @return [ReverseParameters::Parameters]
+    def parameters
+      Parameters.new(params)
+    end
 
+    # Method arguments are the real values passed to (and received by) the function.
+    # @return [ReverseParameters::Arguments]
+    # @param [true, false] blocks_as_values: express block as variable vs a proc passed to the end of a method.
+    # def my_method(&block)
+    # end
+    # ReverseParameters.new(method(:my_method)).arguments(blocks_as_values: true).to_s
+    #   #=> "block"
+    #
+    # # ReverseParameters.new(method(:my_method)).arguments.to_s
+    #   #=> "&block"
+    def arguments(blocks_as_values: false)
+      Arguments.new(params, blocks_as_values: blocks_as_values)
+    end
+
+    private
+    attr_reader :params
+  end
+
   class BaseCollection
     include Enumerable
 
     def initialize(collection, **options)
       @collection = collection.map { |state, name| item_class.new(state: state, name: name, **options) }
@@ -72,16 +81,16 @@
   class Arguments < BaseCollection
 
     class Arg < BaseCollection::Item
       def to_s
         case state
-          when :key, :keyreq
-            "#{name}: #{name}"
-          when :block
-            block(name)
-          else
-            name
+        when :key, :keyreq
+          "#{name}: #{name}"
+        when :block
+          block(name)
+        else
+          name
         end.to_s
       end
 
       def block_as_value
         case state
@@ -120,23 +129,23 @@
   end
   class Parameters < BaseCollection;
     class Param < BaseCollection::Item
       def to_s
         case state
-          when :req
-            name
-          when :rest
-            "*#{name}"
-          when :keyrest
-            "**#{name}"
-          when :opt
-            "#{name}=nil"
-          when :keyreq
-            "#{name}:"
-          when :key
-            "#{name}: nil"
-          when :block
-            "&#{name}"
+        when :req
+          name
+        when :rest
+          "*#{name}"
+        when :keyrest
+          "**#{name}"
+        when :opt
+          "#{name}=nil"
+        when :keyreq
+          "#{name}:"
+        when :key
+          "#{name}: nil"
+        when :block
+          "&#{name}"
         end.to_s
       end
     end
 
     def item_class