1
package pflag
2

3
import (
4
	"encoding"
5
	"fmt"
6
	"reflect"
7
)
8

9
// following is copied from go 1.23.4 flag.go
10
type textValue struct{ p encoding.TextUnmarshaler }
11

12
func newTextValue(val encoding.TextMarshaler, p encoding.TextUnmarshaler) textValue {
13
	ptrVal := reflect.ValueOf(p)
14
	if ptrVal.Kind() != reflect.Ptr {
15
		panic("variable value type must be a pointer")
16
	}
17
	defVal := reflect.ValueOf(val)
18
	if defVal.Kind() == reflect.Ptr {
19
		defVal = defVal.Elem()
20
	}
21
	if defVal.Type() != ptrVal.Type().Elem() {
22
		panic(fmt.Sprintf("default type does not match variable type: %v != %v", defVal.Type(), ptrVal.Type().Elem()))
23
	}
24
	ptrVal.Elem().Set(defVal)
25
	return textValue{p}
26
}
27

28
func (v textValue) Set(s string) error {
29
	return v.p.UnmarshalText([]byte(s))
30
}
31

32
func (v textValue) Get() interface{} {
33
	return v.p
34
}
35

36
func (v textValue) String() string {
37
	if m, ok := v.p.(encoding.TextMarshaler); ok {
38
		if b, err := m.MarshalText(); err == nil {
39
			return string(b)
40
		}
41
	}
42
	return ""
43
}
44

45
//end of copy
46

47
func (v textValue) Type() string {
48
	return reflect.ValueOf(v.p).Type().Name()
49
}
50

51
// GetText set out, which implements encoding.UnmarshalText, to the value of a flag with given name
52
func (f *FlagSet) GetText(name string, out encoding.TextUnmarshaler) error {
53
	flag := f.Lookup(name)
54
	if flag == nil {
55
		return fmt.Errorf("flag accessed but not defined: %s", name)
56
	}
57
	if flag.Value.Type() != reflect.TypeOf(out).Name() {
58
		return fmt.Errorf("trying to get %s value of flag of type %s", reflect.TypeOf(out).Name(), flag.Value.Type())
59
	}
60
	return out.UnmarshalText([]byte(flag.Value.String()))
61
}
62

63
// TextVar defines a flag with a specified name, default value, and usage string. The argument p must be a pointer to a variable that will hold the value of the flag, and p must implement encoding.TextUnmarshaler. If the flag is used, the flag value will be passed to p's UnmarshalText method. The type of the default value must be the same as the type of p.
64
func (f *FlagSet) TextVar(p encoding.TextUnmarshaler, name string, value encoding.TextMarshaler, usage string) {
65
	f.VarP(newTextValue(value, p), name, "", usage)
66
}
67

68
// TextVarP is like TextVar, but accepts a shorthand letter that can be used after a single dash.
69
func (f *FlagSet) TextVarP(p encoding.TextUnmarshaler, name, shorthand string, value encoding.TextMarshaler, usage string) {
70
	f.VarP(newTextValue(value, p), name, shorthand, usage)
71
}
72

73
// TextVar defines a flag with a specified name, default value, and usage string. The argument p must be a pointer to a variable that will hold the value of the flag, and p must implement encoding.TextUnmarshaler. If the flag is used, the flag value will be passed to p's UnmarshalText method. The type of the default value must be the same as the type of p.
74
func TextVar(p encoding.TextUnmarshaler, name string, value encoding.TextMarshaler, usage string) {
75
	CommandLine.VarP(newTextValue(value, p), name, "", usage)
76
}
77

78
// TextVarP is like TextVar, but accepts a shorthand letter that can be used after a single dash.
79
func TextVarP(p encoding.TextUnmarshaler, name, shorthand string, value encoding.TextMarshaler, usage string) {
80
	CommandLine.VarP(newTextValue(value, p), name, shorthand, usage)
81
}
82

83