values.go 6.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315
  1. // mauflag - An extendable command-line argument parser for Golang
  2. // Copyright (C) 2016 Maunium
  3. //
  4. // This program is free software: you can redistribute it and/or modify
  5. // it under the terms of the GNU General Public License as published by
  6. // the Free Software Foundation, either version 3 of the License, or
  7. // (at your option) any later version.
  8. // This program is distributed in the hope that it will be useful,
  9. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11. // GNU General Public License for more details.
  12. // You should have received a copy of the GNU General Public License
  13. // along with this program. If not, see <http://www.gnu.org/licenses/>.
  14. package mauflag
  15. import (
  16. "errors"
  17. "strconv"
  18. "strings"
  19. )
  20. // Value represents the value a flag has
  21. type Value interface {
  22. // Set or add the data in the given string to this value
  23. Set(string) error
  24. // Name should return the type name of this value or (if this value is an array,
  25. // a map or something of that sort) the type name of one object in this value
  26. Name() string
  27. }
  28. type boolValue bool
  29. func (val *boolValue) Name() string {
  30. return "boolean"
  31. }
  32. func (val *boolValue) Set(newVal string) error {
  33. b, err := strconv.ParseBool(newVal)
  34. if err == nil {
  35. *val = boolValue(b)
  36. }
  37. return err
  38. }
  39. type stringValue string
  40. func (val *stringValue) Name() string {
  41. return "string"
  42. }
  43. func (val *stringValue) Set(newVal string) error {
  44. *val = stringValue(newVal)
  45. return nil
  46. }
  47. type ssMapValue map[string]string
  48. func (val *ssMapValue) Name() string {
  49. return "string-string pair"
  50. }
  51. func (val *ssMapValue) Set(newVal string) error {
  52. index := strings.IndexRune(newVal, '=')
  53. if index == -1 {
  54. index = strings.IndexRune(newVal, ',')
  55. if index == -1 {
  56. index = strings.IndexRune(newVal, ':')
  57. if index == -1 {
  58. return errors.New("No pair separator found")
  59. }
  60. }
  61. }
  62. (*val)[newVal[:index]] = newVal[:index+1]
  63. return nil
  64. }
  65. type stringArrayValue []string
  66. func (val *stringArrayValue) Name() string {
  67. return "string"
  68. }
  69. func (val *stringArrayValue) Set(newVal string) error {
  70. *val = append(*val, newVal)
  71. return nil
  72. }
  73. type intArrayValue []int
  74. func (val *intArrayValue) Name() string {
  75. return "integer"
  76. }
  77. func (val *intArrayValue) Set(newVal string) error {
  78. i, err := strconv.Atoi(newVal)
  79. if err != nil {
  80. return err
  81. }
  82. *val = append(*val, i)
  83. return nil
  84. }
  85. type int64ArrayValue []int64
  86. func (val *int64ArrayValue) Name() string {
  87. return "64-bit integer"
  88. }
  89. func (val *int64ArrayValue) Set(newVal string) error {
  90. i, err := strconv.ParseInt(newVal, 10, 64)
  91. if err != nil {
  92. return err
  93. }
  94. *val = append(*val, i)
  95. return nil
  96. }
  97. type intValue int
  98. func (val *intValue) Name() string {
  99. return "integer"
  100. }
  101. func (val *intValue) Set(newVal string) error {
  102. i, err := strconv.ParseInt(newVal, 10, 32)
  103. if err == nil {
  104. *val = intValue(i)
  105. }
  106. return err
  107. }
  108. type uintValue uint
  109. func (val *uintValue) Name() string {
  110. return "unsigned integer"
  111. }
  112. func (val *uintValue) Set(newVal string) error {
  113. i, err := strconv.ParseUint(newVal, 10, 32)
  114. if err == nil {
  115. *val = uintValue(i)
  116. }
  117. return err
  118. }
  119. type int8Value int8
  120. func (val *int8Value) Name() string {
  121. return "8-bit integer"
  122. }
  123. func (val *int8Value) Set(newVal string) error {
  124. i, err := strconv.ParseInt(newVal, 10, 8)
  125. if err == nil {
  126. *val = int8Value(i)
  127. }
  128. return err
  129. }
  130. type uint8Value uint8
  131. func (val *uint8Value) Name() string {
  132. return "unsigned 8-bit integer"
  133. }
  134. func (val *uint8Value) Set(newVal string) error {
  135. i, err := strconv.ParseUint(newVal, 10, 8)
  136. if err == nil {
  137. *val = uint8Value(i)
  138. }
  139. return err
  140. }
  141. type byteValue byte
  142. func (val *byteValue) Name() string {
  143. return "rune (unsigned 8-bit integer)"
  144. }
  145. func (val *byteValue) Set(newVal string) error {
  146. i, err := strconv.ParseUint(newVal, 10, 8)
  147. if err == nil {
  148. *val = byteValue(i)
  149. }
  150. return err
  151. }
  152. type int16Value int16
  153. func (val *int16Value) Name() string {
  154. return "16-bit integer"
  155. }
  156. func (val *int16Value) Set(newVal string) error {
  157. i, err := strconv.ParseInt(newVal, 10, 16)
  158. if err == nil {
  159. *val = int16Value(i)
  160. }
  161. return err
  162. }
  163. type uint16Value uint16
  164. func (val *uint16Value) Name() string {
  165. return "unsigned 16-bit integer"
  166. }
  167. func (val *uint16Value) Set(newVal string) error {
  168. i, err := strconv.ParseUint(newVal, 10, 16)
  169. if err == nil {
  170. *val = uint16Value(i)
  171. }
  172. return err
  173. }
  174. type int32Value int32
  175. func (val *int32Value) Name() string {
  176. return "32-bit integer"
  177. }
  178. func (val *int32Value) Set(newVal string) error {
  179. i, err := strconv.ParseInt(newVal, 10, 32)
  180. if err == nil {
  181. *val = int32Value(i)
  182. }
  183. return err
  184. }
  185. type runeValue rune
  186. func (val *runeValue) Name() string {
  187. return "rune (signed 32-bit integer)"
  188. }
  189. func (val *runeValue) Set(newVal string) error {
  190. i, err := strconv.ParseInt(newVal, 10, 32)
  191. if err == nil {
  192. *val = runeValue(i)
  193. }
  194. return err
  195. }
  196. type uint32Value uint32
  197. func (val *uint32Value) Name() string {
  198. return "unsigned 32-bit integer"
  199. }
  200. func (val *uint32Value) Set(newVal string) error {
  201. i, err := strconv.ParseUint(newVal, 10, 32)
  202. if err == nil {
  203. *val = uint32Value(i)
  204. }
  205. return err
  206. }
  207. type int64Value int64
  208. func (val *int64Value) Name() string {
  209. return "64-bit integer"
  210. }
  211. func (val *int64Value) Set(newVal string) error {
  212. i, err := strconv.ParseInt(newVal, 10, 64)
  213. if err == nil {
  214. *val = int64Value(i)
  215. }
  216. return err
  217. }
  218. type uint64Value uint64
  219. func (val *uint64Value) Name() string {
  220. return "unsigned 64-bit integer"
  221. }
  222. func (val *uint64Value) Set(newVal string) error {
  223. i, err := strconv.ParseUint(newVal, 10, 64)
  224. if err == nil {
  225. *val = uint64Value(i)
  226. }
  227. return err
  228. }
  229. type float32Value float32
  230. func (val *float32Value) Name() string {
  231. return "32-bit float"
  232. }
  233. func (val *float32Value) Set(newVal string) error {
  234. i, err := strconv.ParseFloat(newVal, 32)
  235. if err == nil {
  236. *val = float32Value(i)
  237. }
  238. return err
  239. }
  240. type float64Value float64
  241. func (val *float64Value) Name() string {
  242. return "64-bit float"
  243. }
  244. func (val *float64Value) Set(newVal string) error {
  245. i, err := strconv.ParseFloat(newVal, 64)
  246. if err == nil {
  247. *val = float64Value(i)
  248. }
  249. return err
  250. }