fix quicklook

feedback.go

@@ -4,6 +4,7 @@
 package aw
 
 import (
+	"bytes"
 	"encoding/json"
 	"fmt"
 	"log"
@@ -295,7 +296,7 @@ func (it *Item) MarshalJSON() ([]byte, error) {
 	} else if len(it.arg) > 1 {
 		v.Arg = it.arg
 	}
-	return json.Marshal(v)
+	return Marshal(v)
 }
 
 // itemText encapsulates the copytext and largetext values for a result Item.
@@ -404,7 +405,7 @@ func (m *Modifier) MarshalJSON() ([]byte, error) {
 		v.Arg = m.arg
 	}
 
-	return json.Marshal(v)
+	return Marshal(v)
 }
 
 // Feedback represents the results for an Alfred Script Filter.
@@ -479,7 +480,7 @@ func (fb *Feedback) NewItem(title string) *Item {
 // MarshalJSON serializes Feedback to Alfred's JSON format.
 // You shouldn't need to call this: use Send() instead.
 func (fb *Feedback) MarshalJSON() ([]byte, error) {
-	return json.Marshal(&struct {
+	return Marshal(&struct {
 		Variables map[string]string `json:"variables,omitempty"`
 		Rerun     float64           `json:"rerun,omitempty"`
 		Items     []*Item           `json:"items"`
@@ -499,7 +500,7 @@ func (fb *Feedback) Send() error {
 		log.Printf("Feedback already sent. Ignoring.")
 		return nil
 	}
-	output, err := json.MarshalIndent(fb, "", "  ")
+	output, err := MarshalIndent(fb, "", "  ")
 	if err != nil {
 		return fmt.Errorf("Error generating JSON : %w", err)
 	}
@@ -635,7 +636,7 @@ func (a *ArgVars) MarshalJSON() ([]byte, error) {
 		if len(a.arg) == 0 {
 			return []byte(`""`), nil
 		}
-		return json.Marshal(a.arg[0])
+		return Marshal(a.arg[0])
 	}
 
 	v := struct {
@@ -651,9 +652,30 @@ func (a *ArgVars) MarshalJSON() ([]byte, error) {
 		v.Arg = a.arg
 	}
 
-	return json.Marshal(&struct {
+	return Marshal(&struct {
 		Root interface{} `json:"alfredworkflow"`
 	}{
 		Root: v,
 	})
 }
+
+func Marshal(v interface{}) ([]byte, error) {
+	buffer := &bytes.Buffer{}
+	encoder := json.NewEncoder(buffer)
+	encoder.SetEscapeHTML(false)
+	err := encoder.Encode(v)
+	return buffer.Bytes(), err
+}
+
+func MarshalIndent(v interface{}, prefix, indent string) ([]byte, error) {
+	b, err := Marshal(v)
+	if err != nil {
+		return nil, err
+	}
+	var buf bytes.Buffer
+	err = json.Indent(&buf, b, prefix, indent)
+	if err != nil {
+		return nil, err
+	}
+	return buf.Bytes(), nil
+}