replicate 1.1 → 1.2

data/bin/replicate

@@ -6,12 +6,14 @@
 #/ expression. The --load form reads dump data from stdin and loads into the
 #/ current environment.
 #/
-#/ Options:
-#/   -r, --require        Require the library. Often used with 'config/environment'.
+#/ Mode selection:
 #/   -d, --dump           Dump the repository and all related objects to stdout.
 #/   -l, --load           Load dump file data from stdin.
-#/   -i, --keep-id        Use replicated ids when loading dump file.
 #/
+#/ Options:
+#/   -r, --require        Require the library. Often used with 'config/environment'.
+#/   -i, --keep-id        Use replicated ids when loading dump file.
+#/   -f, --force          Allow loading in production environments.
 #/   -v, --verbose        Write more status output.
 #/   -q, --quiet          Write less status output.
 $stderr.sync = true
@@ -23,10 +25,12 @@ verbose  = false
 quiet    = false
 keep_id  = false
 out      = $stdout
+force    = false
 
 # parse arguments
 file = __FILE__
 usage = lambda { exec "grep ^#/<'#{file}'|cut -c4-" }
+original_argv = ARGV.dup
 ARGV.options do |opts|
   opts.on("-d", "--dump")      { mode = :dump }
   opts.on("-l", "--load")      { mode = :load }
@@ -34,20 +38,17 @@ ARGV.options do |opts|
   opts.on("-v", "--verbose")   { verbose = true }
   opts.on("-q", "--quiet")     { quiet = true }
   opts.on("-i", "--keep-id")   { keep_id = true }
+  opts.on("--force")           { force = true }
   opts.on_tail("-h", "--help", &usage)
   opts.parse!
 end
 
-# load rails environment and replicator lib.
+# load replicate lib and setup AR
 require 'replicate'
-
 if defined?(ActiveRecord::Base)
+  Replicate::AR
   ActiveRecord::Base.replicate_id = keep_id
-  # hack to enable AR query cache
-  ActiveRecord::ConnectionAdapters::QueryCache.
-    send :attr_writer, :query_cache, :query_cache_enabled
-  ActiveRecord::Base.connection.send(:query_cache=, {})
-  ActiveRecord::Base.connection.send(:query_cache_enabled=, true)
+  ActiveRecord::Base.connection.enable_query_cache!
 end
 
 # dump mode means we're reading records from the database here and writing to
@@ -64,9 +65,14 @@ if mode == :dump
 # load mode means we're reading objects from stdin and creating them under
 # the current environment.
 elsif mode == :load
-  Replicate::Loader.new do |loader|
-    loader.log_to $stderr, verbose, quiet
-    loader.read $stdin
+  if Replicate.production_environment? && !force
+    abort "error: refusing to load in production environment\n" +
+          "       manual override: #{File.basename($0)} --force #{original_argv.join(' ')}"
+  else
+    Replicate::Loader.new do |loader|
+      loader.log_to $stderr, verbose, quiet
+      loader.read $stdin
+    end
   end
 
 # mode not set means no -l or -d arg was given. show usage and bail.

data/lib/replicate.rb

@@ -6,5 +6,19 @@ module Replicate
   autoload :Status,  'replicate/status'
   autoload :AR,      'replicate/active_record'
 
+  # Determine if this is a production looking environment. Used in bin/replicate
+  # to safeguard against loading in production.
+  def self.production_environment?
+    if defined?(Rails) && Rails.respond_to?(:env)
+      Rails.env.to_s == 'production'
+    elsif defined?(RAILS_ENV)
+      RAILS_ENV == 'production'
+    elsif ENV['RAILS_ENV']
+      ENV['RAILS_ENV'] == 'production'
+    elsif ENV['RACK_ENV']
+      ENV['RAILS_ENV'] == 'production'
+    end
+  end
+
   AR if defined?(::ActiveRecord::Base)
 end

data/lib/replicate/active_record.rb

@@ -1,3 +1,6 @@
+require 'active_record'
+require 'active_record/version'
+
 module Replicate
   # ActiveRecord::Base instance methods used to dump replicant objects for the
   # record and all 1:1 associations. This module implements the replicant_id
@@ -209,6 +212,7 @@ module Replicate
           end
         end
       end
+
     end
 
     # Special object used to dump the list of associated ids for a
@@ -249,8 +253,25 @@ module Replicate
       end
     end
 
+    # Backport connection.enable_query_cache! for Rails 2.x
+    require 'active_record/connection_adapters/abstract/query_cache'
+    query_cache = ::ActiveRecord::ConnectionAdapters::QueryCache
+    if !query_cache.methods.any? { |m| m.to_sym == :enable_query_cache! }
+      query_cache.module_eval do
+        attr_writer :query_cache, :query_cache_enabled
+
+        def enable_query_cache!
+          @query_cache ||= {}
+          @query_cache_enabled = true
+        end
+
+        def disable_query_cache!
+          @query_cache_enabled = false
+        end
+      end
+    end
+
     # Load active record and install the extension methods.
-    require 'active_record'
     ::ActiveRecord::Base.send :include, InstanceMethods
     ::ActiveRecord::Base.send :extend,  ClassMethods
     ::ActiveRecord::Base.replicate_associations = []

data/test/active_record_test.rb

@@ -349,4 +349,9 @@ class ActiveRecordTest < Test::Unit::TestCase
     user = User.find(user.id)
     assert_equal timestamp, user.updated_at
   end
+
+  def test_enabling_active_record_query_cache
+    ActiveRecord::Base.connection.enable_query_cache!
+    ActiveRecord::Base.connection.disable_query_cache!
+  end
 end

metadata

@@ -1,12 +1,12 @@
 --- !ruby/object:Gem::Specification 
 name: replicate
 version: !ruby/object:Gem::Version 
-  hash: 13
+  hash: 11
   prerelease: 
   segments: 
   - 1
-  - 1
-  version: "1.1"
+  - 2
+  version: "1.2"
 platform: ruby
 authors: 
 - Ryan Tomayko