feat: add gen enum command

2024-12-19 15:00:58 +08:00
parent 797ff49fc7
commit 80ab62534d
15 changed files with 2674 additions and 5 deletions
--- a/pkg/utils/generator/template_funcs.go
+++ b/pkg/utils/generator/template_funcs.go
@@ -0,0 +1,126 @@
+package generator
+
+import (
+	"fmt"
+	"strconv"
+	"strings"
+)
+
+// Stringify returns a string that is all of the enum value names concatenated without a separator
+func Stringify(e Enum, forceLower bool) (ret string, err error) {
+	for _, val := range e.Values {
+		if val.Name != skipHolder {
+			next := val.RawName
+			if forceLower {
+				next = strings.ToLower(next)
+			}
+			ret = ret + next
+		}
+	}
+	return
+}
+
+// Mapify returns a map that is all of the indexes for a string value lookup
+func Mapify(e Enum) (ret string, err error) {
+	strName := fmt.Sprintf(`_%sName`, e.Name)
+	ret = fmt.Sprintf("map[%s]string{\n", e.Name)
+	index := 0
+	for _, val := range e.Values {
+		if val.Name != skipHolder {
+			nextIndex := index + len(val.Name)
+			ret = fmt.Sprintf("%s%s: %s[%d:%d],\n", ret, val.PrefixedName, strName, index, nextIndex)
+			index = nextIndex
+		}
+	}
+	ret = ret + `}`
+	return
+}
+
+// Unmapify returns a map that is all of the indexes for a string value lookup
+func Unmapify(e Enum, lowercase bool) (ret string, err error) {
+	if e.Type == "string" {
+		return UnmapifyStringEnum(e, lowercase)
+	}
+	strName := fmt.Sprintf(`_%sName`, e.Name)
+	ret = fmt.Sprintf("map[string]%s{\n", e.Name)
+	index := 0
+	for _, val := range e.Values {
+		if val.Name != skipHolder {
+			nextIndex := index + len(val.Name)
+			ret = fmt.Sprintf("%s%s[%d:%d]: %s,\n", ret, strName, index, nextIndex, val.PrefixedName)
+			if lowercase {
+				ret = fmt.Sprintf("%sstrings.ToLower(%s[%d:%d]): %s,\n", ret, strName, index, nextIndex, val.PrefixedName)
+			}
+			index = nextIndex
+		}
+	}
+	ret = ret + `}`
+	return
+}
+
+// Unmapify returns a map that is all of the indexes for a string value lookup
+func UnmapifyStringEnum(e Enum, lowercase bool) (ret string, err error) {
+	var builder strings.Builder
+	_, err = builder.WriteString("map[string]" + e.Name + "{\n")
+	if err != nil {
+		return
+	}
+	for _, val := range e.Values {
+		if val.Name != skipHolder {
+			_, err = builder.WriteString(fmt.Sprintf("%q:%s,\n", val.ValueStr, val.PrefixedName))
+			if err != nil {
+				return
+			}
+			if lowercase && strings.ToLower(val.ValueStr) != val.ValueStr {
+				_, err = builder.WriteString(fmt.Sprintf("%q:%s,\n", strings.ToLower(val.ValueStr), val.PrefixedName))
+				if err != nil {
+					return
+				}
+			}
+		}
+	}
+	builder.WriteByte('}')
+	ret = builder.String()
+	return
+}
+
+// Namify returns a slice that is all of the possible names for an enum in a slice
+func Namify(e Enum) (ret string, err error) {
+	if e.Type == "string" {
+		return namifyStringEnum(e)
+	}
+	strName := fmt.Sprintf(`_%sName`, e.Name)
+	ret = "[]string{\n"
+	index := 0
+	for _, val := range e.Values {
+		if val.Name != skipHolder {
+			nextIndex := index + len(val.Name)
+			ret = fmt.Sprintf("%s%s[%d:%d],\n", ret, strName, index, nextIndex)
+			index = nextIndex
+		}
+	}
+	ret = ret + "}"
+	return
+}
+
+// Namify returns a slice that is all of the possible names for an enum in a slice
+func namifyStringEnum(e Enum) (ret string, err error) {
+	ret = "[]string{\n"
+	for _, val := range e.Values {
+		if val.Name != skipHolder {
+			ret = fmt.Sprintf("%sstring(%s),\n", ret, val.PrefixedName)
+		}
+	}
+	ret = ret + "}"
+	return
+}
+
+func Offset(index int, enumType string, val EnumValue) (strResult string) {
+	if strings.HasPrefix(enumType, "u") {
+		// Unsigned
+		return strconv.FormatUint(val.ValueInt.(uint64)-uint64(index), 10)
+	} else {
+		// Signed
+		return strconv.FormatInt(val.ValueInt.(int64)-int64(index), 10)
+	}
+}