105 lines
2.2 KiB
Go
105 lines
2.2 KiB
Go
package render
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/a-h/templ"
|
|
"github.com/drhin/logger"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type Renderer interface {
|
|
Render(ctx context.Context, w http.ResponseWriter, t templ.Component)
|
|
|
|
JSON(w http.ResponseWriter, data any)
|
|
JSONObj(w http.ResponseWriter, message string, data any)
|
|
JSONOk(w http.ResponseWriter, message string)
|
|
JSONErr(w http.ResponseWriter, message string)
|
|
}
|
|
|
|
type render struct {
|
|
log *logger.Logger
|
|
}
|
|
|
|
func NewRender(log *logger.Logger) Renderer {
|
|
return &render{
|
|
log: log,
|
|
}
|
|
|
|
}
|
|
|
|
func (r *render) Render(ctx context.Context, w http.ResponseWriter, t templ.Component) {
|
|
if err := t.Render(ctx, w); err != nil {
|
|
r.log.Error(err.Error(), err,
|
|
zap.String("templ render", fmt.Sprintf("%v", t)),
|
|
)
|
|
}
|
|
}
|
|
|
|
func (r *render) JSON(w http.ResponseWriter, data any) {
|
|
respond(w, data)
|
|
}
|
|
|
|
func (r *render) JSONObj(w http.ResponseWriter, message string, data any) {
|
|
respond(w, Response{Success: true, Message: message, Data: data})
|
|
}
|
|
|
|
func (r *render) JSONOk(w http.ResponseWriter, message string) {
|
|
respond(w, Response{Success: true, Message: message})
|
|
}
|
|
|
|
func (r *render) JSONErr(w http.ResponseWriter, message string) {
|
|
respond(w, Response{Success: false, Message: message})
|
|
}
|
|
|
|
type Response struct {
|
|
Success bool `json:"success"`
|
|
Message string `json:"msg"`
|
|
Data any `json:"data"`
|
|
}
|
|
|
|
type ResponseTree struct {
|
|
Status ResponseTreeStatus `json:"status"`
|
|
Data any `json:"data"`
|
|
}
|
|
|
|
type ResponseTreeStatus struct {
|
|
Code int `json:"code"`
|
|
Message string `json:"message"`
|
|
}
|
|
|
|
type ResponseList struct {
|
|
Code int `json:"code"`
|
|
Message string `json:"msg"`
|
|
Count int64 `json:"count"`
|
|
Data any `json:"data"`
|
|
}
|
|
|
|
func NewResponseList(count int64, data any) ResponseList {
|
|
return ResponseList{
|
|
Code: 0,
|
|
Message: "OK",
|
|
Count: count,
|
|
Data: data,
|
|
}
|
|
}
|
|
|
|
func respond(w http.ResponseWriter, data any) {
|
|
v, err := json.Marshal(data)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
w.WriteHeader(http.StatusOK)
|
|
_, err = w.Write(v)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|