walles/moar
1
1
Fork 0
mirror of https://github.com/walles/moar.git synced 2024-10-10 18:37:51 +03:00
Code Issues Projects Releases Wiki Activity

Accept +1234 argument in any place

Browse Source 
Fixes the issue reported with PR #158:
https://github.com/walles/moar/pull/158#issue-1918313499
This commit is contained in:
Johan Walles 2023-09-29 22:42:41 +02:00
parent 242c050adf
commit faafc49471
2 changed files with 15 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
moar.go
View File

@ -279,8 +279,7 @@ func tryOpen(filename string) error {
// line number, and returns the remaining args.
//
// Returns 0 on no target line number specified.
func getTargetLineNumberOneBased(flagSet *flag.FlagSet) (int, []string) {
args := flagSet.Args()
func getTargetLineNumberOneBased(args []string) (int, []string) {
for i, arg := range args {
if !strings.HasPrefix(arg, "+") {
continue
@ -378,7 +377,9 @@ func main() {
flags = append(strings.Fields(moarEnv), flags...)
}
err := flagSet.Parse(flags)
targetLineNumberOneBased, remainingArgs := getTargetLineNumberOneBased(flags)
err := flagSet.Parse(remainingArgs)
if err != nil {
if err == flag.ErrHelp {
printUsage(os.Stdout, flagSet, false)
@ -408,10 +409,8 @@ func main() {
TimestampFormat: time.StampMicro,
})
targetLineNumberOneBased, remainingArgs := getTargetLineNumberOneBased(flagSet)
if len(remainingArgs) > 1 {
fmt.Fprintln(os.Stderr, "ERROR: Expected exactly one filename, or data piped from stdin, got:", remainingArgs)
if len(flagSet.Args()) > 1 {
fmt.Fprintln(os.Stderr, "ERROR: Expected exactly one filename, or data piped from stdin, got:", flagSet.Args())
fmt.Fprintln(os.Stderr)
printUsage(os.Stderr, flagSet, true)
@ -421,8 +420,8 @@ func main() {
stdinIsRedirected := !term.IsTerminal(int(os.Stdin.Fd()))
stdoutIsRedirected := !term.IsTerminal(int(os.Stdout.Fd()))
var inputFilename *string
if len(remainingArgs) == 1 {
word := remainingArgs[0]
if len(flagSet.Args()) == 1 {
word := flagSet.Args()[0]
inputFilename = &word
// Need to check before twin.NewScreen() below, otherwise the screen

7
test.sh
View File

@ -61,6 +61,13 @@ if ./moar does-not-exist >&/dev/null; then
exit 1
fi
echo Testing not crashing with different argument orders...
./moar +123 moar.go >/dev/null
./moar moar.go +123 >/dev/null
./moar +123 --trace moar.go >/dev/null
./moar --trace +123 moar.go >/dev/null
./moar --trace moar.go +123 >/dev/null
echo Test --version...
./moar --version >/dev/null # Should exit with code 0
diff -u <(./moar --version) <(git describe --tags --dirty --always)