fix set with explicit label

terrastruct · Nov 14, 2024 · 51950f0 · 51950f0
1 parent 7b9f2d1
commit 51950f0
Show file tree

Hide file tree

Showing 3 changed files with 228 additions and 1 deletion.
diff --git a/d2oracle/edit.go b/d2oracle/edit.go
@@ -491,7 +491,12 @@ func _set(g *d2graph.Graph, baseAST *d2ast.Map, key string, tag, value *string)
 			m = obj.Map
 		}
 
-		if (obj.Label.MapKey != nil && writeableLabelMK) && m == nil && (!found || reserved || len(mk.Edges) > 0) {
+		if (obj.Label.MapKey != nil && writeableLabelMK) && m == nil && (!found || reserved || len(mk.Edges) > 0) &&
+			// Label is not set like `hey.label: mylabel`
+			// This should only work when label is set like `hey: mylabel`
+			(obj.Label.MapKey.Key == nil ||
+				len(obj.Label.MapKey.Key.Path) == 0 ||
+				obj.Label.MapKey.Key.Path[len(obj.Label.MapKey.Key.Path)-1].Unbox().ScalarString() != "label") {
 			m2 := &d2ast.Map{
 				Range: d2ast.MakeRange(",1:0:0-1:0:0"),
 			}

diff --git a/d2oracle/edit_test.go b/d2oracle/edit_test.go
@@ -1023,6 +1023,16 @@ square.style.opacity: 0.2
 			key:   `square.top`,
 			value: go2.Pointer(`200`),
 			exp: `square: {top: 200}
+`,
+		},
+		{
+			name: "labeled_set_position",
+			text: `hey.label: what
+`,
+			key:   `hey.top`,
+			value: go2.Pointer(`200`),
+			exp: `hey.label: what
+hey.top: 200
 `,
 		},
 		{

diff --git a/testdata/d2oracle/TestSet/labeled_set_position.exp.json b/testdata/d2oracle/TestSet/labeled_set_position.exp.json