1 # This file is part of NIT ( http://www.nitlanguage.org ).
3 # Copyright 2008 Floréal Morandat <morandat@lirmm.fr>
4 # Copyright 2008 Jean Privat <jean@pryen.org>
6 # This file is free software, which comes along with NIT. This software is
7 # distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
8 # without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
9 # PARTICULAR PURPOSE. You can modify it is you want, provided this header
10 # is kept unaltered, and a notification of the changes is added.
11 # You are allowed to redistribute it and sell it, alone or is a part of
14 # Super class of all option's class
16 # Names for the option (including long and short ones)
17 readable attr _names
: Array[String]
19 # Type of the value of the option
22 # Human readable description of the option
23 readable attr _helptext
: String
25 # Is this option mandatory?
26 readable writable attr _mandatory
: Bool
28 # context where the option is located
29 readable writable attr _context
: OptionContext
31 # Current value of this option
32 readable writable attr _value
: VALUE
34 # Default value of this option
35 readable writable attr _default_value
: VALUE
38 init init_opt
(help
: String, default
: VALUE, names
: Array[String])
41 _names
= new Array[String]
47 _default_value
= default
50 # Add new aliases for this option
51 meth add_aliases
(names
: String...) do _names
.add_all
(names
)
53 # An help text for this option with default settings
54 redef meth to_s
do return pretty
(2)
56 # A pretty print for this help
57 meth pretty
(off
: Int): String
60 text
.append
(_names
.join
(", "))
62 var rest
= off
- text
.length
63 if rest
> 0 then text
.append
(" " * rest
)
65 #text.append(pretty_default)
69 meth pretty_default
: String
71 if default_value
!= null then
72 return " ({default_value})"
77 # Consume parameters for this option
78 protected meth read_param
(it
: Iterator[String]) is abstract
83 init(text
: String) do init_opt
(text
, null, null)
85 redef meth pretty
(off
) do return to_s
87 redef meth to_s
do return helptext
92 redef type VALUE: Bool
94 init(help
: String, names
: String...) do init_opt
(help
, false, names
)
96 redef meth read_param
(it
) do value
= true
99 # Option with one mandatory parameter
100 class OptionParameter
102 protected meth convert
(str
: String): VALUE is abstract
104 redef meth read_param
(it
)
106 assert context
!= null
108 value
= convert
(it
.item
)
115 init init_opt
(h
, d
, n
) do super
119 special OptionParameter
120 redef type VALUE: String
122 init(help
: String, names
: String...) do init_opt
(help
, null, names
)
124 redef meth convert
(str
) do return str
128 special OptionParameter
129 redef type VALUE: Int
130 attr _enum
: Array[String]
132 init(enum: Array[String], help
: String, default
: Int, names
: String...)
134 assert enum != null and enum.length
> 0
136 init_opt
("{help} <{enum.join(", ")}>", default
, names
)
139 redef meth convert
(str
)
141 var id
= _enum
.index_of
(str
)
145 redef meth pretty_default
147 if default_value
!= null then
148 return " ({_enum[default_value]})"
156 special OptionParameter
157 redef type VALUE: Int
159 init(help
: String, default
: Int, names
: String...) do init_opt
(help
, default
, names
)
161 redef meth convert
(str
) do return str
.to_i
165 special OptionParameter
166 redef type VALUE: Array[String]
168 init(help
: String, names
: String...)
170 _values
= new Array[String]
171 init_opt
(help
, _values
, names
)
174 attr _values
: Array[String]
175 redef meth convert
(str
)
183 readable attr _options
: Array[Option]
184 readable attr _rest
: Array[String]
186 attr _optmap
: Map[String, Option]
196 if lmax
< l
then lmax
= l
200 print
(i
.pretty
(lmax
))
204 # Parse ans assign options everywhere is the argument list
205 meth parse
(argv
: Collection[String])
207 var it
= argv
.iterator
211 protected meth parse_intern
(it
: Iterator[String])
217 while parseargs
and it
.is_ok
do
221 rest
.add_all
(it
.to_a
)
224 if _optmap
.has_key
(str
) then
225 var opt
= _optmap
[str
]
236 meth add_option
(opts
: Option...)
246 _options
= new Array[Option]
247 _optmap
= new HashMap[String, Option]
248 _rest
= new Array[String]