Add support for tracking the remote address of the client causing the…

… audit (closes #10)

lib/acts_as_audited/audit_sweeper.rb

@@ -4,6 +4,7 @@ class AuditSweeper < ActionController::Caching::Sweeper
 
   def before_create(audit)
     audit.user ||= current_user
+    audit.remote_address = controller.try(:request).try(:remote_ip)
   end
 
   def current_user

lib/generators/acts_as_audited/templates/add_remote_address_to_audits.rb

@@ -0,0 +1,10 @@
+class <%= migration_class_name %> < ActiveRecord::Migration
+  def self.up
+    add_column :audits, :remote_address, :string
+  end
+
+  def self.down
+    remove_column :audits, :remote_address
+  end
+end
+

lib/generators/acts_as_audited/templates/install.rb

@@ -10,6 +10,7 @@ def self.up
       t.column :audited_changes, :text
       t.column :version, :integer, :default => 0
       t.column :comment, :string
+      t.column :remote_address, :string
       t.column :created_at, :datetime
     end
 

lib/generators/acts_as_audited/upgrade_generator.rb

@@ -39,6 +39,10 @@ def migrations_to_be_applied
         if columns.include?( 'changes' )
           yield :rename_changes_to_audited_changes
         end
+
+        unless columns.include?( 'remote_address' )
+          yield :add_remote_address_to_audits
+        end
       end
     end
   end

spec/audit_sweeper_spec.rb

@@ -48,6 +48,16 @@ def update_user
 
       assigns(:company).audits.last.user.should == user
     end
+
+    it "should record the remote address responsible for the change" do
+      request.env['REMOTE_ADDR'] = "1.2.3.4"
+      controller.send(:current_user=, user)
+
+      post :audit
+
+      assigns(:company).audits.last.remote_address.should == '1.2.3.4'
+    end
+
   end
 
   describe "POST update_user" do
@@ -59,5 +69,6 @@ def update_user
         post :update_user
       }.to_not change( Audit, :count )
     end
+
   end
 end

spec/db/schema.rb

@@ -24,6 +24,7 @@
     t.column :audited_changes, :text
     t.column :version, :integer, :default => 0
     t.column :comment, :string
+    t.column :remote_address, :string
     t.column :created_at, :datetime
   end
 

test/db/version_3.rb

@@ -0,0 +1,19 @@
+ActiveRecord::Schema.define do
+  create_table :audits, :force => true do |t|
+    t.column :auditable_id, :integer
+    t.column :auditable_type, :string
+    t.column :user_id, :integer
+    t.column :user_type, :string
+    t.column :username, :string
+    t.column :action, :string
+    t.column :audited_changes, :text
+    t.column :version, :integer, :default => 0
+    t.column :comment, :string
+    t.column :created_at, :datetime
+  end
+
+  add_index :audits, [:auditable_id, :auditable_type], :name => 'auditable_index'
+  add_index :audits, [:user_id, :user_type], :name => 'user_index'
+  add_index :audits, :created_at
+end
+

test/upgrade_generator_test.rb

@@ -30,4 +30,14 @@ class UpgradeGeneratorTest < Rails::Generators::TestCase
       assert_match /rename_column :audits, :changes, :audited_changes/, content
     end
   end
+
+  test "should add a 'remote_address' to audits table" do
+    load_schema 3
+
+    run_generator %w(upgrade)
+
+    assert_migration "db/migrate/add_remote_address_to_audits.rb" do |content|
+      assert_match /add_column :audits, :remote_address, :string/, content
+    end
+  end
 end