enum.go.tmpl

{{- define "enum" -}}

{{- $enumName := .Name -}}
{{- $enumType := .EnumType -}}

type {{$enumName}} {{$enumType}}

const (
{{- range .Fields}}
	{{$enumName}}_{{.Name}} {{$enumName}} = {{.Value}}
{{- end}}
)

var {{$enumName}}_name = map[{{$enumType}}]string{
{{- range .Fields}}
	{{.Value}}: "{{.Name}}",
{{- end}}
}

var {{$enumName}}_value = map[string]{{$enumType}}{
{{- range .Fields}}
	"{{.Name}}": {{.Value}},
{{- end}}
}

func (x {{$enumName}}) String() string {
	return {{$enumName}}_name[{{$enumType}}(x)]
}

func (x {{$enumName}}) MarshalJSON() ([]byte, error) {
	buf := bytes.NewBufferString(`"`)
	buf.WriteString({{$enumName}}_name[{{$enumType}}(x)])
	buf.WriteString(`"`)
	return buf.Bytes(), nil
}

func (x *{{$enumName}}) UnmarshalJSON(b []byte) error {
	var j string
	err := json.Unmarshal(b, &j)
	if err != nil {
		return err
	}
	*x = {{$enumName}}({{$enumName}}_value[j])
	return nil
}

{{- end -}}