Added handling for crutch dependencies (#13)

apolloio · Sep 10, 2024 · 754c612 · 754c612
1 parent 023147d
commit 754c612
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/chewy/type/crutch.rb b/lib/chewy/type/crutch.rb
@@ -15,7 +15,12 @@ def initialize(type, collection)
           @type._crutches.each_key do |name|
             singleton_class.class_eval <<-METHOD, __FILE__, __LINE__ + 1
               def #{name}
-                @#{name} ||= @type._crutches[:#{name}].call @collection
+                execution_block = @type._crutches[:#{name}]
+                if execution_block.arity == 1
+                  @#{name} ||= execution_block.call @collection
+                else
+                  @#{name} ||= execution_block.call @collection, self
+                end
               end
             METHOD
           end