Support many sources

.ruby-version

@@ -0,0 +1 @@
+2.7.1

.travis.yml

@@ -1,5 +1,8 @@
 sudo: false
 language: ruby
 rvm:
-  - 2.2.3
-before_install: gem install bundler -v 1.12.2
+  - 2.4.4
+  - 2.5.5
+  - 2.6.2
+  - 2.7.1
+script: bundle exec rspec

dump-hook.gemspec

@@ -29,9 +29,9 @@ Gem::Specification.new do |spec|
 
   spec.add_dependency 'timecop'
 
-  spec.add_development_dependency "bundler", "~> 1.12"
-  spec.add_development_dependency "rake", "~> 10.0"
-  spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_development_dependency "bundler", ">= 2.2.10"
+  spec.add_development_dependency "rake", ">= 12.3.3"
+  spec.add_development_dependency "rspec"
   spec.add_development_dependency "sequel"
   spec.add_development_dependency "pg"
   spec.add_development_dependency "mysql2"

lib/dump_hook.rb

@@ -1,5 +1,8 @@
 require "dump_hook/version"
+require "dump_hook/hooks/mysql"
+require "dump_hook/hooks/postgres"
 require "timecop"
+require "ostruct"
 
 module DumpHook
   class Settings
@@ -11,31 +14,47 @@ class Settings
                   :username,
                   :password,
                   :host,
-                  :port
+                  :port,
+                  :sources,
+                  :with_sources
 
     def initialize
-      @database = 'please set database'
       @database_type = 'postgres'
       @dumps_location = 'tmp/dump_hook'
       @remove_old_dumps = true
+      @with_sources = false
+      @sources = {}
     end
   end
 
   class << self
     attr_accessor :settings
-  end
+    attr_accessor :hooks
 
-  def self.setup
-    self.settings = Settings.new
-    yield(settings)
+    def setup
+      self.settings = Settings.new
+      yield(settings)
+      unless settings.sources.empty?
+        settings.with_sources = true
+      end
+      unless settings.database.nil?
+        single_source = { type: settings.database_type.to_sym,
+                          database: settings.database,
+                          username: settings.username,
+                          password: settings.password,
+                          host: settings.host,
+                          port: settings.port }
+        self.settings.sources[settings.database_type.to_sym] = single_source
+      end
+    end
   end
 
   def execute_with_dump(name, opts={}, &block)
     created_on = opts[:created_on]
     actual = opts[:actual] || settings.actual
     create_dirs_if_not_exists
     filename = full_filename(name, created_on, actual)
-    if File.exists?(filename)
+    if File.exist?(filename)
       restore_dump(filename)
     else
       if created_on
@@ -54,30 +73,36 @@ def settings
   end
 
   def store_dump(filename)
-    case settings.database_type
-      when 'postgres'
-        args = ['-a', '-x', '-O', '-f', filename, '-Fc', '-T', 'schema_migrations']
-        args.concat(pg_connection_args)
-        Kernel.system("pg_dump", *args)
-      when 'mysql'
-        args = mysql_connection_args
-        args << "--compress"
-        args.concat ["--result-file", filename]
-        args.concat ["--ignore-table", "#{settings.database}.schema_migrations"]
-        Kernel.system("mysqldump", *args)
+    FileUtils.mkdir_p(filename)
+    settings.sources.each do |name, parameters|
+      filename_with_namespace = File.join(filename, "#{name}.dump")
+      connection_settings = OpenStruct.new(parameters.slice(:database, :username, :password, :port, :host))
+      dumper = case parameters[:type]
+               when :postgres
+                 Hooks::Postgres.new(connection_settings)
+               when :mysql
+                 Hooks::MySql.new(connection_settings)
+               else
+                 raise "Unsupported type of source"
+               end
+      dumper.dump(filename_with_namespace)
     end
   end
 
   def restore_dump(filename)
