To gitea and beyond, let's go(-yco)

internal/middleware/logging.go

@@ -0,0 +1,53 @@
+package middleware
+
+import (
+	"log"
+	"net/http"
+	"time"
+)
+
+func Logging(debug bool) func(http.Handler) http.Handler {
+	return func(next http.Handler) http.Handler {
+		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			start := time.Now()
+
+			wrapped := &responseWriter{ResponseWriter: w, statusCode: http.StatusOK}
+
+			next.ServeHTTP(wrapped, r)
+
+			duration := time.Since(start)
+
+			if debug {
+				log.Printf(
+					"%s %s %d %v %s",
+					r.Method,
+					r.URL.Path,
+					wrapped.statusCode,
+					duration,
+					r.UserAgent(),
+				)
+			} else {
+				if wrapped.statusCode >= 400 || duration > time.Second {
+					log.Printf(
+						"%s %s %d %v %s",
+						r.Method,
+						r.URL.Path,
+						wrapped.statusCode,
+						duration,
+						r.UserAgent(),
+					)
+				}
+			}
+		})
+	}
+}
+
+type responseWriter struct {
+	http.ResponseWriter
+	statusCode int
+}
+
+func (rw *responseWriter) WriteHeader(code int) {
+	rw.statusCode = code
+	rw.ResponseWriter.WriteHeader(code)
+}