xoptparse 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d4581fc5af988d4643c8cadc47f9d685c11bfc93c9c7e467136eb521d4926338
-  data.tar.gz: 7a126c2f9f468d61fc7bc72cdc62d6694f78b8505c0d937a54c9731340c6b5cb
+  metadata.gz: ed8ba0e138b8c1d6e68ca65d8e022ef51f3456647121207904074ffa8d53ac9f
+  data.tar.gz: e3ffc6e2caa748fadeb8b9a6f3129491d2997c8edf89780689bc6eac02c07489
 SHA512:
-  metadata.gz: b3e8e688e2e5499d38dbecd687526d9b8fd9b0f47ce2b2dd2244518f8bd0917a8dfb22a4f49e02b42de7e8bc613d9b8add4dbd77738491b48ae792fb82bfa655
-  data.tar.gz: 12a596b13d62628cca1b31fcb92eef894673a0bb2487dd2624075ae498fe597f3d62e8094cce5149b91a02a3b12036aabf107c5d69eb408f2e3e5d1344a7892e
+  metadata.gz: f160dfefb34a43a2b9f60ec775679e810613665edba3138857836467285eb5d8ce58c664687bd823322c619ea32a8c84ddab328b7e6a432c5f784dfffdff9727
+  data.tar.gz: '0950deea41a448199bc90aa1fb65365940fee26c3e3dd994c27d62e8e5677aaaf160e2234a91e24539ae36ecbc498f17aded45674e50e694c1ce4599c45ed1d5'

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    xoptparse (0.1.0)
+    xoptparse (0.2.0)
 
 GEM
   remote: https://rubygems.org/

data/lib/xoptparse.rb

@@ -4,108 +4,45 @@ require 'xoptparse/version'
 require 'optparse'
 
 class XOptionParser < ::OptionParser
-  class << self
-    def valid_arg_switch_ranges?(ranges)
-      ranges.inject(0) do |pt, r| # prev_type = req: 0, opt: 1, rest: 2, after_req: 3
-        t = r.end.nil? ? 2 : 1 - r.begin
-        next pt.zero? ? 0 : 3 if t.zero?
-        next t if pt < 2 && pt <= t
-
-        return false
-      end
-      true
-    end
-
-    def calc_arg_switch_ranges_counts(ranges)
-      req_count = 0
-      opt_count = 0
-      rest_req_count = nil
-      last_req_count = 0
-      ranges.each do |r|
-        if r.end.nil?
-          rest_req_count = r.begin
-        elsif r.begin.zero?
-          opt_count += 1
-        elsif opt_count.positive? || rest_req_count
-          last_req_count += 1
-        else
-          req_count += 1
-        end
-      end
-      [req_count, opt_count, rest_req_count, last_req_count]
-    end
-  end
-
-  attr_reader :description
-
-  def initialize(description = nil, *args)
+  def initialize(description = nil, *args, &block)
     @commands = {}
-    @arg_stack = [[], []]
-    @description = description
     @banner_usage = 'Usage: '
     @banner_options = '[options]'
     @banner_command = '<command>'
-    super(nil, *args)
+    super(nil, *args) do |opt|
+      if description
+        opt.separator ''
+        opt.separator description
+      end
+      block&.call(opt)
+    end
+  end
+
+  def select(*args, &block)
+    Enumerator.new do |y|
+      visit(:each_option) { |el| y << el }
+    end.select(*args, &block)
   end
+  private :select
 
   def no_options
-    visit(:summarize, {}, {}) { return false }
-    true
+    select { |sw| sw.is_a?(::OptionParser::Switch) }.all? { |sw| !(sw.short || sw.long) }
   end
   private :no_options
 
-  def banner # rubocop:disable Metrics/AbcSize
+  def banner
     return @banner if @banner
 
     banner = +"#{@banner_usage}#{program_name}"
     banner << " #{@banner_options}" unless no_options
     visit(:add_banner, banner)
     banner << " #{@banner_command}" unless @commands.empty?
-    @arg_stack.flatten(1).each do |sw|
-      banner << " #{sw.short.first}"
-    end
-    banner << "\n\n#{description}" if description
 
     banner
   end
 
-  def summarize(to = [], width = @summary_width, max = width - 1, indent = @summary_indent, &blk) # rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity
-    nl = "\n"
-    blk ||= proc { |l| to << (l.index(nl, -1) ? l : l + nl) }
-
-    no_opt = @arg_stack.flatten(1).empty? && no_options
-    blk.call("\nOptions:") if to.is_a?(String) && !no_opt
-
-    res = super(to, width, max, indent, &blk)
-    @arg_stack.flatten(1).each do |sw|
-      sw.summarize({}, {}, width, max, indent, &blk)
-    end
-
-    unless @commands.empty?
-      blk.call("\nCommands:") if to.is_a?(String)
-      @commands.each do |name, command|
-        sw = Switch::NoArgument.new(nil, nil, [name], nil, nil, command[1] ? [command[1]] : [], nil)
-        sw.summarize({}, {}, width, max, indent, &blk)
-      end
-    end
-
-    res
-  end
-
-  def define_opt_switch_values(target, swvs)
-    case target
-    when :tail
-      base.append(*swvs)
-    when :head
-      top.prepend(*swvs)
-    else
-      top.append(*swvs)
-    end
-  end
-  private :define_opt_switch_values
-
   def search_arg_switch_atype(sw0)
