dustcart 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4d97b71787f1d65b045c0a12aa49bbfb3c188574
-  data.tar.gz: 6e9eec49566a1d84ba6aa59f41ab1d2e7a4f7263
+  metadata.gz: 803c7cadc641f88d2592bd95bf795972fbebcb75
+  data.tar.gz: 7ce783b1ba4b835c1f2ee7b8d9dbf304f00ef352
 SHA512:
-  metadata.gz: 972e8c1b5db68163ed5f751260e2b0dd49fe8c9b53cc8a5c5baa9a610f370ffad6933c92ca30a3b648d17af53703118ab292c2122f9e159bb27c60132b13bfe3
-  data.tar.gz: 80d23388d7cfb4ba87fb0d21df73144e1db4506496d87983ea0c0939d8f66530a545e27114f519eb4c7acd825c6cd82e88c6cdd76e1610bad4c2545b61ceea3b
+  metadata.gz: 7fc2e4e4134815ecd506695bed407a2846f9d705a2d8cb028e1b510af1b19a52c3a7e0ea5000c0fc49127a660fa83381d4c22b6f2d9cd5f2d0d98ea89c2086f5
+  data.tar.gz: 01114180e209608f15000c31e2e3c41665276220e0c876e383f0e28a9801372cdb55edf480ca1ccd382b7bebc243adf58d7e44c441eab839b575c0f64b9c68f0

data/dustcart.gemspec

@@ -12,7 +12,7 @@ Gem::Specification.new do |spec|
   spec.summary       = 'S3 integrated backup tool.'
   spec.description   = 'CLI tool to send files/db_dump to S3.'
   spec.homepage      = 'https://github.com/tamano/dustcart'
-  spec.license       = "MIT"
+  spec.license       = 'MIT'
 
   spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
   spec.bindir        = 'exe'

data/lib/dustcart/group.rb

@@ -3,8 +3,10 @@ module Dustcart
   class Group
     INPUT_RESOURCES = [
       'dustcart/input/base',
+      'dustcart/input/file_base',
       'dustcart/input/directory',
-      'dustcart/input/file'
+      'dustcart/input/file',
+      'dustcart/input/postgres'
     ].freeze
 
     OUTPUT_RESOURCES = [

data/lib/dustcart/input/base.rb

@@ -3,19 +3,18 @@ module Dustcart
     module Input
       # base class for input
       class Base < Resource::Base
-        attr_reader :from
+        define_attribute :label
         alias_method :to_dir, :dump_dir
 
-        def initialize(to_dir, from, &block)
+        def initialize(to_dir, &block)
           super(to_dir, &block)
-          @from = from
         end
 
         def precheck
           super
 
-          raise <<-EOS.unindent unless Object::File.exists?(from)
-            target(#{from}) does not exists
+          raise <<-EOS.unindent if attributes.key?(:label) && label !~ /^[\w.]+$/
+            label should be word characters ([a-zA-Z0-9_.]+)
           EOS
         end
       end

data/lib/dustcart/input/directory.rb

@@ -2,19 +2,13 @@ module Dustcart
   module Resource
     module Input
       # input: directory
-      class Directory < Base
-        define_attribute :label
-
+      class Directory < FileBase
         def precheck
           super
 
           raise <<-EOS.unindent unless Object::File.directory?(from)
             target(#{from}) is not a directory
           EOS
-
-          raise <<-EOS.unindent if attributes.key?(:label) && label !~ /^\w+$/
-            label should be word characters ([a-zA-Z0-9_]+)
-          EOS
         end
 
         def run

data/lib/dustcart/input/file.rb

@@ -2,7 +2,7 @@ module Dustcart
   module Resource
     module Input
       # input: file
-      class File < Base
+      class File < FileBase
         define_attribute :label
 
         def precheck
@@ -11,10 +11,6 @@ module Dustcart
           raise <<-EOS.unindent unless Object::File.file?(from)
             target(#{from}) is not a regular file
           EOS
-
-          raise <<-EOS.unindent if attributes.key?(:label) && label !~ /^\w+$/
-            label should be word characters ([a-zA-Z0-9_]+)
-          EOS
         end
 
         def run

data/lib/dustcart/input/file_base.rb

@@ -0,0 +1,23 @@
+module Dustcart
+  module Resource
+    module Input
+      # base class for file system input
+      class FileBase < Base
+        attr_reader :from
+
+        def initialize(to_dir, from, &block)
+          super(to_dir, &block)
+          @from = from
+        end
+
+        def precheck
+          super
+
+          raise <<-EOS.unindent unless Object::File.exists?(from)
+            target(#{from}) does not exists
+          EOS
+        end
+      end
+    end
+  end
+end

data/lib/dustcart/input/postgres.rb

@@ -0,0 +1,104 @@
+require 'helper/system_command'
+require 'shellwords'
+
+module Dustcart
+  module Resource
+    module Input
+      # input: postgresql
+      class Postgres < Base
+        define_attribute :host
+        define_attribute :port
+        define_attribute :user
+        define_attribute :pass
+        attr_reader :database_name
+
+        COMMAND = 'pg_dump'.freeze
+
+        def initialize(to_dir, database_name, &block)
+          super(to_dir, &block)
+          @database_name = database_name
+
+          host  '127.0.0.1'     unless host
+          port  '5432'          unless port
+          label 'database.dump' unless label
+        end
+
+        def precheck
+          super
+
+          requirements
+
+          raise <<-EOS.unindent unless SystemCommand.exists?(COMMAND)
+            #{COMMAND} command is not installed
+          EOS
+
+          raise <<-EOS.unindent unless database_name
+            'database_name' is required
+          EOS
+        end
+
+        def run
+          super
+
+          create_dump
+        end
+
+        private
+
+        def requirements
+          %i(host port user pass label).each do |var|
+            raise <<-EOS.unindent unless attributes.key?(var)
+              '#{var}' is required
+            EOS
+          end
+        end
+
+        def create_dump
+          pipeline = CommandPipeline.new
+          pipeline.add(
+            "#{password_env} #{COMMAND} #{options} #{Shellwords.escape(database_name)}"
+          )
+          pipeline.run
+        end
+
+        def password_env
+          "PGPASSWORD=#{Shellwords.escape(pass)}"
+        end
+
+        # ignore :reek:FeatureEnvy:
+        def options
+          options = []
+          options << general_options
+          options << user_options
+          options << target_options
+          options.join(' ')
+        end
+
+        # ignore :reek:FeatureEnvy:
+        def general_options
+          options = []
+          options << '--format=custom'
+          options << '--blobs'
+          options << "--file=#{Shellwords.escape(to_dir)}/#{Shellwords.escape(label)}"
+          options.join(' ')
+        end
+
+        # ignore :reek:FeatureEnvy:
+        def user_options
+          options = []
+          options << "--host=#{Shellwords.escape(host)}"
+          options << "--port=#{Shellwords.escape(port)}"
+          options << "--username=#{Shellwords.escape(user)}"
+          options.join(' ')
+        end
+
+        # ignore :reek:FeatureEnvy:
+        def target_options
+          options = []
+          options << "--file=#{Shellwords.escape(to_dir)}/#{Shellwords.escape(label)}"
+          options.join(' ')
+        end
+      end
+    end
+  end
+end

data/lib/dustcart/version.rb

@@ -1,3 +1,3 @@
 module Dustcart
-  VERSION = '0.1.1'.freeze
+  VERSION = '0.2.0'.freeze
 end

data/lib/helper/system_command.rb

@@ -0,0 +1,43 @@
+require 'open3'
+require 'shellwords'
+
+# class to call system command
+class SystemCommand
+  class << self
+    def exists?(command_name)
+      exec_command("which #{Shellwords.escape(command_name)}")
+      true
+    rescue
+      false
+    end
+
+    def exec_command(command)
+      output, error, status = Open3.capture3(command)
+
+      raise <<-EOS.unindent unless status.exitstatus.zero?
+        can not execute command: '#{command}'
+        Output: #{output}
+        Error: #{error}
+      EOS
+    end
+  end
+end
+
+# pipeline for command list
+class CommandPipeline
+  attr_reader :pipeline
+
+  def initialize
+    @pipeline = []
+  end
+
+  def add(command)
+    @pipeline << command
+  end
+
+  def run
+    @pipeline.each do |command|
+      SystemCommand.exec_command(command)
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dustcart
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Yuya TAMANO
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-08-24 00:00:00.000000000 Z
+date: 2016-09-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -179,10 +179,13 @@ files:
 - lib/dustcart/input/base.rb
 - lib/dustcart/input/directory.rb
 - lib/dustcart/input/file.rb
+- lib/dustcart/input/file_base.rb
+- lib/dustcart/input/postgres.rb
 - lib/dustcart/output/amazon_s3.rb
 - lib/dustcart/output/base.rb
 - lib/dustcart/resource_base.rb
 - lib/dustcart/version.rb
+- lib/helper/system_command.rb
 - lib/helper/zip_file_generator.rb
 homepage: https://github.com/tamano/dustcart
 licenses: