pg_audit_log 0.1.2 → 0.1.3

data/lib/generators/pg_audit_log/templates/lib/tasks/pg_audit_log.rake

@@ -26,6 +26,12 @@ namespace :pg_audit_log do
     export_development_structure
   end
 
+  desc "Check all tables that are missing triggers (fails if any are)"
+  task :check => :environment do
+    tables = PgAuditLog::Triggers.tables_without_triggers
+    raise(PgAuditLog::Triggers::MissingTriggers, tables) if tables.any?
+  end
+
   private
 
   def export_development_structure

data/lib/pg_audit_log/triggers.rb

@@ -1,10 +1,30 @@
 module PgAuditLog
   class Triggers < PgAuditLog::ActiveRecord
+    class MissingTriggers < Exception
+      def initialize(tables)
+        @tables = tables
+      end
+      def message
+        "Missing PgAuditLog triggers for the following tables: #{@tables.join(', ')}"
+      end
+    end
+
     class << self
       def tables
         connection.tables - (PgAuditLog::IGNORED_TABLES + [PgAuditLog::Entry.table_name])
       end
 
+      def tables_without_triggers
+        tables_with_triggers = connection.select_values <<-SQL
+          SELECT tables.relname as table_name
+          FROM pg_trigger triggers, pg_class tables
+          WHERE triggers.tgrelid = tables.oid
+          AND tables.relname !~ '^pg_'
+          AND triggers.tgname LIKE '#{trigger_prefix}%'
+        SQL
+        tables - tables_with_triggers
+      end
+
       def install
         tables.each do |table|
           create_for_table(table)
@@ -60,8 +80,12 @@ module PgAuditLog
         execute "ALTER TABLE #{table_name} DISABLE TRIGGER #{trigger_name_for_table(table_name)}"
       end
 
+      def trigger_prefix
+        "audit_"
+      end
+
       def trigger_name_for_table(table_name)
-        "audit_#{table_name}"
+        "#{trigger_prefix}#{table_name}"
       end
     end
   end

data/lib/pg_audit_log/version.rb

@@ -1,3 +1,3 @@
 module PgAuditLog
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

data/spec/triggers_spec.rb

@@ -9,6 +9,22 @@ describe PgAuditLog::Triggers do
     table {}
   end
 
+  with_model :table_without_triggers do
+    table {}
+  end
+
+  describe ".tables_without_triggers" do
+    before do
+      PgAuditLog::Triggers.create_for_table(TableWithTriggers.table_name)
+    end
+
+    it "should return an array of all tables that do not have an audit trigger installed" do
+      PgAuditLog::Triggers.tables_without_triggers.should_not include(TableWithTriggers.table_name)
+      PgAuditLog::Triggers.tables_without_triggers.should include(TableWithoutTriggers.table_name)
+    end
+
+  end
+
   describe ".install" do
     it "should work" do
       ->{

metadata

@@ -2,7 +2,7 @@
 name: pg_audit_log
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors: 
 - Case Commons, LLC
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-04-15 00:00:00 -04:00
+date: 2011-04-19 00:00:00 -04:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -118,7 +118,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.6.2
+rubygems_version: 1.6.1
 signing_key: 
 specification_version: 3
 summary: postgresql only database-level audit logging of all databases changes