-    @stack.reverse_each do |el|
+    visit(:tap) do |el|
       el.atype.each do |klass, atype|
         next unless atype[1] == sw0.conv
         return [nil, nil] if klass == Object
@@ -118,150 +55,130 @@ class XOptionParser < ::OptionParser
   end
   private :search_arg_switch_atype
 
-  def fix_arg_switch(sw0) # rubocop:disable Metrics/AbcSize
+  def fix_arg_switch(sw0)
     pattern, conv = search_arg_switch_atype(sw0)
-    sw0.instance_variable_set(:@pattern, pattern)
-    sw0.instance_variable_set(:@conv, conv)
-    sw0.instance_variable_set(:@short, [sw0.desc.shift])
-
-    # arg pattern example:
-    # * req req => 1..1, 1..1
-    # * req [opt] => 1..1, 0..1
-    # * req... => 1..nil
-    # * [opt...] => 0..nil
-    # * req [opt] req... => 1..1, 0..1, 1..nil
-    # * req [opt...] => 1..1, 0..nil
-    # * req [opt...] req => 1..1, 0..nil, 1..1
-    ranges = sw0.short.first.scan(/(?:\[\s*(.*?)\s*\]|(\S+))/).map do |opt, req|
-      (opt ? 0 : 1)..((opt || req).end_with?('...') ? nil : 1)
-    end
-    unless self.class.valid_arg_switch_ranges?(ranges)
-      raise ArgumentError, "unsupported argument format: #{sw0.short.first.inspect}"
-    end
-
-    sw0.instance_variable_set(:@ranges, ranges)
-    sw0.define_singleton_method(:ranges) { @ranges }
-    sw0
+    Switch::SimpleArgument.new(pattern, conv, nil, nil, sw0.desc[0], sw0.desc[1..], sw0.block)
   end
   private :fix_arg_switch
 
-  def valid_arg_switch(sw0)
-    ranges = @arg_stack.flatten(1).map(&:ranges).flatten(1)
-    unless self.class.valid_arg_switch_ranges?(ranges)
-      raise ArgumentError, "unsupported argument format: #{sw0.short.first.inspect}"
-    end
-
-    sw0
-  end
-  private :valid_arg_switch
-
-  def define_arg_switch(target, sw0)
-    case target
-    when :tail
-      @arg_stack[1].append(sw0)
-    when :head
-      @arg_stack[0].prepend(sw0)
-    else
-      @arg_stack[0].append(sw0)
-    end
-
-    valid_arg_switch(sw0)
-  end
-  private :define_arg_switch
-
-  def define_at(target, *opts, &block)
-    sw = make_switch(opts, block || proc {})
+  def make_switch(opts, block = nil)
+    sw = super(opts, block || proc {})
     sw0 = sw[0]
-    if sw0.short || sw0.long
-      define_opt_switch_values(target, sw)
-    else
-      sw0 = fix_arg_switch(sw0)
-      define_arg_switch(target, sw0)
-    end
-    sw0
-  end
-  private :define_at
-
-  def define(*args, &block)
-    define_at(:body, *args, &block)
-  end
-  alias def_option define
-
-  def define_head(*args, &block)
-    define_at(:head, *args, &block)
-  end
-  alias def_head_option define_head
+    return sw if sw0.short || sw0.long
 
-  def define_tail(*args, &block)
-    define_at(:tail, *args, &block)
+    sw0 = fix_arg_switch(sw0)
+    long = sw0.arg.scan(/(?:\[\s*(.*?)\s*\]|(\S+))/).flatten.compact
+    [sw0, nil, long]
   end
-  alias def_tail_option define_tail
-
-  def parse_arguments(original_argv) # rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity
-    arg_sws = @arg_stack.flatten(1)
-    return original_argv if arg_sws.empty?
-
-    req_count, opt_count, rest_req_count, last_req_count =
-      self.class.calc_arg_switch_ranges_counts(arg_sws.map(&:ranges).flatten(1))
 
-    argv_min = req_count + last_req_count + (rest_req_count || 0)
-    raise MissingArgument, original_argv.join(' ').to_s if original_argv.size < argv_min
+  def parse_arguments(argv) # rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity
+    arg_sws = select { |sw| sw.is_a?(Switch::SimpleArgument) }
+    return argv if arg_sws.empty?
 
