cloner 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 084c4eb25834594c2bf7ab958a545f6b85a918c7
-  data.tar.gz: 653be534290038354984642b050a681c6bece755
+  metadata.gz: 766c1d83045885a8d0583a7d68996e10e285f445
+  data.tar.gz: 30a73229c201aaee192ff864d34b986209c78530
 SHA512:
-  metadata.gz: 9887f526f47f8482302a99d0e203d88e792369867dba64fc1c7985f5bc05a726157fc2eb2fb9c118f2e2ebfc1a00770a135b1a55b0a3b02991df8d33b307837b
-  data.tar.gz: 86e4e6c4fe618380ae1aff54f4d3406527b6170fc41d2cbecacd3323743cf4cff09598d959fec25105f591fa5d611aa2e579e7b1d9ca0879b46cdda4578da5e4
+  metadata.gz: 10ae7e947d725997f0e382107aa2db7eeeb32c9fe097b341b8ad3f8e49cf3d1d7e01a4fea20bedf94d18df74bad91cd69693bc86d009f02c5a18951f4245c981
+  data.tar.gz: 63d80240f8d443a070ebbf6f7a699ef1587ab1159e6bd65ac256641cf5b85e34c15e98d1c5c1eccda0a33fd31fc8e4129457b31783c99342a5af39cb4374da4e

data/lib/cloner/internal.rb

@@ -1,6 +1,7 @@
 module Cloner::Internal
   extend ActiveSupport::Concern
   include Cloner::MongoDB
+  include Cloner::Postgres
   include Cloner::SSH
   include Cloner::RSync
 
@@ -19,7 +20,11 @@ module Cloner::Internal
   end
 
   def clone_db
-    clone_mongodb
+    if defined?(Mongoid)
+      clone_mongodb
+    else
+      clone_pg
+    end
   end
 end
 

data/lib/cloner/mongodb.rb

@@ -61,7 +61,7 @@ module Cloner::MongoDB
   end
 
   def mongodb_dump_copy
-    `mkdir -p #{mongodb_path}`
+    FileUtils.mkdir_p(mongodb_path)
     rsync("#{remote_dump_path}/#{mongodb_r_conf['database']}", mongodb_path)
   end
 

data/lib/cloner/postgres.rb

@@ -0,0 +1,82 @@
+module Cloner::Postgres
+  extend ActiveSupport::Concern
+
+  def ar_conf
+    @conf ||= begin
+      YAML.load_file(Rails.root.join('config', 'database.yml'))[Rails.env]
+    end
+  end
+
+  def ar_to
+    ar_conf['database']
+  end
+
+  def ar_r_conf
+    @ar_r_conf ||= begin
+      Net::SSH.start(ssh_host, ssh_user, ssh_opts) do |ssh|
+        ret = ssh_exec!(ssh, "cat #{remote_app_path}/config/database.yml")
+        check_ssh_err(ret)
+        YAML.load(ret[0])[env_from]
+      end
+    end
+  end
+
+  def pg_local_auth
+    if ar_conf['password'].nil?
+      ""
+    else
+      "PGPASSWORD='#{ar_conf['password']}' "
+    end
+  end
+
+  def pg_remote_auth
+    if ar_r_conf['password'].nil?
+      ""
+    else
+      "PGPASSWORD='#{ar_r_conf['password']}' "
+    end
+  end
+
+  def pg_dump_remote
+    puts "backup remote DB via ssh"
+    Net::SSH.start(ssh_host, ssh_user, ssh_opts) do |ssh|
+      ssh.exec!("rm -R #{remote_dump_path}")
+      ret = ssh_exec!(ssh, "mkdir -p #{remote_dump_path}")
+      check_ssh_err(ret)
+      dump = pg_remote_auth + "pg_dump -Fc -U #{ar_r_conf['username']} #{ar_r_conf['database']} > #{remote_dump_path}/tmp.bak"
+      puts dump if verbose?
+      ret = ssh_exec!(ssh, dump)
+      check_ssh_err(ret)
+    end
+  end
+
+  def pg_dump_restore
+    puts "restoring DB"
+    restore = pg_local_auth + "pg_restore -Fc -U #{ar_conf['username']} -d #{ar_to} #{pg_path}/tmp.bak"
+    puts restore if verbose?
+    pipe = IO.popen(restore)
+    while (line = pipe.gets)
+      print line if verbose?
+    end
+    ret = $?.to_i
+    if ret != 0 
+      puts "Error: local command exited with #{ret}"
+    end
+  end
+
+  def pg_path
+    Rails.root.join("tmp", "dump").to_s
+  end
+
+  def pg_dump_copy
+    FileUtils.mkdir_p(pg_path)
+    `mkdir -p #{pg_path}`
+    rsync("#{remote_dump_path}/", pg_path)
+  end
+
+  def clone_pg
+    pg_dump_remote()
+    pg_dump_copy()
+    pg_dump_restore()
+  end
+end

data/lib/cloner/version.rb

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

data/lib/cloner.rb

@@ -2,6 +2,7 @@ require "cloner/version"
 require 'thor'
 require 'active_support/concern'
 require "cloner/mongodb"
+require "cloner/postgres"
 require "cloner/ssh"
 require "cloner/rsync"
 require "cloner/internal"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cloner
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - glebtv
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-10-01 00:00:00.000000000 Z
+date: 2014-10-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -99,6 +99,7 @@ files:
 - lib/cloner/base.rb
 - lib/cloner/internal.rb
 - lib/cloner/mongodb.rb
+- lib/cloner/postgres.rb
 - lib/cloner/rsync.rb
 - lib/cloner/ssh.rb
 - lib/cloner/version.rb