pg_export 0.6.1 → 0.7.0

data/lib/pg_export/lib/pg_export/transactions/import_dump_interactively.rb

@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+
+# auto_register: false
+
+require 'dry/transaction'
+
+require 'pg_export/import'
+require 'pg_export/container'
+
+class PgExport
+  module Transactions
+    class ImportDumpInteractively
+      include Dry::Transaction(container: PgExport::Container)
+      include Import[
+        'adapters.bash_adapter',
+        'repositories.ftp_dump_repository',
+        'repositories.ftp_dump_file_repository',
+        'ui_input'
+      ]
+
+      step :open_ftp_connection, with: 'operations.open_ftp_connection'
+      step :fetch_dumps_from_ftp
+      step :select_dump
+      step :download_dump_from_ftp
+      step :close_ftp_connection
+      step :decrypt_dump, with: 'operations.decrypt_dump'
+      step :select_database
+      step :restore
+
+      private
+
+      def fetch_dumps_from_ftp(database_name:, ftp_adapter:)
+        dumps = ftp_dump_repository.all(database_name: database_name, ftp_adapter: ftp_adapter)
+        return Failure(message: 'No dumps') if dumps.none?
+
+        Success(ftp_adapter: ftp_adapter, dumps: dumps)
+      end
+
+      def select_dump(dumps:, ftp_adapter:)
+        dump = ui_input.select_dump(dumps)
+        Success(dump: dump, ftp_adapter: ftp_adapter)
+      end
+
+      def download_dump_from_ftp(dump:, ftp_adapter:)
+        dump.file = ftp_dump_file_repository.by_name(name: dump.name, ftp_adapter: ftp_adapter)
+        Success(dump: dump, ftp_adapter: ftp_adapter)
+      end
+
+      def close_ftp_connection(dump:, ftp_adapter:)
+        Thread.new { ftp_adapter.close_ftp }
+        Success(dump: dump)
+      end
+
+      def select_database(dump:)
+        name = ui_input.enter_database_name(dump.database)
+        Success(dump: dump, database: name)
+      end
+
+      def restore(dump:, database:)
+        bash_adapter.pg_restore(dump.file, database)
+        Success({})
+      rescue bash_adapter.class::PgRestoreError => e
+        Failure(message: e.to_s)
+      end
+    end
+  end
+end

data/lib/pg_export/lib/pg_export/types.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+require 'dry-types'
+require 'pg_export/lib/pg_export/value_objects/dump_file'
+
+class PgExport
+  module Types
+    include Dry::Types.module
+
+    DumpName = Strict::String.constrained(format: /.+_20[0-9]{6}_[0-9]{6}\Z/)
+    DumpType = Types::Coercible::String.enum('plain', 'encrypted')
+    DumpFile = Types.Instance(PgExport::ValueObjects::DumpFile)
+  end
+end

data/lib/pg_export/lib/pg_export/ui/interactive/input.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require 'tty-prompt'
+
+class PgExport
+  module Ui
+    module Interactive
+      class Input
+        def select_dump(dumps)
+          idx = prompt.select('Select dump to import:') do |menu|
+            menu.enum '.'
+            dumps.each_with_index do |d, i|
+              menu.choice(d.to_s, i)
+            end
+          end
+
+          dumps[idx]
+        end
+
+        def enter_database_name(default = nil)
+          puts 'To which database would you like to restore the downloaded dump?'
+          prompt.ask('Enter a local database name:') do |q|
+            q.required(true)
+            q.default(default) if default
+          end
+        end
+
+        private
+
+        def prompt
+          TTY::Prompt.new
+        end
+      end
+    end
+  end
+end

data/lib/pg_export/lib/pg_export/ui/plain/input.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+class PgExport
+  module Ui
+    module Plain
+      class Input
+        def select_dump(dumps)
+          dumps[0]
+        end
+
+        def enter_database_name(default)
+          default
+        end
+      end
+    end
+  end
+end

data/lib/pg_export/lib/pg_export/value_objects/dump_file.rb

@@ -0,0 +1,70 @@
+# frozen_string_literal: true
+
+require 'tempfile'
+
+class PgExport
+  module ValueObjects
+    class DumpFile
+      def initialize(file = Tempfile.new)
+        @file = file
+      end
+
+      def path
+        file.path
+      end
+
+      def size
+        file.size
+      end
+
+      def rewind
+        file.rewind
+      end
+
+      def read
+        file.read
+      end
+
+      def copy(cipher:)
+        cipher.reset
+        new_self = self.class.new
+        new_self.write do |f|
+          each_chunk do |chunk|
+            f << cipher.update(chunk)
+          end
+          f << cipher.final
+        end
+
+        new_self
+      end
+
+      def write(&block)
+        File.open(path, 'w', &block)
+      end
+
+      def each_chunk
+        File.open(path, 'r') do |file|
+          yield file.read(CHUNK_SIZE) until file.eof?
+        end
+      end
+
+      def size_human
+        MAPPING.each_pair { |e, s| return "#{(size.to_f / (s / 1024)).round(2)}#{e}" if size < s }
+      end
+
+      private
+
+      CHUNK_SIZE = (2**16).freeze
+      MAPPING = {
+        'B'  => 1024,
+        'kB' => 1024 * 1024,
+        'MB' => 1024 * 1024 * 1024,
+        'GB' => 1024 * 1024 * 1024 * 1024,
+        'TB' => 1024 * 1024 * 1024 * 1024 * 1024
+      }.freeze
+      private_constant :CHUNK_SIZE, :MAPPING
+
+      attr_reader :file
+    end
+  end
+end

