formatter_with_request_body.go

package httplog

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"

	"github.com/TylerBrock/colorjson"
)

// Copyright 2022 Jack Rudenko. MadAppGang. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.

// RequestBodyLogFormatter format function with JSON body output or text
func RequestBodyLogFormatter(param LogFormatterParams) string {
	var blueColor, yellowColor, greenColor, redColor, resetColor string
	if param.IsOutputColor() {
		blueColor = blue
		yellowColor = yellow
		greenColor = green
		redColor = red
		resetColor = param.ResetColor()
	}

	var body []byte
	if param.Request.Body != nil {
		// get request body
		var err error
		body, err = ioutil.ReadAll(param.Request.Body)
		if err != nil {
			return fmt.Sprintf("===\n%s ERROR READING BODY: %s %s\n===\n", redColor, err.Error(), resetColor)
		}
		// let's bring back the request body to the next listener
		param.Request.Body = ioutil.NopCloser(bytes.NewBuffer(body))
	}

	if len(body) == 0 {
		return fmt.Sprintf("===\n%s EMPTY BODY %s\n===\n", yellowColor, resetColor)
	}

	var bodyJSON map[string]interface{}
	err := json.Unmarshal(body, &bodyJSON)
	if err != nil {
		// it is not a json
		text := bytes.ToValidUTF8(body, nil)
		return fmt.Sprintf("===\n%s TEXT BODY:%s\n%s\n===\n", blueColor, resetColor, string(text))
	}

	f := colorjson.NewFormatter()
	f.Indent = 2
	s, _ := f.Marshal(bodyJSON)
	return fmt.Sprintf("===\n%s JSON BODY:%s\n%s\n===\n", greenColor, resetColor, string(s))
}

// DefaultLogFormatterWithHeadersAndBody is a combination of default log formatter, header log formatter and json body
var DefaultLogFormatterWithRequestHeadersAndBody = ChainLogFormatter(DefaultLogFormatter, RequestHeaderLogFormatter, RequestBodyLogFormatter)