mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-29 18:36:30 +02:00
f719d885b7
* authenticate: remove unused paths, generate cipher on boot - internal/httputil: add JSON renderer - internal/httputil: remove unused query param store and references Signed-off-by: Bobby DeSimone <bobbydesimone@gmail.com>
66 lines
2 KiB
Go
66 lines
2 KiB
Go
package httputil
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
)
|
|
|
|
// HealthCheck is a simple healthcheck handler that responds to GET and HEAD
|
|
// http requests.
|
|
func HealthCheck(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodGet && r.Method != http.MethodHead {
|
|
http.Error(w, http.StatusText(http.StatusMethodNotAllowed), http.StatusMethodNotAllowed)
|
|
return
|
|
}
|
|
w.Header().Set("Content-Type", "text/plain")
|
|
w.WriteHeader(http.StatusOK)
|
|
if r.Method == http.MethodGet {
|
|
fmt.Fprintln(w, http.StatusText(http.StatusOK))
|
|
}
|
|
}
|
|
|
|
// Redirect wraps the std libs's redirect method indicating that pomerium is
|
|
// the origin of the response.
|
|
func Redirect(w http.ResponseWriter, r *http.Request, url string, code int) {
|
|
w.Header().Set(HeaderPomeriumResponse, "true")
|
|
http.Redirect(w, r, url, code)
|
|
}
|
|
|
|
// RenderJSON replies to the request with the specified struct as JSON and HTTP code.
|
|
// It does not otherwise end the request; the caller should ensure no further
|
|
// writes are done to w.
|
|
// The error message should be application/json.
|
|
func RenderJSON(w http.ResponseWriter, code int, v interface{}) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.Header().Set("X-Content-Type-Options", "nosniff")
|
|
b := new(bytes.Buffer)
|
|
if err := json.NewEncoder(b).Encode(v); err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
fmt.Fprintf(b, `{"error":"%s"}`, err)
|
|
} else {
|
|
w.WriteHeader(code)
|
|
}
|
|
fmt.Fprint(w, b)
|
|
}
|
|
|
|
// The HandlerFunc type is an adapter to allow the use of
|
|
// ordinary functions as HTTP handlers. If f is a function
|
|
// with the appropriate signature, HandlerFunc(f) is a
|
|
// Handler that calls f.
|
|
//
|
|
// adapted from std library to suppport error wrapping
|
|
type HandlerFunc func(http.ResponseWriter, *http.Request) error
|
|
|
|
// ServeHTTP calls f(w, r) error.
|
|
func (f HandlerFunc) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
if err := f(w, r); err != nil {
|
|
var e *HTTPError
|
|
if !errors.As(err, &e) {
|
|
e = &HTTPError{http.StatusInternalServerError, err}
|
|
}
|
|
e.ErrorResponse(w, r)
|
|
}
|
|
}
|