require 'ipaddr'
require 'set'
require 'andand'
require 'active_support'
require 'active_support/version'
%w{
  active_support/core_ext/object/blank
}.each do |active_support_3_requirement|
  require active_support_3_requirement
end if ActiveSupport::VERSION::MAJOR == 3
require 'active_record'
require 'fast_timestamp'
require 'honeypot/ipaddr_ext'
require 'honeypot/remote_request'
require 'honeypot/remote_host'
require 'honeypot/best_guess_routeable_remote_ip'

require 'honeypot/railtie' if defined? ::Rails::Railtie

module Honeypot
  def self.included(base)
    base.class_eval do
      has_many :remote_requests, :as => :requestable, :dependent => :destroy
      has_many :remote_hosts, :through => :remote_requests, :uniq => true
    end
  end
  
  def log_action_dispatch_request(request)
    log_remote_request request.env['honeypot.best_guess_routeable_remote_ip'], request.url, request.referer
  end
  
  def log_rack_env(env)
    request = ::Rack::Request.new env
    log_remote_request request.env['honeypot.best_guess_routeable_remote_ip'], request.url, request.referer
  end
  
  def log_remote_request(ip_address, url, referer)
    remote_host = RemoteHost.find_or_create_by_ip_address ip_address
    remote_request = remote_requests.find_or_create_by_remote_host_id remote_host.id
    remote_request.last_http_referer = referer
    remote_request.last_request_uri = url
    remote_request.increment :hits
    remote_request.save!
    true
  end
  
  def related_requestables(seen_remote_host_ids = Array.new)
    set = Set.new
    conditions = seen_remote_host_ids.present? ? [ "remote_hosts.id NOT IN (?)", seen_remote_host_ids ] : nil
    remote_hosts.where(conditions).find_in_batches do |batch|
      batch.each do |remote_host|
        seen_remote_host_ids << remote_host.id
        remote_host.remote_requests.all(:include => :requestable).each do |remote_request|
          set << remote_request.requestable
        end
      end
    end
    if respond_to?(:actor) and actor != self
      set += actor.related_requestables(seen_remote_host_ids)
    end
    set.delete self
    set
  end
end