#
# This file is free software, which comes along with NIT. This software is
# distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
-# without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
+# without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
# PARTICULAR PURPOSE. You can modify it is you want, provided this header
# is kept unaltered, and a notification of the changes is added.
# You are allowed to redistribute it and sell it, alone or is a part of
# Super class of all option's class
abstract class Option
# Names for the option (including long and short ones)
- readable var _names: Array[String]
+ var names: Array[String]
# Type of the value of the option
type VALUE: nullable Object
# Human readable description of the option
- readable var _helptext: String
+ var helptext: String
# Gathering errors during parsing
- readable var _errors: Array[String]
+ var errors: Array[String] = new Array[String]
# Is this option mandatory?
- readable writable var _mandatory: Bool
+ var mandatory: Bool = false is writable
- # Has this option been read?
- readable var _read:Bool
+ # Is this option hidden from `usage`?
+ var hidden: Bool = false is writable
- # Current value of this option
- writable var _value: nullable VALUE
+ # Has this option been read?
+ var read: Bool = false is writable
# Current value of this option
- fun value: VALUE do return _value.as(VALUE)
+ var value: VALUE is writable
# Default value of this option
- readable writable var _default_value: nullable VALUE
+ var default_value: VALUE is writable
# Create a new option
- init init_opt(help: String, default: nullable VALUE, names: nullable Array[String])
+ init(help: String, default: VALUE, names: nullable Array[String]) is old_style_init do
+ init_opt(help, default, names)
+ end
+
+ # Init option `helptext`, `default_value` and `names`.
+ #
+ # Also set current `value` to `default`.
+ fun init_opt(help: String, default: VALUE, names: nullable Array[String])
do
if names == null then
- _names = new Array[String]
+ self.names = new Array[String]
else
- _names = names.to_a
+ self.names = names.to_a
end
- _helptext = help
- _mandatory = false
- _read = false
- _default_value = default
- _value = default
- _errors = new Array[String]
+ helptext = help
+ default_value = default
+ value = default
end
# Add new aliases for this option
- fun add_aliases(names: String...) do _names.add_all(names)
-
+ fun add_aliases(names: String...) do names.add_all(names)
+
# An help text for this option with default settings
redef fun to_s do return pretty(2)
-
+
# A pretty print for this help
fun pretty(off: Int): String
do
- var text = new Buffer.from(" ")
- text.append(_names.join(", "))
+ var text = new FlatBuffer.from(" ")
+ text.append(names.join(", "))
text.append(" ")
var rest = off - text.length
if rest > 0 then text.append(" " * rest)
return text.to_s
end
+ # Pretty print the default value.
fun pretty_default: String
do
var dv = default_value
- if dv != null then return " ({dv})"
+ if dv != null then return " ({dv.to_s})"
return ""
end
# Consume parameters for this option
- protected fun read_param(it: Iterator[String])
+ protected fun read_param(opts: OptionContext, it: Iterator[String])
do
- _read = true
+ read = true
end
end
+# Not really an option. Just add a line of text when displaying the usage
class OptionText
super Option
- init(text: String) do init_opt(text, null, null)
+
+ # Init a new OptionText with `text`.
+ init(text: String) is old_style_init do super(text, null, null)
redef fun pretty(off) do return to_s
redef fun to_s do return helptext
end
+# A boolean option, `true` when present, `false` if not
class OptionBool
super Option
redef type VALUE: Bool
- init(help: String, names: String...) do init_opt(help, false, names)
+ # Init a new OptionBool with a `help` message and `names`.
+ init(help: String, names: String...) is old_style_init do super(help, false, names)
- redef fun read_param(it)
+ redef fun read_param(opts, it)
do
super
value = true
end
end
+# A count option. Count the number of time this option is present
class OptionCount
super Option
redef type VALUE: Int
- init(help: String, names: String...) do init_opt(help, 0, names)
+ # Init a new OptionCount with a `help` message and `names`.
+ init(help: String, names: String...) is old_style_init do super(help, 0, names)
- redef fun read_param(it)
+ redef fun read_param(opts, it)
do
super
value += 1
# Option with one parameter (mandatory by default)
abstract class OptionParameter
super Option
+
+ # Convert `str` to a value of type `VALUE`.
protected fun convert(str: String): VALUE is abstract
# Is the parameter mandatory?
- readable writable var _parameter_mandatory: Bool
+ var parameter_mandatory = true is writable
- redef fun read_param(it)
+ redef fun read_param(opts, it)
do
super
- if it.is_ok and it.item.first != '-' then
+
+ var ok = it.is_ok
+ if ok and not parameter_mandatory and not it.item.is_empty and it.item.chars.first == '-' then
+ # The next item may looks like a known command
+ # Only check if `not parameter_mandatory`
+ for opt in opts.options do
+ if opt.names.has(it.item) then
+ # The next item is a known command
+ ok = false
+ break
+ end
+ end
+ end
+
+ if ok then
value = convert(it.item)
it.next
else
- if _parameter_mandatory then
- _errors.add("Parameter expected for option {names.first}.")
- end
+ errors.add("Parameter expected for option {names.first}.")
end
end
-
- init init_opt(h, d, n)
- do
- super
- _parameter_mandatory = true
- end
end
+# An option with a `String` as parameter
class OptionString
super OptionParameter
redef type VALUE: nullable String
- init(help: String, names: String...) do init_opt(help, null, names)
+ # Init a new OptionString with a `help` message and `names`.
+ init(help: String, names: String...) is old_style_init do super(help, null, names)
redef fun convert(str) do return str
end
+# An option to choose from an enumeration
+#
+# Declare an enumeration option with all its possible values as an array.
+# Once the arguments are processed, `value` is set as the index of the selected value, if any.
class OptionEnum
super OptionParameter
redef type VALUE: Int
- var _values: Array[String]
- init(values: Array[String], help: String, default: Int, names: String...)
- do
+ # Values in the enumeration.
+ var values: Array[String]
+
+ # Init a new OptionEnum from `values` with a `help` message and `names`.
+ #
+ # `default` is the index of the default value in `values`.
+ init(values: Array[String], help: String, default: Int, names: String...) is old_style_init do
assert values.length > 0
- _values = values.to_a
- init_opt("{help} <{values.join(", ")}>", default, names)
+ self.values = values.to_a
+ super("{help} <{values.join(", ")}>", default, names)
end
redef fun convert(str)
do
- var id = _values.index_of(str)
+ var id = values.index_of(str)
if id == -1 then
- var e = "Unrecognized value for option {_names.join(", ")}.\n"
- e += "Expected values are: {_values.join(", ")}."
- _errors.add(e)
+ var e = "Unrecognized value for option {names.join(", ")}.\n"
+ e += "Expected values are: {values.join(", ")}."
+ errors.add(e)
end
return id
end
- fun value_name: String = _values[value]
+ # Get the value name from `values`.
+ fun value_name: String do return values[value]
redef fun pretty_default
do
- if default_value != null then
- return " ({_values[default_value.as(not null)]})"
- else
- return ""
- end
- end
+ return " ({values[default_value]})"
+ end
end
+# An option with an Int as parameter
class OptionInt
super OptionParameter
redef type VALUE: Int
- init(help: String, default: Int, names: String...) do init_opt(help, default, names)
-
+ # Init a new OptionInt with a `help` message, a `default` value and `names`.
+ init(help: String, default: Int, names: String...) is old_style_init do
+ super(help, default, names)
+ end
+
redef fun convert(str) do return str.to_i
end
+# An option with a Float as parameter
+class OptionFloat
+ super OptionParameter
+ redef type VALUE: Float
+
+ # Init a new OptionFloat with a `help` message, a `default` value and `names`.
+ init(help: String, default: Float, names: String...) is old_style_init do
+ super(help, default, names)
+ end
+
+ redef fun convert(str) do return str.to_f
+end
+
+# An option with an array as parameter
+# `myprog -optA arg1 -optA arg2` is giving an Array `["arg1", "arg2"]`
class OptionArray
super OptionParameter
redef type VALUE: Array[String]
- init(help: String, names: String...)
- do
- _values = new Array[String]
- init_opt(help, _values, names)
+ # Init a new OptionArray with a `help` message and `names`.
+ init(help: String, names: String...) is old_style_init do
+ values = new Array[String]
+ super(help, values, names)
end
- var _values: Array[String]
+ private var values: Array[String]
redef fun convert(str)
do
- _values.add(str)
- return _values
+ values.add(str)
+ return values
end
end
+# Context where the options process
class OptionContext
- readable var _options: Array[Option]
- readable var _rest: Array[String]
- readable var _errors: Array[String]
+ # Options present in the context
+ var options = new Array[Option]
- var _optmap: Map[String, Option]
-
+ # Rest of the options after `parse` is called
+ var rest = new Array[String]
+
+ # Errors found in the context after parsing
+ var context_errors = new Array[String]
+
+ private var optmap = new HashMap[String, Option]
+
+ # Add one or more options to the context
+ fun add_option(opts: Option...) do
+ options.add_all(opts)
+ end
+
+ # Display all the options available
fun usage
do
var lmax = 1
- for i in _options do
+ for i in options do
var l = 3
for n in i.names do
l += n.length + 2
end
if lmax < l then lmax = l
end
-
- for i in _options do
- print(i.pretty(lmax))
+
+ for i in options do
+ if not i.hidden then
+ print(i.pretty(lmax))
+ end
end
end
- # Parse ans assign options everywhere is the argument list
+ # Parse and assign options everywhere in the argument list
fun parse(argv: Collection[String])
do
var it = argv.iterator
parse_intern(it)
end
+ # Must all option be given before the first argument?
+ #
+ # When set to `false` (the default), options of the command line are
+ # all parsed until the end of the list of arguments or until "--" is met (in this case "--" is discarded).
+ #
+ # When set to `true` options are parsed until the first non-option is met.
+ var options_before_rest = false is writable
+
+ # Parse the command line
protected fun parse_intern(it: Iterator[String])
do
var parseargs = true
build
- var rest = _rest
+ var rest = rest
while parseargs and it.is_ok do
var str = it.item
if str == "--" then
+ it.next
rest.add_all(it.to_a)
parseargs = false
else
# We're looking for packed short options
- if str.last_index_of('-') == 0 and str.length > 2 then
+ if str.chars.last_index_of('-') == 0 and str.length > 2 then
var next_called = false
- for i in [1..str.length] do
- var short_opt = "-" + str[i].to_s
- if _optmap.has_key(short_opt) then
- var option = _optmap[short_opt]
+ for i in [1..str.length[ do
+ var short_opt = "-" + str.chars[i].to_s
+ if optmap.has_key(short_opt) then
+ var option = optmap[short_opt]
if option isa OptionParameter then
it.next
next_called = true
end
- option.read_param(it)
+ option.read_param(self, it)
end
end
if not next_called then it.next
else
- if _optmap.has_key(str) then
- var opt = _optmap[str]
+ if optmap.has_key(str) then
+ var opt = optmap[str]
it.next
- opt.read_param(it)
+ opt.read_param(self, it)
else
rest.add(it.item)
it.next
+ if options_before_rest then
+ rest.add_all(it.to_a)
+ parseargs = false
+ end
end
end
end
end
- for opt in _options do
+ for opt in options do
if opt.mandatory and not opt.read then
- _errors.add("Mandatory option {opt.names.join(", ")} not found.")
+ context_errors.add("Mandatory option {opt.names.join(", ")} not found.")
end
end
end
- fun add_option(opts: Option...)
- do
- for opt in opts do
- _options.add(opt)
- end
- end
-
- init
- do
- _options = new Array[Option]
- _optmap = new HashMap[String, Option]
- _rest = new Array[String]
- _errors = new Array[String]
- end
-
private fun build
do
- for o in _options do
+ for o in options do
for n in o.names do
- _optmap[n] = o
+ optmap[n] = o
end
end
end
- fun get_errors: Array[String]
+ # Options parsing errors.
+ fun errors: Array[String]
do
- var errors: Array[String] = new Array[String]
-
- errors.add_all(_errors)
-
- for o in _options do
+ var errors = new Array[String]
+ errors.add_all context_errors
+ for o in options do
for e in o.errors do
errors.add(e)
end
end
-
return errors
end
end