working

Author: alixander

d2js/js.go

@@ -8,21 +8,92 @@ import (
 	"strings"
 	"syscall/js"
 
+	"oss.terrastruct.com/d2/d2ast"
 	"oss.terrastruct.com/d2/d2compiler"
 	"oss.terrastruct.com/d2/d2format"
+	"oss.terrastruct.com/d2/d2oracle"
 	"oss.terrastruct.com/d2/d2parser"
 	"oss.terrastruct.com/d2/d2target"
 	"oss.terrastruct.com/d2/lib/urlenc"
 )
 
 func main() {
+	js.Global().Set("d2GetRefRanges", js.FuncOf(jsGetRefRanges))
 	js.Global().Set("d2Compile", js.FuncOf(jsCompile))
 	js.Global().Set("d2Parse", js.FuncOf(jsParse))
 	js.Global().Set("d2Encode", js.FuncOf(jsEncode))
 	js.Global().Set("d2Decode", js.FuncOf(jsDecode))
 	select {}
 }
 
+type jsRefRanges struct {
+	Ranges     []d2ast.Range `json:"ranges"`
+	ParseError string        `json:"parseError"`
+	UserError  string        `json:"userError"`
+	D2Error    string        `json:"d2Error"`
+}
+
+func jsGetRefRanges(this js.Value, args []js.Value) interface{} {
+	dsl := args[0].String()
+	key := args[1].String()
+
+	mk, err := d2parser.ParseMapKey(key)
+	if err != nil {
+		ret := jsRefRanges{D2Error: err.Error()}
+		str, _ := json.Marshal(ret)
+		return string(str)
+	}
+
+	g, err := d2compiler.Compile("", strings.NewReader(dsl), &d2compiler.CompileOptions{
+		UTF16: true,
+	})
+	var pe d2parser.ParseError
+	if err != nil {
+		if errors.As(err, &pe) {
+			serialized, _ := json.Marshal(err)
+			// TODO
+			ret := jsRefRanges{ParseError: string(serialized)}
+			str, _ := json.Marshal(ret)
+			return string(str)
+		}
+		ret := jsRefRanges{D2Error: err.Error()}
+		str, _ := json.Marshal(ret)
+		return string(str)
+	}
+
+	var ranges []d2ast.Range
+	if len(mk.Edges) == 1 {
+		edge := d2oracle.GetEdge(g, key)
+		if edge == nil {
+			ret := jsRefRanges{D2Error: "edge not found"}
+			str, _ := json.Marshal(ret)
+			return string(str)
+		}
+
+		for _, ref := range edge.References {
+			ranges = append(ranges, ref.MapKey.Range)
+		}
+	} else {
+		obj := d2oracle.GetObj(g, key)
+		if obj == nil {
+			ret := jsRefRanges{D2Error: "obj not found"}
+			str, _ := json.Marshal(ret)
+			return string(str)
+		}
+
+		for _, ref := range obj.References {
+			ranges = append(ranges, ref.Key.Range)
+		}
+	}
+
+	resp := jsRefRanges{
+		Ranges: ranges,
+	}
+
+	str, _ := json.Marshal(resp)
+	return string(str)
+}
+
 type jsObject struct {
 	Result    string `json:"result"`
 	UserError string `json:"userError"`