X-Git-Url: http://nitlanguage.org diff --git a/lib/opts.nit b/lib/opts.nit index f383a32..fa5ce15 100644 --- a/lib/opts.nit +++ b/lib/opts.nit @@ -29,23 +29,28 @@ abstract class Option var errors: Array[String] = new Array[String] # Is this option mandatory? - var mandatory: Bool writable = false + var mandatory: Bool = false is writable + + # Is this option hidden from `usage`? + var hidden: Bool = false is writable # Has this option been read? - var read: Bool writable = false + var read: Bool = false is writable # Current value of this option - var value: VALUE writable + var value: VALUE is writable # Default value of this option - var default_value: VALUE writable + var default_value: VALUE is writable # Create a new option - init(help: String, default: VALUE, names: nullable Array[String]) - do + 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 @@ -60,10 +65,10 @@ abstract class Option # Add new aliases for this option 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 @@ -77,10 +82,11 @@ abstract class Option 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 @@ -91,20 +97,25 @@ abstract class Option end end +# Not really an option. Just add a line of text when displaying the usage class OptionText super Option - init(text: String) do super(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 super(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) do @@ -113,11 +124,13 @@ class OptionBool 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 super(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) do @@ -129,15 +142,17 @@ end # 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? - var parameter_mandatory: Bool writable = true + var parameter_mandatory: Bool = true is writable redef fun read_param(it) do super - if it.is_ok and it.item.chars.first != '-' then + if it.is_ok and (it.item.is_empty or it.item.chars.first != '-') then value = convert(it.item) it.next else @@ -148,22 +163,31 @@ abstract class OptionParameter end end +# An option with a String as parameter class OptionString super OptionParameter redef type VALUE: nullable String - init(help: String, names: String...) do super(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 with an enum as parameter +# In the code, declaring an option enum (-e) with an enum like `["zero", "one", "two"] +# In the command line, typing `myprog -e one` is giving 1 as value class OptionEnum super OptionParameter redef type VALUE: Int + + # Values in the enumeration. var values: Array[String] - init(values: Array[String], help: String, default: Int, names: String...) - do + # 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 self.values = values.to_a super("{help} <{values.join(", ")}>", default, names) @@ -180,42 +204,49 @@ class OptionEnum return id end + # 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 super(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(help: String, default: Float, names: String...) do super(help, default, names) - + # 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 + # 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 @@ -228,13 +259,25 @@ class OptionArray end end +# Context where the options process class OptionContext - var options: Array[Option] - var rest: Array[String] - var errors: Array[String] + # Options present in the context + var options = new Array[Option] + + # Rest of the options after `parse` is called + var rest = new Array[String] + + # Errors found in the context after parsing + var errors = new Array[String] - private var optmap: Map[String, Option] - + 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 @@ -245,19 +288,30 @@ class OptionContext end if lmax < l then lmax = l end - + for i in options do - print(i.pretty(lmax)) + 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 @@ -274,7 +328,7 @@ class OptionContext # We're looking for packed short options if str.chars.last_index_of('-') == 0 and str.length > 2 then var next_called = false - for i in [1..str.length] do + 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] @@ -294,6 +348,10 @@ class OptionContext else rest.add(it.item) it.next + if options_before_rest then + rest.add_all(it.to_a) + parseargs = false + end end end end @@ -306,21 +364,6 @@ class OptionContext 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 @@ -330,18 +373,16 @@ class OptionContext end end + # Options parsing errors. fun get_errors: Array[String] do - var errors: Array[String] = new Array[String] - + var errors = new Array[String] errors.add_all(errors) - for o in options do for e in o.errors do errors.add(e) end end - return errors end end