package options

import (
	"flag"
	"os"
)

func Parse(usage func(), set func(*flag.FlagSet)) *flag.FlagSet {
	args := os.Args
	if len(args) < 2 {
		args = append(args, "")
	}

	fs := flag.NewFlagSet(args[1], flag.ExitOnError)
	fs.Usage = func() {
		usage()
	}
	set(fs)
	fs.Parse(args[2:])
	return fs
}

func AddString(fs *flag.FlagSet, opt *string, short, long, value string) {
	fs.StringVar(opt, short, value, "")
	fs.StringVar(opt, long, value, "")
}

func AddUint64(fs *flag.FlagSet, opt *uint64, short, long string, value uint64) {
	fs.Uint64Var(opt, short, value, "")
	fs.Uint64Var(opt, long, value, "")
}