61 lines
1 KiB
Go
61 lines
1 KiB
Go
package d2ir
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"oss.terrastruct.com/d2/d2parser"
|
|
)
|
|
|
|
// Query is only for tests and debugging.
|
|
func (m *Map) Query(idStr string) (na []Node, _ error) {
|
|
k, err := d2parser.ParseMapKey(idStr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if k.Key != nil {
|
|
f := m.GetField(k.Key.IDA()...)
|
|
if f == nil {
|
|
return nil, nil
|
|
}
|
|
if len(k.Edges) == 0 {
|
|
na = append(na, f)
|
|
return na, nil
|
|
}
|
|
m = f.Map()
|
|
if m == nil {
|
|
return nil, nil
|
|
}
|
|
}
|
|
|
|
eida := NewEdgeIDs(k)
|
|
for _, eid := range eida {
|
|
ea := m.GetEdges(eid)
|
|
for _, e := range ea {
|
|
if k.EdgeKey == nil {
|
|
na = append(na, e)
|
|
} else if e.Map_ != nil {
|
|
f := e.Map_.GetField(k.EdgeKey.IDA()...)
|
|
if f != nil {
|
|
na = append(na, f)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return na, nil
|
|
}
|
|
|
|
func (m *Map) QueryOne(idStr string) (Node, error) {
|
|
na, err := m.Query(idStr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(na) == 0 {
|
|
return nil, nil
|
|
}
|
|
if len(na) > 1 {
|
|
return nil, fmt.Errorf("expected only one query result but got: %#v", err)
|
|
}
|
|
return na[0], nil
|
|
}
|