copy_db_from_prod 0.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 63d0b8b5714cc8f1d676813327671ba4da1caa8d
+  data.tar.gz: f9ec3a35d3dc57fd6354a61d0d75c93089b81537
+SHA512:
+  metadata.gz: 5d00d39a1aff1e76c59ea40f71402b9a8d75548c9c7e4efc3e5ce7c5aa2b4657d5df3a8d66b8d3e4bbfc5f41d18db2b278e75b936bd8e39668a9918df97ca7eb
+  data.tar.gz: 21580ba4fe202ceef232837e64bd40c859fc4faeb889b466f9dc4377fb319dad838a253c4ea9eae6430eb1c5d2da8db2ba112d34f22e79ceb186bc15b1831fbd

data/bin/copy_db_from_prod

@@ -0,0 +1,64 @@
+#!/usr/bin/env ruby
+
+require 'copy_db_from_prod'
+
+def echo_wrong_args
+  puts("Wrong args")
+end
+
+def echo_help
+  puts("copy_db_from_prod -h <host> -u <user> -p <path>
+    <user> - default 'deploy'
+    <path> - default '/projects/insales3', without / at the end")
+end
+
+def schema_only(params)
+  p = CopyDbFromProd.new(params)
+  p.load_schema
+  puts('Load schema success')
+end
+
+class Wraper
+  attr_accessor :shell_args, :host
+
+  def initialize(args)
+    @shell_args = args
+    shell_args.each_with_index do |e, i|
+      if '-h' == e
+        @host = shell_args[i + 1] if '-' != shell_args[i + 1][0]
+      elsif '-u' == e
+        @user = shell_args[i + 1] if '-' != shell_args[i + 1][0]
+      elsif '-p' == e
+        @path = shell_args[i + 1] if '-' != shell_args[i + 1][0]
+      end
+    end
+  end
+
+  def include?(one)
+    @exec ||= shell_args.include?(one)
+    shell_args.include?(one)
+  end
+
+  def exec?
+    @exec
+  end
+
+  def user
+    @user ||= 'deploy'
+  end
+
+  def path
+    @path ||= '/projects/insales3'
+  end
+end
+
+w = Wraper.new(ARGV)
+
+if ARGV.empty?
+  echo_help
+else
+  schema_only(w) if w.include?('--schema-only')
+  CopyDbFromProd.hi if w.include?('--hi')
+  echo_help if w.include?('--help')
+  echo_wrong_args if !w.exec?
+end

data/lib/copy_db_from_prod.rb

@@ -0,0 +1,57 @@
+require 'net/ssh'
+require 'yaml'
+require 'pg'
+
+class CopyDbFromProd
+  attr_accessor :cf
+
+  def self.hi
+    puts "Hello world!"
+  end
+
+  def load_conf(args)
+    @cf = { host: args.host, user: args.user,
+            keys: [File.open("#{ENV['HOME']}/.ssh/id_rsa", 'r').read],
+            deploy_to: args.path }
+  end
+
+  def initialize(args)
+    load_conf(args)
+  end
+
+  def download_schema
+    Net::SSH.start(cf[:host], cf[:user], key_data: cf[:keys], keys_only: TRUE) do |ssh|
+      @prod_conf = YAML.load(ssh.exec!("cat #{cf[:deploy_to]}/config/database.yml"))['production']
+      prod_data = ssh.exec!(dump_schema)
+      if prod_data.include?('pg_dump') || prod_data.include?('warn')
+        raise "error: pg_dump dosen't work correct \n #{prod_data}"
+      else
+        return Zlib::GzipReader.new(StringIO.new(ssh.exec!(prod_data))).read
+      end
+    end
+  end
+
+  def dump_schema
+    cmd = []
+    { password: 'export PGPASSWORD=#{val} &&', database: 'pg_dump #{val}', host: '-h #{val}', port: '-p #{val}',
+      username: '-U #{val}' }.each_pair do |key, val|
+      cmd.push(val.sub('#{val}', @prod_conf[key.to_s].to_s)) if @prod_conf.key?(key.to_s)
+    end
+    "#{cmd.join(' ')} --schema-only --no-owner --no-privileges -Z9"
+  end
+
+  def load_schema
+    @dev_conf = YAML.load(File.open('config/database.yml', 'r').read)['development']
+    conn = PGconn.open(host: @dev_conf['host'],
+                     dbname: 'postgres',
+                       user: @dev_conf['username'],
+                   password: @dev_conf['password'])
+    conn.exec("DROP DATABASE IF EXISTS #{@dev_conf['database']}")
+    conn.exec("CREATE DATABASE #{@dev_conf['database']}")
+    conn = PGconn.open(host: @dev_conf['host'],
+                     dbname: @dev_conf['database'],
+                       user: @dev_conf['username'],
+                   password: @dev_conf['password'])
+    conn.exec(download_schema)
+  end
+end

metadata

@@ -0,0 +1,74 @@
+--- !ruby/object:Gem::Specification
+name: copy_db_from_prod
+version: !ruby/object:Gem::Version
+  version: 0.0.0
+platform: ruby
+authors:
+- vgulaev
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-08-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A simple hello world gem
+email: vgulaev@yandex.ru
+executables:
+- copy_db_from_prod
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/copy_db_from_prod
+- lib/copy_db_from_prod.rb
+homepage: http://rubygems.org/gems/copy_db_from_prod
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: Hola!
+test_files: []