module TheAudit
  module Base
    extend ActiveSupport::Concern

    included do
      include ThePagination::Concern

      belongs_to :user
      belongs_to :obj, polymorphic: true

      include BaseSorts

      scope :audit_scope, ->(params){
        by_ip(params)
        .by_holder(params)
        .by_referer(params)
        .by_user_id(params)
        .by_date_range(params)
        .by_user_agent(params)
        .by_controller_action(params)
        .simple_sort(params)
      }

      scope :by_user_id, ->(params){
        uid = params[:user_id]
        return nil if uid.blank?
        where(user_id: uid)
      }

      scope :by_referer, ->(params){
        referer = params[:referer]
        return nil if referer.blank?
        where(referer: referer)
      }

      scope :by_ip, ->(params){
        ip = params[:ip]
        return nil if ip.blank?
        where(ip: ip)
      }

      scope :by_user_agent, ->(params){
        ua = params[:user_agent]
        return nil if ua.blank?
        where(user_agent: ua)
      }

      scope :by_holder, ->(params){
        return nil if params[:obj_id].blank? && params[:obj_type].blank?
        _type, id = params[:obj_type], params[:obj_id]

        return where(obj_type: _type) if id.blank?
        return where(obj_id: id)      if _type.blank?

        where(obj_type: _type).where(obj_id: id)
      }

      scope :by_date_range, ->(params){
        from, to = params[:date_start], params[:date_end]
        return nil if from.blank? && to.blank?
        return where("audits.created_at > ?", from.to_datetime) if to.blank?
        return where("audits.created_at < ?", to.to_datetime)   if from.blank?
        return where("audits.created_at > ? AND audits.created_at < ?", from.to_datetime, to.to_datetime)
      }

      scope :by_controller_action, ->(params){
        ca = params[:controller_action]
        return nil if ca.blank?

        ctrl, act = ca.split '-'
        return nil if act.blank?

        where(controller_name: ctrl).where(action_name: act)
      }

      scope :with_users, ->{ includes(:user) }
    end

    def user_name
      user.try(:email) || user.try(:login) || user.try(:username)
    end

    def user_path context
      "/users/#{ user.to_param }"
    end

    def user_id_builder controller
      controller.try(:current_user).try(:id)
    end

    def unescape str
      return nil if str.blank?
      res_str = CGI::unescape str

      # try to catch:
      # invalid byte sequence in UTF-8
      # for search requests form old IE6-8 with cp-1251
      begin
        res_str =~ //
      rescue ArgumentError
        res_str = CGI.unescape(str).force_encoding('windows-1251').encode
      end

      res_str
    end

    def init controller, object = nil, data = {}
      self.user_id = self.user_id_builder(controller)

      self.obj             = object
      self.action_name     = controller.action_name
      self.controller_name = controller.controller_name

      self.data = data.to_json unless data.blank?

      if r = controller.request
        self.ip          = r.ip
        self.user_agent  = r.user_agent
        self.bot         = TheAudit.is_bot?(r.user_agent)

        self.remote_ip   = r.remote_ip
        self.remote_addr = r.remote_addr
        self.remote_host = r.remote_host

        self.fullpath    = self.unescape(r.fullpath) || ''
        self.referer     = self.unescape(r.referer)  || :direct_visit
      end

      self
    end
  end
end