45 lines
1.2 KiB
Go
45 lines
1.2 KiB
Go
package http
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"git.sharkk.net/Sky/Moonshark/core/logger"
|
|
)
|
|
|
|
// StatusColors for different status code ranges
|
|
const (
|
|
colorGreen = "\033[32m" // 2xx - Success
|
|
colorCyan = "\033[36m" // 3xx - Redirection
|
|
colorYellow = "\033[33m" // 4xx - Client Errors
|
|
colorRed = "\033[31m" // 5xx - Server Errors
|
|
colorReset = "\033[0m" // Reset color
|
|
colorGray = "\033[90m"
|
|
)
|
|
|
|
// LogRequest logs an HTTP request with custom formatting
|
|
func LogRequest(statusCode int, r *http.Request, duration time.Duration) {
|
|
statusColor := getStatusColor(statusCode)
|
|
|
|
// Use the logger's raw message writer to bypass the standard format
|
|
logger.LogRaw("%s%s%s %s%d %s%s %s %s(%v)%s",
|
|
colorGray, time.Now().Format(logger.TimeFormat()), colorReset,
|
|
statusColor, statusCode, r.Method, colorReset, r.URL.Path, colorGray, duration, colorReset)
|
|
}
|
|
|
|
// getStatusColor returns the ANSI color code for a status code
|
|
func getStatusColor(code int) string {
|
|
switch {
|
|
case code >= 200 && code < 300:
|
|
return colorGreen
|
|
case code >= 300 && code < 400:
|
|
return colorCyan
|
|
case code >= 400 && code < 500:
|
|
return colorYellow
|
|
case code >= 500:
|
|
return colorRed
|
|
default:
|
|
return ""
|
|
}
|
|
}
|