forked from mirror/enumer
Compare commits
No commits in common. "master" and "dan/apple-m1" have entirely different histories.
master
...
dan/apple-
|
@ -4,8 +4,9 @@ matrix:
|
||||||
allow_failures:
|
allow_failures:
|
||||||
- go: master
|
- go: master
|
||||||
include:
|
include:
|
||||||
- go: 1.17.x
|
- go: 1.15.x
|
||||||
- go: 1.18.x
|
- go: 1.16.x
|
||||||
|
- go: master
|
||||||
env:
|
env:
|
||||||
global:
|
global:
|
||||||
- GO111MODULE=on
|
- GO111MODULE=on
|
||||||
|
@ -20,10 +21,9 @@ deploy:
|
||||||
file:
|
file:
|
||||||
- cli/build/enumer.linux-amd64.tar.gz
|
- cli/build/enumer.linux-amd64.tar.gz
|
||||||
- cli/build/enumer.darwin-amd64.tar.gz
|
- cli/build/enumer.darwin-amd64.tar.gz
|
||||||
- cli/build/enumer.darwin-arm64.tar.gz
|
|
||||||
- cli/build/enumer.windows-amd64.exe.tar.gz
|
- cli/build/enumer.windows-amd64.exe.tar.gz
|
||||||
- cli/build/sha256sum.txt
|
- cli/build/sha256sum.txt
|
||||||
on:
|
on:
|
||||||
go: 1.17.x
|
go: 1.16.x
|
||||||
repo: dmarkham/enumer
|
repo: dmarkham/enumer
|
||||||
tags: true
|
tags: true
|
||||||
|
|
16
README.md
16
README.md
|
@ -1,10 +1,10 @@
|
||||||
# Enumer [![GoDoc](https://godoc.org/git.internal/re/enumer?status.svg)](https://godoc.org/git.internal/re/enumer) [![Go Report Card](https://goreportcard.com/badge/git.internal/re/enumer)](https://goreportcard.com/report/git.internal/re/enumer) [![GitHub Release](https://img.shields.io/github/release/dmarkham/enumer.svg)](https://git.internal/re/enumer/releases)[![Build Status](https://travis-ci.com/dmarkham/enumer.svg?branch=master)](https://travis-ci.com/dmarkham/enumer)
|
# Enumer [![GoDoc](https://godoc.org/github.com/dmarkham/enumer?status.svg)](https://godoc.org/github.com/dmarkham/enumer) [![Go Report Card](https://goreportcard.com/badge/github.com/dmarkham/enumer)](https://goreportcard.com/report/github.com/dmarkham/enumer) [![GitHub Release](https://img.shields.io/github/release/dmarkham/enumer.svg)](https://github.com/dmarkham/enumer/releases)[![Build Status](https://travis-ci.com/dmarkham/enumer.svg?branch=master)](https://travis-ci.com/dmarkham/enumer)
|
||||||
|
|
||||||
|
|
||||||
Enumer is a tool to generate Go code that adds useful methods to Go enums (constants with a specific type).
|
Enumer is a tool to generate Go code that adds useful methods to Go enums (constants with a specific type).
|
||||||
It started as a fork of [Rob Pike’s Stringer tool](https://godoc.org/golang.org/x/tools/cmd/stringer)
|
It started as a fork of [Rob Pike’s Stringer tool](https://godoc.org/golang.org/x/tools/cmd/stringer)
|
||||||
maintained by [Álvaro López Espinosa](https://github.com/alvaroloes/enumer).
|
maintained by [Álvaro López Espinosa](https://github.com/alvaroloes/enumer).
|
||||||
This was again forked here as (https://git.internal/re/enumer) picking up where Álvaro left off.
|
This was again forked here as (https://github.com/dmarkham/enumer) picking up where Álvaro left off.
|
||||||
|
|
||||||
|
|
||||||
```
|
```
|
||||||
|
@ -14,7 +14,7 @@ Usage of ./enumer:
|
||||||
Enumer [flags] -type T [directory]
|
Enumer [flags] -type T [directory]
|
||||||
Enumer [flags] -type T files... # Must be a single package
|
Enumer [flags] -type T files... # Must be a single package
|
||||||
For more information, see:
|
For more information, see:
|
||||||
http://godoc.org/git.internal/re/enumer
|
http://godoc.org/github.com/dmarkham/enumer
|
||||||
Flags:
|
Flags:
|
||||||
-addprefix string
|
-addprefix string
|
||||||
transform each item name by adding a prefix. Default: ""
|
transform each item name by adding a prefix. Default: ""
|
||||||
|
@ -38,8 +38,6 @@ Flags:
|
||||||
transform each item name by removing a prefix. Default: ""
|
transform each item name by removing a prefix. Default: ""
|
||||||
-type string
|
-type string
|
||||||
comma-separated list of type names; must be set
|
comma-separated list of type names; must be set
|
||||||
-values
|
|
||||||
if true, alternative string values method will be generated. Default: false
|
|
||||||
-yaml
|
-yaml
|
||||||
if true, yaml marshaling methods will be generated. Default: false
|
if true, yaml marshaling methods will be generated. Default: false
|
||||||
```
|
```
|
||||||
|
@ -195,12 +193,6 @@ name := MyTypeValue.String() // name => "my_type_value"
|
||||||
|
|
||||||
## How to use
|
## How to use
|
||||||
|
|
||||||
For a module-aware repo with `enumer` in the `go.mod` file, generation can be called by adding the following to a `.go` source file:
|
|
||||||
|
|
||||||
```golang
|
|
||||||
//go:generate go run git.internal/re/enumer -type=YOURTYPE
|
|
||||||
```
|
|
||||||
|
|
||||||
There are four boolean flags: `json`, `text`, `yaml` and `sql`. You can use any combination of them (i.e. `enumer -type=Pill -json -text`),
|
There are four boolean flags: `json`, `text`, `yaml` and `sql`. You can use any combination of them (i.e. `enumer -type=Pill -json -text`),
|
||||||
|
|
||||||
For enum string representation transformation the `transform` and `trimprefix` flags
|
For enum string representation transformation the `transform` and `trimprefix` flags
|
||||||
|
@ -213,8 +205,6 @@ it is transformed). If a name doesn't have the prefix it will be passed unchange
|
||||||
|
|
||||||
If a prefix is provided via the `addprefix` flag, it will be added to the start of each name (after trimming and after transforming).
|
If a prefix is provided via the `addprefix` flag, it will be added to the start of each name (after trimming and after transforming).
|
||||||
|
|
||||||
The boolean flag `values` will additionally create an alternative string values method `Values() []string` to fullfill the `EnumValues` interface of [ent](https://entgo.io/docs/schema-fields/#enum-fields).
|
|
||||||
|
|
||||||
## Inspiring projects
|
## Inspiring projects
|
||||||
|
|
||||||
- [Álvaro López Espinosa](https://github.com/alvaroloes/enumer)
|
- [Álvaro López Espinosa](https://github.com/alvaroloes/enumer)
|
||||||
|
|
12
enumer.go
12
enumer.go
|
@ -58,18 +58,6 @@ func (i %[1]s) IsA%[1]s() bool {
|
||||||
}
|
}
|
||||||
`
|
`
|
||||||
|
|
||||||
// Arguments to format are:
|
|
||||||
// [1]: type name
|
|
||||||
const altStringValuesMethod = `func (%[1]s) Values() []string {
|
|
||||||
return %[1]sStrings()
|
|
||||||
}
|
|
||||||
`
|
|
||||||
|
|
||||||
func (g *Generator) buildAltStringValuesMethod(typeName string) {
|
|
||||||
g.Printf("\n")
|
|
||||||
g.Printf(altStringValuesMethod, typeName)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (g *Generator) buildBasicExtras(runs [][]Value, typeName string, runsThreshold int) {
|
func (g *Generator) buildBasicExtras(runs [][]Value, typeName string, runsThreshold int) {
|
||||||
// At this moment, either "g.declareIndexAndNameVars()" or "g.declareNameVars()" has been called
|
// At this moment, either "g.declareIndexAndNameVars()" or "g.declareNameVars()" has been called
|
||||||
|
|
||||||
|
|
|
@ -2,7 +2,7 @@
|
||||||
|
|
||||||
## Steps
|
## Steps
|
||||||
|
|
||||||
1. Go get enumer `go get -u git.internal/re/enumer`
|
1. Go get enumer `go get -u github.com/dmarkham/enumer`
|
||||||
2. `go generate` This should create `pill_enumer.go`
|
2. `go generate` This should create `pill_enumer.go`
|
||||||
3. `go run *.go` to see it in action
|
3. `go run *.go` to see it in action
|
||||||
4. `go mod tidy` to remove the deps for `enumer` once your happy.
|
4. `go mod tidy` to remove the deps for `enumer` once your happy.
|
||||||
|
|
|
@ -2,7 +2,7 @@ package main
|
||||||
|
|
||||||
import "fmt"
|
import "fmt"
|
||||||
|
|
||||||
//go:generate go run git.internal/re/enumer -type=Pill -json
|
//go:generate go run github.com/dmarkham/enumer -type=Pill -json
|
||||||
type Pill int
|
type Pill int
|
||||||
|
|
||||||
const (
|
const (
|
||||||
|
|
11
go.mod
11
go.mod
|
@ -1,13 +1,8 @@
|
||||||
module git.internal/re/enumer
|
module github.com/dmarkham/enumer
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/pascaldekloe/name v1.0.0
|
github.com/pascaldekloe/name v1.0.0
|
||||||
golang.org/x/tools v0.1.12
|
golang.org/x/tools v0.0.0-20200221224223-e1da425f72fd
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
go 1.14
|
||||||
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4 // indirect
|
|
||||||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f // indirect
|
|
||||||
)
|
|
||||||
|
|
||||||
go 1.17
|
|
||||||
|
|
30
go.sum
30
go.sum
|
@ -1,28 +1,16 @@
|
||||||
github.com/pascaldekloe/name v1.0.0 h1:n7LKFgHixETzxpRv2R77YgPUFo85QHGZKrdaYm7eY5U=
|
github.com/pascaldekloe/name v1.0.0 h1:n7LKFgHixETzxpRv2R77YgPUFo85QHGZKrdaYm7eY5U=
|
||||||
github.com/pascaldekloe/name v1.0.0/go.mod h1:Z//MfYJnH4jVpQ9wkclwu2I2MkHmXTlT9wR5UZScttM=
|
github.com/pascaldekloe/name v1.0.0/go.mod h1:Z//MfYJnH4jVpQ9wkclwu2I2MkHmXTlT9wR5UZScttM=
|
||||||
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
|
|
||||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||||
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
|
||||||
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4 h1:6zppjxzCulZykYSLyVDYbneBfbaBIQPYMevg0bEwv2s=
|
golang.org/x/mod v0.1.1-0.20191105210325-c90efee705ee h1:WG0RUwxtNT4qqaXX3DPA8zHFNm/D9xaBpxzHt1WcA/E=
|
||||||
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
golang.org/x/mod v0.1.1-0.20191105210325-c90efee705ee/go.mod h1:QqPTAvyqsEbceGzBzNggFXnrqF1CaUcvgkdR5Ot7KZg=
|
||||||
|
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||||
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
|
||||||
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
|
|
||||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
|
||||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
|
||||||
golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
|
||||||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f h1:v4INt8xihDGvnrfjMDVXGxw9wrfxYyCjk0KbXjhR55s=
|
|
||||||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
|
||||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
|
||||||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
|
||||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||||
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
golang.org/x/tools v0.0.0-20200221224223-e1da425f72fd h1:hHkvGJK23seRCflePJnVa9IMv8fsuavSCWKd11kDQFs=
|
||||||
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
golang.org/x/tools v0.0.0-20200221224223-e1da425f72fd/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=
|
||||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898 h1:/atklqdjdhuosWIl6AIbOeHJjicWYPqR9bpxqxYG2pA=
|
||||||
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
golang.org/x/tools v0.1.12 h1:VveCTK38A2rkS8ZqFY25HIDFscX5X9OoEhJd3quQmXU=
|
|
||||||
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
|
||||||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
|
||||||
|
|
|
@ -315,45 +315,45 @@ const (
|
||||||
|
|
||||||
func TestGolden(t *testing.T) {
|
func TestGolden(t *testing.T) {
|
||||||
for _, test := range golden {
|
for _, test := range golden {
|
||||||
runGoldenTest(t, test, false, false, false, false, false, false, true, "", "")
|
runGoldenTest(t, test, false, false, false, false, false, false, "", "")
|
||||||
}
|
}
|
||||||
for _, test := range goldenJSON {
|
for _, test := range goldenJSON {
|
||||||
runGoldenTest(t, test, true, false, false, false, false, false, false, "", "")
|
runGoldenTest(t, test, true, false, false, false, false, false, "", "")
|
||||||
}
|
}
|
||||||
for _, test := range goldenText {
|
for _, test := range goldenText {
|
||||||
runGoldenTest(t, test, false, false, false, true, false, false, false, "", "")
|
runGoldenTest(t, test, false, false, false, true, false, false, "", "")
|
||||||
}
|
}
|
||||||
for _, test := range goldenYAML {
|
for _, test := range goldenYAML {
|
||||||
runGoldenTest(t, test, false, true, false, false, false, false, false, "", "")
|
runGoldenTest(t, test, false, true, false, false, false, false, "", "")
|
||||||
}
|
}
|
||||||
for _, test := range goldenSQL {
|
for _, test := range goldenSQL {
|
||||||
runGoldenTest(t, test, false, false, true, false, false, false, false, "", "")
|
runGoldenTest(t, test, false, false, true, false, false, false, "", "")
|
||||||
}
|
}
|
||||||
for _, test := range goldenJSONAndSQL {
|
for _, test := range goldenJSONAndSQL {
|
||||||
runGoldenTest(t, test, true, false, true, false, false, false, false, "", "")
|
runGoldenTest(t, test, true, false, true, false, false, false, "", "")
|
||||||
}
|
}
|
||||||
for _, test := range goldenGQLGen {
|
for _, test := range goldenGQLGen {
|
||||||
runGoldenTest(t, test, false, false, false, false, false, true, false, "", "")
|
runGoldenTest(t, test, false, false, false, false, false, true, "", "")
|
||||||
}
|
}
|
||||||
for _, test := range goldenTrimPrefix {
|
for _, test := range goldenTrimPrefix {
|
||||||
runGoldenTest(t, test, false, false, false, false, false, false, false, "Day", "")
|
runGoldenTest(t, test, false, false, false, false, false, false, "Day", "")
|
||||||
}
|
}
|
||||||
for _, test := range goldenTrimPrefixMultiple {
|
for _, test := range goldenTrimPrefixMultiple {
|
||||||
runGoldenTest(t, test, false, false, false, false, false, false, false, "Day,Night", "")
|
runGoldenTest(t, test, false, false, false, false, false, false, "Day,Night", "")
|
||||||
}
|
}
|
||||||
for _, test := range goldenWithPrefix {
|
for _, test := range goldenWithPrefix {
|
||||||
runGoldenTest(t, test, false, false, false, false, false, false, false, "", "Day")
|
runGoldenTest(t, test, false, false, false, false, false, false, "", "Day")
|
||||||
}
|
}
|
||||||
for _, test := range goldenTrimAndAddPrefix {
|
for _, test := range goldenTrimAndAddPrefix {
|
||||||
runGoldenTest(t, test, false, false, false, false, false, false, false, "Day", "Night")
|
runGoldenTest(t, test, false, false, false, false, false, false, "Day", "Night")
|
||||||
}
|
}
|
||||||
for _, test := range goldenLinecomment {
|
for _, test := range goldenLinecomment {
|
||||||
runGoldenTest(t, test, false, false, false, false, true, false, false, "", "")
|
runGoldenTest(t, test, false, false, false, false, true, false, "", "")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func runGoldenTest(t *testing.T, test Golden,
|
func runGoldenTest(t *testing.T, test Golden,
|
||||||
generateJSON, generateYAML, generateSQL, generateText, linecomment, generateGQLGen, generateValuesMethod bool,
|
generateJSON, generateYAML, generateSQL, generateText, linecomment, generateGQLGen bool,
|
||||||
trimPrefix string, prefix string) {
|
trimPrefix string, prefix string) {
|
||||||
|
|
||||||
var g Generator
|
var g Generator
|
||||||
|
@ -382,7 +382,7 @@ func runGoldenTest(t *testing.T, test Golden,
|
||||||
if len(tokens) != 3 {
|
if len(tokens) != 3 {
|
||||||
t.Fatalf("%s: need type declaration on first line", test.name)
|
t.Fatalf("%s: need type declaration on first line", test.name)
|
||||||
}
|
}
|
||||||
g.generate(tokens[1], generateJSON, generateYAML, generateSQL, generateText, generateGQLGen, "noop", trimPrefix, prefix, linecomment, generateValuesMethod)
|
g.generate(tokens[1], generateJSON, generateYAML, generateSQL, generateText, generateGQLGen, "noop", trimPrefix, prefix, linecomment)
|
||||||
got := string(g.format())
|
got := string(g.format())
|
||||||
if got != loadGolden(test.name) {
|
if got != loadGolden(test.name) {
|
||||||
// Use this to help build a golden text when changes are needed
|
// Use this to help build a golden text when changes are needed
|
||||||
|
|
15
stringer.go
15
stringer.go
|
@ -5,7 +5,7 @@
|
||||||
// Enumer is a tool to generate Go code that adds useful methods to Go enums (constants with a specific type).
|
// Enumer is a tool to generate Go code that adds useful methods to Go enums (constants with a specific type).
|
||||||
// It started as a fork of Rob Pike’s Stringer tool
|
// It started as a fork of Rob Pike’s Stringer tool
|
||||||
//
|
//
|
||||||
// Please visit http://git.internal/re/enumer for a comprehensive documentation
|
// Please visit http://github.com/dmarkham/enumer for a comprehensive documentation
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
@ -50,7 +50,6 @@ var (
|
||||||
yaml = flag.Bool("yaml", false, "if true, yaml marshaling methods will be generated. Default: false")
|
yaml = flag.Bool("yaml", false, "if true, yaml marshaling methods will be generated. Default: false")
|
||||||
text = flag.Bool("text", false, "if true, text marshaling methods will be generated. Default: false")
|
text = flag.Bool("text", false, "if true, text marshaling methods will be generated. Default: false")
|
||||||
gqlgen = flag.Bool("gqlgen", false, "if true, GraphQL marshaling methods for gqlgen will be generated. Default: false")
|
gqlgen = flag.Bool("gqlgen", false, "if true, GraphQL marshaling methods for gqlgen will be generated. Default: false")
|
||||||
altValuesFunc = flag.Bool("values", false, "if true, alternative string values method will be generated. Default: false")
|
|
||||||
output = flag.String("output", "", "output file name; default srcdir/<type>_string.go")
|
output = flag.String("output", "", "output file name; default srcdir/<type>_string.go")
|
||||||
transformMethod = flag.String("transform", "noop", "enum item name transformation method. Default: noop")
|
transformMethod = flag.String("transform", "noop", "enum item name transformation method. Default: noop")
|
||||||
trimPrefix = flag.String("trimprefix", "", "transform each item name by removing a prefix. Default: \"\"")
|
trimPrefix = flag.String("trimprefix", "", "transform each item name by removing a prefix. Default: \"\"")
|
||||||
|
@ -71,7 +70,7 @@ func Usage() {
|
||||||
_, _ = fmt.Fprintf(os.Stderr, "\tEnumer [flags] -type T [directory]\n")
|
_, _ = fmt.Fprintf(os.Stderr, "\tEnumer [flags] -type T [directory]\n")
|
||||||
_, _ = fmt.Fprintf(os.Stderr, "\tEnumer [flags] -type T files... # Must be a single package\n")
|
_, _ = fmt.Fprintf(os.Stderr, "\tEnumer [flags] -type T files... # Must be a single package\n")
|
||||||
_, _ = fmt.Fprintf(os.Stderr, "For more information, see:\n")
|
_, _ = fmt.Fprintf(os.Stderr, "For more information, see:\n")
|
||||||
_, _ = fmt.Fprintf(os.Stderr, "\thttp://godoc.org/git.internal/re/enumer\n")
|
_, _ = fmt.Fprintf(os.Stderr, "\thttp://godoc.org/github.com/dmarkham/enumer\n")
|
||||||
_, _ = fmt.Fprintf(os.Stderr, "Flags:\n")
|
_, _ = fmt.Fprintf(os.Stderr, "Flags:\n")
|
||||||
flag.PrintDefaults()
|
flag.PrintDefaults()
|
||||||
}
|
}
|
||||||
|
@ -135,7 +134,7 @@ func main() {
|
||||||
|
|
||||||
// Run generate for each type.
|
// Run generate for each type.
|
||||||
for _, typeName := range typs {
|
for _, typeName := range typs {
|
||||||
g.generate(typeName, *json, *yaml, *sql, *text, *gqlgen, *transformMethod, *trimPrefix, *addPrefix, *linecomment, *altValuesFunc)
|
g.generate(typeName, *json, *yaml, *sql, *text, *gqlgen, *transformMethod, *trimPrefix, *addPrefix, *linecomment)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Format the output.
|
// Format the output.
|
||||||
|
@ -415,7 +414,7 @@ func (g *Generator) prefixValueNames(values []Value, prefix string) {
|
||||||
// generate produces the String method for the named type.
|
// generate produces the String method for the named type.
|
||||||
func (g *Generator) generate(typeName string,
|
func (g *Generator) generate(typeName string,
|
||||||
includeJSON, includeYAML, includeSQL, includeText, includeGQLGen bool,
|
includeJSON, includeYAML, includeSQL, includeText, includeGQLGen bool,
|
||||||
transformMethod string, trimPrefix string, addPrefix string, lineComment bool, includeValuesMethod bool) {
|
transformMethod string, trimPrefix string, addPrefix string, lineComment bool) {
|
||||||
values := make([]Value, 0, 100)
|
values := make([]Value, 0, 100)
|
||||||
for _, file := range g.pkg.files {
|
for _, file := range g.pkg.files {
|
||||||
file.lineComment = lineComment
|
file.lineComment = lineComment
|
||||||
|
@ -462,10 +461,6 @@ func (g *Generator) generate(typeName string,
|
||||||
default:
|
default:
|
||||||
g.buildMap(runs, typeName)
|
g.buildMap(runs, typeName)
|
||||||
}
|
}
|
||||||
if includeValuesMethod {
|
|
||||||
g.buildAltStringValuesMethod(typeName)
|
|
||||||
}
|
|
||||||
|
|
||||||
g.buildNoOpOrderChangeDetect(runs, typeName)
|
g.buildNoOpOrderChangeDetect(runs, typeName)
|
||||||
|
|
||||||
g.buildBasicExtras(runs, typeName, runsThreshold)
|
g.buildBasicExtras(runs, typeName, runsThreshold)
|
||||||
|
@ -790,6 +785,8 @@ const stringOneRun = `func (i %[1]s) String() string {
|
||||||
// [2]: lowest defined value for type, as a string
|
// [2]: lowest defined value for type, as a string
|
||||||
// [3]: size of index element (8 for uint8 etc.)
|
// [3]: size of index element (8 for uint8 etc.)
|
||||||
// [4]: less than zero check (for signed types)
|
// [4]: less than zero check (for signed types)
|
||||||
|
/*
|
||||||
|
*/
|
||||||
const stringOneRunWithOffset = `func (i %[1]s) String() string {
|
const stringOneRunWithOffset = `func (i %[1]s) String() string {
|
||||||
i -= %[2]s
|
i -= %[2]s
|
||||||
if %[4]si >= %[1]s(len(_%[1]sIndex)-1) {
|
if %[4]si >= %[1]s(len(_%[1]sIndex)-1) {
|
||||||
|
|
|
@ -12,10 +12,6 @@ func (i Day) String() string {
|
||||||
return _DayName[_DayIndex[i]:_DayIndex[i+1]]
|
return _DayName[_DayIndex[i]:_DayIndex[i+1]]
|
||||||
}
|
}
|
||||||
|
|
||||||
func (Day) Values() []string {
|
|
||||||
return DayStrings()
|
|
||||||
}
|
|
||||||
|
|
||||||
// An "invalid array index" compiler error signifies that the constant values have changed.
|
// An "invalid array index" compiler error signifies that the constant values have changed.
|
||||||
// Re-run the stringer command to generate them again.
|
// Re-run the stringer command to generate them again.
|
||||||
func _DayNoOp() {
|
func _DayNoOp() {
|
||||||
|
|
|
@ -29,10 +29,6 @@ func (i Gap) String() string {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (Gap) Values() []string {
|
|
||||||
return GapStrings()
|
|
||||||
}
|
|
||||||
|
|
||||||
// An "invalid array index" compiler error signifies that the constant values have changed.
|
// An "invalid array index" compiler error signifies that the constant values have changed.
|
||||||
// Re-run the stringer command to generate them again.
|
// Re-run the stringer command to generate them again.
|
||||||
func _GapNoOp() {
|
func _GapNoOp() {
|
||||||
|
|
|
@ -13,10 +13,6 @@ func (i Num) String() string {
|
||||||
return _NumName[_NumIndex[i]:_NumIndex[i+1]]
|
return _NumName[_NumIndex[i]:_NumIndex[i+1]]
|
||||||
}
|
}
|
||||||
|
|
||||||
func (Num) Values() []string {
|
|
||||||
return NumStrings()
|
|
||||||
}
|
|
||||||
|
|
||||||
// An "invalid array index" compiler error signifies that the constant values have changed.
|
// An "invalid array index" compiler error signifies that the constant values have changed.
|
||||||
// Re-run the stringer command to generate them again.
|
// Re-run the stringer command to generate them again.
|
||||||
func _NumNoOp() {
|
func _NumNoOp() {
|
||||||
|
|
|
@ -13,10 +13,6 @@ func (i Number) String() string {
|
||||||
return _NumberName[_NumberIndex[i]:_NumberIndex[i+1]]
|
return _NumberName[_NumberIndex[i]:_NumberIndex[i+1]]
|
||||||
}
|
}
|
||||||
|
|
||||||
func (Number) Values() []string {
|
|
||||||
return NumberStrings()
|
|
||||||
}
|
|
||||||
|
|
||||||
// An "invalid array index" compiler error signifies that the constant values have changed.
|
// An "invalid array index" compiler error signifies that the constant values have changed.
|
||||||
// Re-run the stringer command to generate them again.
|
// Re-run the stringer command to generate them again.
|
||||||
func _NumberNoOp() {
|
func _NumberNoOp() {
|
||||||
|
|
|
@ -25,10 +25,6 @@ func (i Prime) String() string {
|
||||||
return fmt.Sprintf("Prime(%d)", i)
|
return fmt.Sprintf("Prime(%d)", i)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (Prime) Values() []string {
|
|
||||||
return PrimeStrings()
|
|
||||||
}
|
|
||||||
|
|
||||||
// An "invalid array index" compiler error signifies that the constant values have changed.
|
// An "invalid array index" compiler error signifies that the constant values have changed.
|
||||||
// Re-run the stringer command to generate them again.
|
// Re-run the stringer command to generate them again.
|
||||||
func _PrimeNoOp() {
|
func _PrimeNoOp() {
|
||||||
|
|
|
@ -23,10 +23,6 @@ func (i Unum) String() string {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (Unum) Values() []string {
|
|
||||||
return UnumStrings()
|
|
||||||
}
|
|
||||||
|
|
||||||
// An "invalid array index" compiler error signifies that the constant values have changed.
|
// An "invalid array index" compiler error signifies that the constant values have changed.
|
||||||
// Re-run the stringer command to generate them again.
|
// Re-run the stringer command to generate them again.
|
||||||
func _UnumNoOp() {
|
func _UnumNoOp() {
|
||||||
|
|
Loading…
Reference in New Issue