File size: 1,565 Bytes
1f10f31 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 | package render
import (
"encoding/json"
"net/http"
"github.com/invopop/yaml"
)
const (
ContentTypeKey = "Content-Type"
ContentTypeJSONValue = "application/json"
ContentTypeYAMLValue = "application/yaml"
defaultStatusCode = http.StatusOK
)
type config struct {
statusCode int
}
type Option func(w http.ResponseWriter, c *config)
func WithStatus(status int) Option {
return func(_ http.ResponseWriter, c *config) {
c.statusCode = status
}
}
func WithHeader(key, value string) Option {
return func(w http.ResponseWriter, _ *config) {
w.Header().Set(key, value)
}
}
func WithContentType(contentType string) Option {
return WithHeader(ContentTypeKey, contentType)
}
// RenderJSON renders json object with returned apierrors
func RenderJSON[O any](w http.ResponseWriter, o O, opts ...Option) error {
c := &config{
statusCode: defaultStatusCode,
}
for _, opt := range opts {
opt(w, c)
}
if w.Header().Get(ContentTypeKey) == "" {
w.Header().Set(ContentTypeKey, ContentTypeJSONValue)
}
w.WriteHeader(c.statusCode)
body, err := json.Marshal(o)
if err != nil {
return err
}
_, err = w.Write(body)
return err
}
func RenderYAML[O any](w http.ResponseWriter, o O, opts ...Option) error {
c := &config{
statusCode: defaultStatusCode,
}
for _, opt := range opts {
opt(w, c)
}
if w.Header().Get(ContentTypeKey) == "" {
w.Header().Set(ContentTypeKey, ContentTypeYAMLValue)
}
w.WriteHeader(c.statusCode)
body, err := yaml.Marshal(o)
if err != nil {
return err
}
_, err = w.Write(body)
return err
}
|