capistrano-db-pull 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e69cec59066ebbbfc828f42903230ee0448c9322
-  data.tar.gz: bb0fc6405d575a91229b436b622a82f479b24ed3
+  metadata.gz: a027c0c483289dc7577a0c642bea055255e583e9
+  data.tar.gz: b9330f0a2cf9f0d7acbf2351703faf49e2d8ac93
 SHA512:
-  metadata.gz: 4c9b54d76b8decb4df3b4717200dae8af5c52b1a1d74e12e4c94f7457c845b8eb12c64e7859da49c413298c910a8f7bcb07402b64079bbfe8eadf8a9d22b0e4a
-  data.tar.gz: 9f1a439e96fcb283dc278a5ef4d80f4cb8617387973cdd16a21e08812ae51dd4a29be04503939e5a241e4e856943b379762f683d620f921875e0425f2e78b9ce
+  metadata.gz: 4c84368adc629e9ae81fe9f3c4be62e72008d9c909ac918cff290bb3ea75206946d4a383fee6449504d5721fd67ee5c1e99b3d95e91f235dfe827379c14e95de
+  data.tar.gz: f9354ef4d04d3bc9d5395c823f5b0ea628747480e611eff4c46197e1b7c592a9fdf71c301383f04fcbdf5e11cca08e0ad112f09f017250291de13c60a56ccec8

data/capistrano-db-pull.gemspec

@@ -2,7 +2,7 @@ $:.push File.expand_path('../lib', __FILE__)
 
 Gem::Specification.new do |s|
   s.name        = 'capistrano-db-pull'
-  s.version     = '0.0.2'
+  s.version     = '0.0.3'
   s.licenses    = ['BSD-2-Clause']
 
   s.summary     = 'Download remote database to local database'

data/lib/{capistrano/db/database.rb → capistrano-db-pull/application.rb}

@@ -1,4 +1,4 @@
-module Database
+module Application
   class Base
     attr_accessor :capistrano, :config
 
@@ -14,9 +14,17 @@ module Database
       adapter == 'sqlite3'
     end
 
+    def mysql?
+      adapter == 'mysql2' || adapter == 'mysql'
+    end
+
     def adapter
       @config['adapter'].downcase
     end
+
+    def database
+      @config['database']
+    end
   end
 
   class Remote < Base

data/lib/capistrano-db-pull/database.rb

@@ -0,0 +1,36 @@
+module Database
+  def self.factory application
+    if application.postgresql?
+      Database::PostgreSQL.new
+    elsif application.sqlite3?
+      Database::SQLite3.new
+    elsif application.mysql?
+      Database::MySQL.new
+    else
+      raise 'Not implemented'
+    end
+  end
+
+  class Base
+
+    def dump
+      raise 'Not implemented'
+    end
+
+    def load filename
+      raise 'Not implemented'
+    end
+  end
+
+  class PostgreSQL < Base
+    def dump
+      "pg_dump --data-only --exclude-table=schema_migrations --column-inserts"
+    end
+  end
+
+  class Sqlite3 < Base
+  end
+
+  class MySQL < Base
+  end
+end

data/lib/capistrano/db/pull.rb

@@ -1,3 +1,4 @@
-require 'capistrano/db/database'
+require 'capistrano-db-pull/application'
+require 'capistrano-db-pull/database'
 
 load File.expand_path('../../tasks/db-pull.rake', __FILE__)

data/lib/capistrano/tasks/db-pull.rake

@@ -1,9 +1,12 @@
 namespace :db do
   task :pull do
+    remote = nil
     on roles(:db) do
-      remote = Database::Remote.new(self, fetch(:stage) || 'production')
+      remote = Application::Remote.new(self, fetch(:stage) || 'production')
       if remote.postgresql?
-        execute "pg_dump --data-only --exclude-table=schema_migrations --column-inserts | gzip -9 > #{fetch(:application)}.sql.gz"
+        execute "#{Database.factory(remote).dump} | gzip -9 > #{fetch(:application)}.sql.gz"
+      elsif remote.mysql?
+        execute "mysqldump --skip-opt --no-create-info #{remote.database} | gzip -9 > #{fetch(:application)}.sql.gz"
       else
         raise "Remote database adapter '#{remote.adapter}' is currently unsupported"
       end
@@ -11,8 +14,8 @@ namespace :db do
       execute "rm #{fetch(:application)}.sql.gz"
     end
 
-    local = Database::Local.new(self)
-    if local.sqlite3?
+    local = Application::Local.new(self)
+    if remote.postgresql? && local.sqlite3?
       system "echo 'BEGIN;' > #{fetch(:application)}.sql"
       system "gunzip -c #{fetch(:application)}.sql.gz | sed '/^SET/ d' |\
         sed '/^SELECT pg_catalog.setval/ d' |\
@@ -23,14 +26,18 @@ namespace :db do
         sed \"s/true)/'t')/g\" |\
         sed \"s/false)/'f')/g\" >> #{fetch(:application)}.sql"
       system "echo 'END;' >> #{fetch(:application)}.sql"
-
-      system "bin/rake db:drop && bin/rake db:schema:load &&
-          cat #{fetch(:application)}.sql | sqlite3 db/development.sqlite3"
-
-      system "rm #{fetch(:application)}.sql"
+    elsif remote.mysql? && local.sqlite3?
+      system "gunzip -c #{fetch(:application)}.sql.gz |
+          sed 's/\\`//g' |
+          sed \"s/\\\'/\'\'/g\" |
+          sed 's/\\\"/\"/g' > #{fetch(:application)}.sql"
     else
       raise "Local database adapter '#{local.adapter}' is currently unsupported"
     end
+    system "bin/rake db:drop && bin/rake db:schema:load &&
+        cat #{fetch(:application)}.sql | sqlite3 db/development.sqlite3"
+
+    system "rm #{fetch(:application)}.sql"
     system "rm #{fetch(:application)}.sql.gz"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: capistrano-db-pull
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Frank Groeneveld
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-02-19 00:00:00.000000000 Z
+date: 2017-03-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: capistrano
@@ -37,7 +37,8 @@ files:
 - LICENSE
 - capistrano-db-pull.gemspec
 - lib/capistrano-db-pull.rb
-- lib/capistrano/db/database.rb
+- lib/capistrano-db-pull/application.rb
+- lib/capistrano-db-pull/database.rb
 - lib/capistrano/db/pull.rb
 - lib/capistrano/tasks/db-pull.rake
 homepage: https://github.com/ivaldi/capistrano-db-pull
@@ -60,7 +61,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.5.1
+rubygems_version: 2.5.2
 signing_key: 
 specification_version: 4
 summary: Download remote database to local database