enum.go

package golang

import (
	"strings"
	"unicode"

	"github.com/sqlc-dev/sqlc/internal/codegen/golang/opts"
)

type Constant struct {
	Name  string
	Type  string
	Value string
}

type Enum struct {
	Name      string
	Comment   string
	Constants []Constant
	NameTags  map[string]string
	ValidTags map[string]string
}

func (e Enum) NameTag() string {
	return TagsToString(e.NameTags)
}

func (e Enum) ValidTag() string {
	return TagsToString(e.ValidTags)
}

func enumReplacer(r rune) rune {
	if strings.ContainsRune("-/:_", r) {
		return '_'
	} else if (r >= 'a' && r <= 'z') ||
		(r >= 'A' && r <= 'Z') ||
		(r >= '0' && r <= '9') {
		return r
	} else {
		return -1
	}
}

// EnumReplace removes all non ident symbols (all but letters, numbers and
// underscore) and returns valid ident name for provided name.
func EnumReplace(value string) string {
	return strings.Map(enumReplacer, value)
}

// EnumValueName removes all non ident symbols (all but letters, numbers and
// underscore) and converts snake case ident to camel case.
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)
	}

	return strings.Join(parts, "")
}

func titleFirst(s string) string {
	if len(s) == 0 {
		return s
	}

	r := []rune(s)
	r[0] = unicode.ToUpper(r[0])

	return string(r)
}