module Danbooru
  module Extensions
    module ActiveRecord
      extend ActiveSupport::Concern
      
      module ClassMethods
        def without_timeout
          connection.execute("SET STATEMENT_TIMEOUT = 0") unless Rails.env == "test"
          yield
        ensure
          connection.execute("SET STATEMENT_TIMEOUT = 5000") unless Rails.env == "test"
        end

        def with_timeout(n)
          connection.execute("SET STATEMENT_TIMEOUT = #{n}") unless Rails.env == "test"
          yield
        ensure
          connection.execute("SET STATEMENT_TIMEOUT = 5000") unless Rails.env == "test"
        end
      end
      
      %w(execute select_value select_values select_all).each do |method_name|
        define_method("#{method_name}_sql") do |sql, *params|
          connection.__send__(method_name, self.class.sanitize_sql_array([sql, *params]))
        end

        self.class.__send__(:define_method, "#{method_name}_sql") do |sql, *params|
          connection.__send__(method_name, sanitize_sql_array([sql, *params]))
        end
      end
      
      def arbitrary_sql_order_clause(ids, table_name = nil)
        table_name = self.class.table_name if table_name.nil?
        
        if ids.empty?
          return "#{table_name}.id desc"
        end
        
        conditions = []
        
        ids.each_with_index do |x, n|
          conditions << "when #{x} then #{n}"
        end
        
        "case #{table_name}.id " + conditions.join(" ") + " end"
      end
      
      def serializable_hash(options = {})
        options = {:except => [:uploader_ip_addr, :updater_ip_addr, :creator_ip_addr, :ip_addr]}.merge(options ||= {})
        super(options)
      end
    end
  end
end

class ActiveRecord::Base
  class << self    
    public :sanitize_sql_array
  end

  include Danbooru::Extensions::ActiveRecord
end