data/lib/pg_export/system/boot/config.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+PgExport::Container.boot :config do
+  init do
+    require 'pg_export/configuration'
+  end
+
+  start do
+    register(:config, memoize: true) { PgExport::Configuration.build(ENV) }
+  end
+end

data/lib/pg_export/system/boot/interactive.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+PgExport::Container.boot(:interactive) do
+  init do
+    require 'pg_export/lib/pg_export/transactions/import_dump_interactively'
+  end
+
+  start do
+    use :main
+
+    # type = 'plain'
+    type = 'interactive'
+
+    transaction = PgExport::Transactions::ImportDumpInteractively.new(ui_input: target["ui.#{type}.input"])
+
+    unless target[:config].logger_muted?
+      use :logger
+
+      %i[
+        open_ftp_connection
+        fetch_dumps_from_ftp
+        download_dump_from_ftp
+        decrypt_dump
+        restore
+      ].each do |step|
+        transaction.subscribe(step => target["listeners.#{type}.#{step}"])
+      end
+    end
+
+    register('transactions.import_dump_interactively', memoize: true) { transaction }
+  end
+end

data/lib/pg_export/system/boot/logger.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+PgExport::Container.boot :logger do
+  init do
+    require 'pg_export/build_logger'
+  end
+
+  start do
+    use :config
+
+    register(:logger, memoize: true) do
+      PgExport::BuildLogger.call(stream: $stdout, format: target[:config].logger_format)
+    end
+  end
+end

data/lib/pg_export/system/boot/plain.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+PgExport::Container.boot(:plain) do
+  init do
+    require 'pg_export/lib/pg_export/transactions/export_dump'
+  end
+
+  start do
+    use :main
+
+    transaction = PgExport::Transactions::ExportDump.new
+
+    unless target[:config].logger_muted?
+      use :logger
+
+      type = 'plain'
+      # type = 'interactive'
+
+      %i[
+        build_dump
+        encrypt_dump
+        open_ftp_connection
+        upload_dump_to_ftp
+        remove_old_dumps_from_ftp
+        close_ftp_connection
+      ].each do |step|
+        transaction.subscribe(step => target["listeners.#{type}.#{step}"])
+      end
+    end
+
+    register('transactions.export_dump', memoize: true) { transaction }
+  end
+end

data/lib/pg_export/version.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 class PgExport
-  VERSION = '0.6.1'.freeze
+  VERSION = '0.7.0'
 end

data/lib/pg_export.rb

@@ -1,29 +1,32 @@
-require 'pg_export/version'
-require 'pg_export/configuration'
-require 'pg_export/boot_container'
-require 'pg_export/roles/interactive'
-require 'pg_export/errors'
-require 'pg_export/roles/validatable'
+# frozen_string_literal: true
+
+require 'pry'
+require 'pg_export/container'
 
 class PgExport
-  include Roles::Validatable
-
-  def initialize(**args)
-    config = Configuration.new(**args)
-    extend Roles::Interactive if config.interactive
-    @container = BootContainer.call(config.to_h)
-  rescue Dry::Struct::Error => e
-    raise ArgumentError, e
+  class InitializationError < StandardError; end
+
+  class << self
+    def interactive
+      PgExport::Container.start(:interactive)
+      new(transaction: PgExport::Container['transactions.import_dump_interactively'])
+    end
+
+    def plain
+      PgExport::Container.start(:plain)
+      new(transaction: PgExport::Container['transactions.export_dump'])
+    end
+  end
+
+  def initialize(transaction:)
+    @transaction = transaction
   end
 
-  def call(database_name, keep_dumps)
-    container[:create_and_export_dump].call(
-      validate_database_name(database_name),
-      validate_keep_dumps(keep_dumps)
-    )
+  def call(database_name, &block)
+    transaction.call(database_name: database_name, &block)
   end
 
   private
 
-  attr_reader :container
+  attr_reader :transaction
 end

data/pg_export.gemspec

@@ -1,5 +1,6 @@
-# coding: utf-8
-lib = File.expand_path('../lib', __FILE__)
+# frozen_string_literal: true
+
+lib = File.expand_path('lib', __dir__)
 $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 require 'pg_export/version'
 
@@ -18,16 +19,18 @@ Gem::Specification.new do |spec|
   spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
   spec.executables   = ['pg_export']
   spec.require_paths = ['lib']
-  spec.required_ruby_version = '>= 2.2.0'
+  spec.required_ruby_version = '>= 2.3.0'
 
-  spec.add_dependency 'cli_spinnable', '~> 0.2'
-  spec.add_dependency 'dry-types', '~> 0.11.1'
-  spec.add_dependency 'dry-struct', '~> 0.3.1'
+  spec.add_dependency 'dry-initializer', '~> 2.5.0'
+  spec.add_dependency 'dry-system', '~> 0.10.0'
+  spec.add_dependency 'dry-transaction', '~> 0.13.0'
+  spec.add_dependency 'tty-prompt'
+  spec.add_dependency 'tty-spinner'
 
-  spec.add_development_dependency 'bundler', '~> 1.10'
-  spec.add_development_dependency 'rubocop', '~> 0.44'
+  spec.add_development_dependency 'bundler', '~> 1.16'
+  spec.add_development_dependency 'pg', '~> 0.21'
+  spec.add_development_dependency 'pry'
   spec.add_development_dependency 'rake', '~> 10.0'
   spec.add_development_dependency 'rspec', '~> 3.4'
-  spec.add_development_dependency 'pg', '~> 0.19'
-  spec.add_development_dependency 'pry'
+  spec.add_development_dependency 'rubocop', '~> 0.59.2'
 end