mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-19 20:32:35 +02:00
108 lines
3.1 KiB
Go
108 lines
3.1 KiB
Go
package gphttp
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"syscall"
|
|
|
|
"github.com/yusing/go-proxy/internal/gperr"
|
|
"github.com/yusing/go-proxy/internal/net/gphttp/httpheaders"
|
|
)
|
|
|
|
// ServerError is for handling server errors.
|
|
//
|
|
// It logs the error and returns http.StatusInternalServerError to the client.
|
|
// Status code can be specified as an argument.
|
|
func ServerError(w http.ResponseWriter, r *http.Request, err error, code ...int) {
|
|
switch {
|
|
case err == nil,
|
|
errors.Is(err, context.Canceled),
|
|
errors.Is(err, syscall.EPIPE),
|
|
errors.Is(err, syscall.ECONNRESET):
|
|
return
|
|
}
|
|
LogError(r).Msg(err.Error())
|
|
if httpheaders.IsWebsocket(r.Header) {
|
|
return
|
|
}
|
|
if len(code) == 0 {
|
|
code = []int{http.StatusInternalServerError}
|
|
}
|
|
http.Error(w, http.StatusText(code[0]), code[0])
|
|
}
|
|
|
|
// ClientError is for responding to client errors.
|
|
//
|
|
// It returns http.StatusBadRequest with reason to the client.
|
|
// Status code can be specified as an argument.
|
|
//
|
|
// For JSON marshallable errors (e.g. gperr.Error), it returns the error details as JSON.
|
|
// Otherwise, it returns the error details as plain text.
|
|
func ClientError(w http.ResponseWriter, r *http.Request, err error, code ...int) {
|
|
if len(code) == 0 {
|
|
code = []int{http.StatusBadRequest}
|
|
}
|
|
w.WriteHeader(code[0])
|
|
accept := GetAccept(r.Header)
|
|
switch {
|
|
case accept.AcceptJSON():
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(err)
|
|
case accept.AcceptMarkdown():
|
|
w.Header().Set("Content-Type", "text/markdown")
|
|
w.Write(gperr.Markdown(err))
|
|
default:
|
|
w.Header().Set("Content-Type", "text/plain")
|
|
w.Write(gperr.Plain(err))
|
|
}
|
|
}
|
|
|
|
// JSONError returns a JSON response of gperr.Error with the given status code.
|
|
func JSONError(w http.ResponseWriter, err gperr.Error, code int) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.WriteHeader(code)
|
|
json.NewEncoder(w).Encode(err)
|
|
}
|
|
|
|
// BadRequest returns a Bad Request response with the given error message.
|
|
func BadRequest(w http.ResponseWriter, err string, code ...int) {
|
|
if len(code) == 0 {
|
|
code = []int{http.StatusBadRequest}
|
|
}
|
|
w.WriteHeader(code[0])
|
|
w.Write([]byte(err))
|
|
}
|
|
|
|
// Unauthorized returns an Unauthorized response with the given error message.
|
|
func Unauthorized(w http.ResponseWriter, err string) {
|
|
BadRequest(w, err, http.StatusUnauthorized)
|
|
}
|
|
|
|
// Forbidden returns a Forbidden response with the given error message.
|
|
func Forbidden(w http.ResponseWriter, err string) {
|
|
BadRequest(w, err, http.StatusForbidden)
|
|
}
|
|
|
|
// NotFound returns a Not Found response with the given error message.
|
|
func NotFound(w http.ResponseWriter, err string) {
|
|
BadRequest(w, err, http.StatusNotFound)
|
|
}
|
|
|
|
func MissingKey(w http.ResponseWriter, k string) {
|
|
BadRequest(w, k+" is required", http.StatusBadRequest)
|
|
}
|
|
|
|
func InvalidKey(w http.ResponseWriter, k string) {
|
|
BadRequest(w, k+" is invalid", http.StatusBadRequest)
|
|
}
|
|
|
|
func KeyAlreadyExists(w http.ResponseWriter, k, v string) {
|
|
BadRequest(w, fmt.Sprintf("%s %q already exists", k, v), http.StatusBadRequest)
|
|
}
|
|
|
|
func ValueNotFound(w http.ResponseWriter, k, v string) {
|
|
BadRequest(w, fmt.Sprintf("%s %q not found", k, v), http.StatusNotFound)
|
|
}
|