blog-server/internal/handler/sys_user.go

package handler

import (
	"net/http"
	"server/internal/middleware"
	"server/internal/model/common"
	"server/internal/model/request"
	"server/internal/pkg/errs"
	"server/internal/pkg/httputil"
	"server/internal/service"
)

type SysUserHandler struct {
	userService *service.SysUserService
}

func NewSysUserHandler(userService *service.SysUserService) *SysUserHandler {
	return &SysUserHandler{
		userService: userService,
	}
}

func (h *SysUserHandler) ListPage(w http.ResponseWriter, r *http.Request) {
	pagination := httputil.Pagination(r)

	params := request.SearchSysUserParams{
		Pagination: *pagination,
		Username:   r.URL.Query().Get("username"),
	}

	list, total, err := h.userService.ListPage(r.Context(), params)
	if err != nil {
		httputil.Fail(w, err)
		return
	}

	resp := common.PageResponse{
		Page:     pagination.Page,
		PageSize: pagination.PageSize,
		List:     list,
		Total:    total,
	}

	httputil.OkWithPage(w, &resp)
}

func (h *SysUserHandler) GetUserInfo(w http.ResponseWriter, r *http.Request) {
	claims, ok := middleware.GetClaims(r.Context())

	if !ok || claims.UserID == 0 {
		httputil.Fail(w, errs.ErrUnauthorized)
		return
	}

	isAdmin, _ := r.Context().Value(middleware.IsAdminKey).(bool)
	uid := claims.UserID

	user, err := h.userService.GetUserInfo(r.Context(), uid, isAdmin)
	if err != nil {
		httputil.Fail(w, err)
		return
	}

	httputil.Ok(w, user)
}

func (h *SysUserHandler) GetRoles(w http.ResponseWriter, r *http.Request) {
	id, err := httputil.URLParamInt32(r, "id")
	if err != nil {
		httputil.Fail(w, err)
		return
	}

	list, err := h.userService.GetRoles(r.Context(), id)
	if err != nil {
		httputil.Fail(w, err)
		return
	}

	httputil.Ok(w, list)
}

func (h *SysUserHandler) Create(w http.ResponseWriter, r *http.Request) {
	var req request.CreateSysUserRequest

	if err := httputil.BindJson(r, &req); err != nil {
		httputil.Fail(w, err)
		return
	}

	if err := h.userService.Create(r.Context(), req); err != nil {
		httputil.Fail(w, err)
		return
	}

	httputil.Ok(w)
}

func (h *SysUserHandler) Update(w http.ResponseWriter, r *http.Request) {
	var req request.UpdateSysUserRequest

	id, err := httputil.URLParamInt32(r, "id")
	if err != nil {
		httputil.Fail(w, err)
		return
	}

	if err = httputil.BindJson(r, &req); err != nil {
		httputil.Fail(w, err)
		return
	}

	if err = h.userService.Update(r.Context(), id, req); err != nil {
		httputil.Fail(w, err)
		return
	}

	httputil.Ok(w)
}

func (h *SysUserHandler) SetRoles(w http.ResponseWriter, r *http.Request) {
	var req request.SetSysUserRolesRequest

	id, err := httputil.URLParamInt32(r, "id")
	if err != nil {
		httputil.Fail(w, err)
		return
	}

	if err = httputil.BindJson(r, &req); err != nil {
		httputil.Fail(w, err)
		return
	}

	if err = h.userService.SetRoles(r.Context(), id, req); err != nil {
		httputil.Fail(w, err)
		return
	}

	httputil.Ok(w)

}

func (h *SysUserHandler) UpdatePassword(w http.ResponseWriter, r *http.Request) {
	var req request.UpdateSysUserPassword

	id, err := httputil.URLParamInt32(r, "id")
	if err != nil {
		httputil.Fail(w, err)
		return
	}

	if err = httputil.BindJson(r, &req); err != nil {
		httputil.Fail(w, err)
		return
	}

	if err = h.userService.UpdatePassword(r.Context(), id, req); err != nil {
		httputil.Fail(w, err)
		return
	}

	httputil.Ok(w)
}

func (h *SysUserHandler) Delete(w http.ResponseWriter, r *http.Request) {
	id, err := httputil.URLParamInt32(r, "id")
	if err != nil {
		httputil.Fail(w, err)
		return
	}

	if err = h.userService.Delete(r.Context(), id); err != nil {
		httputil.Fail(w, err)
		return
	}

	httputil.Ok(w)
}