package main

import (
	"net/http"
	"time"

	"github.com/rs/zerolog/log"
	"github.com/urfave/negroni"
)

// type loggingResponseWriter struct {
// 	http.ResponseWriter
// 	statusCode int
// }
//
// func NewLoggingResponseWriter(w http.ResponseWriter) *loggingResponseWriter {
// 	return &loggingResponseWriter{w, http.StatusOK}
// }
//
// func (lrw *loggingResponseWriter) WriteHeader(code int) {
// 	lrw.statusCode = code
// 	lrw.ResponseWriter.WriteHeader(code)
// }

func RequestLogger(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		start := time.Now()

		// lrw := NewLoggingResponseWriter(w)
		lrw := negroni.NewResponseWriter(w)
		next.ServeHTTP(lrw, r)

		log.Info().Str("host", r.Host).Str("uri", r.RequestURI).Str("method", r.Method).Int("status", lrw.Status()).Int("size", lrw.Size()).Str("duration", time.Since(start).String()).Msg("Received Request")
	})
}