module Rakismet
  module Model
   
    def self.included(base)
      base.class_eval do
        attr_accessor :akismet_response
        class_inheritable_hash :akismet_attrs
        extend ClassMethods
        include InstanceMethods
        self.rakismet_attrs
      end
    end
   
    module ClassMethods
      def rakismet_attrs(args={})
        self.akismet_attrs ||= {}
        [:comment_type, :author, :author_url, :author_email, :content].each do |field|
           # clunky, but throwing around +type+ will break your heart
           fieldname = field.to_s =~ %r(^comment_) ? field : "comment_#{field}".intern
           self.akismet_attrs[fieldname] = args.delete(field) || field
        end
        [:user_ip, :user_agent, :referrer].each do |field|
          self.akismet_attrs[field] = args.delete(field)
        end
        args.each_pair do |f,v|
          self.akismet_attrs[f] = v
        end
      end
    end
    
    module InstanceMethods
      def spam?
        data = akismet_data

        unless Rakismet::Base.rakismet_binding.nil?
          { :referrer => 'request.referer', :user_ip => 'request.remote_ip',
            :user_agent => 'request.user_agent' }.each_pair do |k,v|
              data[k] = eval(v, Rakismet::Base.rakismet_binding) || ''
          end
        end

        self.akismet_response = Rakismet::Base.akismet_call('comment-check', data)
        self.akismet_response == 'true'
      end

      def spam!
        Rakismet::Base.akismet_call('submit-spam', akismet_data)
      end

      def ham!
        Rakismet::Base.akismet_call('submit-ham', akismet_data)
      end

      private

        def akismet_data
          self.class.akismet_attrs.keys.inject({}) do |data,attr|
            data.merge attr =>  if self.class.akismet_attrs[attr].is_a?(Proc)
                                  instance_eval(&self.class.akismet_attrs[attr])
                                elsif !self.class.akismet_attrs[attr].nil? && respond_to?(self.class.akismet_attrs[attr])
                                  send(self.class.akismet_attrs[attr])
                                elsif !self.class.akismet_attrs[attr].nil?
                                  self.class.akismet_attrs[attr]
                                elsif respond_to?(attr)
                                  send(attr)
                                end
          end.delete_if { |k,v| v.blank? }
        end
    end
    
  end
end