support json response colorful fix astaxie#12

Author: astaxie

bat.go

@@ -192,7 +192,7 @@ func main() {
 			}
 			if printV == "A" || printV == "b" {
 				body := formatResponseBody(res, httpreq, pretty)
-				fmt.Println(ColorfulResponse(body))
+				fmt.Println(ColorfulResponse(body, res.Header.Get("Content-Type")))
 			}
 		} else {
 			body := formatResponseBody(res, httpreq, pretty)

color.go

@@ -19,7 +19,11 @@ const (
 )
 
 func Color(str string, color uint8) string {
-	return fmt.Sprintf("\033[%dm%s%s", color, str, EndColor)
+	return fmt.Sprintf("%s%s%s", ColorStart(color), str, EndColor)
+}
+
+func ColorStart(color uint8) string {
+	return fmt.Sprintf("\033[%dm", color)
 }
 
 func ColorfulRequest(str string) string {
@@ -43,6 +47,96 @@ func ColorfulRequest(str string) string {
 	return strings.Join(lines, "\n")
 }
 
-func ColorfulResponse(str string) string {
+func ColorfulResponse(str, contenttype string) string {
+	if strings.Contains(contenttype, contentJsonRegex) {
+		str = ColorfulJson(str)
+	} else {
+		str = ColorfulHTML(str)
+	}
+	return str
+}
+
+func ColorfulJson(str string) string {
+	var rsli []rune
+	var key, val, startcolor, endcolor, startsemicolon bool
+	for _, char := range []rune(str) {
+		switch char {
+		case ' ':
+			rsli = append(rsli, char)
+		case '{':
+			startcolor = true
+			key = true
+			val = false
+			rsli = append(rsli, char)
+		case '}':
+			startcolor = false
+			endcolor = false
+			key = false
+			val = false
+			rsli = append(rsli, char)
+		case '"':
+			if startcolor {
+				rsli = append(rsli, char)
+				if key {
+					rsli = append(rsli, []rune(ColorStart(Magenta))...)
+				} else if val {
+					rsli = append(rsli, []rune(ColorStart(Cyan))...)
+				}
+				startsemicolon = true
+				key = false
+				val = false
+				startcolor = false
+			} else {
+				rsli = append(rsli, []rune(EndColor)...)
+				rsli = append(rsli, char)
+				endcolor = true
+				startsemicolon = false
+			}
+		case ',':
+			if !startsemicolon {
+				startcolor = true
+				key = true
+				val = false
+				if !endcolor {
+					rsli = append(rsli, []rune(EndColor)...)
+					endcolor = true
+				}
+			}
+			rsli = append(rsli, char)
+		case ':':
+			if !startsemicolon {
+				key = false
+				val = true
+				startcolor = true
+				if !endcolor {
+					rsli = append(rsli, []rune(EndColor)...)
+					endcolor = true
+				}
+			}
+			rsli = append(rsli, char)
+		case '\n', '\r', '[', ']':
+			rsli = append(rsli, char)
+		default:
+			if !startsemicolon {
+				if key && startcolor {
+					rsli = append(rsli, []rune(ColorStart(Magenta))...)
+					key = false
+					startcolor = false
+					endcolor = false
+				}
+				if val && startcolor {
+					rsli = append(rsli, []rune(ColorStart(Cyan))...)
+					val = false
+					startcolor = false
+					endcolor = false
+				}
+			}
+			rsli = append(rsli, char)
+		}
+	}
+	return string(rsli)
+}
+
+func ColorfulHTML(str string) string {
 	return Color(str, Green)
 }