Some modifications to the devise login tracker gem

.gitignore

@@ -15,3 +15,5 @@ spec/reports
 test/tmp
 test/version_tmp
 tmp
+*.swp
+

lib/devise/hooks/login_tracker.rb

@@ -4,14 +4,14 @@
   if record.respond_to?(:mark_login!)
     login_record = record.mark_login!(warden.request)
     scope = opts[:scope]
-    warden.session["#{scope}.login_id"] = login_record.id
+    warden.session(scope)['login_id'] = login_record.id
   end
 end
 
 Warden::Manager.before_logout do |record, warden, opts|
   if record.respond_to?(:mark_logout!)
     scope = opts[:scope]
-    login_record_id = warden.session["#{scope}.login_id"]
+    login_record_id = warden.session(scope)['login_id']
     record.mark_logout!(login_record_id) if login_record_id
   end
 end

lib/devise/models/login_tracker.rb

@@ -11,28 +11,20 @@ module LoginTracker
       # to login
       # @return [UserLogin]
       def mark_login!(request)
-        login_class.create(attrs_for_login(request))
+        logins.create( ip_address: request.remote_ip,
+                      user_agent: request.user_agent,
+                      signed_in_at: Time.now
+                     )
       end
 
       # Marks the time when user has logged out on given login record ID.
       #
       # @param [Fixnum, String] login_record_id ID of login record.
       def mark_logout!(login_record_id)
-        login_record = login_class.find(login_record_id)
+        login_record = logins.find(login_record_id)
         login_record.update_column :signed_out_at, Time.now
       end
 
-      protected
-
-      def attrs_for_login(request)
-        { user_id: id, ip_address: request.remote_ip,
-          user_agent: request.user_agent,
-          signed_in_at: Time.now }
-      end
-
-      def login_class
-        "#{self.class.name}Login".safe_constantize
-      end
 
     end
   end

lib/generators/devise_login_tracker/templates/migration.rb

@@ -3,8 +3,8 @@ class DeviseCreate<%= table_name.camelize.singularize %>Logins < ActiveRecord::M
   def up
     create_table :<%= table_name.singularize %>_logins do |t|
       t.integer  :<%= table_name.classify.foreign_key  %>
-      t.string :ip_address
-      t.string :user_agent
+      t.inet     :ip_address
+      t.string   :user_agent
       t.datetime :signed_in_at
       t.datetime :signed_out_at
     end