You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
1.9 KiB
95 lines
1.9 KiB
package input
|
|
|
|
func NewDefinition() *Definition {
|
|
return &Definition{
|
|
options: make(map[string]Option),
|
|
args: make(map[string]Argument),
|
|
short: make(map[string]string),
|
|
}
|
|
}
|
|
|
|
type Definition struct {
|
|
options map[string]Option
|
|
posOpt []string
|
|
args map[string]Argument
|
|
posArgs []string
|
|
short map[string]string
|
|
}
|
|
|
|
func (d *Definition) Options() []string {
|
|
return d.posOpt
|
|
}
|
|
|
|
func (d *Definition) Arguments() []string {
|
|
return d.posArgs
|
|
}
|
|
|
|
func (d *Definition) SetOption(name, description string, opts ...func(*Option)) *Definition {
|
|
return d.SetOptions(NewOption(name, description, opts...))
|
|
}
|
|
|
|
func (d *Definition) SetOptions(opts ...Option) *Definition {
|
|
for _, opt := range opts {
|
|
if _, has := d.options[opt.Name]; !has {
|
|
d.posOpt = append([]string{opt.Name}, d.posOpt...)
|
|
}
|
|
|
|
d.options[opt.Name] = opt
|
|
if opt.HasShort() {
|
|
d.short[opt.Short] = opt.Name
|
|
}
|
|
}
|
|
|
|
return d
|
|
}
|
|
|
|
func (d *Definition) SetArgument(name, description string, opts ...func(*Argument)) *Definition {
|
|
return d.SetArguments(NewArgument(name, description, opts...))
|
|
}
|
|
|
|
func (d *Definition) SetArguments(args ...Argument) *Definition {
|
|
for _, arg := range args {
|
|
if _, ok := d.args[arg.Name]; !ok {
|
|
d.posArgs = append(d.posArgs, arg.Name)
|
|
}
|
|
|
|
d.args[arg.Name] = arg
|
|
}
|
|
|
|
return d
|
|
}
|
|
|
|
func (d *Definition) Argument(pos int) (Argument, error) {
|
|
if len(d.posArgs) == 0 {
|
|
return Argument{}, ErrNoArgs
|
|
}
|
|
|
|
lastPos := len(d.posArgs) - 1
|
|
if lastPos < pos {
|
|
arg := d.args[d.posArgs[lastPos]]
|
|
if arg.IsArray() {
|
|
return arg, nil
|
|
}
|
|
|
|
return Argument{}, ErrToManyArgs
|
|
}
|
|
|
|
return d.args[d.posArgs[pos]], nil
|
|
}
|
|
|
|
func (d *Definition) ShortOption(short string) (Option, error) {
|
|
name, ok := d.short[short]
|
|
if !ok {
|
|
return Option{}, ErrNotFound
|
|
}
|
|
|
|
return d.Option(name)
|
|
}
|
|
|
|
func (d *Definition) Option(name string) (Option, error) {
|
|
if opt, ok := d.options[name]; ok {
|
|
return opt, nil
|
|
}
|
|
|
|
return Option{}, ErrNotFound
|
|
}
|
|
|