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"
                                 },