Merge pull request #22 from bentranter/add-json-indent-method

Add JSONIndent method

context.go

@@ -167,6 +167,16 @@ func (ctx *Context) JSON(obj interface{}) {
 	ctx.Send(json)
 }
 
+// JSONIndent Sends a JSON response, indenting the JSON as desired.
+func (ctx *Context) JSONIndent(obj interface{}, prefix, indent string) {
+	jsonIndented, err := json.MarshalIndent(obj, prefix, indent)
+	if err != nil {
+		panic(err)
+	}
+	ctx.SetHeader("Content-Type", "application/json")
+	ctx.Send(jsonIndented)
+}
+
 // Send the response immediately. Set `ctx.IsSent` to `true` to make
 // sure that the response won't be sent twice.
 func (ctx *Context) Send(body interface{}) {

context_test.go

@@ -385,3 +385,28 @@ func TestJSON(t *testing.T) {
 		assertEqual(t, w.HeaderMap.Get("Content-Type"), `application/json`)
 	}
 }
+
+func TestJSONIndent(t *testing.T) {
+	cases := []struct {
+		input  map[string]interface{}
+		output string
+	}{
+		{
+			map[string]interface{}{"status": "success", "code": 200},
+			`{
+  "code": 200,
+  "status": "success"
+}`,
+		},
+	}
+
+	for _, c := range cases {
+		r := makeTestHTTPRequest(nil, "GET", "/")
+		w := httptest.NewRecorder()
+		app := New()
+		ctx := NewContext(r, w, app)
+		ctx.JSONIndent(c.input, "", "  ")
+		assertEqual(t, w.Body.String(), c.output)
+		assertEqual(t, w.HeaderMap.Get("Content-Type"), `application/json`)
+	}
+}