59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
|
package d2graph
|
||
|
|
|
||
|
|
import "oss.terrastruct.com/d2/d2target"
|
||
|
|
|
||
|
|
func (obj *Object) IsSequenceDiagram() bool {
|
||
|
|
return obj != nil && obj.Attributes.Shape.Value == d2target.ShapeSequenceDiagram
|
||
|
|
}
|
||
|
|
|
||
|
|
func (obj *Object) outerSequenceDiagram() *Object {
|
||
|
|
for obj != nil {
|
||
|
|
obj = obj.Parent
|
||
|
|
if obj.IsSequenceDiagram() {
|
||
|
|
return obj
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// notes are descendant of actors with no edges and no children
|
||
|
|
func (obj *Object) IsSequenceDiagramNote() bool {
|
||
|
|
sd := obj.outerSequenceDiagram()
|
||
|
|
if sd == nil {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
return !obj.hasEdgeRef() && !obj.ContainsAnyEdge(obj.Graph.Edges) && len(obj.ChildrenArray) == 0
|
||
|
|
}
|
||
|
|
|
||
|
|
func (obj *Object) hasEdgeRef() bool {
|
||
|
|
for _, ref := range obj.References {
|
||
|
|
if ref.MapKey != nil && len(ref.MapKey.Edges) > 0 {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
func (obj *Object) ContainsAnyEdge(edges []*Edge) bool {
|
||
|
|
for _, e := range edges {
|
||
|
|
if e.ContainedBy(obj) {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *Edge) ContainedBy(obj *Object) bool {
|
||
|
|
for _, ref := range e.References {
|
||
|
|
curr := ref.ScopeObj
|
||
|
|
for curr != nil {
|
||
|
|
if curr == obj {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
curr = curr.Parent
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return false
|
||
|
|
}
|