brancher 0.1.2 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 338f02687959d178804d912e5b7055f2e764a644
-  data.tar.gz: 85e3e43519492a9614418763f0be94c890bdd524
+  metadata.gz: c616af20efc6517103ceebed2bd1120d4c0fc5ad
+  data.tar.gz: 317c28a2aa798faf3261b5c0e63cb4354dea04b7
 SHA512:
-  metadata.gz: d241ee0923aca862c6ef840cdaced23daa31ac6c758cc0e9a2295901b4cae4c5d4947f5f019c98507f527c4119177a54a91c56cb319b891623bb56d5dba1e989
-  data.tar.gz: 49fc4fd0579356f4b7c844f617355104270949b26e1830cbe5284e5fd92d091a7b94dc4b108f5b8bc9586645747f8b98ce665344fe41ac29a06a7330a0c83c9a
+  metadata.gz: 76e80a391937a051a3f39a46b17ff249e44669606fe102e99befb2b82c034f281b667612adb1bbb6beee8b7e9ac1fde0cea801bb96b59b92170497abc29fe07f
+  data.tar.gz: 3e0c8ce75d8d71fcfe3885d701f5ed6467e38e13c8ac4426186832b01e394f007623b5bee0cc1e94f2bebe0cb5124d882a8fb85f06e6c2fe133dd8c85ebf9a3b

data/README.md

@@ -1,6 +1,7 @@
 # Brancher
 
 [![Build Status](https://travis-ci.org/naoty/brancher.svg)](https://travis-ci.org/naoty/brancher)
+[![Code Climate](https://codeclimate.com/github/naoty/brancher/badges/gpa.svg)](https://codeclimate.com/github/naoty/brancher)
 
 Brancher is a rubygem to switch databases connected with ActiveRecord by Git branch.
 
@@ -22,6 +23,22 @@ And then execute:
 $ bundle
 ```
 
+## Configuration
+
+```ruby
+# config/environments/development.rb
+
+Brancher.configure do |c|
+  # if branch is "master" or "develop", database name has no suffix.
+  c.except_branches << "master"
+  c.except_branches << "develop"
+
+  # if auto_copy is true and database does not exist,
+  # copy database from no suffix name database to suffixed name one.
+  c.auto_copy = true
+end
+```
+
 ## Contributing
 
 1. Fork it ( https://github.com/[my-github-username]/brancher/fork )

data/lib/brancher.rb

@@ -1,6 +1,15 @@
 $LOAD_PATH.unshift(__dir__) unless $LOAD_PATH.include?(__dir__)
 
+require 'active_support/configurable'
+
+module Brancher
+  include ActiveSupport::Configurable
+  config.except_branches ||= []
+  config.auto_copy ||= false
+end
+
 require "brancher/database_configuration_renaming"
 require "brancher/database_rename_service"
+require "brancher/auto_copying"
 require "brancher/railtie"
-require "brancher/version"
+require "brancher/version"

data/lib/brancher/auto_copying.rb

@@ -0,0 +1,65 @@
+module Brancher
+  module AutoCopying
+    def new_connection
+      done = false
+
+      begin
+        super
+      rescue
+        raise if done
+        raise unless Brancher.config.auto_copy
+
+        executor = Executor.new(spec.config)
+        executor.auto_copy
+        done = true
+        retry
+      end
+    end
+
+    class Executor
+      attr_reader :config
+
+      def initialize(config)
+        @config = config
+      end
+
+      def auto_copy
+        suffix = Brancher::DatabaseRenameService.suffix
+        return unless suffix
+
+        database_name = config[:database]
+        original_database_name = database_name.gsub(Regexp.new(suffix), "")
+
+        case config[:adapter]
+        when /mysql/
+          mysql_copy(original_database_name, database_name)
+        when /postgresql/
+          pg_copy(original_database_name, database_name)
+        end
+      end
+
+      def mysql_copy(original_database_name, database_name)
+        system("bundle", "exec", "rake", "db:create")
+
+        cmd = ["mysqldump", "-u", config[:username]]
+        cmd.concat(["-h", config[:host]]) if config[:host].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 << database_name
+        system(cmd.join(" "))
+      end
+
+      def pg_copy(original_database_name, database_name)
+        env = {}
+        env["PGUSER"] = config[:username] if config[:username].present?
+        env["PGPASSWORD"] = config[:password] if config[:password].present?
+        env["PGHOST"] = config[:host] if config[:host].present?
+
+        system(env, "createdb", "-T", original_database_name, database_name)
+      end
+    end
+  end
+end

data/lib/brancher/database_rename_service.rb

@@ -13,18 +13,21 @@ module Brancher
       configurations
     end
 
+    def suffix
+      return nil if current_branch.blank?
+      return nil if Brancher.config.except_branches.include?(current_branch)
+
+      "_#{current_branch}"
+    end
+
     private
 
     def env
       Rails.env
     end
 
-    def suffix
-      "_#{current_branch}"
-    end
-
     def current_branch
       @current_branch ||= `git rev-parse --abbrev-ref HEAD`.chomp
     end
   end
-end
+end

data/lib/brancher/railtie.rb

@@ -4,6 +4,7 @@ module Brancher
   class Railtie < Rails::Railtie
     initializer "brancher.rename_database", before: "active_record.initialize_database" do
       Rails::Application::Configuration.send(:prepend, DatabaseConfigurationRenaming)
+      ActiveRecord::ConnectionAdapters::ConnectionPool.send(:prepend, AutoCopying)
     end
 
     rake_tasks do
@@ -14,4 +15,4 @@ module Brancher
       end
     end
   end
-end
+end

data/lib/brancher/version.rb

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

metadata

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