module Ahoy
  module Model

    def ahoy_visit
      class_eval do

        belongs_to :user, polymorphic: true

        before_create :set_traffic_source
        before_create :set_utm_parameters
        before_create :set_technology
        before_create :set_location

        def set_traffic_source
          referring_domain = Addressable::URI.parse(referrer).host.first(255) rescue nil
          self.referring_domain = referring_domain if respond_to?(:referring_domain=)
          # performance hack for referer-parser
          search_keyword = Ahoy.referrer_parser.parse(referrer)[1].first(255) rescue nil
          self.search_keyword = search_keyword.presence if respond_to?(:search_keyword=)
          true
        end

        def set_utm_parameters
          landing_uri = Addressable::URI.parse(landing_page) rescue nil
          if landing_uri
            query_values = landing_uri.query_values || {}
            %w[utm_source utm_medium utm_term utm_content utm_campaign].each do |name|
              self[name] = query_values[name] if respond_to?(:"#{name}=")
            end
          end
          true
        end

        def set_technology
          if respond_to?(:user_agent)
            browser = Browser.new(ua: user_agent)

            self.browser = browser.name if respond_to?(:browser=)

            # TODO add more
            self.os =
              if browser.android?
                "Android"
              elsif browser.ios?
                "iOS"
              elsif browser.windows_phone?
                "Windows Phone"
              elsif browser.blackberry?
                "Blackberry"
              elsif browser.chrome_os?
                "Chrome OS"
              elsif browser.mac?
                "Mac"
              elsif browser.windows?
                "Windows"
              elsif browser.linux?
                "Linux"
              end if respond_to?(:os=)

            self.device_type =
              if browser.tv?
                "TV"
              elsif browser.console?
                "Console"
              elsif browser.tablet?
                "Tablet"
              elsif browser.mobile?
                "Mobile"
              else
                "Desktop"
              end if respond_to?(:device_type=)
          end
          true
        end

        def set_location
          if respond_to?(:ip) and [:country=, :region=, :city=].any?{|method| respond_to?(method) }
            location = Geocoder.search(ip).first rescue nil
            if location
              self.country = location.country.presence if respond_to?(:country=)
              self.region = location.state.presence if respond_to?(:region=)
              self.city = location.city.presence if respond_to?(:city=)
            end
          end
          true
        end

      end # end class_eval
    end

    def visitable
      class_eval do
        belongs_to :visit

        before_create :set_visit

        def set_visit
          if !self.class.column_names.include?("visit_id")
            raise "Add a visit_id column to this table to use visitable"
          else
            self.visit ||= RequestStore.store[:ahoy_controller].try(:send, :current_visit)
          end
        end
      end
    end

  end
end