culturecode_stagehand 0.9.2 → 0.10.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0f0f8f7ad6db23b1f57b4489d7b835421c55f4e9
-  data.tar.gz: ddf42219103d4f9db984f15381d59fd797ce6834
+  metadata.gz: 8e8e470e10c36d55ed09a55577353f992f4dce33
+  data.tar.gz: 8ff3559aa8264aba97054676c4829cd8930bad1d
 SHA512:
-  metadata.gz: 268b7cb9e19ce1e64cb726e7e9c0fbf573910ede523c9b8aa20dbabf37f6c77f42322349408ccec4db507a6be0a50b84e157ab56f4550dc4c0509bb4123c5565
-  data.tar.gz: 7d15177fd390976858af7cd8cb0c35efbf2cea8c3018ded4e7b38df24a2e70395bb9d7dcbfdfac255201fe27e7be00699bc49a72f689e5c2d8364567415f5c31
+  metadata.gz: 72c42cf30acae539d585e1d8c07abcc8a1bb48c2ac6d0d540e324aa92407bcff9f86acfb86d433abf5fadf7e672472c7ff1694c5e764116d63cfbdaab1111514
+  data.tar.gz: c2da536ed22c74a74db0ba20f41e0891898d02f779cfc870e2d5f46d9aaf7f2e3dd34ed08b0e5917a07e2a36571293dac382ee17f9b4e2d9db83efab1fbb0b93

data/lib/stagehand/configuration.rb

@@ -20,6 +20,11 @@ module Stagehand
       !!Rails.configuration.x.stagehand.ghost_mode
     end
 
+    # Allow unsynchronized writes directly to the production database? A warning will be logged if set to true.
+    def allow_unsynced_production_writes?
+      !!Rails.configuration.x.stagehand.allow_unsynced_production_writes
+    end
+
     # Returns true if the production and staging connections are the same.
     # Use case: Front-end devs may not have a second database set up as they are only concerned with the front end
     def single_connection?

data/lib/stagehand/connection_adapter_extensions.rb

@@ -0,0 +1,76 @@
+module Stagehand
+  module Connection
+    def self.with_production_writes(model, &block)
+      state = model.connection.readonly?
+      model.connection.readonly!(false)
+      return block.call
+    ensure
+      model.connection.readonly!(state)
+    end
+
+    module AdapterExtensions
+      def self.prepended(base)
+        base.set_callback :checkout, :after, :update_readonly_state
+        base.set_callback :checkin, :before, :clear_readonly_state
+      end
+
+      def exec_insert(*)
+        handle_readonly_writes!
+        super
+      end
+
+      def exec_update(*)
+        handle_readonly_writes!
+        super
+      end
+
+      def exec_delete(*)
+        handle_readonly_writes!
+        super
+      end
+
+      def allow_writes(&block)
+        state = readonly?
+        readonly!(true)
+        return block.call
+      ensure
+        readonly!(state)
+      end
+
+      def readonly!(state = true)
+        @readonly = state
+      end
+
+      def readonly?
+        !!@readonly
+      end
+
+      private
+
+      def update_readonly_state
+        readonly! unless Configuration.single_connection? || @config[:database] != Database.production_database_name
+      end
+
+      def clear_readonly_state
+        readonly!(false)
+      end
+
+      def handle_readonly_writes!
+        if !readonly?
+          return
+        elsif Configuration.allow_unsynced_production_writes?
+          Rails.logger.warn "Writing directly to production database"
+        else
+          raise(UnsyncedProductionWrite, "Attempted to write directly to production database")
+        end
+      end
+    end
+  end
+
+
+  # EXCEPTIONS
+
+  class UnsyncedProductionWrite < StandardError; end
+end
+
+ActiveRecord::Base.connection.class.prepend(Stagehand::Connection::AdapterExtensions)

data/lib/stagehand/engine.rb

@@ -12,6 +12,7 @@ module Stagehand
       require "stagehand/cache"
       require "stagehand/key"
       require "stagehand/database"
+      require "stagehand/connection_adapter_extensions"
       require "stagehand/controller_extensions"
       require "stagehand/active_record_extensions"
       require "stagehand/staging"

data/lib/stagehand/production.rb

@@ -20,17 +20,21 @@ module Stagehand
 
       return unless attributes.present?
 
-      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.save if is_new
+      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.save if is_new
+        end
       end
     end
 
     def delete(staging_record, table_name = nil)
-      matching(staging_record, table_name).delete_all
+      Connection.with_production_writes(Record) do
+        matching(staging_record, table_name).delete_all
+      end
     end
 
     def exists?(staging_record, table_name = nil)

data/lib/stagehand/version.rb

@@ -1,3 +1,3 @@
 module Stagehand
-  VERSION = "0.9.2"
+  VERSION = "0.10.0"
 end

data/lib/tasks/stagehand_tasks.rake

@@ -19,9 +19,11 @@ namespace :stagehand do
   def rake_both_databases(task, stagehand_task = task.gsub(':','_'))
     task(stagehand_task => :environment) do
       Stagehand::Database.each do |connection_name|
-        puts "#{connection_name}"
-        Rake::Task[task].reenable
-        Rake::Task[task].invoke
+        Stagehand::Connection.with_production_writes(ActiveRecord::Base) do
+          puts "#{connection_name}"
+          Rake::Task[task].reenable
+          Rake::Task[task].invoke
+        end
       end
       Rake::Task[task].clear
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: culturecode_stagehand
 version: !ruby/object:Gem::Version
-  version: 0.9.2
+  version: 0.10.0
 platform: ruby
 authors:
 - Nicholas Jakobsen
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-09-12 00:00:00.000000000 Z
+date: 2016-09-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -71,6 +71,7 @@ files:
 - lib/stagehand/auditor.rb
 - lib/stagehand/cache.rb
 - lib/stagehand/configuration.rb
+- lib/stagehand/connection_adapter_extensions.rb
 - lib/stagehand/controller_extensions.rb
 - lib/stagehand/database.rb
 - lib/stagehand/engine.rb