culturecode_stagehand 0.13.2 → 0.13.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b27211ffdd21569aa717eae6259506d5ea30523e
-  data.tar.gz: b9f10019544e6238472afb6652cc3e02da0196e3
+  metadata.gz: 7ed3d3abf2a64c62d90182193fe80714d2bd8418
+  data.tar.gz: 4825e8a8e5621fc84020f57269589a4cea1ee04b
 SHA512:
-  metadata.gz: c52ebc2e370e7b29eccd9f920563c1cc5a115cb19b471d40584c57b9e735991b204f80242cb0c0812424eb58e9a95b98c8af462435bad97b2eb9e25a2865b4e9
-  data.tar.gz: df36c9ed6419bc708e9996006645ad5529cbcb5d3df2629a228ccff28c5467992f0f885c5788d7e3ff2ff6c301d6ad5139e768d8d48d090be667f73ade0a31ca
+  metadata.gz: 5d4b00fbdf65ce78f0d930732b0cb1477e0fcb7d94c5f58008483c613fa7a72a14390e480b9d48912c45a3d43ba9f4f21707d11cc235b810ca3731fddf0c403e
+  data.tar.gz: edb00a7aa78d1c5f16f8eab425447790b53de432568531453908f7ce38c218abc617d47324b77239f043dc485327c32df730cf049305dde641b7b7bee4f82655

data/lib/stagehand/production.rb

@@ -20,12 +20,17 @@ module Stagehand
 
       return unless attributes.present?
 
+      write(staging_record, attributes, table_name)
+    end
+
+    def write(staging_record, attributes, table_name = nil)
       Connection.with_production_writes(Record) do
         is_new = matching(staging_record, table_name).update_all(attributes).zero?
 
         # Ensure we always return a record, even when updating instead of creating
         Record.new.tap do |record|
           record.assign_attributes(attributes)
+          record.id = Stagehand::Key.generate(staging_record, :table_name => table_name).last unless record.id
           record.save if is_new
         end
       end

data/lib/stagehand/schema.rb

@@ -29,12 +29,7 @@ module Stagehand
 
     def add_stagehand!(options = {})
       ActiveRecord::Schema.define do
-        table_names = ActiveRecord::Base.connection.tables
-        table_names -= UNTRACKED_TABLES
-        table_names -= Array(options[:except]).collect(&:to_s)
-        table_names &= Array(options[:only]).collect(&:to_s) if options[:only].present?
-
-        table_names.each do |table_name|
+        Stagehand::Schema.send :each_table, options do |table_name|
           Stagehand::Schema.send :create_operation_trigger, table_name, 'insert', 'NEW'
           Stagehand::Schema.send :create_operation_trigger, table_name, 'update', 'NEW'
           Stagehand::Schema.send :create_operation_trigger, table_name, 'delete', 'OLD'
@@ -44,10 +39,7 @@ module Stagehand
 
     def remove_stagehand!(options = {})
       ActiveRecord::Schema.define do
-        table_names = ActiveRecord::Base.connection.tables
-        table_names &= Array(options[:only]).collect(&:to_s) if options[:only].present?
-
-        table_names.each do |table_name|
+        Stagehand::Schema.send :each_table, options do |table_name|
           Stagehand::Schema.send :drop_trigger, table_name, 'insert'
           Stagehand::Schema.send :drop_trigger, table_name, 'update'
           Stagehand::Schema.send :drop_trigger, table_name, 'delete'
@@ -69,6 +61,17 @@ module Stagehand
 
     private
 
+    def each_table(options = {})
+      table_names = ActiveRecord::Base.connection.tables
+      table_names -= UNTRACKED_TABLES
+      table_names -= Array(options[:except]).collect(&:to_s)
+      table_names &= Array(options[:only]).collect(&:to_s) if options[:only].present?
+
+      table_names.each do |table_name|
+        yield table_name
+      end
+    end
+
     # Create trigger to initialize session using a function
     def create_session_trigger
       drop_trigger(:stagehand_commit_entries, :insert)

data/lib/stagehand/version.rb

@@ -1,3 +1,3 @@
 module Stagehand
-  VERSION = "0.13.2"
+  VERSION = "0.13.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: culturecode_stagehand
 version: !ruby/object:Gem::Version
-  version: 0.13.2
+  version: 0.13.3
 platform: ruby
 authors:
 - Nicholas Jakobsen
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-04-11 00:00:00.000000000 Z
+date: 2017-05-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails