podman

1
package server
2

3
import (
4
	"bufio"
5
	"errors"
6
	"io"
7
	"net"
8
	"net/http"
9
	"time"
10

11
	"github.com/gorilla/mux"
12
	"github.com/sirupsen/logrus"
13
)
14

15
type responseWriter struct {
16
	http.ResponseWriter
17
}
18

19
var apiLogger = &logrus.Logger{
20
	Formatter: &logrus.TextFormatter{
21
		DisableColors:          true,
22
		DisableLevelTruncation: true,
23
		FullTimestamp:          true,
24
		QuoteEmptyFields:       true,
25
		TimestampFormat:        time.RFC3339,
26
	},
27
	Level: logrus.TraceLevel,
28
	Out:   logrus.StandardLogger().Out,
29
}
30

31
func (l responseWriter) Write(b []byte) (int, error) {
32
	apiLogger.WithFields(logrus.Fields{
33
		"API":            "response",
34
		"X-Reference-Id": l.Header().Get("X-Reference-Id"),
35
	}).Trace(string(b))
36
	return l.ResponseWriter.Write(b)
37
}
38

39
func (l responseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
40
	if wrapped, ok := l.ResponseWriter.(http.Hijacker); ok {
41
		return wrapped.Hijack()
42
	}
43

44
	return nil, nil, errors.New("ResponseWriter does not support hijacking")
45
}
46

47
func (l responseWriter) Header() http.Header {
48
	return l.ResponseWriter.Header()
49
}
50

51
func (l responseWriter) WriteHeader(statusCode int) {
52
	l.ResponseWriter.WriteHeader(statusCode)
53
}
54

55
func (l responseWriter) Flush() {
56
	if wrapped, ok := l.ResponseWriter.(http.Flusher); ok {
57
		wrapped.Flush()
58
	}
59
}
60

61
func loggingHandler() mux.MiddlewareFunc {
62
	return func(h http.Handler) http.Handler {
63
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
64
			annotated := apiLogger.WithFields(logrus.Fields{
65
				"API":            "request",
66
				"X-Reference-Id": r.Header.Get("X-Reference-Id"),
67
			})
68
			r.Body = io.NopCloser(
69
				io.TeeReader(r.Body, annotated.WriterLevel(logrus.TraceLevel)))
70

71
			w = responseWriter{ResponseWriter: w}
72
			h.ServeHTTP(w, r)
73
		})
74
	}
75
}
76