capistrano-sync 0.0.1

data/README

@@ -0,0 +1,15 @@
+== capistrano-sync
+
+Recipies for download and replace database or directories from production or staging to development.
+
+Defined three task:
+
+* cap sync:db [RAILS_ENV=env]
+* cap sync:dir FROM=path_relative_to_deploy TO=path
+* cap sync:dir:public_system
+
+== Installation
+
+Add to you Gmefile:
+
+    gem "capistrano-sync", :git => "git://github.com/dima-exe/capistrano-sync.git"

data/Rakefile

@@ -0,0 +1,13 @@
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gemspec|
+    gemspec.name = "capistrano-sync"
+    gemspec.summary = "Sync db or directories recipes for Capistrano"
+    gemspec.description = "Sync db or directories recipes for Capistrano"
+    gemspec.email = "dima.exe@gmail.com"
+    gemspec.homepage = "http://github.com/dima-exe/capistrano-sync"
+    gemspec.authors = ["Dmitry Galinsky"]
+  end
+rescue LoadError
+  puts "Jeweler not available. Install it with: gem install jeweler"
+end

data/VERSION

@@ -0,0 +1 @@
+0.0.1

data/capistrano-sync.gemspec

@@ -0,0 +1,41 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{capistrano-sync}
+  s.version = "0.0.1"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Dmitry Galinsky"]
+  s.date = %q{2011-10-06}
+  s.description = %q{Sync db or directories recipes for Capistrano}
+  s.email = %q{dima.exe@gmail.com}
+  s.extra_rdoc_files = [
+    "README"
+  ]
+  s.files = [
+    "README",
+    "Rakefile",
+    "VERSION",
+    "capistrano-sync.gemspec",
+    "lib/capistrano-sync.rb",
+    "lib/capistrano-sync/sync.rb"
+  ]
+  s.homepage = %q{http://github.com/dima-exe/capistrano-sync}
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.7}
+  s.summary = %q{Sync db or directories recipes for Capistrano}
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+    else
+    end
+  else
+  end
+end
+

data/lib/capistrano-sync/sync.rb

