Add text-transform attribute to the style block

This provides an option to override the functionality provided by the
theme, or apply `text-transform` outside of the theme.

The functionality of the `text-transform` is as below:

- `text-transform: none` - will disable **any** transformation (like the uppercasing by `terminal` theme)
- `text-transform: uppercase` (uppercase not upper as per your message) - will force all characters into uppercase.
- `text-transform: lowercase` - will force all characters into lowercase.
- `text-transform: capitalize` - will uppercase the first letter of every word

In addition, this commit introduces:
- helper methods on the `d2graph.Style` struct to determine the type of
  `text-transform` to be applied.
- `ApplyTextTransform` method on the `d2graph.Attributes` which will
  transform the `Label.Value` to the correct text case.

d2compiler/compile.go

@@ -439,6 +439,8 @@ func compileStyleFieldInit(attrs *d2graph.Attributes, f *d2ir.Field) {
 		attrs.Left = &d2graph.Scalar{MapKey: f.LastPrimaryKey()}
 	case "double-border":
 		attrs.Style.DoubleBorder = &d2graph.Scalar{MapKey: f.LastPrimaryKey()}
+	case "text-transform":
+		attrs.Style.TextTransform = &d2graph.Scalar{MapKey: f.LastPrimaryKey()}
 	}
 }
 

d2graph/d2graph.go

