d2parse

Author: alixander

d2js/js.go

@@ -11,11 +11,13 @@ import (
 	"oss.terrastruct.com/d2/d2compiler"
 	"oss.terrastruct.com/d2/d2format"
 	"oss.terrastruct.com/d2/d2parser"
+	"oss.terrastruct.com/d2/d2target"
 	"oss.terrastruct.com/d2/lib/urlenc"
 )
 
 func main() {
 	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 {}
@@ -27,6 +29,68 @@ type jsObject struct {
 	D2Error   string `json:"d2Error"`
 }
 
+type jsParseResponse struct {
+	DSL        string            `json:"dsl"`
+	Texts      []*d2target.MText `json:"texts"`
+	ParseError string            `json:"parseError"`
+	UserError  string            `json:"userError"`
+	D2Error    string            `json:"d2Error"`
+}
+
+func jsParse(this js.Value, args []js.Value) interface{} {
+	dsl := args[0].String()
+	themeID := args[1].Int()
+
+	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)
+			ret := jsParseResponse{ParseError: string(serialized)}
+			str, _ := json.Marshal(ret)
+			return string(str)
+		}
+		ret := jsParseResponse{D2Error: err.Error()}
+		str, _ := json.Marshal(ret)
+		return string(str)
+	}
+
+	if len(g.Layers) > 0 || len(g.Scenarios) > 0 || len(g.Steps) > 0 {
+		ret := jsParseResponse{UserError: "layers, scenarios, and steps are not yet supported. Coming soon."}
+		str, _ := json.Marshal(ret)
+		return string(str)
+	}
+
+	for _, o := range g.Objects {
+		if (o.Attributes.Top == nil) != (o.Attributes.Left == nil) {
+			ret := jsParseResponse{UserError: `keywords "top" and "left" currently must be used together`}
+			str, _ := json.Marshal(ret)
+			return string(str)
+		}
+	}
+
+	err = g.ApplyTheme(int64(themeID))
+	if err != nil {
+		ret := jsParseResponse{D2Error: err.Error()}
+		str, _ := json.Marshal(ret)
+		return string(str)
+	}
+
+	resp := jsParseResponse{
+		Texts: g.Texts(),
+	}
+
+	newDSL := d2format.Format(g.AST)
+	if dsl != newDSL {
+		resp.DSL = newDSL
+	}
+
+	str, _ := json.Marshal(resp)
+	return string(str)
+}
+
 // TODO error passing
 // TODO recover panics
 func jsCompile(this js.Value, args []js.Value) interface{} {