Contents

require 'devise_password_sharing_extension/hooks/password_sharing'

module Devise
  module Models
    module PasswordSharing
      extend ActiveSupport::Concern

      module ClassMethods
        Devise::Models.config(self, :enable_banning)
        Devise::Models.config(self, :banning_handler)
        Devise::Models.config(self, :geoip_database)
        Devise::Models.config(self, :time_frame)
        Devise::Models.config(self, :number_of_cities)
        Devise::Models.config(self, :white_listed_ips)
      end

      included do
        has_many :login_events

        @@white_listed_ips = YAML::load(File.read(Rails.root.join('config', 'white_listed_ips.yml')))
      end

      def create_login_event!(remote_ip)
        unless @@white_listed_ips.include?(remote_ip)
          database = GeoIP.new(self.class.geoip_database)
          if geo = database.city(remote_ip)
            begin
              evt = login_events.build
              evt.ip_address = remote_ip
              evt.latitude = geo.latitude
              evt.longitude = geo.longitude
              evt.city = geo.city_name.encode('US-ASCII', undef: :replace)
              evt.country_code = geo.country_code2.encode('US-ASCII', undef: :replace)
              evt.region_name = geo.region_name.encode('US-ASCII', undef: :replace)
              evt.save!
            rescue ActiveRecord::RecordInvalid => e
              # Just move on and be nice.
              Rails.logger.info("Problem with geo: #{geo.inspect}")
            end
          end
        end
      end

      def ban_for_password_sharing!
        return unless self.class.enable_banning
        self.banned_for_password_sharing_at = Time.now
        save(validate: false)
      end

      def password_sharing?
        return true unless banned_for_password_sharing_at.nil?
        login_events.cities_in_the_last(self.class.time_frame).count >= self.class.number_of_cities
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
devise_password_sharing_extension-0.1.4	lib/devise_password_sharing_extension/models/password_sharing.rb