pg_export 0.7.7 → 1.0.0.rc5

data/lib/pg_export/lib/pg_export/factories/ftp_gateway_factory.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+# auto_register: false
+
+require 'pg_export/lib/pg_export/gateways/ftp'
+require 'pg_export/import'
+
+class PgExport
+  module Factories
+    class FtpGatewayFactory
+      include Import['config']
+
+      def gateway
+        ::PgExport::Gateways::Ftp.new(
+          host: config.gateway_host,
+          user: config.gateway_user,
+          password: config.gateway_password
+        )
+      end
+    end
+  end
+end

data/lib/pg_export/lib/pg_export/factories/ssh_gateway_factory.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+# auto_register: false
+
+require 'pg_export/lib/pg_export/gateways/ssh'
+require 'pg_export/import'
+
+class PgExport
+  module Factories
+    class SshGatewayFactory
+      include Import['config']
+
+      def gateway
+        ::PgExport::Gateways::Ssh.new(
+          host: config.gateway_host,
+          user: config.gateway_user,
+          password: config.gateway_password
+        )
+      end
+    end
+  end
+end

data/lib/pg_export/lib/pg_export/{adapters/ftp_adapter.rb → gateways/ftp.rb}

@@ -5,28 +5,31 @@
 require 'net/ftp'
 
 class PgExport
-  module Adapters
-    class FtpAdapter
+  module Gateways
+    class Ftp
       CHUNK_SIZE = (2**16).freeze
 
       def initialize(host:, user:, password:)
         @host, @user, @password, @logger = host, user, password
-        ObjectSpace.define_finalizer(self, proc { ftp.close if @ftp })
       end
 
-      def open_ftp
+      def open
         @ftp = Net::FTP.new(host, user, password)
         @ftp.passive = true
         @ftp
       end
 
-      def close_ftp
+      def welcome
+        open.welcome
+      end
+
+      def close
         @ftp&.close
       end
 
-      def list(regex_string)
+      def list(name)
         ftp
-          .list(regex_string)
+          .list([name, '*'].join('_'))
           .map { |row| extracted_meaningful_attributes(row) }
           .sort_by { |item| item[:name] }
           .reverse
@@ -49,7 +52,7 @@ class PgExport
       end
 
       def ftp
-        @ftp ||= open_ftp
+        @ftp ||= open
       end
 
       private

data/lib/pg_export/lib/pg_export/gateways/ssh.rb

@@ -0,0 +1,83 @@
+# frozen_string_literal: true
+
+# auto_register: false
+
+require 'ed25519'
+require 'net/ssh'
+require 'net/scp'
+
+class PgExport
+  module Gateways
+    class Ssh
+      CHUNK_SIZE = (2**16).freeze
+
+      def initialize(host:, user:, password:)
+        @host, @user, @password, @logger = host, user, password
+      end
+
+      def open
+        @ssh =
+          if password.nil?
+            Net::SSH.start(host, user)
+          else
+            Net::SSH.start(host, user, password)
+          end
+      end
+
+      def welcome
+        open.exec!('hostname')
+      end
+
+      def close
+        @ssh&.close
+      end
+
+      def list(name)
+        grep =
+          if name.nil? || name.empty?
+            ''
+          else
+            " | grep #{name}"
+          end
+
+        ssh
+          .exec!("ls -l#{grep}")
+          .split("\n").map { |row| extract_meaningful_attributes(row) }
+          .reject { |item| item[:name].nil? }
+          .sort_by { |item| item[:name] }
+          .reverse
+      end
+
+      def delete(name)
+        # @TODO
+      end
+
+      def persist(file, name)
+        ssh.scp.upload(file.path, name).wait
+      end
+
+      def get(file, name)
+        ssh.scp.download(name, file.path).wait
+      end
+
+      def to_s
+        host
+      end
+
+      def ssh
+        @ssh ||= open
+      end
+
+      private
+
+      attr_reader :host, :user, :password
+
+      def extract_meaningful_attributes(item)
+        MEANINGFUL_KEYS.zip(item.split(' ').values_at(8, 4)).to_h
+      end
+
+      MEANINGFUL_KEYS = %i[name size].freeze
+      private_constant :MEANINGFUL_KEYS
+    end
+  end
+end

data/lib/pg_export/lib/pg_export/listeners/interactive/build_dump.rb

@@ -6,12 +6,12 @@ class PgExport
   module Listeners
     class Interactive
       class BuildDump < InteractiveListener
-        def on_step(step_name:, args:)
-          @spinner = build_spinner("Dumping database #{args.first[:database_name]}")
+        def on_step(event)
+          @spinner = build_spinner("Dumping database #{event[:args].first[:database_name]}")
         end
 
-        def on_step_succeeded(step_name:, args:, value:)
-          @spinner.success([success, value[:dump]].join(' '))
+        def on_step_succeeded(event)
+          @spinner.success([success, event[:value][:dump]].join(' '))
         end
       end
     end

data/lib/pg_export/lib/pg_export/listeners/interactive/{close_ftp_connection.rb → close_connection.rb}

@@ -5,7 +5,7 @@ require_relative '../interactive_listener'
 class PgExport
   module Listeners
     class Interactive
-      class CloseFtpConnection < InteractiveListener
+      class CloseConnection < InteractiveListener
         def on_step(*)
           @spinner = build_spinner('Closing FTP')
         end

data/lib/pg_export/lib/pg_export/listeners/interactive/decrypt_dump.rb

@@ -10,8 +10,8 @@ class PgExport
           @spinner = build_spinner('Decrypting')
         end
 
-        def on_step_succeeded(step_name:, args:, value:)
-          @spinner.success([success, value[:dump]].join(' '))
+        def on_step_succeeded(event)
+          @spinner.success([success, event[:value][:dump]].join(' '))
         end
       end
     end

data/lib/pg_export/lib/pg_export/listeners/interactive/{download_dump_from_ftp.rb → download_dump.rb}

@@ -5,13 +5,13 @@ require_relative '../interactive_listener'
 class PgExport
   module Listeners
     class Interactive
-      class DownloadDumpFromFtp < InteractiveListener
+      class DownloadDump < InteractiveListener
         def on_step(*)
           @spinner = build_spinner('Downloading')
         end
 
-        def on_step_succeeded(step_name:, args:, value:)
-          @spinner.success([success, value[:dump]].join(' '))
+        def on_step_succeeded(event)
+          @spinner.success([success, event[:value][:dump]].join(' '))
         end
       end
     end

data/lib/pg_export/lib/pg_export/listeners/interactive/encrypt_dump.rb

@@ -10,8 +10,8 @@ class PgExport
           @spinner = build_spinner('Encrypting')
         end
 
-        def on_step_succeeded(step_name:, args:, value:)
-          @spinner.success([success, value[:dump]].join(' '))
+        def on_step_succeeded(event)
+          @spinner.success([success, event[:value][:dump]].join(' '))
         end
       end
     end

data/lib/pg_export/lib/pg_export/listeners/interactive/{fetch_dumps_from_ftp.rb → fetch_dumps.rb}

@@ -5,7 +5,7 @@ require_relative '../interactive_listener'
 class PgExport
   module Listeners
     class Interactive
-      class FetchDumpsFromFtp < InteractiveListener
+      class FetchDumps < InteractiveListener
         def on_step(*)
           @spinner = build_spinner('Fetching dumps')
         end

data/lib/pg_export/lib/pg_export/listeners/interactive/{open_ftp_connection.rb → open_connection.rb}

@@ -5,9 +5,9 @@ require_relative '../interactive_listener'
 class PgExport
   module Listeners
     class Interactive
-      class OpenFtpConnection < InteractiveListener
+      class OpenConnection < InteractiveListener
         def on_step(*)
-          @spinner = build_spinner('Opening ftp connection')
+          @spinner = build_spinner('Opening connection')
         end
 
         def on_step_succeeded(*)

data/lib/pg_export/lib/pg_export/listeners/interactive/{remove_old_dumps_from_ftp.rb → remove_old_dumps.rb}

@@ -5,14 +5,14 @@ require_relative '../interactive_listener'
 class PgExport
   module Listeners
     class Interactive
-      class RemoveOldDumpsFromFtp < InteractiveListener
-        def on_step(step_name:, args:)
-          @spinner = build_spinner("Checking for old dumps on #{args.first[:ftp_adapter]}")
+      class RemoveOldDumps < InteractiveListener
+        def on_step(event)
+          @spinner = build_spinner("Checking for old dumps on #{event[:args].first[:gateway]}")
         end
 
-        def on_step_succeeded(step_name:, args:, value:)
-          if value[:removed_dumps].any?
-            @spinner.success([success, value[:removed_dumps].map { |filename| "    #{filename} removed" }].join("\n"))
+        def on_step_succeeded(event)
+          if event[:value][:removed_dumps].any?
+            @spinner.success([success, event[:value][:removed_dumps].map { |filename| "    #{filename} removed" }].join("\n"))
           else
             @spinner.success([success, 'nothing to remove'].join(' '))
           end

data/lib/pg_export/lib/pg_export/listeners/interactive/restore.rb

@@ -6,8 +6,8 @@ class PgExport
   module Listeners
     class Interactive
       class Restore < InteractiveListener
-        def on_step(step_name:, args:)
-          @spinner = build_spinner("Restoring dump to database #{args.first[:database]}")
+        def on_step(event)
+          @spinner = build_spinner("Restoring dump to database #{event[:args].first[:database]}")
         end
 
         def on_step_succeeded(*)

data/lib/pg_export/lib/pg_export/listeners/interactive/select_database.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require_relative '../interactive_listener'
+
+class PgExport
+  module Listeners
+    class Interactive
+      class SelectDatabase < InteractiveListener
+      end
+    end
+  end
+end

data/lib/pg_export/lib/pg_export/listeners/interactive/select_dump.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require_relative '../interactive_listener'
+
+class PgExport
+  module Listeners
+    class Interactive
+      class SelectDump < InteractiveListener
+      end
+    end
+  end
+end

data/lib/pg_export/lib/pg_export/listeners/interactive/{upload_dump_to_ftp.rb → upload_dump.rb}

@@ -5,9 +5,9 @@ require_relative '../interactive_listener'
 class PgExport
   module Listeners
     class Interactive
-      class UploadDumpToFtp < InteractiveListener
-        def on_step(step_name:, args:)
-          @spinner = build_spinner("Uploading #{args.first[:dump]} to #{args.first[:ftp_adapter]}")
+      class UploadDump < InteractiveListener
+        def on_step(event)
+          @spinner = build_spinner("Uploading #{event[:args].first[:dump]} to #{event[:args].first[:gateway]}")
         end
 
         def on_step_succeeded(*)

data/lib/pg_export/lib/pg_export/listeners/interactive_listener.rb

@@ -8,8 +8,8 @@ require 'tty-spinner'
 class PgExport
   module Listeners
     class InteractiveListener
-      def on_step_failed(step_name:, args:, value:)
-        @spinner.error([error, self.class.red(value[:message])].join("\n"))
+      def on_step_failed(event)
+        @spinner.error([error, self.class.red(event[:value][:message])].join("\n"))
       end
 
       class << self

data/lib/pg_export/lib/pg_export/listeners/plain/build_dump.rb

@@ -6,8 +6,8 @@ class PgExport
   module Listeners
     class Plain
       class BuildDump < PlainListener
-        def on_step_succeeded(step_name:, args:, value:)
-          logger.info("Dump database #{value[:dump].database} to #{value[:dump]}")
+        def on_step_succeeded(event)
+          logger.info("Dump database #{event[:value][:dump].database} to #{event[:value][:dump]}")
         end
       end
     end

data/lib/pg_export/lib/pg_export/listeners/plain/{close_ftp_connection.rb → close_connection.rb}

@@ -5,9 +5,9 @@ require_relative '../plain_listener'
 class PgExport
   module Listeners
     class Plain
-      class CloseFtpConnection < PlainListener
+      class CloseConnection < PlainListener
         def on_step_succeeded(*)
-          logger.info('Close FTP')
+          logger.info('Close connection')
         end
       end
     end

data/lib/pg_export/lib/pg_export/listeners/plain/decrypt_dump.rb

@@ -6,8 +6,8 @@ class PgExport
   module Listeners
     class Plain
       class DecryptDump < PlainListener
-        def on_step_succeeded(step_name:, args:, value:)
-          logger.info("Decrypt #{value[:dump]}")
+        def on_step_succeeded(event)
+          logger.info("Decrypt #{event[:value][:dump]}")
         end
       end
     end

data/lib/pg_export/lib/pg_export/listeners/plain/{download_dump_from_ftp.rb → download_dump.rb}

@@ -5,9 +5,9 @@ require_relative '../plain_listener'
 class PgExport
   module Listeners
     class Plain
-      class DownloadDumpFromFtp < PlainListener
-        def on_step_succeeded(step_name:, args:, value:)
-          logger.info("Download #{value[:dump]}")
+      class DownloadDump < PlainListener
+        def on_step_succeeded(event)
+          logger.info("Download #{event[:value][:dump]}")
         end
       end
     end

data/lib/pg_export/lib/pg_export/listeners/plain/encrypt_dump.rb

@@ -6,8 +6,8 @@ class PgExport
   module Listeners
     class Plain
       class EncryptDump < PlainListener
-        def on_step_succeeded(step_name:, args:, value:)
-          logger.info("Encrypt #{value[:dump]}")
+        def on_step_succeeded(event)
+          logger.info("Encrypt #{event[:value][:dump]}")
         end
       end
     end

data/lib/pg_export/lib/pg_export/listeners/plain/fetch_dumps.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require_relative '../plain_listener'
+
+class PgExport
+  module Listeners
+    class Plain
+      class FetchDumps < PlainListener
+        def on_step_succeeded(event)
+          logger.info("Fetch dumps (#{event[:value][:dumps].count} items)")
+        end
+      end
+    end
+  end
+end

data/lib/pg_export/lib/pg_export/listeners/plain/open_connection.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require_relative '../plain_listener'
+
+class PgExport
+  module Listeners
+    class Plain
+      class OpenConnection < PlainListener
+        def on_step_succeeded(event)
+          logger.info("Connect to #{event[:value][:gateway]}")
+        end
+      end
+    end
+  end
+end