kubecolor · prune998 · Aug 21, 2024 · Aug 13, 2024
@@ -2,7 +2,12 @@
 
 import (
 	"fmt"
+	"io"
+	"os"
 	"strings"
+	"text/tabwriter"
+	"unicode"
+	"unicode/utf8"
 )
 
 type YAMLPrinter struct {
@@ -16,31 +21,34 @@
 }
 
 func (p *YAMLPrinter) printCategory(category Category, path []string) error {
+	tw := tabwriter.NewWriter(os.Stdout, 0, 0, 1, ' ', 0)
 	for _, field := range category.Fields {
 		newPath := append(path, field.Name)
 		switch field.Type {
 		case "Color":
-			p.printField(field, "color", newPath)
+			p.printField(tw, field, "color", newPath)
 		case "ColorSlice":
-			p.printField(field, "color[]", newPath)
+			p.printField(tw, field, "color[]", newPath)
 		default:
 			sub, ok := p.findCategory(field.Type)
 			if !ok {
 				return fmt.Errorf("invalid category field type: %q", field.Type)
 			}
-			fmt.Printf("%s%s:\n",
+			fmt.Fprintf(tw, "%s%s:\n",
 				strings.Repeat("  ", len(path)),
-				strings.ToLower(field.Name),
+				p.formatName(field),
 			)
+			tw.Flush()
 			if err := p.printCategory(sub, newPath); err != nil {
 				return err
 			}
 		}
 	}
+	tw.Flush()
 	return nil
 }
 
-func (p *YAMLPrinter) printField(field Field, typeString string, path []string) error {
+func (p *YAMLPrinter) printField(w io.Writer, field Field, typeString string, path []string) error {
 	fallback := p.formatFallback(field)
 	desc := strings.ReplaceAll(field.Comment, "\n", " ")
 	if fallback != "" {
@@ -53,7 +61,7 @@
 		desc = " " + desc
 	}
 
-	fmt.Printf("%s%s: %s # (%s)%s\n",
+	fmt.Fprintf(w, "%s%s: %s\t# (%s)%s\n",
 		strings.Repeat("  ", len(path)-1),
 		p.formatName(field),
 		p.viper.GetString(strings.Join(path, ".")),
@@ -77,11 +85,37 @@
 }
 
 func (p *YAMLPrinter) formatName(field Field) string {
-	lower := strings.ToLower(field.Name)
+	lower := camelCase(field.Name)
+
 	switch lower {
 	case "true", "false", "null":
 		return fmt.Sprintf(`"%s"`, lower)
 	default:
 		return lower
 	}
 }
+
+func camelCase(s string) string {
+	if isAllUppercase(s) {
+		return strings.ToLower(s)
+	}
+	return firstLetterLowercase(s)
+}
+
+func firstLetterLowercase(s string) string {
+	var sb strings.Builder
+	sb.Grow(len(s))
+	firstRune, size := utf8.DecodeRuneInString(s)
+	sb.WriteRune(unicode.ToLower(firstRune))
+	sb.WriteString(s[size:])
+	return sb.String()
+}
+
+func isAllUppercase(s string) bool {
+	for _, r := range s {
+		if unicode.IsLower(r) {
+			return false
+		}
+	}
+	return true
+}