From 7f13be1524c2ea53cd1867138398b5a9c0e9f61c Mon Sep 17 00:00:00 2001
From: Gavin Nishizawa <gavin@terrastruct.com>
Date: Fri, 13 Jan 2023 20:03:39 -0800
Subject: [PATCH] account for leading/trailing newlines when measuring code
 shapes

---
 d2graph/d2graph.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/d2graph/d2graph.go b/d2graph/d2graph.go
index 21361304a..28ea03ef9 100644
--- a/d2graph/d2graph.go
+++ b/d2graph/d2graph.go
@@ -1027,6 +1027,27 @@ func GetTextDimensions(mtexts []*d2target.MText, ruler *textmeasure.Ruler, t *d2
 		var h int
 		if t.Language != "" {
 			w, h = ruler.Measure(d2fonts.SourceCodePro.Font(t.FontSize, d2fonts.FONT_STYLE_REGULAR), t.Text)
+
+			// count empty leading and trailing lines since ruler will not be able to measure it
+			lines := strings.Split(t.Text, "\n")
+			leadingLines := 0
+			for _, line := range lines {
+				if strings.TrimSpace(line) == "" {
+					leadingLines++
+				} else {
+					break
+				}
+			}
+			trailingLines := 0
+			for i := len(lines) - 1; i >= 0; i-- {
+				if strings.TrimSpace(lines[i]) == "" {
+					trailingLines++
+				} else {
+					break
+				}
+			}
+			h += t.FontSize * (leadingLines + trailingLines)
+
 			// padding
 			w += 12
 			h += 12