d2/d2lsp/d2lsp.go

// d2lsp contains functions useful for IDE clients
package d2lsp

import (
	"fmt"
	"strings"

	"oss.terrastruct.com/d2/d2ir"
	"oss.terrastruct.com/d2/d2parser"
	"oss.terrastruct.com/d2/lib/memfs"
)

func GetFieldRefs(path string, fs map[string]string, key string, boardPath []string) (refs []d2ir.Reference, _ error) {
	if _, ok := fs[path]; !ok {
		return nil, fmt.Errorf(`"%s" not found`, path)
	}
	r := strings.NewReader(fs[path])
	ast, err := d2parser.Parse(path, r, nil)
	if err != nil {
		return nil, err
	}

	mfs, err := memfs.New(fs)
	if err != nil {
		return nil, err
	}

	mk, err := d2parser.ParseMapKey(key)
	if err != nil {
		return nil, err
	}
	if mk.Key == nil {
		return nil, fmt.Errorf(`"%s" is invalid`, key)
	}

	ir, _, err := d2ir.Compile(ast, &d2ir.CompileOptions{
		FS: mfs,
	})
	if err != nil {
		return nil, err
	}

	ir = ir.FindBoardRoot(boardPath)
	if ir == nil {
		return nil, fmt.Errorf(`board "%v" not found`, boardPath)
	}

	var f *d2ir.Field
	curr := ir
	for _, p := range mk.Key.Path {
		f = curr.GetField(p.Unbox().ScalarString())
		if f == nil {
			return nil, nil
		}
		curr = f.Map()
	}
	for _, ref := range f.References {
		refs = append(refs, ref)
	}
	return refs, nil
}