lib/sn_foil/contexts/update_context.rb in snfoil-0.3.0 vs lib/sn_foil/contexts/update_context.rb in snfoil-0.4.0

@@ -13,17 +13,24 @@
         include SetupContext
         include ChangeContext
       end
 
       class_methods do
-        attr_reader :i_before_update_hooks, :i_after_update_hooks, :i_after_update_success_hooks, :i_after_update_failure_hooks
+        attr_reader :i_setup_update_hooks, :i_before_update_hooks, :i_after_update_hooks,
+                    :i_after_update_success_hooks, :i_after_update_failure_hooks
         def update(id:, params:, user: nil, **options)
           new(user).update(**options, id: id, params: params)
         end
 
+        def setup_update(method = nil, **options, &block)
+          raise ArgumentError, '#setup_update requires either a method name or a block' if method.nil? && block.nil?
+
+          (@i_setup_update_hooks ||= []) << { method: method, block: block, if: options[:if], unless: options[:unless] }
+        end
+
         def before_update(method = nil, **options, &block)
-          raise ArgumentError, '#on_update requires either a method name or a block' if method.nil? && block.nil?
+          raise ArgumentError, '#before_update requires either a method name or a block' if method.nil? && block.nil?
 
           (@i_before_update_hooks ||= []) << { method: method, block: block, if: options[:if], unless: options[:unless] }
         end
 
         def after_update(method = nil, **options, &block)
@@ -54,11 +61,11 @@
         options.merge! object: object
       end
 
       def update(**options)
         options[:action] = :update
-        options = setup_change(setup_update(**options))
+        options = before_setup_update_object(**options)
         options = setup_update_object(**options)
         authorize(options[:object], :update?, **options)
         options = update_hooks(**options)
         unwrap_object(options[:object])
       end
@@ -81,10 +88,14 @@
 
       def after_update_failure(**options)
         options
       end
 
+      def setup_update_hooks
+        self.class.i_setup_update_hooks || []
+      end
+
       def before_update_hooks
         self.class.i_before_update_hooks || []
       end
 
       def after_update_hooks
@@ -98,9 +109,18 @@
       def after_update_failure_hooks
         self.class.i_after_update_failure_hooks || []
       end
 
       private
+
+      def before_setup_update_object(**options)
+        options = setup_update(**options)
+        options = setup_update_hooks.reduce(options) { |opts, hook| run_hook(hook, opts) }
+        options = setup_change(**options)
+        options = setup_change_hooks.reduce(options) { |opts, hook| run_hook(hook, opts) }
+        options = setup(**options)
+        setup_hooks.reduce(options) { |opts, hook| run_hook(hook, opts) }
+      end
 
       # This method is private to help protect the order of execution of hooks
       def update_hooks(options)
         options = before_update_save(options)
         options = if options[:object].save