refactor code changes into compile.go

This commit is contained in:
Daniel Suh 2024-05-31 00:55:58 -04:00 committed by Alexander Wang
parent a5bf9c1b4d
commit 95d886b936
No known key found for this signature in database
GPG key ID: BE3937D0D52D8927

View file

@ -776,13 +776,8 @@ func (c *compiler) compileEdge(obj *d2graph.Object, e *d2ir.Edge) {
c.compileLabel(&edge.Attributes, e)
}
for _, field := range e.Map_.Fields {
if field.Name == "link" && edge.Label.Value == "" {
edge.Label.Value = field.Primary_.String()
}
}
if e.Map() != nil {
c.compileEdgeLinks(edge, e.Map())
c.compileEdgeMap(edge, e.Map())
}
@ -841,6 +836,17 @@ func (c *compiler) compileEdgeMap(edge *d2graph.Edge, m *d2ir.Map) {
}
}
func (c *compiler) compileEdgeLinks(edge *d2graph.Edge, m *d2ir.Map) {
if m == nil {
return
}
for _, field := range m.Fields {
if field.Name == "link" && edge.Label.Value == "" {
edge.Label.Value = field.Primary_.String()
}
}
}
func (c *compiler) compileEdgeField(edge *d2graph.Edge, f *d2ir.Field) {
keyword := strings.ToLower(f.Name.ScalarString())
_, isStyleReserved := d2ast.StyleKeywords[keyword]