mysqlaudit 0.0.3 → 0.0.4

data/bin/mysqlaudit

@@ -53,25 +53,27 @@ end
 
 command :rollback do |c|
   c.description = 'Rollback transaction cath in audit table'
-  c.syntax = 'audits rollback --host 127.0.0.1 --user root [--pass admin] --schema sakila --table foo --statement insert'
+  c.syntax = 'audits rollback --host 127.0.0.1 --user root [--pass admin] --schema sakila --table foo [--statement all]'
   c.option '--host STRING', String, 'Host'
   c.option '--user STRING', String, 'User'
   c.option '--pass STRING', String, 'Password'
   c.option '--schema STRING', String, 'Schema name'
   c.option '--table STRING', String, 'Table name'
-  c.option '--statement STRING', String, 'Statement operation: insert, update, delete'
+  c.option '--statement STRING', String, 'Statement operation: all, insert, update, delete'
   c.action do |args, options|
-    if !options.statement.nil?
-      if [:insert, :update, :delete].include? options.statement.to_sym
-        Mysqlaudit::Actions.new({host: options.host,
-                                 user: options.user,
-                                 password: options.pass,
-                                 schema: options.schema,
-                                 table: options.table,
-                                 statement: options.statement.to_sym}).rollback
-      else
-        puts "Invalid values on argument: --statement."
-      end
+    if options.statement.nil?
+      options.statement = :all
+    end
+
+    if [:all, :insert, :update, :delete].include? options.statement.to_sym
+      Mysqlaudit::Actions.new({host: options.host,
+                               user: options.user,
+                               password: options.pass,
+                               schema: options.schema,
+                               table: options.table,
+                               statement: options.statement.to_sym}).rollback
+    else
+      puts "Invalid values on argument: --statement."
     end
   end
 end

data/lib/mysqlaudit/audit.rb

@@ -77,7 +77,7 @@ SQL
     end
 
     def get_columns(table)
-      sql       = "SELECT column_name AS name FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA='#{$schema}' AND TABLE_NAME='#{table}' AND column_key <> 'PRI';"
+      sql        = "SELECT column_name AS name FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA='#{$schema}' AND TABLE_NAME='#{table}' AND column_key <> 'PRI';"
       sql_result = @mysql.query(sql)
       sql_result.map { |table| table['name']}
     end
@@ -124,40 +124,57 @@ SQL
     end
 
     def has_trigger(table, actionName)
-      sql       = "SELECT true AS has FROM information_schema.triggers WHERE trigger_schema = '#{$schema}' AND trigger_name = 'audit_#{table}_#{actionName}';"
+      sql = "SELECT true AS has FROM information_schema.triggers WHERE trigger_schema = '#{$schema}' AND trigger_name = 'audit_#{table}_#{actionName}';"
       @mysql.query(sql).count == 1
     end
 
-    def rollback(table, statement)
+    def insert(table)
+      sql  = "SELECT CONCAT('/* ', trigger_at, ' */ DELETE FROM #{table} WHERE id = ', MAX(primary_key),';') AS `row` "
+      sql << "FROM audits "
+      sql << "WHERE `type`  = 'I' AND `table` = '#{table}' GROUP BY primary_key "
+      sql << "ORDER BY trigger_at, primary_key, `column` ASC;"
+    end
+
+    def update(table)
+      sql  = "SELECT CONCAT('/* ', trigger_at, ' */ UPDATE users SET ', `column`, ' = ', QUOTE(old), ' WHERE id = ', primary_key, ';') AS `row`"
+      sql << "FROM audits WHERE `type` = 'U' AND `table` = '#{table}' ORDER BY trigger_at, primary_key, `column` ASC;"
+    end
+
+    def delete(table)
       columns = get_columns(table)
       key     = get_primary_key(table)
 
+      sql = "SELECT CONCAT('/* ', trigger_at, ' */ INSERT INTO #{table} (#{key}, #{columns.join(', ')}) VALUES (', primary_key, ', ', "
+      columns.each do | column |
+        sql << " QUOTE(MAX(IF(`column` = '#{column}', `old`, NULL))), ', ', "
+      end
+      sql  = sql.chomp(" ', ', ")
+      sql << "');'"
+      sql << ") AS `row` FROM audits WHERE `type`  = 'D' AND `table` = '#{table}' GROUP BY primary_key ORDER BY trigger_at, primary_key, `column` ASC;"
+    end
+
+    def rollback(table, statement)
+      sqls = []
+
       case statement
+        when :all
+          sqls << insert(table)
+          sqls << update(table)
+          sqls << delete(table)
         when :insert
-          sql = <<SQL
-SELECT CONCAT('/* ', trigger_at, ' */ DELETE FROM #{table} WHERE id = ', MAX(primary_key),';') AS `row`
-FROM audits
-WHERE `type`  = 'I'
-  AND `table` = '#{table}'
-GROUP BY primary_key
-ORDER BY trigger_at, primary_key, `column` ASC;
-SQL
-        when :delete
-          sql = "SELECT CONCAT('/* ', trigger_at, ' */ INSERT INTO #{table} (#{key}, #{columns.join(', ')}) VALUES (', primary_key, ', ', "
-          columns.each do | column |
-            sql << " QUOTE(MAX(IF(`column` = '#{column}', `old`, NULL))), ', ', "
-          end
-          sql  = sql.chomp(" ', ', ")
-          sql << "');'"
-          sql << ") AS `row` FROM audits WHERE `type`  = 'D' AND `table` = '#{table}' GROUP BY primary_key ORDER BY trigger_at, primary_key, `column` ASC;"
+          sqls << insert(table)
         when :update
-          sql = "SELECT CONCAT('/* ', trigger_at, ' */ UPDATE users SET ',  `column`, ' = ', QUOTE(old), ' WHERE id = ', primary_key, ';') AS `row`"
-          sql << "FROM audits WHERE `type` = 'U' AND `table` = '#{table}' ORDER BY trigger_at, primary_key, `column` ASC;"
+          sqls << update(table)
+        when :delete
+          sqls << delete(table)
       end
 
-      sql_result = @mysql.query(sql)
-      sql_result.each(:as => :array) do |row|
-        puts row
-      end    end
+      sqls.each do | sql |
+        sql_result = @mysql.query(sql)
+        sql_result.each(:as => :array) do |row|
+          puts row
+        end
+      end
+    end
   end
 end

data/lib/mysqlaudit/version.rb

@@ -1,3 +1,3 @@
 module Mysqlaudit
-  VERSION = '0.0.3'
+  VERSION = '0.0.4'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mysqlaudit
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-07-19 00:00:00.000000000 Z
+date: 2013-07-24 00:00:00.000000000 Z
 dependencies: []
 description: MySQL tool for audit all tables with triggers.
 email: nicola51980@gmail.com