string_array.go 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. package pflag
  2. import (
  3. "fmt"
  4. "strings"
  5. )
  6. var _ = fmt.Fprint
  7. // -- stringArray Value
  8. type stringArrayValue struct {
  9. value *[]string
  10. changed bool
  11. }
  12. func newStringArrayValue(val []string, p *[]string) *stringArrayValue {
  13. ssv := new(stringArrayValue)
  14. ssv.value = p
  15. *ssv.value = val
  16. return ssv
  17. }
  18. func (s *stringArrayValue) Set(val string) error {
  19. if !s.changed {
  20. *s.value = []string{val}
  21. s.changed = true
  22. } else {
  23. *s.value = append(*s.value, val)
  24. }
  25. return nil
  26. }
  27. func (s *stringArrayValue) Type() string {
  28. return "stringArray"
  29. }
  30. func (s *stringArrayValue) String() string {
  31. str, _ := writeAsCSV(*s.value)
  32. return "[" + str + "]"
  33. }
  34. func stringArrayConv(sval string) (interface{}, error) {
  35. sval = strings.Trim(sval, "[]")
  36. // An empty string would cause a array with one (empty) string
  37. if len(sval) == 0 {
  38. return []string{}, nil
  39. }
  40. return readAsCSV(sval)
  41. }
  42. // GetStringArray return the []string value of a flag with the given name
  43. func (f *FlagSet) GetStringArray(name string) ([]string, error) {
  44. val, err := f.getFlagType(name, "stringArray", stringArrayConv)
  45. if err != nil {
  46. return []string{}, err
  47. }
  48. return val.([]string), nil
  49. }
  50. // StringArrayVar defines a string flag with specified name, default value, and usage string.
  51. // The argument p points to a []string variable in which to store the values of the multiple flags.
  52. // The value of each argument will not try to be separated by comma
  53. func (f *FlagSet) StringArrayVar(p *[]string, name string, value []string, usage string) {
  54. f.VarP(newStringArrayValue(value, p), name, "", usage)
  55. }
  56. // StringArrayVarP is like StringArrayVar, but accepts a shorthand letter that can be used after a single dash.
  57. func (f *FlagSet) StringArrayVarP(p *[]string, name, shorthand string, value []string, usage string) {
  58. f.VarP(newStringArrayValue(value, p), name, shorthand, usage)
  59. }
  60. // StringArrayVar defines a string flag with specified name, default value, and usage string.
  61. // The argument p points to a []string variable in which to store the value of the flag.
  62. // The value of each argument will not try to be separated by comma
  63. func StringArrayVar(p *[]string, name string, value []string, usage string) {
  64. CommandLine.VarP(newStringArrayValue(value, p), name, "", usage)
  65. }
  66. // StringArrayVarP is like StringArrayVar, but accepts a shorthand letter that can be used after a single dash.
  67. func StringArrayVarP(p *[]string, name, shorthand string, value []string, usage string) {
  68. CommandLine.VarP(newStringArrayValue(value, p), name, shorthand, usage)
  69. }
  70. // StringArray defines a string flag with specified name, default value, and usage string.
  71. // The return value is the address of a []string variable that stores the value of the flag.
  72. // The value of each argument will not try to be separated by comma
  73. func (f *FlagSet) StringArray(name string, value []string, usage string) *[]string {
  74. p := []string{}
  75. f.StringArrayVarP(&p, name, "", value, usage)
  76. return &p
  77. }
  78. // StringArrayP is like StringArray, but accepts a shorthand letter that can be used after a single dash.
  79. func (f *FlagSet) StringArrayP(name, shorthand string, value []string, usage string) *[]string {
  80. p := []string{}
  81. f.StringArrayVarP(&p, name, shorthand, value, usage)
  82. return &p
  83. }
  84. // StringArray defines a string flag with specified name, default value, and usage string.
  85. // The return value is the address of a []string variable that stores the value of the flag.
  86. // The value of each argument will not try to be separated by comma
  87. func StringArray(name string, value []string, usage string) *[]string {
  88. return CommandLine.StringArrayP(name, "", value, usage)
  89. }
  90. // StringArrayP is like StringArray, but accepts a shorthand letter that can be used after a single dash.
  91. func StringArrayP(name, shorthand string, value []string, usage string) *[]string {
  92. return CommandLine.StringArrayP(name, shorthand, value, usage)
  93. }