From 76b6f1796d37435e3bf8575c0612ab8cc26deddd Mon Sep 17 00:00:00 2001
From: Jordan Orelli <jordanorelli@gmail.com>
Date: Fri, 30 Oct 2015 11:40:13 -0400
Subject: [PATCH] pretty print in the client

this makes sure that source documents are pretty-printed, as ES won't
pretty print source documents if you just supply the ?pretty query param
---
 main.go | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/main.go b/main.go
index 7a6229e..d247272 100644
--- a/main.go
+++ b/main.go
@@ -3,10 +3,12 @@ package main
 import (
 	"bufio"
 	"bytes"
+	"encoding/json"
 	"flag"
 	"fmt"
 	"github.com/jordanorelli/moon"
 	"io"
+	"io/ioutil"
 	"net/http"
 	"os"
 	"strings"
@@ -74,10 +76,6 @@ func (r *repl) run() {
 
 		req.Header["Content-Type"] = []string{"application/x-www-form-urlencoded"}
 
-		q := req.URL.Query()
-		q.Set("pretty", "")
-		req.URL.RawQuery = q.Encode()
-
 		res, err := client.Do(req)
 		if err != nil {
 			r.errorf("error sending http request: %v", err)
@@ -114,7 +112,19 @@ func (r *repl) dumpResponse(res *http.Response) {
 		w = r.out1
 	}
 
-	io.Copy(w, res.Body)
+	b, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		r.errorf("error reading response body: %v", err)
+		return
+	}
+
+	var buf bytes.Buffer
+	if err := json.Indent(&buf, b, "", "  "); err != nil {
+		r.errorf("error pretty-printing json: %v", err)
+		return
+	}
+
+	io.Copy(w, &buf)
 	w.Write([]byte("\n"))
 }