capistrano-db-pull 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 23e7fa198b7f768db665cdf3f2817607065e14a3
-  data.tar.gz: 4cb1921a81df400f094149ce22a152037ea89bb4
+  metadata.gz: e69cec59066ebbbfc828f42903230ee0448c9322
+  data.tar.gz: bb0fc6405d575a91229b436b622a82f479b24ed3
 SHA512:
-  metadata.gz: 92fb33b8b8d30ba9e178a2fd69bb82c5f58b2fd172e17a1b303f358bbc6c1c832e0baf72c55a40f4c95d93faf078c3575407489a42cb09ca67d8af6ab5ec3882
-  data.tar.gz: 8973045627fadb3538b5605e9c32eafff1dc2a95d6acd19c7ab2664afb438fdab3e96c672f56af9a9367e3f9f461b39534b079732ba4334a1079c080f41728e2
+  metadata.gz: 4c9b54d76b8decb4df3b4717200dae8af5c52b1a1d74e12e4c94f7457c845b8eb12c64e7859da49c413298c910a8f7bcb07402b64079bbfe8eadf8a9d22b0e4a
+  data.tar.gz: 9f1a439e96fcb283dc278a5ef4d80f4cb8617387973cdd16a21e08812ae51dd4a29be04503939e5a241e4e856943b379762f683d620f921875e0425f2e78b9ce

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.1'
+  s.version     = '0.0.2'
   s.licenses    = ['BSD-2-Clause']
 
   s.summary     = 'Download remote database to local database'

data/lib/capistrano/db/database.rb

@@ -0,0 +1,38 @@
+module Database
+  class Base
+    attr_accessor :capistrano, :config
+
+    def initialize(instance)
+      @capistrano = instance
+    end
+
+    def postgresql?
+      adapter == 'pg' || adapter == 'postgresql'
+    end
+
+    def sqlite3?
+      adapter == 'sqlite3'
+    end
+
+    def adapter
+      @config['adapter'].downcase
+    end
+  end
+
+  class Remote < Base
+    def initialize(instance, stage)
+      super(instance)
+      config = @capistrano.capture(
+          "cat #{@capstrano.current_path}/config/database.yml")
+      @config = YAML.load(ERB.new(config).result)[stage.to_s]
+    end
+  end
+
+  class Local < Base
+    def initialize(instance)
+      super(instance)
+      config = File.read('config/database.yml')
+      @config = YAML.load(ERB.new(config).result)['development']
+    end
+  end
+end

data/lib/capistrano/db/pull.rb

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

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

@@ -1,26 +1,36 @@
 namespace :db do
   task :pull do
     on roles(:db) do
-      execute "pg_dump --data-only --exclude-table=schema_migrations --column-inserts | gzip -9 > #{fetch(:application)}.sql.gz"
+      remote = Database::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"
+      else
+        raise "Remote database adapter '#{remote.adapter}' is currently unsupported"
+      end
       download! "#{fetch(:application)}.sql.gz", "#{fetch(:application)}.sql.gz"
       execute "rm #{fetch(:application)}.sql.gz"
     end
 
-    system "echo 'BEGIN;' > #{fetch(:application)}.sql"
-    system "gunzip -c #{fetch(:application)}.sql.gz | sed '/^SET/ d' |\
-      sed '/^SELECT pg_catalog.setval/ d' |\
-      sed \"s/ true,/ 't',/g\" |\
-      sed \"s/ false,/ 'f',/g\" |\
-      sed \"s/(true/('t'/g\" |\
-      sed \"s/(false/('f'/g\" |\
-      sed \"s/true)/'t')/g\" |\
-      sed \"s/false)/'f')/g\" >> #{fetch(:application)}.sql"
-    system "echo 'END;' >> #{fetch(:application)}.sql"
+    local = Database::Local.new(self)
+    if local.sqlite3?
+      system "echo 'BEGIN;' > #{fetch(:application)}.sql"
+      system "gunzip -c #{fetch(:application)}.sql.gz | sed '/^SET/ d' |\
+        sed '/^SELECT pg_catalog.setval/ d' |\
+        sed \"s/ true,/ 't',/g\" |\
+        sed \"s/ false,/ 'f',/g\" |\
+        sed \"s/(true/('t'/g\" |\
+        sed \"s/(false/('f'/g\" |\
+        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 "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"
+    else
+      raise "Local database adapter '#{local.adapter}' is currently unsupported"
+    end
     system "rm #{fetch(:application)}.sql.gz"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: capistrano-db-pull
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Frank Groeneveld
@@ -37,6 +37,7 @@ files:
 - LICENSE
 - capistrano-db-pull.gemspec
 - lib/capistrano-db-pull.rb
+- lib/capistrano/db/database.rb
 - lib/capistrano/db/pull.rb
 - lib/capistrano/tasks/db-pull.rake
 homepage: https://github.com/ivaldi/capistrano-db-pull