CLose #343

formatter.go

@@ -1009,6 +1009,57 @@ var defaultTemplateFuncs = template.FuncMap{
  }
  return color.New(colorAttr).Sprint(input)
  },
+ "colorhttp": func(enable bool, colorName string, isResponse bool, input string) string {
+ if !enable {
+ return input
+ }
+
+ colorAttr := color.Reset
+ present := true
+ var sb strings.Builder
+ if ca, ok := defaultColors[colorName]; ok {
+ colorAttr = ca
+ }
+
+ httpMethod := map[string]bool{
+ "GET": present,
+ "POST": present,
+ "PATCH": present,
+ "DELETE": present,
+ "PUT": present,
+ }
+
+ getColor := func(word string, forceApplyColor bool) color.Attribute {
+ if v := httpMethod[word]; v == present {
+ return colorAttr
+ }
+ if word[len(word)-1] == ':' {
+ return colorAttr
+ }
+ if forceApplyColor {
+ return colorAttr
+ }
+
+ return color.Reset
+ }
+ isFirstLine := true
+ for _, line := range strings.Split(input, "\n") {
+ if sb.Len() != 0 {
+ sb.WriteString("\n")
+ }
+
+ words := strings.Fields(line)
+ wordLen := len(words)
+ for index, word := range words {
+ forceApplyColor := index != wordLen-1 && isResponse && isFirstLine
+ sb.WriteString(color.New(getColor(word, forceApplyColor)).Sprint(word))
+ sb.WriteString(" ")
+ }
+ isFirstLine = false
+ }
+
+ return sb.String()
+ },
  "colordiff": func(enable bool, input string) string {
  if !enable {
  return input
@@ -1098,11 +1149,11 @@ request name: {{ .RequestName | color $.EnableColors "Cyan" }}
 {{- end -}}
 {{- if .HaveRequest }}
 
-request: {{ .Request | indent | trim | color $.EnableColors "HiMagenta" }}
+request: {{ .Request | indent | trim | colorhttp $.EnableColors "HiMagenta" false}}
 {{- end -}}
 {{- if .HaveResponse }}
 
-response: {{ .Response | indent | trim | color $.EnableColors "HiMagenta" }}
+response: {{ .Response | indent | trim | colorhttp $.EnableColors "HiMagenta" true}}
 {{- end -}}
 {{- if .HaveStacktrace }}