139 lines
3.4 KiB
Go
139 lines
3.4 KiB
Go
package handle
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"tinyskin/handle/handelerror"
|
|
"tinyskin/model"
|
|
"tinyskin/service"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-playground/validator/v10"
|
|
|
|
U "tinyskin/utils"
|
|
)
|
|
|
|
type AdminHandel struct {
|
|
handleError *handelerror.HandleError
|
|
adminService *service.AdminService
|
|
validate *validator.Validate
|
|
}
|
|
|
|
func NewAdminHandel(handleError *handelerror.HandleError, adminService *service.AdminService, validate *validator.Validate) *AdminHandel {
|
|
return &AdminHandel{
|
|
handleError: handleError,
|
|
adminService: adminService,
|
|
validate: validate,
|
|
}
|
|
}
|
|
|
|
type tokenValue string
|
|
|
|
const tokenKey = tokenValue("token")
|
|
|
|
func (h *AdminHandel) NeedAuth(handle http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
ctx := r.Context()
|
|
token := h.getTokenbyAuthorization(ctx, w, r)
|
|
if token == "" {
|
|
return
|
|
}
|
|
t, err := h.adminService.Auth(ctx, token)
|
|
if err != nil {
|
|
h.handleError.Service(ctx, w, err)
|
|
return
|
|
}
|
|
r = r.WithContext(context.WithValue(ctx, tokenKey, t))
|
|
handle.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
|
|
func (h *AdminHandel) NeedAdmin(handle http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
ctx := r.Context()
|
|
t := ctx.Value(tokenKey).(*model.TokenClaims)
|
|
err := h.adminService.IsAdmin(ctx, t)
|
|
if err != nil {
|
|
h.handleError.Service(ctx, w, err)
|
|
return
|
|
}
|
|
handle.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
|
|
func (h *AdminHandel) ListUser() http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
ctx := r.Context()
|
|
page := r.FormValue("page")
|
|
pagei := 1
|
|
if page != "" {
|
|
p, err := strconv.Atoi(page)
|
|
if err != nil {
|
|
h.handleError.Error(ctx, w, "page 必须为数字", model.ErrInput, 400, slog.LevelDebug)
|
|
return
|
|
}
|
|
if p == 0 {
|
|
p = 1
|
|
}
|
|
pagei = p
|
|
}
|
|
email := r.FormValue("email")
|
|
name := r.FormValue("name")
|
|
|
|
ul, uc, err := h.adminService.ListUser(ctx, pagei, email, name)
|
|
if err != nil {
|
|
h.handleError.Service(ctx, w, err)
|
|
return
|
|
}
|
|
encodeJson(w, model.API[model.List[model.UserList]]{Data: model.List[model.UserList]{List: ul, Total: uc}})
|
|
}
|
|
}
|
|
|
|
func (h *AdminHandel) EditUser() http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
ctx := r.Context()
|
|
uid := chi.URLParamFromCtx(ctx, "uid")
|
|
if uid == "" {
|
|
h.handleError.Error(ctx, w, "uid 为空", model.ErrInput, 400, slog.LevelDebug)
|
|
return
|
|
}
|
|
uidi, err := strconv.Atoi(uid)
|
|
if err != nil {
|
|
h.handleError.Error(ctx, w, err.Error(), model.ErrInput, 400, slog.LevelDebug)
|
|
return
|
|
}
|
|
|
|
a, err := U.DeCodeBody[model.EditUser](r.Body, h.validate)
|
|
if err != nil {
|
|
h.handleError.Error(ctx, w, err.Error(), model.ErrInput, 400, slog.LevelDebug)
|
|
return
|
|
}
|
|
err = h.adminService.EditUser(ctx, a, uidi)
|
|
if err != nil {
|
|
h.handleError.Service(ctx, w, err)
|
|
return
|
|
}
|
|
encodeJson[any](w, model.API[any]{
|
|
Code: 0,
|
|
})
|
|
}
|
|
}
|
|
|
|
func (h *AdminHandel) getTokenbyAuthorization(ctx context.Context, w http.ResponseWriter, r *http.Request) string {
|
|
auth := r.Header.Get("Authorization")
|
|
if auth == "" {
|
|
h.handleError.Error(ctx, w, "缺少 Authorization", model.ErrAuth, 401, slog.LevelDebug)
|
|
return ""
|
|
}
|
|
al := strings.Split(auth, " ")
|
|
if len(al) != 2 || al[0] != "Bearer" {
|
|
h.handleError.Error(ctx, w, "Authorization 格式错误", model.ErrAuth, 401, slog.LevelDebug)
|
|
return ""
|
|
}
|
|
return al[1]
|
|
}
|