80 lines
1.8 KiB
Go
80 lines
1.8 KiB
Go
package middleware
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/seifghazi/claude-code-monitor/internal/model"
|
|
)
|
|
|
|
func Logging(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
start := time.Now()
|
|
|
|
var bodyBytes []byte
|
|
if r.Body != nil {
|
|
var err error
|
|
bodyBytes, err = io.ReadAll(r.Body)
|
|
if err != nil {
|
|
log.Printf("❌ Error reading request body: %v", err)
|
|
http.Error(w, "Error reading request body", http.StatusBadRequest)
|
|
return
|
|
}
|
|
r.Body.Close()
|
|
r.Body = io.NopCloser(bytes.NewReader(bodyBytes))
|
|
}
|
|
|
|
ctx := context.WithValue(r.Context(), model.BodyBytesKey, bodyBytes)
|
|
r = r.WithContext(ctx)
|
|
|
|
wrapped := &responseWriter{ResponseWriter: w, statusCode: http.StatusOK}
|
|
next.ServeHTTP(wrapped, r)
|
|
|
|
duration := time.Since(start)
|
|
log.Printf("Response: %d %s (took %v)", wrapped.statusCode, http.StatusText(wrapped.statusCode), duration)
|
|
})
|
|
}
|
|
|
|
func formatHeaders(headers http.Header) string {
|
|
headerMap := make(map[string][]string)
|
|
for k, v := range headers {
|
|
headerMap[k] = sanitizeHeaderValue(k, v)
|
|
}
|
|
headerBytes, _ := json.MarshalIndent(headerMap, "", " ")
|
|
return string(headerBytes)
|
|
}
|
|
|
|
func sanitizeHeaderValue(key string, values []string) []string {
|
|
lowerKey := strings.ToLower(key)
|
|
sensitiveHeaders := []string{
|
|
"x-api-key",
|
|
"api-key",
|
|
"authorization",
|
|
"anthropic-api-key",
|
|
"openai-api-key",
|
|
"bearer",
|
|
}
|
|
|
|
for _, sensitive := range sensitiveHeaders {
|
|
if strings.Contains(lowerKey, sensitive) {
|
|
return []string{"[REDACTED]"}
|
|
}
|
|
}
|
|
return values
|
|
}
|
|
|
|
type responseWriter struct {
|
|
http.ResponseWriter
|
|
statusCode int
|
|
}
|
|
|
|
func (rw *responseWriter) WriteHeader(code int) {
|
|
rw.statusCode = code
|
|
rw.ResponseWriter.WriteHeader(code)
|
|
}
|