-    case settings.database_type
-      when 'postgres'
-        args = pg_connection_args
-        args << filename
-        Kernel.system("pg_restore", *args)
-      when 'mysql'
-        args = mysql_connection_args
-        args.concat ["-e", "source #{filename}"]
-        Kernel.system("mysql", *args)
+    FileUtils.mkdir_p(filename)
+    settings.sources.each do |name, parameters|
+      filename_with_namespace = File.join(filename, "#{name}.dump")
+      connection_settings = OpenStruct.new(parameters.slice(:database, :username, :password, :port, :host))
+      dumper = case parameters[:type]
+               when :postgres
+                 Hooks::Postgres.new(connection_settings)
+               when :mysql
+                 Hooks::MySql.new(connection_settings)
+               else
+                 raise "Unsupported type of source"
+               end
+      dumper.restore(filename_with_namespace)
     end
   end
 
@@ -88,25 +113,12 @@ def full_filename(name, created_on, actual)
     elsif actual
       name_with_created_on = "#{name_with_created_on}_actual#{actual}"
     end
-    "#{settings.dumps_location}/#{name_with_created_on}.dump"
+    full_path = "#{settings.dumps_location}/#{name_with_created_on}"
+
+    settings.with_sources ? full_path : "#{full_path}.dump"
   end
 
   def create_dirs_if_not_exists
     FileUtils.mkdir_p(settings.dumps_location)
   end
-
-  def mysql_connection_args
-    args = [settings.database]
-    args.concat ['--user', settings.username] if settings.username
-    args << "--password=#{settings.password}" if settings.password
-    args
-  end
-
-  def pg_connection_args
-    args = ['-d', settings.database]
-    args.concat(['-U', settings.username]) if settings.username
-    args.concat(['-h', settings.host]) if settings.host
-    args.concat(['-p', settings.port]) if settings.port
-    args
-  end
 end

lib/dump_hook/hooks/connectable.rb

@@ -0,0 +1,9 @@
+module DumpHook
+  module Hooks
+    module Connectable
+      def initialize(connection_settings)
+        @connection_settings = connection_settings
+      end
+    end
+  end
+end

lib/dump_hook/hooks/mysql.rb

@@ -0,0 +1,27 @@
+require "dump_hook/hooks/connectable"
+
+module DumpHook
+  module Hooks
+    class MySql
+      include ::DumpHook::Hooks::Connectable
+
+      def dump(filename)
+        args = [@connection_settings.database]
+        args << "--compress"
+        args.concat(["--result-file", filename])
+        args.concat(["--ignore-table", "#{@connection_settings.database}.schema_migrations"])
+        args.concat(['--user', @connection_settings.username]) if @connection_settings.username
+        args << "--password=#{@connection_settings.password}" if @connection_settings.password
+        Kernel.system("mysqldump", *args)
+      end
+
+      def restore(filename)
+        args = [@connection_settings.database]
+        args.concat(["-e", "source #{filename}"])
+        args.concat(['--user', @connection_settings.username]) if @connection_settings.username
+        args << "--password=#{@connection_settings.password}" if @connection_settings.password
+        Kernel.system("mysql", *args)
+      end
+    end
+  end
+end

lib/dump_hook/hooks/postgres.rb

@@ -0,0 +1,27 @@
+require "dump_hook/hooks/connectable"
+
+module DumpHook
+  module Hooks
+    class Postgres
+      include ::DumpHook::Hooks::Connectable
+
+      def dump(filename)
+        args = ['-a', '-x', '-O', '-f', filename, '-Fc', '-T', 'schema_migrations']
+        args.concat(['-d', @connection_settings.database])
+        args.concat(['-U', @connection_settings.username]) if @connection_settings.username
+        args.concat(['-h', @connection_settings.host]) if @connection_settings.host
+        args.concat(['-p', @connection_settings.port.to_s]) if @connection_settings.port
+        Kernel.system("pg_dump", *args)
+      end
+
+      def restore(filename)
+        args = ['-d', @connection_settings.database]
+        args.concat(['-U', @connection_settings.username]) if @connection_settings.username
+        args.concat(['-h', @connection_settings.host]) if @connection_settings.host
+        args.concat(['-p', @connection_settings.port.to_s]) if @connection_settings.port
+        args << filename
+        Kernel.system("pg_restore", *args)
+      end
+    end
+  end
+end

lib/dump_hook/version.rb

@@ -1,3 +1,3 @@
 module DumpHook
-  VERSION = "0.0.1"
+  VERSION = "1.0.0.rc"
 end