flag_boolstr.go

package main

import (
	"github.com/spf13/pflag"
	"strconv"
)

func NewBoolStrFlag(bs *boolStr, name, shorthand, usage string) *pflag.Flag {
	flag := &pflag.Flag{
		Name:        name,
		Shorthand:   shorthand,
		Usage:       usage,
		Value:       bs,
		DefValue:    "false",
		NoOptDefVal: "true",
	}

	return flag
}

type boolStr struct {
	bool bool
	str  string
}

func (bs *boolStr) Set(s string) error {
	b, err := strconv.ParseBool(s)
	if err == nil {
		s = ""
	} else {
		b = true
	}

	*bs = boolStr{
		bool: b,
		str:  s,
	}

	return nil
}

func (bs *boolStr) Type() string {
	return "bool|str"
}

func (bs *boolStr) String() string {
	if bs.bool {
		return bs.str
	}

	return "false"
}