parse.go

package args

import (
	"sort"
	"strings"
)

type Opts []string

func (opts Opts) index(opt string) int {
	off := 1
	if len(opt) > 1 {
		off = 2
	}
	for i, o := range opts {
		if len(o) >= 2 && o[0] == '-' {
			if o[off:] == opt {
				return i
			}
		}
	}
	return -1
}

// Has returns true if opt flag is in opts.
func (opts Opts) Has(opt string) bool {
	return opts.index(opt) != -1
}

// Val return the value of the first occurrence of opt.
func (opts Opts) Val(opt string) string {
	if idx := opts.index(opt); idx != -1 && idx+1 < len(opts) {
		return opts[idx+1]
	}
	return ""
}

// Vals return the values of all occurrences of opt.
func (opts Opts) Vals(opt string) []string {
	var vals []string
	off := 1
	if len(opt) > 1 {
		off = 2
	}
	for i, o := range opts {
		if len(o) >= 2 && o[0] == '-' {
			if o[off:] == opt {
				if i+1 < len(opts) {
					i++
					vals = append(vals, opts[i])
				}
			}
		}
	}
	return vals
}

// Remove removes all occurrences of opt and return true if found.
func (opts *Opts) Remove(opt string) bool {
	found := false
	for idx := opts.index(opt); idx != -1 && idx < len(*opts); idx = opts.index(opt) {
		*opts = append((*opts)[:idx], (*opts)[idx+1:]...)
		found = true
	}
	return found
}

// Parse converts an HTTPie like argv into a list of curl options.
func Parse(argv Opts) (opts Opts) {
	isForm := argv.Has("F") || argv.Has("form")
	if isForm {
		argv.Remove("F")
		argv.Remove("form")
	}
	args := []string{}
	sort.Strings(curlLongValues)
	more := true
	for i := 1; i < len(argv); i++ {
		arg := argv[i]
		if !more || len(arg) < 2 || arg[0] != '-' {
			args = append(args, arg)
			continue
		}
		if arg == "--" {
			// Enf of opts marker
			more = false
			continue
		}
		if arg[1] == '-' {
			opts = append(opts, arg)
			if longHasValue(arg[2:]) && i+1 < len(argv) {
				opts = append(opts, argv[i+1])
				i++
			}
			continue
		}
		// Parse componed short args
		for j := 1; j < len(arg); j++ {
			opts = append(opts, string([]byte{'-', arg[j]}))
			if strings.IndexByte(curlShortValues, arg[j]) != -1 {
				// Short arg as value, it must be last in compound.
				// The value is either the remaining or the next arg.
				if j == len(arg)-1 {
					if i+1 < len(argv) {
						opts = append(opts, argv[i+1])
						i++
					}
				} else {
					opts = append(opts, arg[j+1:])
				}
				break
			}
		}
	}
	if len(args) > 0 {
		postMode := PostModeJSON
		if isForm {
			postMode = PostModeFORM
		}
		opts = append(opts, parseFancyArgs(args, postMode)...)
	}
	return
}

func longHasValue(arg string) bool {
	i := sort.SearchStrings(curlLongValues, arg)
	return i < len(curlLongValues) && curlLongValues[i] == arg
}