luciano/d2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

update getters

Browse source
This commit is contained in:
Alexander Wang 2023-06-21 15:43:26 -07:00
parent 25b68c6140
commit e6db07d61d
No known key found for this signature in database
GPG key ID: D89FA31966BDBECE
1 changed files with 31 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
d2oracle/get.go
View file

@ -88,7 +88,12 @@ func ReplaceBoardNode(ast, ast2 *d2ast.Map, boardPath []string) bool {
return false return false
} }
func GetChildrenIDs(g *d2graph.Graph, absID string) (ids []string, _ error) { func GetChildrenIDs(g *d2graph.Graph, boardPath []string, absID string) (ids []string, _ error) {
g = GetBoardGraph(g, boardPath)
if g == nil {
return nil, fmt.Errorf("board at path %v not found", boardPath)
}
mk, err := d2parser.ParseMapKey(absID) mk, err := d2parser.ParseMapKey(absID)
if err != nil { if err != nil {
return nil, err return nil, err
@ -105,7 +110,12 @@ func GetChildrenIDs(g *d2graph.Graph, absID string) (ids []string, _ error) {
return ids, nil return ids, nil
} }
func GetParentID(g *d2graph.Graph, absID string) (string, error) { func GetParentID(g *d2graph.Graph, boardPath []string, absID string) (string, error) {
g = GetBoardGraph(g, boardPath)
if g == nil {
return "", fmt.Errorf("board at path %v not found", boardPath)
}
mk, err := d2parser.ParseMapKey(absID) mk, err := d2parser.ParseMapKey(absID)
if err != nil { if err != nil {
return "", err return "", err
@ -118,13 +128,23 @@ func GetParentID(g *d2graph.Graph, absID string) (string, error) {
return obj.Parent.AbsID(), nil return obj.Parent.AbsID(), nil
} }
func GetObj(g *d2graph.Graph, absID string) *d2graph.Object { func GetObj(g *d2graph.Graph, boardPath []string, absID string) *d2graph.Object {
g = GetBoardGraph(g, boardPath)
if g == nil {
return nil
}
mk, _ := d2parser.ParseMapKey(absID) mk, _ := d2parser.ParseMapKey(absID)
obj, _ := g.Root.HasChild(d2graph.Key(mk.Key)) obj, _ := g.Root.HasChild(d2graph.Key(mk.Key))
return obj return obj
} }
func GetEdge(g *d2graph.Graph, absID string) *d2graph.Edge { func GetEdge(g *d2graph.Graph, boardPath []string, absID string) *d2graph.Edge {
g = GetBoardGraph(g, boardPath)
if g == nil {
return nil
}
for _, e := range g.Edges { for _, e := range g.Edges {
if e.AbsID() == absID { if e.AbsID() == absID {
return e return e
@ -133,7 +153,12 @@ func GetEdge(g *d2graph.Graph, absID string) *d2graph.Edge {
return nil return nil
} }
func GetObjOrder(g *d2graph.Graph) []string { func GetObjOrder(g *d2graph.Graph, boardPath []string) ([]string, error) {
g = GetBoardGraph(g, boardPath)
if g == nil {
return nil, fmt.Errorf("board at path %v not found", boardPath)
}
var order []string var order []string
queue := []*d2graph.Object{g.Root} queue := []*d2graph.Object{g.Root}
@ -148,7 +173,7 @@ func GetObjOrder(g *d2graph.Graph) []string {
} }
} }
return order return order, nil
} }
func IsLabelKeyID(key, label string) bool { func IsLabelKeyID(key, label string) bool {