module CouchPotato
  module Persistence
    module Callbacks
      def self.included(base)
        base.extend ClassMethods
        
        base.class_eval do
          attr_accessor :skip_callbacks
          def self.callbacks
            @@callbacks ||= {}
            @@callbacks[self.name] ||= {:before_validation_on_create => [], 
              :before_validation_on_update => [], :before_validation_on_save => [], :before_create => [], 
              :after_create => [], :before_update => [], :after_update => [],
              :before_save => [], :after_save => [],
              :before_destroy => [], :after_destroy => []}
          end
        end
      end
      
      def save_without_callbacks
        self.skip_callbacks = true
        result = save
        self.skip_callbacks = false
        result
      end
      
      private
      
      def run_callbacks(name)
        return if skip_callbacks
        self.class.callbacks[name].each do |callback|
          self.send callback
        end
      end
      
      module ClassMethods
        def before_validation_on_create(*names)
          names.each do |name|
            callbacks[:before_validation_on_create] << name
          end
        end
        
        def before_validation_on_update(*names)
          names.each do |name|
            callbacks[:before_validation_on_update] << name
          end
        end
        
        def before_validation_on_save(*names)
          names.each do |name|
            callbacks[:before_validation_on_save] << name
          end
        end
        
        def before_create(*names)
          names.each do |name|
            callbacks[:before_create] << name
          end
        end
        
        def before_save(*names)
          names.each do |name|
            callbacks[:before_save] << name
          end
        end
        
        def before_update(*names)
          names.each do |name|
            callbacks[:before_update] << name
          end
        end
        
        def before_destroy(*names)
          names.each do |name|
            callbacks[:before_destroy] << name
          end
        end

        def after_update(*names)
          names.each do |name|
            callbacks[:after_update] << name
          end
        end
        
        def after_save(*names)
          names.each do |name|
            callbacks[:after_save] << name
          end
        end
        
        def after_create(*names)
          names.each do |name|
            callbacks[:after_create] << name
          end
        end

        def after_destroy(*names)
          names.each do |name|
            callbacks[:after_destroy] << name
          end
        end
      end
    end
  end
end