Module: CLI::Kit::Args::Tokenizer

Extended by:
T::Sig
Defined in:
lib/cli/kit/args/tokenizer.rb

Defined Under Namespace

Classes: InvalidCharInShortOption, InvalidShortOption, Token

Constant Summary collapse

Error =
Class.new(Args::Error)

Class Method Summary collapse

Methods included from T::Sig

sig

Class Method Details

.tokenize(raw_args) ⇒ Object



64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
# File 'lib/cli/kit/args/tokenizer.rb', line 64

def tokenize(raw_args)
  args = []

  mode = :init

  raw_args.each do |arg|
    case mode
    when :unparsed
      args << Token::UnparsedArgument.new(arg)
    when :init
      case arg
      when '--'
        mode = :unparsed
      when /\A--./
        name, value = arg.split('=', 2)
        args << Token::LongOptionName.new(T.must(T.must(name)[2..-1]))
        if value
          args << Token::OptionValue.new(value)
        end
      when /\A-./
        args.concat(tokenize_short_option(T.must(arg[1..-1])))
      else
        args << if args.last.is_a?(Token::OptionName)
          Token::OptionValueOrPositionalArgument.new(arg)
        else
          Token::PositionalArgument.new(arg)
        end
      end
    end
  end

  args
end

.tokenize_short_option(arg) ⇒ Object



99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
# File 'lib/cli/kit/args/tokenizer.rb', line 99

def tokenize_short_option(arg)
  args = []
  mode = :init
  number = +''
  arg.each_char do |char|
    case mode
    when :numeric
      case char
      when /[0-9]/
        number << char
      else
        raise(InvalidShortOption, arg)
      end
    when :init
      case char
      when /[a-zA-Z]/
        args << Token::ShortOptionName.new(char)
      when /[0-9]/
        mode = :numeric
        number << char
      else
        raise(InvalidCharInShortOption.new(arg, char))
      end
    end
  end
  if number != ''
    args << Token::OptionValue.new(number)
  end
  args
end