71 lines
1.1 KiB
Go
71 lines
1.1 KiB
Go
package arguments
|
|
|
|
import (
|
|
"bytes"
|
|
"flag"
|
|
"io/ioutil"
|
|
"strings"
|
|
)
|
|
|
|
var FlagName = "command"
|
|
|
|
type FlagSet struct {
|
|
*flag.FlagSet
|
|
}
|
|
|
|
func NewFlagSet() *FlagSet {
|
|
fs := flag.NewFlagSet(FlagName, flag.ContinueOnError)
|
|
fs.SetOutput(ioutil.Discard)
|
|
|
|
return &FlagSet{fs}
|
|
}
|
|
|
|
func (fs *FlagSet) Usage() string {
|
|
var buf bytes.Buffer
|
|
|
|
fs.FlagSet.SetOutput(&buf)
|
|
fs.FlagSet.Usage()
|
|
fs.FlagSet.SetOutput(ioutil.Discard)
|
|
|
|
return buf.String()
|
|
}
|
|
|
|
type Flag struct {
|
|
command string
|
|
arguments []string
|
|
}
|
|
|
|
func (f *Flag) ParseContent(arguments []string) error {
|
|
// trim the command out
|
|
f.command, f.arguments = arguments[0], arguments[1:]
|
|
return nil
|
|
}
|
|
|
|
func (f *Flag) Usage() string {
|
|
return "[flags] arguments..."
|
|
}
|
|
|
|
func (f *Flag) Command() string {
|
|
return f.command
|
|
}
|
|
|
|
func (f *Flag) Args() []string {
|
|
return f.arguments
|
|
}
|
|
|
|
func (f *Flag) Arg(n int) string {
|
|
if n < 0 || n >= len(f.arguments) {
|
|
return ""
|
|
}
|
|
|
|
return f.arguments[n]
|
|
}
|
|
|
|
func (f *Flag) String() string {
|
|
return strings.Join(f.arguments, " ")
|
|
}
|
|
|
|
func (f *Flag) With(fs *flag.FlagSet) error {
|
|
return fs.Parse(f.arguments)
|
|
}
|