-    argv_max = rest_req_count ? nil : argv_min + opt_count
-    argv = original_argv.slice!(0...argv_max)
+    sws_ranges = arg_sws.map(&:ranges).flatten(1)
+    req_count = sws_ranges.sum(&:begin)
+    raise MissingArgument, argv.join(' ') if argv.size < req_count
 
-    opt_size = [argv.size - argv_min, opt_count].min
-    req_argv = argv[0...req_count] + argv[(argv.size - last_req_count)...argv.size]
-    opt_argv = argv[req_count...(req_count + opt_size)] # + ([nil] * (opt_count - opt_size))
-    rest_argv = argv[(req_count + opt_size)...(argv.size - last_req_count)]
+    opt_count = sws_ranges.sum(&:size) - sws_ranges.size
+    opt_index = argv[req_count...].index { |arg| @commands.include?(arg) } unless @commands.empty?
+    opt_count = [opt_count, opt_index || Float::INFINITY, argv.size - req_count].min
 
     arg_sws.each do |sw|
       conv = proc { |v| sw.send(:conv_arg, *sw.send(:parse_arg, v))[2] }
       a = sw.ranges.map do |r|
         if r.end.nil?
-          rest_argv.map(&conv)
+          rest_size = r.begin + opt_count
+          req_count -= r.begin
+          opt_count = 0
+          argv.slice!(0...rest_size).map(&conv)
         elsif r.begin.zero?
-          conv.call(opt_argv.shift)
+          next conv.call(nil) if opt_count.zero?
+
+          opt_count -= 1
+          conv.call(argv.shift)
         else
-          conv.call(req_argv.shift)
+          req_count -= 1
+          conv.call(argv.shift)
         end
       end
       sw.block.call(*a)
     end
 
-    original_argv
+    argv
   end
   private :parse_arguments
 
-  def order!(*args, **kwargs) # rubocop:disable Metrics/AbcSize
-    if @commands.empty?
-      argv = super(*args, **kwargs)
-      return block_given? ? argv : parse_arguments(argv)
-    end
+  def parse_in_order(*args, &nonopt)
+    argv = []
+    rest = if nonopt
+             super(*args, &argv.method(:<<))
+           else
+             argv = super(*args)
+           end
+    parse_arguments(argv).map(&nonopt)
+    rest
+  end
+  private :parse_in_order
+
+  def order!(*args, **kwargs)
+    return super(*args, **kwargs) if @commands.empty?
 
-    argv = super(*args, **kwargs) { |a| throw :terminate, a }
+    argv = super(*args, **kwargs, &nil)
     return argv if argv.empty?
 
     name = argv.shift
-    cmd = @commands[name]
-    return cmd.first.call.send(block_given? ? :permute! : :order!, *args, **kwargs) if cmd
+    sw = @commands[name]
+    return sw.block.call.send(block_given? ? :permute! : :order!, *args, **kwargs) if sw
 
     puts "#{program_name}:" \
          "'#{name}' is not a #{program_name} command. See '#{program_name} --help'."
     exit
   end
 
-  def permute!(*args, **kwargs)
-    parse_arguments(super(*args, **kwargs))
-  end
-
   def command(name, desc = nil, *args, &block)
-    @commands[name.to_s] = [proc do
+    sw0 = Switch::SummarizeArgument.new(nil, nil, nil, nil, name.to_s, desc ? [desc] : [], nil) do
       self.class.new(desc, *args) do |opt|
         opt.program_name = "#{program_name} #{name}"
-        block.call(opt) if block
+        block&.call(opt)
       end
-    end, desc]
+    end
+    top.append(sw0, nil, [sw0.arg])
+    @commands[name.to_s] = sw0
     nil
   end
+
+  class Switch < ::OptionParser::Switch
+    class SummarizeArgument < NoArgument
+      undef_method :add_banner
+
+      def summarize(*args)
+        original_arg = arg
+        @short = arg.scan(/\[\s*.*?\s*\]|\S+/)
+        @arg = nil
+        res = super(*args)
+        @arg = original_arg
+        @short = nil
+        res
+      end
+
+      def match_nonswitch?(*args)
+        super(*args) if @pattern.is_a?(Regexp)
+      end
+    end
+
+    class SimpleArgument < SummarizeArgument
+      attr_reader :ranges
+
+      def initialize(*args)
+        super(*args)
+        @ranges = arg.scan(/\[\s*(.*?)\s*\]|(\S+)/).map do |opt, req|
+          (opt ? 0 : 1)..((opt || req).end_with?('...') ? nil : 1)
+        end
+      end
+
+      def add_banner(to)
+        to << " #{arg}"
+      end
+    end
+  end
 end

data/lib/xoptparse/version.rb

@@ -3,5 +3,5 @@
 require 'optparse'
 
 class XOptionParser < ::OptionParser
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: xoptparse
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - ofk
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-07-11 00:00:00.000000000 Z
+date: 2020-07-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler