Error handling. Vendoring
Signed-off-by: Ettore Dreucci <ettore.dreucci@gmail.com>
This commit is contained in:
21
vendor/github.com/dixonwille/wmenu/LICENSE
generated
vendored
Normal file
21
vendor/github.com/dixonwille/wmenu/LICENSE
generated
vendored
Normal file
@@ -0,0 +1,21 @@
|
||||
The MIT License (MIT)
|
||||
|
||||
Copyright (c) 2016 Will Dixon
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
167
vendor/github.com/dixonwille/wmenu/README.md
generated
vendored
Normal file
167
vendor/github.com/dixonwille/wmenu/README.md
generated
vendored
Normal file
@@ -0,0 +1,167 @@
|
||||
# WMenu[](https://travis-ci.org/dixonwille/wmenu) [](https://codecov.io/gh/dixonwille/wmenu)
|
||||
|
||||
Package wmenu creates menus for cli programs. It uses wlog for it's interface
|
||||
with the command line. It uses os.Stdin, os.Stdout, and os.Stderr with
|
||||
concurrency by default. wmenu allows you to change the color of the different
|
||||
parts of the menu. This package also creates it's own error structure so you can
|
||||
type assert if you need to. wmenu will validate all responses before calling any function. It will also figure out which function should be called so you don't have to.
|
||||
|
||||
[](https://asciinema.org/a/4lv3ash3ubtnsclindvzdf320)
|
||||
|
||||
## Import
|
||||
|
||||
I try and keep up with my tags. To use the version and stable it is recommended to use `govendor` or another vendoring tool that allows you to build your project for specific tags.
|
||||
|
||||
```sh
|
||||
govendor fetch github.com/dixonwille/wmenu@v4
|
||||
```
|
||||
|
||||
The above will grab the latest v4 at that time and mark it. It will then be stable for you to use.
|
||||
|
||||
I will try to support as many versions as possable but please be patient.
|
||||
|
||||
### V1.0.0 - Major Release [](https://goreportcard.com/report/gopkg.in/dixonwille/wmenu.v1) [](https://godoc.org/gopkg.in/dixonwille/wmenu.v1)
|
||||
|
||||
### V2.0.0 - Allowing an interface to be passed in for options [](https://goreportcard.com/report/gopkg.in/dixonwille/wmenu.v2) [](https://godoc.org/gopkg.in/dixonwille/wmenu.v2)
|
||||
|
||||
### V3.0.0 - Pass in the option to that option's function [](https://goreportcard.com/report/gopkg.in/dixonwille/wmenu.v3) [](https://godoc.org/gopkg.in/dixonwille/wmenu.v3)
|
||||
|
||||
### V4.0.0 - Now have an Action that supports multiple options [](https://goreportcard.com/report/gopkg.in/dixonwille/wmenu.v4) [](https://godoc.org/gopkg.in/dixonwille/wmenu.v4)
|
||||
|
||||
|
||||
## Features
|
||||
|
||||
* Force single selection
|
||||
* Allow multiple selection
|
||||
* Change the delimiter
|
||||
* Change the color of different parts of the menu
|
||||
* Easily see which option(s) are default
|
||||
* Change the symbol used for default option(s)
|
||||
* Ask simple yes and no questions
|
||||
* Validate all responses before calling any functions
|
||||
* With yes and no can accept:
|
||||
* yes, Yes, YES, y, Y
|
||||
* no, No, NO, n, N
|
||||
* Figure out which Action should be called (Options, Default, or Multiple Action)
|
||||
* Re-ask question if invalid response up to a certain number of times
|
||||
* Can change max number of times to ask before failing output
|
||||
* Change reader and writer
|
||||
* Clear the screen whenever the menu is brought up
|
||||
* Has its own error structure so you can type assert menu errors
|
||||
|
||||
### V2 - Adds these Features
|
||||
|
||||
* Allowing any interface to be passed through for the options.
|
||||
|
||||
### V3 - Adds these Features
|
||||
|
||||
* Pass the option chosen to that options function
|
||||
|
||||
### V4 - Adds these Features
|
||||
|
||||
* Have one function for both single and multiple select. Allowing the user to an easier way of handeling the request.
|
||||
|
||||
## Usage
|
||||
This is a simple use of the package. (**NOTE: THIS IS A V4 SAMPLE**)
|
||||
|
||||
``` go
|
||||
menu := wmenu.NewMenu("What is your favorite food?")
|
||||
menu.Action(func (opts []wmenu.Opt) error {fmt.Printf(opts[0].Text + " is your favorite food."); return nil})
|
||||
menu.Option("Pizza", nil, true, nil)
|
||||
menu.Option("Ice Cream", nil, false, nil)
|
||||
menu.Option("Tacos", nil, false, func(opt wmenu.Opt) error {
|
||||
fmt.Printf("Tacos are great")
|
||||
})
|
||||
err := menu.Run()
|
||||
if err != nil{
|
||||
log.Fatal(err)
|
||||
}
|
||||
```
|
||||
|
||||
The output would look like this:
|
||||
|
||||
```
|
||||
0) *Pizza
|
||||
1) Ice Cream
|
||||
2) Tacos
|
||||
What is your favorite food?
|
||||
```
|
||||
|
||||
If the user just presses `[Enter]` then the option(s) with the `*` will be selected. This indicates that it is a default function. If they choose `1` then they would see `Ice Cream is your favorite food.`. This used the Action's function because the option selected didn't have a function along with it. But if they choose `2` they would see `Tacos are great`. That option did have a function with it which take precedence over Action.
|
||||
|
||||
You can you also use:
|
||||
|
||||
``` go
|
||||
menu.AllowMultiple()
|
||||
```
|
||||
|
||||
This will allow the user to select multiple options. The default delimiter is a `[space]`, but can be changed by using:
|
||||
|
||||
``` go
|
||||
menu.SetSeperator("some string")
|
||||
```
|
||||
|
||||
Another feature is the ability to ask yes or no questions.
|
||||
|
||||
``` go
|
||||
menu.IsYesNo(0)
|
||||
```
|
||||
|
||||
This will remove any options previously added options and hide the ones used for the menu. It will simply just ask yes or no. Menu will parse and validate the response for you. This option will always call the Action's function and pass in the option that was selected.
|
||||
|
||||
## V3+ - Release
|
||||
Allows the user to pass anything for the value so it can be retrieved later in the function. The following is to show case the power of this.
|
||||
|
||||
> The following was written in V3 but the concept holds for V4. V4 just changed `actFunc` to be `func([]wmenu.Opt) error` instead.
|
||||
|
||||
```go
|
||||
type NameEntity struct {
|
||||
FirstName string
|
||||
LastName string
|
||||
}
|
||||
|
||||
optFunc := func(opt wmenu.Opt) error {
|
||||
fmt.Println("Option 0 was chosen.")
|
||||
return nil
|
||||
}
|
||||
actFunc := func(opt wmenu.Opt) error {
|
||||
name, ok := opt.Value.(NameEntity)
|
||||
if !ok {
|
||||
log.Fatal("Could not cast option's value to NameEntity")
|
||||
}
|
||||
fmt.Printf("%s has an id of %d.\n", opt.Text, opt.ID)
|
||||
fmt.Printf("Hello, %s %s.\n", name.FirstName, name.LastName)
|
||||
return nil
|
||||
}
|
||||
menu := NewMenu("Choose an option.")
|
||||
menu.ChangeReaderWriter(reader, os.Stdout, os.Stderr)
|
||||
menu.Action(actFunc)
|
||||
menu.Option("Option 0", NameEntity{"Bill", "Bob"}, true, optFunc)
|
||||
menu.Option("Option 1", NameEntity{"John", "Doe"}, false, nil)
|
||||
menu.Option("Option 2", NameEntity{"Jane", "Doe"}, false, nil)
|
||||
err := menu.Run()
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
```
|
||||
|
||||
The immediate output would be:
|
||||
|
||||
```
|
||||
Output:
|
||||
0) *Option 0
|
||||
1) Option 1
|
||||
2) Option 2
|
||||
Choose an option.
|
||||
```
|
||||
|
||||
Now if the user pushes `[ENTER]` the output would be `Options 0 was chosen.`. But now if either option 1 or 2 were chosen it would cast the options value to a NameEntity allowing the function to be able to gather both the first name and last name of the NameEntity. If you want though you can just pass in `nil` as the value or even a string (`"hello"`) since both of these implement the empty interface required by value. Just make sure to cast the values so you can use them appropriately.
|
||||
|
||||
## Further Reading
|
||||
This whole package has been documented and has a few examples in:
|
||||
* [godocs V1](https://godoc.org/gopkg.in/dixonwille/wmenu.v1)
|
||||
* [godocs V2](https://godoc.org/gopkg.in/dixonwille/wmenu.v2)
|
||||
* [godocs V3](https://godoc.org/gopkg.in/dixonwille/wmenu.v3)
|
||||
* [godocs V4](https://godoc.org/gopkg.in/dixonwille/wmenu.v4)
|
||||
|
||||
You should read the docs to find all functions and structures at your finger tips.
|
||||
36
vendor/github.com/dixonwille/wmenu/clearScreen.go
generated
vendored
Normal file
36
vendor/github.com/dixonwille/wmenu/clearScreen.go
generated
vendored
Normal file
@@ -0,0 +1,36 @@
|
||||
package wmenu
|
||||
|
||||
import (
|
||||
"os"
|
||||
"os/exec"
|
||||
"runtime"
|
||||
)
|
||||
|
||||
var clear map[string]func()
|
||||
|
||||
func init() {
|
||||
clear = make(map[string]func())
|
||||
clear["linux"] = func() {
|
||||
cmd := exec.Command("clear")
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Run()
|
||||
}
|
||||
clear["darwin"] = func() {
|
||||
cmd := exec.Command("clear")
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Run()
|
||||
}
|
||||
clear["windows"] = func() {
|
||||
cmd := exec.Command("cmd", "/c", "cls")
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Run()
|
||||
}
|
||||
}
|
||||
|
||||
//Clear simply clears the command line interface (os.Stdout only).
|
||||
func Clear() {
|
||||
value, ok := clear[runtime.GOOS]
|
||||
if ok {
|
||||
value()
|
||||
}
|
||||
}
|
||||
86
vendor/github.com/dixonwille/wmenu/error.go
generated
vendored
Normal file
86
vendor/github.com/dixonwille/wmenu/error.go
generated
vendored
Normal file
@@ -0,0 +1,86 @@
|
||||
package wmenu
|
||||
|
||||
import "errors"
|
||||
|
||||
var (
|
||||
//ErrInvalid is returned if a response from user was an invalid option
|
||||
ErrInvalid = errors.New("Invalid response")
|
||||
|
||||
//ErrTooMany is returned if multiSelect is false and a user tries to select multiple options
|
||||
ErrTooMany = errors.New("Too many responses")
|
||||
|
||||
//ErrNoResponse is returned if there were no responses and no action to call
|
||||
ErrNoResponse = errors.New("No response")
|
||||
|
||||
//ErrDuplicate is returned is a user selects an option twice
|
||||
ErrDuplicate = errors.New("Duplicated response")
|
||||
)
|
||||
|
||||
//MenuError records menu errors
|
||||
type MenuError struct {
|
||||
Err error
|
||||
Res string
|
||||
TriesLeft int
|
||||
}
|
||||
|
||||
//Error prints the error in an easy to read string.
|
||||
func (e *MenuError) Error() string {
|
||||
if e.Res != "" {
|
||||
return e.Err.Error() + ": " + e.Res
|
||||
}
|
||||
return e.Err.Error()
|
||||
}
|
||||
|
||||
func newMenuError(err error, res string, tries int) *MenuError {
|
||||
return &MenuError{
|
||||
Err: err,
|
||||
Res: res,
|
||||
TriesLeft: tries,
|
||||
}
|
||||
}
|
||||
|
||||
//IsInvalidErr checks to see if err is of type invalid error returned by menu.
|
||||
func IsInvalidErr(err error) bool {
|
||||
e, ok := err.(*MenuError)
|
||||
if ok && e.Err == ErrInvalid {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
//IsNoResponseErr checks to see if err is of type no response returned by menu.
|
||||
func IsNoResponseErr(err error) bool {
|
||||
e, ok := err.(*MenuError)
|
||||
if ok && e.Err == ErrNoResponse {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
//IsTooManyErr checks to see if err is of type too many returned by menu.
|
||||
func IsTooManyErr(err error) bool {
|
||||
e, ok := err.(*MenuError)
|
||||
if ok && e.Err == ErrTooMany {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
//IsDuplicateErr checks to see if err is of type duplicate returned by menu.
|
||||
func IsDuplicateErr(err error) bool {
|
||||
e, ok := err.(*MenuError)
|
||||
if ok && e.Err == ErrDuplicate {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
//IsMenuErr checks to see if it is a menu err.
|
||||
//This is a general check not a specific one.
|
||||
func IsMenuErr(err error) bool {
|
||||
_, ok := err.(*MenuError)
|
||||
if ok {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
387
vendor/github.com/dixonwille/wmenu/menu.go
generated
vendored
Normal file
387
vendor/github.com/dixonwille/wmenu/menu.go
generated
vendored
Normal file
@@ -0,0 +1,387 @@
|
||||
//Package wmenu creates menus for cli programs.
|
||||
//It uses wlog for it's interface with the command line.
|
||||
//It uses os.Stdin, os.Stdout, and os.Stderr with concurrency by default.
|
||||
//wmenu allows you to change the color of the different parts of the menu.
|
||||
//This package also creates it's own error structure so you can type assert if you need to.
|
||||
//wmenu will validate all responses before calling any function.
|
||||
//It will also figure out which function should be called so you don't have to.
|
||||
package wmenu
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"regexp"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"github.com/mattn/go-isatty"
|
||||
wlog "gopkg.in/dixonwille/wlog.v2"
|
||||
)
|
||||
|
||||
const (
|
||||
y = iota
|
||||
n
|
||||
)
|
||||
|
||||
var (
|
||||
NoColor = os.Getenv("TERM") == "dumb" ||
|
||||
(!isatty.IsTerminal(os.Stdout.Fd()) && !isatty.IsCygwinTerminal(os.Stdout.Fd()))
|
||||
)
|
||||
|
||||
//Menu is used to display options to a user.
|
||||
//A user can then select options and Menu will validate the response and perform the correct action.
|
||||
type Menu struct {
|
||||
question string
|
||||
function func([]Opt) error
|
||||
options []Opt
|
||||
ui wlog.UI
|
||||
multiSeparator string
|
||||
allowMultiple bool
|
||||
loopOnInvalid bool
|
||||
clear bool
|
||||
tries int
|
||||
defIcon string
|
||||
isYN bool
|
||||
ynDef int
|
||||
}
|
||||
|
||||
//NewMenu creates a menu with a wlog.UI as the writer.
|
||||
func NewMenu(question string) *Menu {
|
||||
//Create a default ui to use for menu
|
||||
var ui wlog.UI
|
||||
ui = wlog.New(os.Stdin, os.Stdout, os.Stderr)
|
||||
ui = wlog.AddConcurrent(ui)
|
||||
|
||||
return &Menu{
|
||||
question: question,
|
||||
function: nil,
|
||||
options: nil,
|
||||
ui: ui,
|
||||
multiSeparator: " ",
|
||||
allowMultiple: false,
|
||||
loopOnInvalid: false,
|
||||
clear: false,
|
||||
tries: 3,
|
||||
defIcon: "*",
|
||||
isYN: false,
|
||||
ynDef: 0,
|
||||
}
|
||||
}
|
||||
|
||||
//AddColor will change the color of the menu items.
|
||||
//optionColor changes the color of the options.
|
||||
//questionColor changes the color of the questions.
|
||||
//errorColor changes the color of the question.
|
||||
//Use wlog.None if you do not want to change the color.
|
||||
func (m *Menu) AddColor(optionColor, questionColor, responseColor, errorColor wlog.Color) {
|
||||
if !NoColor {
|
||||
m.ui = wlog.AddColor(questionColor, errorColor, wlog.None, wlog.None, optionColor, responseColor, wlog.None, wlog.None, wlog.None, m.ui)
|
||||
}
|
||||
}
|
||||
|
||||
//ClearOnMenuRun will clear the screen when a menu is ran.
|
||||
//This is checked when LoopOnInvalid is activated.
|
||||
//Meaning if an error occurred then it will clear the screen before asking again.
|
||||
func (m *Menu) ClearOnMenuRun() {
|
||||
m.clear = true
|
||||
}
|
||||
|
||||
//SetSeparator sets the separator to use when multiple options are valid responses.
|
||||
//Default value is a space.
|
||||
func (m *Menu) SetSeparator(sep string) {
|
||||
m.multiSeparator = sep
|
||||
}
|
||||
|
||||
//SetTries sets the number of tries on the loop before failing out.
|
||||
//Default is 3.
|
||||
//Negative values act like 0.
|
||||
func (m *Menu) SetTries(i int) {
|
||||
m.tries = i
|
||||
}
|
||||
|
||||
//LoopOnInvalid is used if an invalid option was given then it will prompt the user again.
|
||||
func (m *Menu) LoopOnInvalid() {
|
||||
m.loopOnInvalid = true
|
||||
}
|
||||
|
||||
//SetDefaultIcon sets the icon used to identify which options will be selected by default
|
||||
func (m *Menu) SetDefaultIcon(icon string) {
|
||||
m.defIcon = icon
|
||||
}
|
||||
|
||||
//IsYesNo sets the menu to a yes/no state.
|
||||
//Does not show options but does ask question.
|
||||
//Will also parse the answer to allow for all variants of yes/no (IE Y yes No ...)
|
||||
//Specify the default value using def. 0 is for yes and 1 is for no.
|
||||
//Both will call the Action function you specified.
|
||||
// Opt{ID: 0, Text: "y"} for yes and Opt{ID: 1, Text: "n"} for no will be passed to the function.
|
||||
func (m *Menu) IsYesNo(def int) {
|
||||
m.isYN = true
|
||||
m.ynDef = def
|
||||
}
|
||||
|
||||
//Option adds an option to the menu for the user to select from.
|
||||
//value is an empty interface that can be used to pass anything through to the function.
|
||||
//title is the string the user will select
|
||||
//isDefault is whether this option is a default option (IE when no options are selected).
|
||||
//function is what is called when only this option is selected.
|
||||
//If function is nil then it will default to the menu's Action.
|
||||
func (m *Menu) Option(title string, value interface{}, isDefault bool, function func(Opt) error) {
|
||||
option := newOption(len(m.options), title, value, isDefault, function)
|
||||
m.options = append(m.options, *option)
|
||||
}
|
||||
|
||||
//Action adds a default action to use in certain scenarios.
|
||||
//If the selected option (by default or user selected) does not have a function applied to it this will be called.
|
||||
//If there are no default options and no option was selected this will be called with an option that has an ID of -1.
|
||||
func (m *Menu) Action(function func([]Opt) error) {
|
||||
m.function = function
|
||||
}
|
||||
|
||||
//AllowMultiple will tell the menu to allow multiple selections.
|
||||
//The menu will fail if this is not called and mulple selections were selected.
|
||||
func (m *Menu) AllowMultiple() {
|
||||
m.allowMultiple = true
|
||||
}
|
||||
|
||||
//ChangeReaderWriter changes where the menu listens and writes to.
|
||||
//reader is where user input is collected.
|
||||
//writer and errorWriter is where the menu should write to.
|
||||
func (m *Menu) ChangeReaderWriter(reader io.Reader, writer, errorWriter io.Writer) {
|
||||
var ui wlog.UI
|
||||
ui = wlog.New(reader, writer, errorWriter)
|
||||
m.ui = ui
|
||||
}
|
||||
|
||||
//Run is used to execute the menu.
|
||||
//It will print to options and question to the screen.
|
||||
//It will only clear the screen if ClearOnMenuRun is activated.
|
||||
//This will validate all responses.
|
||||
//Errors are of type MenuError.
|
||||
func (m *Menu) Run() error {
|
||||
if m.clear {
|
||||
Clear()
|
||||
}
|
||||
valid := false
|
||||
var options []Opt
|
||||
//Loop and on error check if loopOnInvalid is enabled.
|
||||
//If it is Clear the screen and write error.
|
||||
//Then ask again
|
||||
for !valid {
|
||||
//step 1 print options to screen
|
||||
m.print()
|
||||
//step 2 ask question, get and validate response
|
||||
opt, err := m.ask()
|
||||
if err != nil {
|
||||
m.tries = m.tries - 1
|
||||
if !IsMenuErr(err) {
|
||||
err = newMenuError(err, "", m.triesLeft())
|
||||
}
|
||||
if m.loopOnInvalid && m.tries > 0 {
|
||||
if m.clear {
|
||||
Clear()
|
||||
}
|
||||
m.ui.Error(err.Error())
|
||||
} else {
|
||||
return err
|
||||
}
|
||||
} else {
|
||||
options = opt
|
||||
valid = true
|
||||
}
|
||||
}
|
||||
//step 3 call appropriate action with the responses
|
||||
return m.callAppropriate(options)
|
||||
}
|
||||
|
||||
func (m *Menu) callAppropriate(options []Opt) (err error) {
|
||||
if len(options) == 0 {
|
||||
return m.callAppropriateNoOptions()
|
||||
}
|
||||
if len(options) == 1 && options[0].function != nil {
|
||||
return options[0].function(options[0])
|
||||
}
|
||||
return m.function(options)
|
||||
}
|
||||
|
||||
func (m *Menu) callAppropriateNoOptions() (err error) {
|
||||
options := m.getDefault()
|
||||
if len(options) == 0 {
|
||||
return m.function([]Opt{{ID: -1}})
|
||||
}
|
||||
if len(options) == 1 && options[0].function != nil {
|
||||
return options[0].function(options[0])
|
||||
}
|
||||
return m.function(options)
|
||||
}
|
||||
|
||||
//hide options when this is a yes or no
|
||||
func (m *Menu) print() {
|
||||
if !m.isYN {
|
||||
for _, opt := range m.options {
|
||||
icon := m.defIcon
|
||||
if !opt.isDefault {
|
||||
icon = ""
|
||||
}
|
||||
m.ui.Output(fmt.Sprintf("%d) %s%s", opt.ID, icon, opt.Text))
|
||||
}
|
||||
} else {
|
||||
//TODO Allow user to specify what to use as value for YN options
|
||||
m.options = []Opt{}
|
||||
m.Option("y", "yes", m.ynDef == y, nil)
|
||||
m.Option("n", "no", m.ynDef == n, nil)
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Menu) ask() ([]Opt, error) {
|
||||
if m.isYN {
|
||||
if m.ynDef == y {
|
||||
m.question += " (Y/n)"
|
||||
} else {
|
||||
m.question += " (y/N)"
|
||||
}
|
||||
}
|
||||
trim := ""
|
||||
if m.multiSeparator == " " {
|
||||
trim = m.multiSeparator
|
||||
} else {
|
||||
trim = m.multiSeparator + " "
|
||||
}
|
||||
res, err := m.ui.Ask(m.question, trim)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
//Validate responses
|
||||
//Check if no responses are returned and no action to call
|
||||
if res == "" {
|
||||
//get default options
|
||||
opt := m.getDefault()
|
||||
if !m.validOptAndFunc(opt) {
|
||||
return nil, newMenuError(ErrNoResponse, "", m.triesLeft())
|
||||
}
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
var responses []int
|
||||
if !m.isYN {
|
||||
responses, err = m.resToInt(res)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = m.validateResponses(responses)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
} else {
|
||||
responses, err = m.ynResParse(res)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
//Parse responses and return them as options
|
||||
var finalOptions []Opt
|
||||
for _, response := range responses {
|
||||
finalOptions = append(finalOptions, m.options[response])
|
||||
}
|
||||
|
||||
return finalOptions, nil
|
||||
}
|
||||
|
||||
//Converts the response string to a slice of ints, also validates along the way.
|
||||
func (m *Menu) resToInt(res string) ([]int, error) {
|
||||
resStrings := strings.Split(res, m.multiSeparator)
|
||||
//Check if we don't want multiple responses
|
||||
if !m.allowMultiple && len(resStrings) > 1 {
|
||||
return nil, newMenuError(ErrTooMany, "", m.triesLeft())
|
||||
}
|
||||
|
||||
//Convert responses to intigers
|
||||
var responses []int
|
||||
for _, response := range resStrings {
|
||||
//Check if it is an intiger
|
||||
response = strings.Trim(response, " ")
|
||||
r, err := strconv.Atoi(response)
|
||||
if err != nil {
|
||||
return nil, newMenuError(ErrInvalid, response, m.triesLeft())
|
||||
}
|
||||
responses = append(responses, r)
|
||||
}
|
||||
return responses, nil
|
||||
}
|
||||
|
||||
func (m *Menu) ynResParse(res string) ([]int, error) {
|
||||
resStrings := strings.Split(res, m.multiSeparator)
|
||||
if len(resStrings) > 1 {
|
||||
return nil, newMenuError(ErrTooMany, "", m.triesLeft())
|
||||
}
|
||||
re := regexp.MustCompile("^\\s*(?:([Yy])(?:es|ES)?|([Nn])(?:o|O)?)\\s*$")
|
||||
matches := re.FindStringSubmatch(res)
|
||||
if len(matches) < 2 {
|
||||
return nil, newMenuError(ErrInvalid, res, m.triesLeft())
|
||||
}
|
||||
if strings.ToLower(matches[1]) == "y" {
|
||||
return []int{y}, nil
|
||||
}
|
||||
return []int{n}, nil
|
||||
}
|
||||
|
||||
//Check if response is in the range of options
|
||||
//If it is make sure it is not duplicated
|
||||
func (m *Menu) validateResponses(responses []int) error {
|
||||
var tmp []int
|
||||
for _, response := range responses {
|
||||
if response < 0 || len(m.options)-1 < response {
|
||||
return newMenuError(ErrInvalid, strconv.Itoa(response), m.triesLeft())
|
||||
}
|
||||
|
||||
if exist(tmp, response) {
|
||||
return newMenuError(ErrDuplicate, strconv.Itoa(response), m.triesLeft())
|
||||
}
|
||||
|
||||
tmp = append(tmp, response)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
//Simply checks if number exists in the slice
|
||||
func exist(slice []int, number int) bool {
|
||||
for _, s := range slice {
|
||||
if number == s {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
//gets a list of default options
|
||||
func (m *Menu) getDefault() []Opt {
|
||||
var opt []Opt
|
||||
for _, o := range m.options {
|
||||
if o.isDefault {
|
||||
opt = append(opt, o)
|
||||
}
|
||||
}
|
||||
return opt
|
||||
}
|
||||
|
||||
//make sure that there is an action available to be called in certain cases
|
||||
//returns false if it chould not find an action for the number options available
|
||||
func (m *Menu) validOptAndFunc(opt []Opt) bool {
|
||||
if m.function == nil {
|
||||
if len(opt) == 1 && opt[0].function != nil {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func (m *Menu) triesLeft() int {
|
||||
if m.loopOnInvalid && m.tries > 0 {
|
||||
return m.tries
|
||||
}
|
||||
return 0
|
||||
}
|
||||
21
vendor/github.com/dixonwille/wmenu/option.go
generated
vendored
Normal file
21
vendor/github.com/dixonwille/wmenu/option.go
generated
vendored
Normal file
@@ -0,0 +1,21 @@
|
||||
package wmenu
|
||||
|
||||
//Opt is what Menu uses to display options to screen.
|
||||
//Also holds information on what should run and if it is a default option
|
||||
type Opt struct {
|
||||
ID int
|
||||
Text string
|
||||
Value interface{}
|
||||
function func(Opt) error
|
||||
isDefault bool
|
||||
}
|
||||
|
||||
func newOption(id int, text string, value interface{}, def bool, function func(Opt) error) *Opt {
|
||||
return &Opt{
|
||||
ID: id,
|
||||
Text: text,
|
||||
Value: value,
|
||||
isDefault: def,
|
||||
function: function,
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user