brancher 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c616af20efc6517103ceebed2bd1120d4c0fc5ad
-  data.tar.gz: 317c28a2aa798faf3261b5c0e63cb4354dea04b7
+  metadata.gz: e46188837b98d89a415fd14531e6c8596e28e833
+  data.tar.gz: a30cba0b35d679485f32b964a80deecc52ce522c
 SHA512:
-  metadata.gz: 76e80a391937a051a3f39a46b17ff249e44669606fe102e99befb2b82c034f281b667612adb1bbb6beee8b7e9ac1fde0cea801bb96b59b92170497abc29fe07f
-  data.tar.gz: 3e0c8ce75d8d71fcfe3885d701f5ed6467e38e13c8ac4426186832b01e394f007623b5bee0cc1e94f2bebe0cb5124d882a8fb85f06e6c2fe133dd8c85ebf9a3b
+  metadata.gz: d630b069a8244240763bda6357c257e042263ca9289c996f8420918d5da3417ed13e6e6ae525a4576a29492d7c19f99d1701514a529e8130d130a10dabfa7ade
+  data.tar.gz: e65166e71657c1bb0f9d6bfe49404ace55d7c74c913b5bb20eaa1aa5b88ade3050f1fda6e87e8c860467d48ad3a4fec648577f4c6a6f272ed17d06ff5707bcea

data/lib/brancher.rb

@@ -6,9 +6,11 @@ module Brancher
   include ActiveSupport::Configurable
   config.except_branches ||= []
   config.auto_copy ||= false
+  config.max_database_name_length ||= 63
 end
 
 require "brancher/database_configuration_renaming"
+require "brancher/multiple_database_configuration_renaming"
 require "brancher/database_rename_service"
 require "brancher/auto_copying"
 require "brancher/railtie"

data/lib/brancher/auto_copying.rb

@@ -24,11 +24,10 @@ module Brancher
       end
 
       def auto_copy
-        suffix = Brancher::DatabaseRenameService.suffix
-        return unless suffix
+        return if config[:database] == config[:original_database]
 
         database_name = config[:database]
-        original_database_name = database_name.gsub(Regexp.new(suffix), "")
+        original_database_name = config[:original_database]
 
         case config[:adapter]
         when /mysql/
@@ -39,15 +38,15 @@ module Brancher
       end
 
       def mysql_copy(original_database_name, database_name)
-        system("bundle", "exec", "rake", "db:create")
+        ActiveRecord::Tasks::DatabaseTasks.create(config.with_indifferent_access)
 
         cmd = ["mysqldump", "-u", config[:username]]
         cmd.concat(["-h", config[:host]]) if config[:host].present?
-        cmd.concat(["-p", config[:password]]) if config[:password].present?
+        cmd.concat(["-p#{config[:password]}"]) if config[:password].present?
         cmd << original_database_name
         cmd.concat(["|", "mysql", "-u", config[:username]])
         cmd.concat(["-h", config[:host]]) if config[:host].present?
-        cmd.concat(["-p", config[:password]]) if config[:password].present?
+        cmd.concat(["-p#{config[:password]}"]) if config[:password].present?
         cmd << database_name
         system(cmd.join(" "))
       end

data/lib/brancher/database_rename_service.rb

@@ -4,23 +4,28 @@ module Brancher
   module DatabaseRenameService
     extend self
 
-    def rename!(configurations)
-      configuration = configurations[env]
-      database_extname = File.extname(configuration["database"])
-      database_name = configuration["database"].gsub(%r{#{database_extname}$}) { "" }
-      database_name += suffix unless database_name =~ %r{#{suffix}$}
-      configuration["database"] = database_name + database_extname
+    def rename!(configurations, key = env)
+      configuration = configurations[key]
+      configuration["original_database"] = configuration["database"]
+      configuration["database"] = database_name_with_suffix(configuration["database"])
       configurations
     end
 
-    def suffix
-      return nil if current_branch.blank?
-      return nil if Brancher.config.except_branches.include?(current_branch)
+    private
 
-      "_#{current_branch}"
+    def database_name_with_suffix(database)
+      database_extname = File.extname(database)
+      database_name = database.gsub(%r{#{database_extname}$}) { "" }
+      database_name += suffix unless database_name =~ %r{#{suffix}$}
+      database_name += database_extname
+      database_name = database_name.slice(0,Brancher.config.max_database_name_length-22) + [Digest::MD5.digest(database_name)].pack("m0").slice(0,22).gsub(/[^\w]/, '_').downcase if database_name.length > Brancher.config.max_database_name_length
+      database_name
     end
 
-    private
+    def suffix
+      return nil if current_branch.blank? || Brancher.config.except_branches.include?(current_branch)
+      "_#{current_branch}"
+    end
 
     def env
       Rails.env

data/lib/brancher/multiple_database_configuration_renaming.rb

@@ -0,0 +1,15 @@
+module Brancher
+  module MultipleDatabaseConfigurationRenaming
+    module ClassMethods
+      def establish_connection(spec = nil)
+        DatabaseRenameService.rename!(configurations, spec.to_s) if spec && spec.is_a?(Hash).!
+
+        super
+      end
+    end
+
+    def self.prepended(base)
+      base.singleton_class.send(:prepend, ClassMethods)
+    end
+  end
+end

data/lib/brancher/railtie.rb

@@ -4,15 +4,23 @@ module Brancher
   class Railtie < Rails::Railtie
     initializer "brancher.rename_database", before: "active_record.initialize_database" do
       Rails::Application::Configuration.send(:prepend, DatabaseConfigurationRenaming)
+      ActiveRecord::Base.send(:prepend, MultipleDatabaseConfigurationRenaming)
       ActiveRecord::ConnectionAdapters::ConnectionPool.send(:prepend, AutoCopying)
     end
 
     rake_tasks do
       namespace :db do
         task :load_config do
+          require_environment!
           DatabaseRenameService.rename!(ActiveRecord::Base.configurations)
         end
       end
     end
+
+    def require_environment!
+      return unless defined? Rails
+      environemnt = "#{Rails.root}/config/environments/#{Rails.env}.rb"
+      require environemnt if File.exists?(environemnt)
+    end
   end
 end

data/lib/brancher/version.rb

@@ -1,3 +1,3 @@
 module Brancher
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

data/spec/brancher/database_rename_service_spec.rb

@@ -46,7 +46,8 @@ describe Brancher::DatabaseRenameService do
           "adapter" => adapter,
           "pool" => 5,
           "timeout" => 5000,
-          "database" => new_database_name
+          "database" => new_database_name,
+          "original_database" => database_name
         }
       }
     end
@@ -80,5 +81,45 @@ describe Brancher::DatabaseRenameService do
 
       it { is_expected.to eq new_configurations }
     end
+
+    context "when the database name is longer than max_database_name_length" do
+      let(:adapter) do
+        "mysql2"
+      end
+
+      let(:database_name) do
+        "this_is_a_very_long_sample_app_development_database_name_that_exceeds_the_default_set_max_database_name_length"
+      end
+
+      let(:new_database_name) do
+        "this_is_a_very_long_sample_app_developmen#{[Digest::MD5.digest("#{database_name}_#{branch}")].pack('m0').slice(0,22).gsub(/[^\w]/, '_').downcase}"
+      end
+
+      it { is_expected.to eq new_configurations }
+    end
+
+    context "when it connect another database" do
+      let(:adapter) do
+        "mysql2"
+      end
+
+      let(:database_name) do
+        "sample_app_another_db"
+      end
+
+      let(:new_database_name) do
+        "#{database_name}_#{branch}"
+      end
+
+      let(:env) do
+        "another_db"
+      end
+
+      subject do
+        Brancher::DatabaseRenameService.rename!(configurations, env)
+      end
+
+      it { is_expected.to eq new_configurations }
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: brancher
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Naoto Kaneko
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-05-16 00:00:00.000000000 Z
+date: 2015-10-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -112,6 +112,7 @@ files:
 - lib/brancher/auto_copying.rb
 - lib/brancher/database_configuration_renaming.rb
 - lib/brancher/database_rename_service.rb
+- lib/brancher/multiple_database_configuration_renaming.rb
 - lib/brancher/railtie.rb
 - lib/brancher/version.rb
 - spec/brancher/database_rename_service_spec.rb
@@ -143,4 +144,3 @@ summary: Switching databases connected with ActiveRecord by Git branch
 test_files:
 - spec/brancher/database_rename_service_spec.rb
 - spec/spec_helper.rb
-has_rdoc: