eac_tools 0.2.1 → 0.4.0

data/sub/eac_ruby_utils/lib/eac_ruby_utils/custom_format.rb

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+module EacRubyUtils
+  # A formatter like [String.sprintf].
+  class CustomFormat
+    TYPE_PATTERN = /[a-zA-Z]/.freeze
+    SEQUENCE_PATTERN = /(?<!%)%(#{TYPE_PATTERN})/.freeze
+
+    attr_reader :mapping
+
+    def initialize(mapping)
+      @mapping = mapping.transform_keys(&:to_sym).freeze
+    end
+
+    def format(string)
+      ::EacRubyUtils::CustomFormat::String.new(self, string)
+    end
+
+    class String
+      attr_reader :format, :string
+
+      def initialize(format, string)
+        @format = format
+        @string = string
+      end
+
+      def mapping
+        @mapping ||= format.mapping.select do |k, _v|
+          sequences.include?(k)
+        end
+      end
+
+      def sequences
+        @sequences ||= string.scan(SEQUENCE_PATTERN).map(&:first).uniq.map(&:to_sym)
+      end
+
+      def source_object_value(object, method)
+        return object.send(method) if object.respond_to?(method)
+        return object[method] if object.respond_to?('[]')
+
+        raise ::ArgumentError, "Methods \"#{method}\" or \"[]\" not found for #{object}"
+      end
+
+      def with(source_object)
+        r = string
+        mapping.each do |key, method|
+          r = r.gsub(/%#{::Regexp.quote(key)}/, source_object_value(source_object, method).to_s)
+        end
+        r.gsub('%%', '%')
+      end
+    end
+  end
+end

data/sub/eac_ruby_utils/lib/eac_ruby_utils/enum.rb

@@ -0,0 +1,102 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/inflector'
+
+module EacRubyUtils
+  # Similar to Java's enum type (https://docs.oracle.com/javase/tutorial/java/javaOO/enum.html).
+  class Enum
+    class << self
+      def enum(key, *args, &block)
+        value = Value.new(self, key, args, &block)
+        raise ::ArgumentError, "#{self} already has a value with key=#{value.key}" if
+          value_set.include?(value)
+
+        value_set << value.apply
+
+        self
+      end
+
+      def values
+        value_set.map(&:value)
+      end
+
+      private
+
+      def value_set
+        @value_set ||= []
+      end
+    end
+
+    attr_reader :key
+
+    protected
+
+    def initialize(key)
+      @key = key
+    end
+
+    class Value
+      include ::Comparable
+
+      attr_reader :klass, :key, :args, :block
+
+      def initialize(klass, key, args, &block)
+        @klass = klass
+        @key = ::EacRubyUtils::Inflector.variableize(key.to_s).to_sym
+        @args = args
+        @block = block
+      end
+
+      def <=>(other)
+        key <=> other.key
+      end
+
+      def apply
+        define_constant
+        define_class_method
+
+        self
+      end
+
+      # @return [String]
+      def class_method_name
+        key.to_s
+      end
+
+      # @return [String]
+      def constant_name
+        class_method_name.upcase
+      end
+
+      def value
+        @value ||= uncached_value
+      end
+
+      private
+
+      def define_class_method
+        the_value = self
+        klass.define_singleton_method(class_method_name) { the_value.value }
+      end
+
+      def define_constant
+        klass.const_set(constant_name, value)
+      end
+
+      def uncached_value
+        klass.new(key, *value_args)
+      end
+
+      def value_args
+        r = if block
+              block.call
+            elsif args.count == 1 && args.first.is_a?(::Enumerable)
+              args.first
+            else
+              args
+            end
+        r.is_a?(::Enumerable) ? r : [r]
+      end
+    end
+  end
+end

data/sub/eac_ruby_utils/lib/eac_ruby_utils/envs/base_env.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/envs/command'
+require 'eac_ruby_utils/envs/executable'
+require 'eac_ruby_utils/envs/file'
+
+module EacRubyUtils
+  module Envs
+    class BaseEnv
+      def command(*args)
+        ::EacRubyUtils::Envs::Command.new(self, args)
+      end
+
+      # <b>DEPRECATED:</b> Please use <tt>file(file).exist?</tt> instead.
+      def file_exist?(file)
+        self.file(file).exist?
+      end
+
+      def executable(*executable_new_args)
+        ::EacRubyUtils::Envs::Executable.new(self, *executable_new_args)
+      end
+
+      def file(path)
+        ::EacRubyUtils::Envs::File.new(self, path)
+      end
+    end
+  end
+end

data/sub/eac_ruby_utils/lib/eac_ruby_utils/envs/command/concat.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/struct'
+
+module EacRubyUtils
+  module Envs
+    class Command
+      module Concat
+        def concat(operator, other_command)
+          duplicate_by_extra_options(concat: ::EacRubyUtils::Struct.new(
+            operator: operator, command: other_command
+          ))
+        end
+
+        def or(other_command)
+          concat('||', other_command)
+        end
+
+        def pipe(other_command)
+          concat('|', other_command)
+        end
+
+        private
+
+        def append_concat(command)
+          extra_options[:concat].if_present(command) do |v|
+            "#{command} #{v.operator} #{v.command.command}"
+          end
+        end
+      end
+    end
+  end
+end

data/sub/eac_ruby_utils/lib/eac_ruby_utils/envs/command/envvars.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module EacRubyUtils
+  module Envs
+    class Command
+      module Envvars
+        def envvar(name, value)
+          duplicate_by_extra_options(envvars: envvars.merge(name => value))
+        end
+
+        private
+
+        def append_envvars(command)
+          e = envvars.map { |k, v| "#{Shellwords.escape(k)}=#{Shellwords.escape(v)}" }.join(' ')
+          e.present? ? "#{e} #{command}" : command
+        end
+
+        def envvars
+          extra_options[:envvars] ||= {}.with_indifferent_access
+        end
+      end
+    end
+  end
+end

data/sub/eac_ruby_utils/lib/eac_ruby_utils/envs/command/extra_options.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+require 'active_support/core_ext/hash/indifferent_access'
+require 'active_support/core_ext/object/blank'
+require 'shellwords'
+
+module EacRubyUtils
+  module Envs
+    class Command
+      module ExtraOptions
+        def chdir(dir)
+          duplicate_by_extra_options(chdir: dir)
+        end
+
+        def status_result(status_code, result)
+          duplicate_by_extra_options(status_results: status_results.merge(status_code => result))
+        end
+
+        private
+
+        attr_reader :extra_options
+
+        def status_results
+          extra_options[:status_results] ||= {}.with_indifferent_access
+        end
+
+        def append_chdir(command)
+          extra_options[:chdir].present? ? "(cd '#{extra_options[:chdir]}' ; #{command} )" : command
+        end
+      end
+    end
+  end
+end

data/sub/eac_ruby_utils/lib/eac_ruby_utils/envs/command.rb

@@ -0,0 +1,172 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/core_ext'
+require 'eac_ruby_utils/envs/process'
+require 'eac_ruby_utils/envs/spawn'
+require 'pp'
+require 'shellwords'
+
+module EacRubyUtils
+  module Envs
+    class Command
+      require_sub __FILE__, include_modules: true
+
+      def initialize(env, command, extra_options = {})
+        @env = env
+        @extra_options = extra_options.with_indifferent_access
+        if command.count == 1 && command.first.is_a?(Array)
+          @command = command.first
+        elsif command.is_a?(Array)
+          @command = command
+        else
+          raise "Invalid argument command: #{command}|#{command.class}"
+        end
+      end
+
+      def args
+        @command
+      end
+
+      def append(args)
+        duplicate_by_command(@command + args)
+      end
+
+      def prepend(args)
+        duplicate_by_command(args + @command)
+      end
+
+      def to_s
+        "#{@command} [ENV: #{@env}]"
+      end
+
+      def command(options = {})
+        c = @command
+        c = c.map { |x| escape(x) }.join(' ') if c.is_a?(Enumerable)
+        append_command_options(
+          @env.command_line(
+            append_chdir(append_concat(append_envvars(c)))
+          ),
+          options
+        )
+      end
+
+      def execute!(options = {})
+        options[:exit_outputs] = status_results.merge(options[:exit_outputs].presence || {})
+        er = ExecuteResult.new(execute(options), options)
+        return er.result if er.success?
+
+        raise "execute! command failed: #{self}\n#{er.r.pretty_inspect}"
+      end
+
+      def execute(options = {})
+        c = command(options)
+        debug_print("BEFORE: #{c}")
+        t1 = Time.now
+        r = ::EacRubyUtils::Envs::Process.new(c).to_h
+        i = Time.now - t1
+        debug_print("AFTER [#{i}]: #{c}")
+        r
+      end
+
+      def spawn(options = {})
+        c = command(options)
+        debug_print("SPAWN: #{c}")
+        ::EacRubyUtils::Envs::Spawn.new(c)
+      end
+
+      def system!(options = {})
+        return if system(options)
+
+        raise "system! command failed: #{self}"
+      end
+
+      def system(options = {})
+        c = command(options)
+        debug_print(c)
+        Kernel.system(c)
+      end
+
+      protected
+
+      def duplicate(command, extra_options)
+        self.class.new(@env, command, extra_options)
+      end
+
+      private
+
+      attr_reader :extra_options
+
+      def duplicate_by_command(new_command)
+        duplicate(new_command, @extra_options)
+      end
+
+      def duplicate_by_extra_options(set_extra_options)
+        duplicate(@command, @extra_options.merge(set_extra_options))
+      end
+
+      def debug?
+        ENV['DEBUG'].to_s.strip != ''
+      end
+
+      # Print a message if debugging is enabled.
+      def debug_print(message)
+        message = message.to_s
+        puts message.if_respond(:light_red, message) if debug?
+      end
+
+      def append_command_options(command, options)
+        command = options[:input].command + ' | ' + command if options[:input]
+        if options[:input_file]
+          command = "cat #{Shellwords.escape(options[:input_file])}" \
+            " | #{command}"
+        end
+        command += ' > ' + Shellwords.escape(options[:output_file]) if options[:output_file]
+        command
+      end
+
+      def escape(arg)
+        arg = arg.to_s
+        m = /^\@ESC_(.+)$/.match(arg)
+        m ? m[1] : Shellwords.escape(arg)
+      end
+
+      class ExecuteResult
+        attr_reader :r, :options
+
+        def initialize(result, options)
+          @r = result
+          @options = options
+        end
+
+        def result
+          return exit_code_zero_result if exit_code_zero?
+          return expected_error_result if expected_error?
+
+          raise 'Failed!'
+        end
+
+        def success?
+          exit_code_zero? || expected_error?
+        end
+
+        private
+
+        def exit_code_zero?
+          r[:exit_code]&.zero?
+        end
+
+        def exit_code_zero_result
+          r[options[:output] || :stdout]
+        end
+
+        def expected_error_result
+          options[:exit_outputs][r[:exit_code]]
+        end
+
+        def expected_error?
+          options[:exit_outputs].is_a?(Hash) && options[:exit_outputs].key?(r[:exit_code])
+        end
+      end
+    end
+  end
+end

data/sub/eac_ruby_utils/lib/eac_ruby_utils/envs/executable.rb

@@ -0,0 +1,81 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/listable'
+require 'eac_ruby_utils/simple_cache'
+
+module EacRubyUtils
+  module Envs
+    class Executable
+      include ::EacRubyUtils::Listable
+      include ::EacRubyUtils::SimpleCache
+
+      lists.add_symbol :option, :check_args, :auto_validate
+
+      DEFAULT_AUTO_VALIDATE = true
+
+      attr_reader :env, :name, :options
+
+      def initialize(env, name, *check_args)
+        @env = env
+        @name = name
+        self.options = self.class.lists.option.hash_keys_validate!(check_args.extract_options!)
+        options[OPTION_CHECK_ARGS] = check_args unless options.key?(OPTION_CHECK_ARGS)
+        options.freeze
+      end
+
+      def auto_validate?
+        options.key?(OPTION_AUTO_VALIDATE) ? options[OPTION_AUTO_VALIDATE] : DEFAULT_AUTO_VALIDATE
+      end
+
+      def check_args
+        options[OPTION_CHECK_ARGS]
+      end
+
+      def exist?
+        exist
+      end
+
+      def validate
+        return nil if exist?
+
+        "Program \"#{::Shellwords.join(executable_args)}\" not found in environment #{env}"
+      end
+
+      def validate!
+        message = validate
+
+        raise ProgramNotFoundError, message if message
+      end
+
+      def command(*command_args)
+        validate! if auto_validate?
+        env.command(*executable_args, *command_args)
+      end
+
+      def executable_args
+        executable_args_from_envvar || [name]
+      end
+
+      def executable_args_envvar
+        "#{name}_command".variableize.upcase
+      end
+
+      def executable_args_from_envvar
+        ENV[executable_args_envvar].if_present { |v| ::Shellwords.split(v) }
+      end
+
+      private
+
+      attr_writer :options
+
+      def exist_uncached
+        env.command(*executable_args, *check_args).execute!
+        true
+      rescue Errno::ENOENT
+        false
+      end
+
+      class ProgramNotFoundError < ::StandardError; end
+    end
+  end
+end

data/sub/eac_ruby_utils/lib/eac_ruby_utils/envs/file.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module EacRubyUtils
+  module Envs
+    class File
+      attr_reader :env, :path
+
+      def initialize(env, path)
+        @env = env
+        @path = path
+      end
+
+      def exist?
+        env.command('stat', path).execute[:exit_code].zero?
+      end
+
+      def read
+        env.command('cat', path).execute!
+      end
+    end
+  end
+end

data/sub/eac_ruby_utils/lib/eac_ruby_utils/envs/local_env.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/envs/base_env'
+
+module EacRubyUtils
+  module Envs
+    class LocalEnv < ::EacRubyUtils::Envs::BaseEnv
+      def to_s
+        'LOCAL'
+      end
+
+      def command_line(line)
+        line
+      end
+    end
+  end
+end

data/sub/eac_ruby_utils/lib/eac_ruby_utils/envs/process.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require 'open3'
+
+module EacRubyUtils
+  module Envs
+    class Process
+      EXIT_CODE_KEY = :exit_code
+      ERR_KEY = :stderr
+      OUT_KEY = :stdout
+
+      def initialize(command)
+        self.data = { command: command }
+        data[OUT_KEY], data[ERR_KEY], data[EXIT_CODE_KEY] = Open3.capture3(command)
+        data[EXIT_CODE_KEY] = data[EXIT_CODE_KEY].to_i
+      end
+
+      def to_h
+        data.dup
+      end
+
+      private
+
+      attr_accessor :data
+    end
+  end
+end

data/sub/eac_ruby_utils/lib/eac_ruby_utils/envs/spawn.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module EacRubyUtils
+  module Envs
+    class Spawn
+      attr_reader :command, :pid
+
+      def initialize(command)
+        @command = command
+        @pid = ::Process.spawn(command)
+      end
+
+      def kill(signal)
+        ::Process.kill(signal, pid)
+      end
+
+      def kill_at_end(&block)
+        block.call(self)
+      ensure
+        kill('KILL')
+      end
+
+      def to_h
+        { command: command, pid: pid }
+      end
+
+      def wait
+        ::Process.wait pid
+      end
+    end
+  end
+end

data/sub/eac_ruby_utils/lib/eac_ruby_utils/envs/ssh_env/dasho_options.rb

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/patches/module/common_concern'
+require 'eac_ruby_utils/patches/module/simple_cache'
+
+module EacRubyUtils
+  module Envs
+    class SshEnv < ::EacRubyUtils::Envs::BaseEnv
+      module DashoOptions
+        common_concern
+
+        module ClassMethods
+          def add_nodasho_option(name)
+            return if nodasho_options.include?(name)
+
+            nodasho_options << name
+            const_set("#{name.underscore}_option".upcase, name)
+          end
+
+          def nodasho_options
+            @nodasho_options ||= []
+          end
+        end
+
+        module InstanceMethods
+          def ssh_command_line_dasho_args
+            r = []
+            uri.query_values&.each do |k, v|
+              r += ['-o', "#{k}=#{v}"] unless nodasho_options.include?(k)
+            end
+            r
+          end
+
+          def ssh_command_line_nodasho_args
+            nodasho_options.flat_map do |option_name|
+              uri_query_value(option_name).if_present([]) do |option_value|
+                send("ssh_command_line_#{option_name.underscore}_args", option_value)
+              end
+            end
+          end
+
+          def nodasho_options
+            self.class.nodasho_options
+          end
+
+          def uri_query_value(name)
+            uri.query_values.if_present { |v| v[name] }
+          end
+        end
+      end
+    end
+  end
+end