mirror of
https://github.com/johnkerl/miller.git
synced 2026-07-17 16:38:54 +00:00
5eb40c9e7b
* Comment style * IRecordTransformer -> RecordTransfomer * make fmt * else-return style mod * snake-case -> camel-case * Remove redundant err = nil and similar zero-value initializations. * redundant break; * bugfix * neaten * typofix * simplify/standardize init of zero-length slices * Standardize fmt.Fprintf w/ errors * fix double print of "mlr:" * neatening * Uniformize error messages * make docs * avoid shadowing package names * shorten some receiver names
43 lines
1.3 KiB
Go
43 lines
1.3 KiB
Go
package lib
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
// Getoptify expands "-xyz" into "-x -y -z" while leaving "--xyz" intact. This
|
|
// is a keystroke-saver for the user.
|
|
//
|
|
// This is OK to do here globally since Miller is quite consistent (in main,
|
|
// verbs, auxents, and terminals) that multi-character options start with two
|
|
// dashes, e.g. "--csv". (The sole exception is the sort verb's -nf/-nr which
|
|
// are handled specially there.)
|
|
//
|
|
// Additionally, we split "--foo=bar" into "--foo" and "bar".
|
|
func Getoptify(inargs []string) []string {
|
|
expandRegex := regexp.MustCompile("^-[a-zA-Z0-9]+$")
|
|
splitRegex := regexp.MustCompile("^--[^=]+=.+$")
|
|
numberRegex := regexp.MustCompile("^-[0-9]+$")
|
|
outargs := []string{}
|
|
for _, inarg := range inargs {
|
|
if expandRegex.MatchString(inarg) {
|
|
if numberRegex.MatchString(inarg) {
|
|
// Don't expand things like '-12345' which are (likely!) numeric arguments to verbs.
|
|
// Example: 'mlr unsparsify --fill-with -99999'.
|
|
outargs = append(outargs, inarg)
|
|
} else {
|
|
for _, c := range inarg[1:] {
|
|
outargs = append(outargs, "-"+string(c))
|
|
}
|
|
}
|
|
} else if splitRegex.MatchString(inarg) {
|
|
pair := strings.SplitN(inarg, "=", 2)
|
|
InternalCodingErrorIf(len(pair) != 2)
|
|
outargs = append(outargs, pair[0])
|
|
outargs = append(outargs, pair[1])
|
|
} else {
|
|
outargs = append(outargs, inarg)
|
|
}
|
|
}
|
|
return outargs
|
|
}
|