support colorful

Author: astaxie

bat.go

@@ -146,11 +146,11 @@ func main() {
 		}
 		if fi.Mode()&os.ModeDevice == os.ModeDevice {
 			dump := httpreq.DumpRequest()
-			fmt.Println(string(dump))
+			fmt.Println(ColorfulRequest(string(dump)))
 			fmt.Println("")
-			fmt.Println(Color(res.Proto, Red), Color(res.Status, Green))
+			fmt.Println(Color(res.Proto, Magenta), Color(res.Status, Green))
 			for k, v := range res.Header {
-				fmt.Println(k, ":", strings.Join(v, " "))
+				fmt.Println(Color(k, Gray), ":", Color(strings.Join(v, " "), Cyan))
 			}
 			str, err := httpreq.String()
 			if err != nil {
@@ -169,7 +169,7 @@ func main() {
 				}
 				str = string(bstr)
 			}
-			fmt.Println(str)
+			fmt.Println(ColorfulResponse(str))
 		} else {
 			str, err := httpreq.String()
 			if err != nil {

color.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"fmt"
+	"strings"
 )
 
 const (
@@ -20,3 +21,26 @@ const (
 func Color(str string, color uint8) string {
 	return fmt.Sprintf("\033[%dm%s%s", color, str, EndColor)
 }
+
+func ColorfulRequest(str string) string {
+	lines := strings.Split(str, "\n")
+	strs := strings.Split(lines[0], " ")
+	strs[0] = Color(strs[0], Magenta)
+	strs[1] = Color(strs[1], Cyan)
+	strs[2] = Color(strs[2], Magenta)
+	lines[0] = strings.Join(strs, " ")
+	for i, line := range lines[1:] {
+		substr := strings.Split(line, ":")
+		if len(substr) != 2 {
+			continue
+		}
+		substr[0] = Color(substr[0], Gray)
+		substr[1] = Color(substr[1], Cyan)
+		lines[i+1] = strings.Join(substr, ":")
+	}
+	return strings.Join(lines, "\n")
+}
+
+func ColorfulResponse(str string) string {
+	return Color(str, Green)
+}