2018-03-09 21:12:22 +03:00
|
|
|
|
# Enumer [![GoDoc](https://godoc.org/github.com/alvaroloes/enumer?status.svg)](https://godoc.org/github.com/alvaroloes/enumer) [![Go Report Card](https://goreportcard.com/badge/github.com/alvaroloes/enumer)](https://goreportcard.com/report/github.com/alvaroloes/enumer) [![cover.run go](https://cover.run/go/github.com/alvaroloes/enumer.svg?tag=golang-1.10)](https://cover.run/go/github.com/alvaroloes/enumer?tag=golang-1.10)
|
2016-01-21 13:49:12 +03:00
|
|
|
|
Enumer is a tool to generate Go code that adds useful methods to Go enums (constants with a specific type).
|
2016-01-21 02:09:18 +03:00
|
|
|
|
It started as a fork of [Rob Pike’s Stringer tool](https://godoc.org/golang.org/x/tools/cmd/stringer).
|
2015-12-29 16:27:33 +03:00
|
|
|
|
|
2017-02-01 11:49:25 +03:00
|
|
|
|
## Generated functions and methods
|
2016-05-21 15:34:12 +03:00
|
|
|
|
When Enumer is applied to a type, it will generate:
|
2016-01-21 02:09:18 +03:00
|
|
|
|
|
2018-03-09 17:47:25 +03:00
|
|
|
|
* The following basic methods/functions:
|
2018-03-09 17:44:41 +03:00
|
|
|
|
|
|
|
|
|
* Method `String()`: returns the string representation of the enum value. This makes the enum conform
|
2016-01-21 02:09:18 +03:00
|
|
|
|
the `Stringer` interface, so whenever you print an enum value, you'll get the string name instead of a number.
|
2018-03-09 17:44:41 +03:00
|
|
|
|
* Function `<Type>String(s string)`: returns the enum value from its string representation. This is useful
|
2016-12-20 15:57:15 +03:00
|
|
|
|
when you need to read enum values from command line arguments, from a configuration file, or
|
|
|
|
|
from a REST API request... In short, from those places where using the real enum value (an integer) would
|
2016-01-21 02:09:18 +03:00
|
|
|
|
be almost meaningless or hard to trace or use by a human.
|
2018-03-09 17:47:25 +03:00
|
|
|
|
* Function `<Type>Values()`: returns a slice with all the values of the enum
|
2018-03-09 17:44:41 +03:00
|
|
|
|
* Method `IsA<Type>()`: returns true only if the current value is among the values of the enum. Useful for validations.
|
2016-12-20 15:57:15 +03:00
|
|
|
|
* When the flag `json` is provided, two additional methods will be generated, `MarshalJSON()` and `UnmarshalJSON()`. These make
|
|
|
|
|
the enum conform to the `json.Marshaler` and `json.Unmarshaler` interfaces. Very useful to use it in JSON APIs.
|
2018-03-08 23:08:41 +03:00
|
|
|
|
* When the flag `text` is provided, two additional methods will be generated, `MarshalText()` and `UnmarshalText()`. These make
|
|
|
|
|
the enum conform to the `encoding.TextMarshaler` and `encoding.TextUnmarshaler` interfaces.
|
|
|
|
|
**Note:** If you use your enum values as keys in a map and you encode the map as _JSON_, you need this flag set to true to properly
|
|
|
|
|
convert the map keys to json (strings). If not, the numeric values will be used instead
|
2016-12-20 15:57:15 +03:00
|
|
|
|
* When the flag `yaml` is provided, two additional methods will be generated, `MarshalYAML()` and `UnmarshalYAML()`. These make
|
|
|
|
|
the enum conform to the `gopkg.in/yaml.v2.Marshaler` and `gopkg.in/yaml.v2.Unmarshaler` interfaces.
|
2016-10-26 19:12:50 +03:00
|
|
|
|
* When the flag `sql` is provided, the methods for implementing the Scanner and Valuer interfaces will be also generated.
|
2016-10-26 19:12:25 +03:00
|
|
|
|
Useful when storing the enum in a database.
|
2015-12-31 17:36:39 +03:00
|
|
|
|
|
|
|
|
|
For example, if we have an enum type called `Pill`,
|
2015-12-31 17:41:41 +03:00
|
|
|
|
```go
|
2015-12-31 17:36:39 +03:00
|
|
|
|
type Pill int
|
2015-12-29 16:27:33 +03:00
|
|
|
|
|
2015-12-31 17:36:39 +03:00
|
|
|
|
const (
|
|
|
|
|
Placebo Pill = iota
|
|
|
|
|
Aspirin
|
|
|
|
|
Ibuprofen
|
|
|
|
|
Paracetamol
|
|
|
|
|
Acetaminophen = Paracetamol
|
|
|
|
|
)
|
|
|
|
|
```
|
2018-03-09 17:44:41 +03:00
|
|
|
|
executing `enumer -type=Pill -json` will generate a new file with four basic method and two extra for JSON:
|
2015-12-31 17:41:41 +03:00
|
|
|
|
```go
|
2018-03-09 17:44:41 +03:00
|
|
|
|
func (i Pill) String() string {
|
|
|
|
|
//...
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func PillString(s string) (Pill, error) {
|
|
|
|
|
//...
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func PillValues() []Pill {
|
|
|
|
|
//...
|
2015-12-29 16:27:33 +03:00
|
|
|
|
}
|
|
|
|
|
|
2018-03-09 17:44:41 +03:00
|
|
|
|
func (i Pill) IsAPill() bool {
|
|
|
|
|
//...
|
2015-12-29 16:27:33 +03:00
|
|
|
|
}
|
2016-01-21 02:09:18 +03:00
|
|
|
|
|
|
|
|
|
func (i Pill) MarshalJSON() ([]byte, error) {
|
|
|
|
|
//...
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (i *Pill) UnmarshalJSON(data []byte) error {
|
|
|
|
|
//...
|
|
|
|
|
}
|
2015-12-29 16:27:33 +03:00
|
|
|
|
```
|
2015-12-31 17:36:39 +03:00
|
|
|
|
From now on, we can:
|
2015-12-31 17:41:41 +03:00
|
|
|
|
```go
|
2015-12-31 17:36:39 +03:00
|
|
|
|
// Convert any Pill value to string
|
|
|
|
|
var aspirinString string = Aspirin.String()
|
|
|
|
|
// (or use it in any place where a Stringer is accepted)
|
|
|
|
|
fmt.Println("I need ", Paracetamol) // Will print "I need Paracetamol"
|
|
|
|
|
|
|
|
|
|
// Convert a string with the enum name to the corresponding enum value
|
|
|
|
|
pill, err := PillString("Ibuprofen")
|
|
|
|
|
if err != nil {
|
|
|
|
|
fmt.Println("Unrecognized pill: ", err)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
// Now pill == Ibuprofen
|
2016-01-21 02:09:18 +03:00
|
|
|
|
|
2018-03-09 17:44:41 +03:00
|
|
|
|
// Get all the values of the string
|
|
|
|
|
allPills := PillValues()
|
|
|
|
|
fmt.Println(allPills) // Will print [Placebo Aspirin Ibuprofen Paracetamol]
|
|
|
|
|
|
|
|
|
|
// Check if a value belongs to the Pill enum values
|
|
|
|
|
var notAPill Pill = 42
|
|
|
|
|
if (notAPill.IsAPill()) {
|
|
|
|
|
fmt.Println(notAPill, "is not a value of the Pill enum")
|
|
|
|
|
}
|
|
|
|
|
|
2016-01-21 02:15:57 +03:00
|
|
|
|
// Marshal/unmarshal to/from json strings, either directly or automatically when
|
|
|
|
|
// the enum is a field of a struct
|
|
|
|
|
pillJSON := Aspirin.MarshalJSON()
|
|
|
|
|
// Now pillJSON == `"Aspirin"`
|
2015-12-31 17:36:39 +03:00
|
|
|
|
```
|
|
|
|
|
|
2016-01-21 02:09:18 +03:00
|
|
|
|
The generated code is exactly the same as the Stringer tool plus the mentioned additions, so you can use
|
2015-12-31 17:36:39 +03:00
|
|
|
|
**Enumer** where you are already using **Stringer** without any code change.
|
2015-12-29 16:27:33 +03:00
|
|
|
|
|
2017-02-09 14:02:18 +03:00
|
|
|
|
## Transforming the string representation of the enum value
|
2017-02-01 11:49:25 +03:00
|
|
|
|
|
2017-02-09 14:02:18 +03:00
|
|
|
|
By default, Enumer uses the same name of the enum value for generating the string representation (usually CamelCase in Go).
|
2017-02-01 11:49:25 +03:00
|
|
|
|
|
|
|
|
|
```go
|
|
|
|
|
type MyType int
|
|
|
|
|
|
|
|
|
|
...
|
|
|
|
|
|
|
|
|
|
name := MyTypeValue.String() // name => "MyTypeValue"
|
|
|
|
|
```
|
|
|
|
|
|
2017-02-09 14:02:18 +03:00
|
|
|
|
Sometimes you need to use some other string representation format than CamelCase (i.e. in JSON).
|
2017-11-03 00:02:53 +03:00
|
|
|
|
|
2017-02-09 14:02:18 +03:00
|
|
|
|
To transform it from CamelCase to snake_case or kebab-case, you can use the `transform` flag.
|
2017-02-01 11:49:25 +03:00
|
|
|
|
|
2017-02-09 14:02:18 +03:00
|
|
|
|
For example, the command `enumer -type=MyType -json -transform=snake` would generate the following string representation:
|
2017-02-01 11:49:25 +03:00
|
|
|
|
|
|
|
|
|
```go
|
|
|
|
|
name := MyTypeValue.String() // name => "my_type_value"
|
|
|
|
|
```
|
2017-11-03 00:02:53 +03:00
|
|
|
|
**Note**: The transformation only works form CamelCase to snake_case or kebab-case, not the other way around.
|
2017-02-01 11:49:25 +03:00
|
|
|
|
|
2015-12-31 17:36:39 +03:00
|
|
|
|
## How to use
|
2016-01-21 02:09:18 +03:00
|
|
|
|
The usage of Enumer is the same as Stringer, so you can refer to the [Stringer docs](https://godoc.org/golang.org/x/tools/cmd/stringer)
|
|
|
|
|
for more information.
|
2016-01-21 02:15:57 +03:00
|
|
|
|
|
2018-03-08 23:08:41 +03:00
|
|
|
|
There are four boolean flags: `json`, `text`, `yaml` and `sql`. You can use any combination of them (i.e. `enumer -type=Pill -json -text`),
|
|
|
|
|
|
2016-01-13 13:50:18 +03:00
|
|
|
|
|
2018-02-06 03:48:43 +03:00
|
|
|
|
For enum string representation transformation the `transform` and `trimprefix` flags
|
2017-04-10 21:37:23 +03:00
|
|
|
|
were added (i.e. `enumer -type=MyType -json -transform=snake`).
|
|
|
|
|
Possible transform values are `snake` and `kebab` for transformation to snake_case and kebab-case accordingly.
|
2017-02-01 11:49:25 +03:00
|
|
|
|
The default value for `transform` flag is `noop` which means no transformation will be performed.
|
|
|
|
|
|
2018-02-06 15:17:13 +03:00
|
|
|
|
If a prefix is provided via the `trimprefix` flag, it will be trimmed from the start of each name (before
|
2017-04-10 21:37:23 +03:00
|
|
|
|
it is transformed). If a name doesn't have the prefix it will be passed unchanged.
|
|
|
|
|
|
2016-01-21 02:18:14 +03:00
|
|
|
|
## Inspiring projects
|
|
|
|
|
* [Stringer](https://godoc.org/golang.org/x/tools/cmd/stringer)
|
|
|
|
|
* [jsonenums](https://github.com/campoy/jsonenums)
|