Implement rename_enum option (Fix #2129)

Author: heyvito

internal/codegen/golang/enum.go

@@ -3,6 +3,8 @@ package golang
 import (
 "strings"
 "unicode"
+
+"github.com/sqlc-dev/sqlc/internal/codegen/golang/opts"
 )
 
 type Constant struct {
@@ -47,9 +49,16 @@ func EnumReplace(value string) string {
 
 // EnumValueName removes all non ident symbols (all but letters, numbers and
 // underscore) and converts snake case ident to camel case.
-func EnumValueName(value string) string {
+func EnumValueName(value string, options *opts.Options) string {
+if rename := options.RenameEnum[value]; rename != "" {
+return rename
+}
 parts := strings.Split(EnumReplace(value), "_")
 for i, part := range parts {
+if _, found := options.InitialismsMap[part]; found {
+parts[i] = strings.ToUpper(part)
+continue
+}
 parts[i] = titleFirst(part)
 }
 

internal/codegen/golang/opts/options.go

@@ -30,6 +30,7 @@ type Options struct {
 Out string `json:"out" yaml:"out"`
 Overrides []Override `json:"overrides,omitempty" yaml:"overrides"`
 Rename map[string]string `json:"rename,omitempty" yaml:"rename"`
+RenameEnum map[string]string `json:"rename_enum,omitempty" yaml:"rename_enum"`
 SqlPackage string `json:"sql_package" yaml:"sql_package"`
 SqlDriver string `json:"sql_driver" yaml:"sql_driver"`
 OutputBatchFileName string `json:"output_batch_file_name,omitempty" yaml:"output_batch_file_name"`
@@ -50,8 +51,9 @@ type Options struct {
 }
 
 type GlobalOptions struct {
-Overrides []Override `json:"overrides,omitempty" yaml:"overrides"`
-Rename map[string]string `json:"rename,omitempty" yaml:"rename"`
+Overrides []Override `json:"overrides,omitempty" yaml:"overrides"`
+Rename map[string]string `json:"rename,omitempty" yaml:"rename"`
+RenameEnum map[string]string `json:"rename_enum,omitempty" yaml:"rename_enum"`
 }
 
 func Parse(req *plugin.GenerateRequest) (*Options, error) {
@@ -72,6 +74,12 @@ func Parse(req *plugin.GenerateRequest) (*Options, error) {
 }
 maps.Copy(options.Rename, global.Rename)
 }
+if len(global.RenameEnum) > 0 {
+if options.RenameEnum == nil {
+options.RenameEnum = map[string]string{}
+}
+maps.Copy(options.RenameEnum, global.RenameEnum)
+}
 return options, nil
 }
 

internal/codegen/golang/result.go

@@ -40,7 +40,7 @@ func buildEnums(req *plugin.GenerateRequest, options *opts.Options) []Enum {
 
 seen := make(map[string]struct{}, len(enum.Vals))
 for i, v := range enum.Vals {
-value := EnumReplace(v)
+value := EnumValueName(v, options)
 if _, found := seen[value]; found || value == "" {
 value = fmt.Sprintf("value_%d", i)
 }

internal/config/v_two.json

@@ -227,6 +227,14 @@
  }
  }
  },
+ "rename_enum": {
+ "type": "object",
+ "patternProperties": {
+ ".*": {
+ "type": "string"
+ }
+ }
+ },
  "sql_package": {
  "type": "string"
  },