From 1420dab5a92ef4e638e9d2be7fbad7744e13feb9 Mon Sep 17 00:00:00 2001
From: Alexander Wang <alex@terrastruct.com>
Date: Fri, 13 Jan 2023 08:26:38 -0800
Subject: [PATCH] also transfer arrowheads

---
 d2compiler/compile.go                                           | 2 ++
 e2etests/stable_test.go                                         | 1 +
 .../stable/sql_table_tooltip_animated/dagre/board.exp.json      | 2 +-
 .../stable/sql_table_tooltip_animated/dagre/sketch.exp.svg      | 2 +-
 .../stable/sql_table_tooltip_animated/elk/board.exp.json        | 2 +-
 .../stable/sql_table_tooltip_animated/elk/sketch.exp.svg        | 2 +-
 6 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/d2compiler/compile.go b/d2compiler/compile.go
index 38c1f68d6..c89a4853e 100644
--- a/d2compiler/compile.go
+++ b/d2compiler/compile.go
@@ -757,10 +757,12 @@ func flattenContainer(g *d2graph.Graph, obj *d2graph.Object) {
 		// TODO more attributes
 		if e.SrcTableColumnIndex != nil {
 			newEdge.SrcTableColumnIndex = new(int)
+			newEdge.SrcArrowhead = e.SrcArrowhead
 			*newEdge.SrcTableColumnIndex = *e.SrcTableColumnIndex
 		}
 		if e.DstTableColumnIndex != nil {
 			newEdge.DstTableColumnIndex = new(int)
+			newEdge.DstArrowhead = e.DstArrowhead
 			*newEdge.DstTableColumnIndex = *e.DstTableColumnIndex
 		}
 		newEdge.Attributes = e.Attributes
diff --git a/e2etests/stable_test.go b/e2etests/stable_test.go
index 24981d9fe..fb9f7399a 100644
--- a/e2etests/stable_test.go
+++ b/e2etests/stable_test.go
@@ -1813,6 +1813,7 @@ a: {
 
 x.y -> a.b: {
   style.animated: true
+	target-arrowhead.shape: cf-many
 }
 `,
 		},
diff --git a/e2etests/testdata/stable/sql_table_tooltip_animated/dagre/board.exp.json b/e2etests/testdata/stable/sql_table_tooltip_animated/dagre/board.exp.json
index 93aba66a4..08265e940 100644
--- a/e2etests/testdata/stable/sql_table_tooltip_animated/dagre/board.exp.json
+++ b/e2etests/testdata/stable/sql_table_tooltip_animated/dagre/board.exp.json
@@ -152,7 +152,7 @@
       "srcArrow": "none",
       "srcLabel": "",
       "dst": "a",
-      "dstArrow": "triangle",
+      "dstArrow": "cf-many",
       "dstLabel": "",
       "opacity": 1,
       "strokeDash": 0,
diff --git a/e2etests/testdata/stable/sql_table_tooltip_animated/dagre/sketch.exp.svg b/e2etests/testdata/stable/sql_table_tooltip_animated/dagre/sketch.exp.svg
index 4c942441c..12c3e8317 100644
--- a/e2etests/testdata/stable/sql_table_tooltip_animated/dagre/sketch.exp.svg
+++ b/e2etests/testdata/stable/sql_table_tooltip_animated/dagre/sketch.exp.svg
@@ -56,7 +56,7 @@ width="280" height="464" viewBox="-102 -118 280 464"><style type="text/css">
 </svg>
 </g><title>I like turtles</title></g></g><g id="a"><g class="shape" ><rect class="shape" x="0" y="172" width="60" height="72" style="fill:#FFFFFF;stroke:#0A0F25;opacity:1.000000;stroke-width:2;"/><rect class="class_header" x="0.000000" y="172.000000" width="60.000000" height="36.000000" fill="#0A0F25" /><text class="text" x="20.000000" y="199.000000" style="text-anchor:start;font-size:24px;fill:#FFFFFF">a</text><text class="text" x="10.000000" y="231.000000" style="text-anchor:start;font-size:20px;fill:#0D32B2">b</text>
 <text class="text" x="40.000000" y="231.000000" style="text-anchor:start;font-size:20px;fill:#676C7E"></text>
-<text class="text" x="40.000000" y="231.000000" style="text-anchor:end;font-size:20px;fill:#4A6FF3;letter-spacing:2px;"></text><line x1="0.000000" y1="244.000000" x2="60.000000" y2="244.000000" style="stroke-width:2;stroke:#0A0F25" /></g></g><g id="(x -&gt; a)[0]"><marker id="mk-3990223579" markerWidth="10.000000" markerHeight="12.000000" refX="7.000000" refY="6.000000" viewBox="0.000000 0.000000 10.000000 12.000000" orient="auto" markerUnits="userSpaceOnUse"> <polygon class="connection" fill="#0D32B2" stroke-width="2" points="0.000000,0.000000 10.000000,6.000000 0.000000,12.000000" /> </marker><path d="M 30.000000 74.000000 C 30.000000 112.000000 30.000000 132.000000 30.000000 168.000000" class="connection animated-connection" style="fill:none;stroke:#0D32B2;opacity:1.000000;stroke-width:2;stroke-dasharray:10.000000,9.865639;stroke-dashoffset:-198.656393;animation: dashdraw 4.932820s linear infinite;" marker-end="url(#mk-3990223579)" mask="url(#3439718239)"/></g><mask id="3439718239" maskUnits="userSpaceOnUse" x="-100" y="-100" width="280" height="464">
+<text class="text" x="40.000000" y="231.000000" style="text-anchor:end;font-size:20px;fill:#4A6FF3;letter-spacing:2px;"></text><line x1="0.000000" y1="244.000000" x2="60.000000" y2="244.000000" style="stroke-width:2;stroke:#0A0F25" /></g></g><g id="(x -&gt; a)[0]"><marker id="mk-238431170" markerWidth="28.000000" markerHeight="30.000000" refX="25.000000" refY="15.000000" viewBox="0.000000 0.000000 28.000000 30.000000" orient="auto" markerUnits="userSpaceOnUse"> <g class="connection" stroke="#0D32B2" stroke-width="2" fill="white"><circle class="connection" stroke="#0D32B2" stroke-width="2" fill="white" cx="5.000000" cy="15.000000" r="4.000000"/><path d="M25.000000,15.000000 36.000000,15.000000 M10.000000,15.000000 36.000000,0.000000 M10.000000,15.000000 36.000000,30.000000"/></g> </marker><path d="M 30.000000 74.000000 C 30.000000 112.000000 30.000000 132.000000 30.000000 168.000000" class="connection animated-connection" style="fill:none;stroke:#0D32B2;opacity:1.000000;stroke-width:2;stroke-dasharray:10.000000,9.865639;stroke-dashoffset:-198.656393;animation: dashdraw 4.932820s linear infinite;" marker-end="url(#mk-238431170)" mask="url(#833117786)"/></g><mask id="833117786" maskUnits="userSpaceOnUse" x="-100" y="-100" width="280" height="464">
 <rect x="-100" y="-100" width="280" height="464" fill="white"></rect>
 
 </mask><style type="text/css"><![CDATA[
diff --git a/e2etests/testdata/stable/sql_table_tooltip_animated/elk/board.exp.json b/e2etests/testdata/stable/sql_table_tooltip_animated/elk/board.exp.json
index 315f66a6c..e869a4914 100644
--- a/e2etests/testdata/stable/sql_table_tooltip_animated/elk/board.exp.json
+++ b/e2etests/testdata/stable/sql_table_tooltip_animated/elk/board.exp.json
@@ -152,7 +152,7 @@
       "srcArrow": "none",
       "srcLabel": "",
       "dst": "a",
-      "dstArrow": "triangle",
+      "dstArrow": "cf-many",
       "dstLabel": "",
       "opacity": 1,
       "strokeDash": 0,
diff --git a/e2etests/testdata/stable/sql_table_tooltip_animated/elk/sketch.exp.svg b/e2etests/testdata/stable/sql_table_tooltip_animated/elk/sketch.exp.svg
index 0efc20882..ef9d36cc8 100644
--- a/e2etests/testdata/stable/sql_table_tooltip_animated/elk/sketch.exp.svg
+++ b/e2etests/testdata/stable/sql_table_tooltip_animated/elk/sketch.exp.svg
@@ -56,7 +56,7 @@ width="280" height="464" viewBox="-90 -106 280 464"><style type="text/css">
 </svg>
 </g><title>I like turtles</title></g></g><g id="a"><g class="shape" ><rect class="shape" x="12" y="184" width="60" height="72" style="fill:#FFFFFF;stroke:#0A0F25;opacity:1.000000;stroke-width:2;"/><rect class="class_header" x="12.000000" y="184.000000" width="60.000000" height="36.000000" fill="#0A0F25" /><text class="text" x="32.000000" y="211.000000" style="text-anchor:start;font-size:24px;fill:#FFFFFF">a</text><text class="text" x="22.000000" y="243.000000" style="text-anchor:start;font-size:20px;fill:#0D32B2">b</text>
 <text class="text" x="52.000000" y="243.000000" style="text-anchor:start;font-size:20px;fill:#676C7E"></text>
-<text class="text" x="52.000000" y="243.000000" style="text-anchor:end;font-size:20px;fill:#4A6FF3;letter-spacing:2px;"></text><line x1="12.000000" y1="256.000000" x2="72.000000" y2="256.000000" style="stroke-width:2;stroke:#0A0F25" /></g></g><g id="(x -&gt; a)[0]"><marker id="mk-3990223579" markerWidth="10.000000" markerHeight="12.000000" refX="7.000000" refY="6.000000" viewBox="0.000000 0.000000 10.000000 12.000000" orient="auto" markerUnits="userSpaceOnUse"> <polygon class="connection" fill="#0D32B2" stroke-width="2" points="0.000000,0.000000 10.000000,6.000000 0.000000,12.000000" /> </marker><path d="M 42.000000 86.000000 L 42.000000 180.000000" class="connection animated-connection" style="fill:none;stroke:#0D32B2;opacity:1.000000;stroke-width:2;stroke-dasharray:10.000000,9.865639;stroke-dashoffset:-198.656393;animation: dashdraw 4.932820s linear infinite;" marker-end="url(#mk-3990223579)" mask="url(#1608076706)"/></g><mask id="1608076706" maskUnits="userSpaceOnUse" x="-100" y="-100" width="280" height="464">
+<text class="text" x="52.000000" y="243.000000" style="text-anchor:end;font-size:20px;fill:#4A6FF3;letter-spacing:2px;"></text><line x1="12.000000" y1="256.000000" x2="72.000000" y2="256.000000" style="stroke-width:2;stroke:#0A0F25" /></g></g><g id="(x -&gt; a)[0]"><marker id="mk-238431170" markerWidth="28.000000" markerHeight="30.000000" refX="25.000000" refY="15.000000" viewBox="0.000000 0.000000 28.000000 30.000000" orient="auto" markerUnits="userSpaceOnUse"> <g class="connection" stroke="#0D32B2" stroke-width="2" fill="white"><circle class="connection" stroke="#0D32B2" stroke-width="2" fill="white" cx="5.000000" cy="15.000000" r="4.000000"/><path d="M25.000000,15.000000 36.000000,15.000000 M10.000000,15.000000 36.000000,0.000000 M10.000000,15.000000 36.000000,30.000000"/></g> </marker><path d="M 42.000000 86.000000 L 42.000000 180.000000" class="connection animated-connection" style="fill:none;stroke:#0D32B2;opacity:1.000000;stroke-width:2;stroke-dasharray:10.000000,9.865639;stroke-dashoffset:-198.656393;animation: dashdraw 4.932820s linear infinite;" marker-end="url(#mk-238431170)" mask="url(#1359229009)"/></g><mask id="1359229009" maskUnits="userSpaceOnUse" x="-100" y="-100" width="280" height="464">
 <rect x="-100" y="-100" width="280" height="464" fill="white"></rect>
 
 </mask><style type="text/css"><![CDATA[