saseo 0.4.0 → 0.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8d3227155157a0d8e3df890b3e066aea67e73d68
-  data.tar.gz: 6673eb5e80dca1a684f97708e9f40bfced88b188
+  metadata.gz: dd8922670e32782ea4fb6f13bf50d9c2b9ef22e7
+  data.tar.gz: 6bde66f4b1505a25cb04809816203bf885ea4258
 SHA512:
-  metadata.gz: 1f86a332905806cee8e96e3f00104a87fa48de5b1c391e7f9dcaeceaf9d17260e968da2ae1675f31d3fb59e0e5b14418937d55f80d5c1752875c4bd080503a2f
-  data.tar.gz: 68206c9fa409dbdb4430ea92dab8824b7f7d3b9929b89d257506b20ec8f3783bc9317c31fa4ef082b3e21d807dc11f98fa5f951fb5b1cf04c835c2b1391b1b9e
+  metadata.gz: e188da5e002b2247c09f0a7708008141b92cf211f332511d58d9887bde0b8b35980af1dabad9ee776f3dd071f0006bcec455b5921d01994747a85f93f614556d
+  data.tar.gz: 5ef4b0ccc05fafbc654c845f4fcd03a6b8d336980f6ec1b412a57198800a796fa7ba64c03cff7f67f55b9fdb5899e44f2c61cb91484b557b1c731a3f65c21e4e

data/lib/generators/saseo/templates/add_saseo_trigger.rb.erb

@@ -1,14 +1,16 @@
-class AddSaseoTriggerTo<%= @table_name.camelcase %> < ActiveRecord::Migration
+class <%= @target_migration_name.camelcase %> < ActiveRecord::Migration
   def up
     # originally copied from:
     # https://wiki.postgresql.org/wiki/Audit_trigger
-    trigger_definition = <<SQL
-CREATE TRIGGER <%= @table_name %>_saseo_trigger AFTER INSERT OR UPDATE OR DELETE ON <%= @table_name %> FOR EACH ROW EXECUTE PROCEDURE saseo.audit_func();
-SQL
-    execute(trigger_definition)
+      <% @table_names.each do |table_name| %>
+      execute('DROP TRIGGER IF EXISTS <%= table_name %>_saseo_trigger ON <%= table_name %> CASCADE')
+      execute('CREATE TRIGGER <%= table_name %>_saseo_trigger AFTER INSERT OR UPDATE OR DELETE ON <%= table_name %> FOR EACH ROW EXECUTE PROCEDURE saseo.audit_func()')
+     <% end %>
   end
 
   def down
-      execute('DROP TRIGGER <%= @table_name %>_saseo_trigger ON <%= @table_name %> CASCADE;')
+   <%  @table_names.each do |table_name| %>
+   execute('DROP TRIGGER IF EXISTS <%= table_name %>_saseo_trigger ON <%= table_name %> CASCADE')
+   <% end %>
   end
 end

data/lib/generators/saseo/trigger_generator.rb

@@ -17,10 +17,8 @@ module Saseo
       tables = Saseo::Models::Source::Base.connection.tables if tables.include? 'all'
       tables -= [Saseo.config.source_table_name, Saseo.config.table_name]
 
-      tables.each do |table_name|
-        @table_name = table_name
-        add_saseo_migration("add_saseo_trigger")
-      end
+      @table_names = tables
+      add_saseo_migration("add_saseo_trigger")
     end
 
     def self.next_migration_number(dirname)
@@ -30,9 +28,14 @@ module Saseo
     protected
     def add_saseo_migration(template)
       migration_dir = File.expand_path('db/migrate')
+      if @table_names.count > 1
+        @target_migration_name = "#{template}_to_#{@table_names.first}_and_#{@table_names.count - 1}_others"
+      else
+        @target_migration_name = "#{template}_to_#{@table_names.first}"
+      end
 
-      unless self.class.migration_exists?(migration_dir, template)
-        migration_template "#{template}.rb.erb", "db/migrate/#{template}_to_#{@table_name}.rb"
+      unless self.class.migration_exists?(migration_dir, @target_migration_name)
+        migration_template "#{template}.rb.erb", "db/migrate/#{@target_migration_name}.rb"
       else
         warn("ALERT: Migration already exists named '#{template}'." +
                " Please check your migrations directory before re-running")

data/lib/saseo/version.rb

@@ -1,3 +1,3 @@
 module Saseo
-  VERSION = "0.4.0"
+  VERSION = "0.4.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: saseo
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.4.1
 platform: ruby
 authors:
 - Nathan Keyes