@@ -0,0 +1,229 @@
+require 'tempfile'
+require 'action_view'
+require 'capistrano'
+
+module CapistranoSyncTask
+  module Helper
+    def get_ssh_command()
+      server = cap.find_servers.first
+      srv = server.host
+      srv += ":#{server.port}" if server.port
+      ["ssh -l #{cap[:user]} #{srv}", srv]
+    end
+
+    def log(*args)
+      cap.logger.info *args
+    end
+
+    # missing capture task
+    def _capture(command, options={})
+      output = ""
+      cap.invoke_command(command, options.merge(:once => true)) do |ch, stream, data|
+        case stream
+        when :out then output << data
+        when :err then warn "[err :: #{ch[:server]}] #{data}"
+        end
+      end
+      output
+    end
+  end
+
+  class Db
+    include Helper
+    attr_accessor :rails_env, :local_rails_env, :tables, :cap
+
+    def initialize(cap, rails_env, local_rails_env, tables)
+      self.rails_env = rails_env
+      self.local_rails_env = local_rails_env
+      self.tables = tables
+      self.cap = cap
+    end
+
+    def sync
+      remote_config = remote_database_config
+      local_config  = local_database_config
+
+      dump_method = remote_config[:adapter] + "_dump"
+      load_method = local_config[:adapter] + "_load"
+      unless self.private_methods.include?(dump_method)
+        puts "FATAL: Can't dump: unknown adapter #{remote_config[:adapter].inspect}"
+        exit(1)
+      end
+      unless self.private_methods.include?(load_method)
+        puts "FATAL: Can't load: unknown adapter #{local_config[:adapter].inspect}"
+        exit(1)
+      end
+
+      dump_command    = __send__(dump_method, remote_config)
+      load_command    = __send__(load_method, local_config)
+      ssh_cmd, server = get_ssh_command
+
+      log "Drop and create local database"
+      drop_and_create_local_db
+      log "Dump from #{server} and load to local #{local_rails_env} db (see progress)"
+
+      cmd = "#{ssh_cmd} #{dump_command} | pv | #{load_command}"
+      system cmd
+    end
+
+    private
+    def drop_and_create_local_db
+      system "bundle exec rake -q db:drop db:create RAILS_ENV=#{local_rails_env} 2&>1 /dev/null"
+    end
+
+    def remote_database_config
+      unless @remove_database_config
+        remote_config = _capture "cat #{cap.current_path}/config/database.yml"
+        @remove_database_config = load_database_config(remote_config, rails_env)
+      end
+      @remove_database_config
+    end
+
+    def local_database_config
+      unless @local_database_config
+        local_config = IO.read("config/database.yml")
+        @local_database_config = load_database_config(local_config, self.local_rails_env)
+      end
+      @local_database_config
+    end
+
+    def load_database_config(io, env)
+      config = YAML.load(io)[env]
+      raise "Can't read #{env} entry from database.yml" unless config
+      {
+        :adapter  => config["adapter"],
+        :user     => config["username"] || config["user"],
+        :pass     => config["password"],
+        :dbname   => config["database"],
+        :socket   => config["socket"],
+        :host     => config["host"],
+      }
+    end
+
+    def postgresql_dump(config)
+      dump_cmd = "pg_dump"
+      dump_cmd << " --no-owner --no-privileges --disable-triggers --inserts"
+      dump_cmd << " --username=#{config[:user]}" if config[:user]
+      #dump_cmd << " --password=#{config[:pass]}" if config[:pass]
+      dump_cmd << " --host=#{config[:host]}" if config[:host]
+      dump_cmd << " #{config[:dbname]}"
+    end
+
+    def postgresql_load(config)
+      load_cmd = "psql --single-transaction --quiet -o /dev/null"
+      load_cmd << " --username=#{config[:user]}" if config[:user]
+      load_cmd << " --password=#{config[:pass]}" if config[:pass]
+      load_cmd << " --host=#{config[:host]}" if config[:host]
+      load_cmd << " #{config[:dbname]}"
+      load_cmd
+    end
+
+    def mysql2_dump(config)
+      dump_cmd = "mysqldump"
+      dump_cmd << " --quick --single-transaction"
+      dump_cmd << " --user=#{config[:user]}" if config[:user]
+      dump_cmd << " --password=#{config[:pass]}" if config[:pass]
+      dump_cmd << " --socket=#{config[:socket]}" if config[:socket]
+      dump_cmd << " --host=#{config[:host]}" if config[:host]
+      dump_cmd << " --opt #{config[:dbname]} #{config[:tables]}"
+      dump_cmd
+    end
+
+    def mysql2_load(config)
+      load_cmd = "mysql"
+      load_cmd << " --user=#{config[:user]}" if config[:user]
+      #load_cmd << " --password=#{config[:pass]}" if config[:pass]
+      load_cmd << " --socket=#{config[:socket]}" if config[:socket]
+      load_cmd << " --host=#{config[:host]}" if config[:host]
+      load_cmd << " #{config[:dbname]}"
+      load_cmd
+    end
+  end
+
+  class Dir
+    include Helper
+    attr_accessor :from, :to, :cap
+    def initialize(cap, from, to)
+      self.from = from
+      self.to   = to
+      self.cap  = cap
+    end
+
+    def sync
+      check_deps
+      ssh_cmd, server = get_ssh_command
+      log "Download and extract #{server}:#{from} -> local:#{to} (see progress)"
+      cmd = "#{ssh_cmd} \"tar -cC #{from} .\" |pv -s #{total} | tar -x -C #{to}"
+      system cmd
+    end
+
+    private
+    def check_deps
+      system "which pv > /dev/null"
+      unless ($?.to_i == 0)
+        puts "FATAL: pv (Pipe Viewer) command not found, please install 'port install pv' or 'brew install pv'"
+        exit(1)
+      end
+    end
+    def total
+      unless @total
+        log "Calculate files size"
+        @total = _capture("du -sb #{from} | awk '{print $1}'", :once => true).to_i
+        log "TOTAL: #{proxy.number_to_human_size @total}"
+      end
+      @total
+    end
+
+    def proxy
+      @proxy ||= HelperProxy.new
+    end
+
+    def log(*args)
+      cap.logger.info *args
+    end
+  end
+
+  class HelperProxy
+    include ActionView::Helpers::NumberHelper
+  end
+
+  module Capistrano
+    def self.load_into(configuration)
+      configuration.load do
+        namespace :sync do
+          namespace :dir do
+            desc "sync directory"
+            task :default, :roles => :app do
+              from_path = deploy_to + "/" + ENV["FROM"]
+              to_path = ENV["TO"]
+              if !from_path || !to_path
+                puts "Usage cap sync:dir FROM=<..> TO=<..>"
+                exit(1)
+              end
+              s = CapistranoSyncTask::Dir.new(self, from_path, to_path)
+              s.sync
+            end
+
+            desc "sync public/system directory"
+            task :public_system, :roles => :app do
+              s = CapistranoSyncTask::Dir.new(self, "#{deploy_to}/current/public/system/", "public/system/")
+              s.sync
+            end
+          end
+
+          desc "sync databases"
+          task :db, :roles => :db do
+            local_rails_env = ENV["RAILS_ENV"] || "development"
+            tables = ENV["TABLES"] || ""
+            s = CapistranoSyncTask::Db.new(self, rails_env, local_rails_env, tables)
+            s.sync
+          end
+        end
+      end
+    end
+  end
+end
+
+if Capistrano::Configuration.instance
+  CapistranoSyncTask::Capistrano.load_into(Capistrano::Configuration.instance)
+end

data/lib/capistrano-sync.rb

@@ -0,0 +1 @@
+require 'capistrano-sync/sync'

metadata

@@ -0,0 +1,72 @@
+--- !ruby/object:Gem::Specification 
+name: capistrano-sync
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Dmitry Galinsky
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-10-06 00:00:00 +04:00
+default_executable: 
+dependencies: []
+
+description: Sync db or directories recipes for Capistrano
+email: dima.exe@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README
+files: 
+- README
+- Rakefile
+- VERSION
+- capistrano-sync.gemspec
+- lib/capistrano-sync.rb
+- lib/capistrano-sync/sync.rb
+has_rdoc: true
+homepage: http://github.com/dima-exe/capistrano-sync
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Sync db or directories recipes for Capistrano
+test_files: []
+