snapshot of https://github.com/sqlc-dev/sqlc.git (main, 2026-09-16; blobs > 131072B omitted)

internal/codegen/golang/field.go

@@ -0,0 +1,143 @@
+package golang
+
+import (
+	"fmt"
+	"regexp"
+	"sort"
+	"strings"
+
+	"github.com/sqlc-dev/sqlc/internal/codegen/golang/opts"
+	"github.com/sqlc-dev/sqlc/internal/plugin"
+)
+
+type Field struct {
+	Name    string // CamelCased name for Go
+	DBName  string // Name as used in the DB
+	Type    string
+	Tags    map[string]string
+	Comment string
+	Column  *plugin.Column
+	// EmbedFields contains the embedded fields that require scanning.
+	EmbedFields []Field
+}
+
+func (gf Field) Tag() string {
+	return TagsToString(gf.Tags)
+}
+
+func (gf Field) HasSqlcSlice() bool {
+	return gf.Column.IsSqlcSlice
+}
+
+func TagsToString(tags map[string]string) string {
+	if len(tags) == 0 {
+		return ""
+	}
+	tagParts := make([]string, 0, len(tags))
+	for key, val := range tags {
+		tagParts = append(tagParts, fmt.Sprintf("%s:%q", key, val))
+	}
+	sort.Strings(tagParts)
+	return strings.Join(tagParts, " ")
+}
+
+func JSONTagName(name string, options *opts.Options) string {
+	style := options.JsonTagsCaseStyle
+	idUppercase := options.JsonTagsIdUppercase
+	if style == "" || style == "none" {
+		return name
+	} else {
+		return SetJSONCaseStyle(name, style, idUppercase)
+	}
+}
+
+func SetCaseStyle(name string, style string) string {
+	switch style {
+	case "camel":
+		return toCamelCase(name)
+	case "pascal":
+		return toPascalCase(name)
+	case "snake":
+		return toSnakeCase(name)
+	default:
+		panic(fmt.Sprintf("unsupported JSON tags case style: '%s'", style))
+	}
+}
+
+func SetJSONCaseStyle(name string, style string, idUppercase bool) string {
+	switch style {
+	case "camel":
+		return toJsonCamelCase(name, idUppercase)
+	case "pascal":
+		return toPascalCase(name)
+	case "snake":
+		return toSnakeCase(name)
+	default:
+		panic(fmt.Sprintf("unsupported JSON tags case style: '%s'", style))
+	}
+}
+
+var camelPattern = regexp.MustCompile("[^A-Z][A-Z]+")
+
+func toSnakeCase(s string) string {
+	if !strings.ContainsRune(s, '_') {
+		s = camelPattern.ReplaceAllStringFunc(s, func(x string) string {
+			return x[:1] + "_" + x[1:]
+		})
+	}
+	return strings.ToLower(s)
+}
+
+func toCamelCase(s string) string {
+	return toCamelInitCase(s, false)
+}
+
+func toPascalCase(s string) string {
+	return toCamelInitCase(s, true)
+}
+
+func toCamelInitCase(name string, initUpper bool) string {
+	var out strings.Builder
+	for i, p := range strings.Split(name, "_") {
+		if !initUpper && i == 0 {
+			out.WriteString(p)
+			continue
+		}
+		if p == "id" {
+			out.WriteString("ID")
+		} else {
+			out.WriteString(strings.Title(p))
+		}
+	}
+	return out.String()
+}
+
+func toJsonCamelCase(name string, idUppercase bool) string {
+	var out strings.Builder
+	idStr := "Id"
+
+	if idUppercase {
+		idStr = "ID"
+	}
+
+	for i, p := range strings.Split(name, "_") {
+		if i == 0 {
+			out.WriteString(p)
+			continue
+		}
+		if p == "id" {
+			out.WriteString(idStr)
+		} else {
+			out.WriteString(strings.Title(p))
+		}
+	}
+	return out.String()
+}
+
+func toLowerCase(str string) string {
+	if str == "" {
+		return ""
+	}
+
+	return strings.ToLower(str[:1]) + str[1:]
+}