57 lines
1 KiB
Go
57 lines
1 KiB
Go
package d2ir
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"oss.terrastruct.com/d2/d2ast"
|
|
"oss.terrastruct.com/d2/d2graph"
|
|
"oss.terrastruct.com/d2/d2parser"
|
|
)
|
|
|
|
type compiler struct {
|
|
err d2parser.ParseError
|
|
}
|
|
|
|
func (c *compiler) errorf(n d2ast.Node, f string, v ...interface{}) {
|
|
f = "%v: " + f
|
|
v = append([]interface{}{n.GetRange()}, v...)
|
|
c.err.Errors = append(c.err.Errors, d2ast.Error{
|
|
Range: n.GetRange(),
|
|
Message: fmt.Sprintf(f, v...),
|
|
})
|
|
}
|
|
|
|
func Apply(dst *Map, ast *d2ast.Map) error {
|
|
var c compiler
|
|
c.apply(dst, ast)
|
|
return c.err
|
|
}
|
|
|
|
func (c *compiler) apply(dst *Map, ast *d2ast.Map) {
|
|
for _, n := range ast.Nodes {
|
|
if n.MapKey == nil {
|
|
continue
|
|
}
|
|
|
|
c.applyKey(dst, n.MapKey)
|
|
}
|
|
}
|
|
|
|
func (c *compiler) applyKey(dst *Map, k *d2ast.Key) {
|
|
if k.Key != nil && len(k.Key.Path) > 0 {
|
|
f, ok := dst.Ensure(d2graph.Key(k.Key))
|
|
if !ok {
|
|
c.errorf(k.Key, "cannot index into array")
|
|
return
|
|
}
|
|
|
|
if len(k.Edges) == 0 {
|
|
if k.Primary.Unbox() != nil {
|
|
f.Primary = &Scalar{
|
|
parent: f,
|
|
Value: k.Primary.Unbox(),
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|