@@ -9,6 +9,8 @@ import (
 	"strconv"
 	"strings"
 
+	"golang.org/x/text/cases"
+	"golang.org/x/text/language"
 	"oss.terrastruct.com/util-go/go2"
 
 	"oss.terrastruct.com/d2/d2ast"
@@ -131,6 +133,22 @@ type Attributes struct {
 	Constraint Scalar `json:"constraint"`
 }
 
+func (a *Attributes) ApplyTextTransform() {
+	if a.Style.NoTextTransform() {
+		return
+	}
+
+	if a.Style.UppercaseTextTransform() {
+		a.Label.Value = strings.ToUpper(a.Label.Value)
+	}
+	if a.Style.LowercaseTextTransform() {
+		a.Label.Value = strings.ToLower(a.Label.Value)
+	}
+	if a.Style.CapitalizeTextTransform() {
+		a.Label.Value = cases.Title(language.Und).String(a.Label.Value)
+	}
+}
+
 // TODO references at the root scope should have their Scope set to root graph AST
 type Reference struct {
 	Key          *d2ast.KeyPath `json:"key"`
@@ -151,25 +169,42 @@ func (r Reference) InEdge() bool {
 }
 
 type Style struct {
-	Opacity      *Scalar `json:"opacity,omitempty"`
-	Stroke       *Scalar `json:"stroke,omitempty"`
-	Fill         *Scalar `json:"fill,omitempty"`
-	FillPattern  *Scalar `json:"fillPattern,omitempty"`
-	StrokeWidth  *Scalar `json:"strokeWidth,omitempty"`
-	StrokeDash   *Scalar `json:"strokeDash,omitempty"`
-	BorderRadius *Scalar `json:"borderRadius,omitempty"`
-	Shadow       *Scalar `json:"shadow,omitempty"`
-	ThreeDee     *Scalar `json:"3d,omitempty"`
-	Multiple     *Scalar `json:"multiple,omitempty"`
-	Font         *Scalar `json:"font,omitempty"`
-	FontSize     *Scalar `json:"fontSize,omitempty"`
-	FontColor    *Scalar `json:"fontColor,omitempty"`
-	Animated     *Scalar `json:"animated,omitempty"`
-	Bold         *Scalar `json:"bold,omitempty"`
-	Italic       *Scalar `json:"italic,omitempty"`
-	Underline    *Scalar `json:"underline,omitempty"`
-	Filled       *Scalar `json:"filled,omitempty"`
-	DoubleBorder *Scalar `json:"doubleBorder,omitempty"`
+	Opacity       *Scalar `json:"opacity,omitempty"`
+	Stroke        *Scalar `json:"stroke,omitempty"`
+	Fill          *Scalar `json:"fill,omitempty"`
+	FillPattern   *Scalar `json:"fillPattern,omitempty"`
+	StrokeWidth   *Scalar `json:"strokeWidth,omitempty"`
+	StrokeDash    *Scalar `json:"strokeDash,omitempty"`
+	BorderRadius  *Scalar `json:"borderRadius,omitempty"`
+	Shadow        *Scalar `json:"shadow,omitempty"`
+	ThreeDee      *Scalar `json:"3d,omitempty"`
+	Multiple      *Scalar `json:"multiple,omitempty"`
+	Font          *Scalar `json:"font,omitempty"`
+	FontSize      *Scalar `json:"fontSize,omitempty"`
+	FontColor     *Scalar `json:"fontColor,omitempty"`
+	Animated      *Scalar `json:"animated,omitempty"`
+	Bold          *Scalar `json:"bold,omitempty"`
+	Italic        *Scalar `json:"italic,omitempty"`
+	Underline     *Scalar `json:"underline,omitempty"`
+	Filled        *Scalar `json:"filled,omitempty"`
+	DoubleBorder  *Scalar `json:"doubleBorder,omitempty"`
+	TextTransform *Scalar `json:"capsLock,omitempty"`
+}
+
+func (s Style) NoTextTransform() bool {
+	return s.TextTransform != nil && s.TextTransform.Value == "none"
+}
+
+func (s Style) UppercaseTextTransform() bool {
+	return s.TextTransform != nil && s.TextTransform.Value == "uppercase"
+}
+
+func (s Style) LowercaseTextTransform() bool {
+	return s.TextTransform != nil && s.TextTransform.Value == "lowercase"
+}
+
+func (s Style) CapitalizeTextTransform() bool {
+	return s.TextTransform != nil && s.TextTransform.Value == "capitalize"
 }
 
 func (s *Style) Apply(key, value string) error {
@@ -340,6 +375,14 @@ func (s *Style) Apply(key, value string) error {
 			return errors.New(`expected "double-border" to be true or false`)
 		}
 		s.DoubleBorder.Value = value
+	case "text-transform":
+		if s.TextTransform == nil {
+			break
+		}
+		if !go2.Contains(textTransforms, strings.ToLower(value)) {
+			return errors.New(`expected "text-transform" to be one of (none, uppercase, lowercase, capitalize)`)
+		}
+		s.TextTransform.Value = value
 	default:
 		return fmt.Errorf("unknown style key: %s", key)
 	}
@@ -1305,10 +1348,11 @@ func (g *Graph) SetDimensions(mtexts []*d2target.MText, ruler *textmeasure.Ruler
 		}
 
 		if g.Theme != nil && g.Theme.SpecialRules.CapsLock && !strings.EqualFold(obj.Attributes.Shape.Value, d2target.ShapeCode) {
-			if obj.Attributes.Language != "latex" {
+			if obj.Attributes.Language != "latex" && !obj.Attributes.Style.NoTextTransform() {
 				obj.Attributes.Label.Value = strings.ToUpper(obj.Attributes.Label.Value)
 			}
 		}
+		obj.Attributes.ApplyTextTransform()
 
 		labelDims, err := obj.GetLabelSize(mtexts, ruler, fontFamily)
 		if err != nil {
@@ -1427,9 +1471,11 @@ func (g *Graph) SetDimensions(mtexts []*d2target.MText, ruler *textmeasure.Ruler
 			continue
 		}
 
-		if g.Theme != nil && g.Theme.SpecialRules.CapsLock {
+		if g.Theme != nil && g.Theme.SpecialRules.CapsLock && !edge.Attributes.Style.NoTextTransform() {
 			edge.Attributes.Label.Value = strings.ToUpper(edge.Attributes.Label.Value)
 		}
+		edge.Attributes.ApplyTextTransform()
+
 		usedFont := fontFamily
 		if edge.Attributes.Style.Font != nil {
 			f := d2fonts.D2_FONT_TO_FAMILY[edge.Attributes.Style.Font.Value]
@@ -1451,16 +1497,9 @@ func (g *Graph) SetDimensions(mtexts []*d2target.MText, ruler *textmeasure.Ruler
 func (g *Graph) Texts() []*d2target.MText {
 	var texts []*d2target.MText
 
-	capsLock := g.Theme != nil && g.Theme.SpecialRules.CapsLock
-
 	for _, obj := range g.Objects {
 		if obj.Attributes.Label.Value != "" {
 			text := obj.Text()
-			if capsLock && !strings.EqualFold(obj.Attributes.Shape.Value, d2target.ShapeCode) {
-				if obj.Attributes.Language != "latex" {
-					text.Text = strings.ToUpper(text.Text)
-				}
-			}
 			texts = appendTextDedup(texts, text)
 		}
 		if obj.Class != nil {
@@ -1489,9 +1528,6 @@ func (g *Graph) Texts() []*d2target.MText {
 	for _, edge := range g.Edges {
 		if edge.Attributes.Label.Value != "" {
 			text := edge.Text()
-			if capsLock {
-				text.Text = strings.ToUpper(text.Text)
-			}
 			texts = appendTextDedup(texts, text)
 		}
 		if edge.SrcArrowhead != nil && edge.SrcArrowhead.Label.Value != "" {

d2graph/style_helper.go

@@ -0,0 +1,3 @@
+package d2graph
+
+var textTransforms = []string{"none", "uppercase", "lowercase", "capitalize"}