better query escaping

trealla/prolog_test.go

@@ -29,14 +29,17 @@ func TestQuery(t *testing.T) {
 		{
 			name: "member/2",
 			want: trealla.Answer{
-				Query:  `member(X, [1,foo(bar),4.2,"baz",'boop']).`,
+				Query:  `member(X, [1,foo(bar),4.2,"baz",'boop', [q, '"'], '\\', '\n']).`,
 				Result: "success",
 				Answers: []trealla.Solution{
 					{"X": int64(1)},
 					{"X": trealla.Compound{Functor: "foo", Args: []trealla.Term{"bar"}}},
 					{"X": 4.2},
 					{"X": "baz"},
 					{"X": "boop"},
+					{"X": []trealla.Term{"q", `"`}},
+					{"X": `\`},
+					{"X": "\n"},
 				},
 			},
 		},

trealla/trealla.go

@@ -166,10 +166,12 @@ func (pl *prolog) ask(ctx context.Context, query string) (string, error) {
 }
 
 func escapeQuery(query string) string {
-	query = strings.ReplaceAll(query, `"`, `\"`)
+	query = stringEscaper.Replace(query)
 	return fmt.Sprintf(`use_module(library(wasm_toplevel)), wasm_ask("%s")`, query)
 }
 
+var stringEscaper = strings.NewReplacer(`\`, `\\`, `"`, `\"`)
+
 // Option is an optional parameter for New.
 type Option func(*prolog)