projectx/internal/erpserver/handler/system/user.go

package system

import (
	"log"
	"net/http"

	"management/internal/db/model/dto"
	db "management/internal/db/sqlc"
	"management/internal/erpserver/model/form"
	systemmodel "management/internal/erpserver/model/system"
	v1 "management/internal/erpserver/service/v1"
	"management/internal/pkg/binding"
	"management/internal/pkg/convertor"
	"management/internal/pkg/middleware"
	"management/internal/pkg/tpl"
	"management/internal/pkg/tpl/html"
)

// userHandler 是 UserHandler 接口的实现.
type userHandler struct {
	render tpl.Renderer
	mi     middleware.Middleware

	captchasvc    v1.CaptchaService
	usersvc       v1.UserService
	rolesvc       v1.RoleService
	departmentsvc v1.DepartmentService
}

func NewUserHandler(
	render tpl.Renderer,
	mi middleware.Middleware,
	captchasvc v1.CaptchaService,
	usersvc v1.UserService,
	rolesvc v1.RoleService,
	departmentsvc v1.DepartmentService,
) *userHandler {
	return &userHandler{
		render:        render,
		mi:            mi,
		captchasvc:    captchasvc,
		usersvc:       usersvc,
		rolesvc:       rolesvc,
		departmentsvc: departmentsvc,
	}
}

func (h *userHandler) Add(w http.ResponseWriter, r *http.Request) {
	h.render.HTML(w, r, "user/edit.tmpl", map[string]any{
		"Item": &db.SysUser{
			HashedPassword: nil,
		},
	})
}

func (h *userHandler) Edit(w http.ResponseWriter, r *http.Request) {
	vars := r.URL.Query()
	id := convertor.QueryInt[int32](vars, "id", 0)
	user := &systemmodel.User{}
	if id > 0 {
		ctx := r.Context()
		if u, err := h.usersvc.Get(ctx, id); err == nil {
			user.HashedPassword = []byte("********")
			user = u
		}
	}
	h.render.HTML(w, r, "user/edit.tmpl", map[string]any{
		"Item": user,
	})
}

func (h *userHandler) Save(w http.ResponseWriter, r *http.Request) {
	var req form.User
	if err := binding.Form.Bind(r, &req); err != nil {
		h.render.JSONERR(w, binding.ValidatorErrors(err))
		return
	}

	ctx := r.Context()
	if req.DepartmentID > 0 {
		if _, err := h.departmentsvc.Get(ctx, req.DepartmentID); err != nil {
			h.render.JSONERR(w, "部门数据错误")
			return
		}
	}
	if req.RoleID > 0 {
		if _, err := h.rolesvc.Get(ctx, req.RoleID); err != nil {
			h.render.JSONERR(w, "角色数据错误")
			return
		}
	}

	if *req.ID == 0 {
		err := h.usersvc.Create(ctx, &req)
		if err != nil {
			h.render.JSONERR(w, err.Error())
			return
		}

		h.render.JSONOK(w, "添加成功")
	} else {
		err := h.usersvc.Update(ctx, &req)
		if err != nil {
			h.render.JSONERR(w, err.Error())
			return
		}

		h.render.JSONOK(w, "更新成功")
	}
}

func (h *userHandler) List(w http.ResponseWriter, r *http.Request) {
	switch r.Method {
	case http.MethodGet:
		h.render.HTML(w, r, "user/list.tmpl", map[string]any{
			"Statuses": html.NewSelectControls(html.SearchStatuses, "0"),
		})
	case http.MethodPost:
		var q dto.SearchDto
		q.SearchTimeBegin, q.SearchTimeEnd = convertor.DefaultStartTimeAndEndTime(r.PostFormValue("timeBegin"), r.PostFormValue("timeEnd"))
		q.SearchStatus = convertor.ConvertInt(r.PostFormValue("status"), 9999)
		q.SearchName = r.PostFormValue("name")
		q.SearchEmail = r.PostFormValue("email")
		q.SearchID = convertor.ConvertInt[int64](r.PostFormValue("id"), 0)
		q.Page = convertor.ConvertInt(r.PostFormValue("page"), 1)
		q.Rows = convertor.ConvertInt(r.PostFormValue("rows"), 10)
		res, count, err := h.usersvc.List(r.Context(), q)
		if err != nil {
			h.render.JSONERR(w, err.Error())
			return
		}

		data := tpl.ResponseList{
			Code:    0,
			Message: "ok",
			Count:   count,
			Data:    res,
		}
		h.render.JSON(w, data)
	default:
		http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
	}
}

func (h *userHandler) Profile(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	user := h.mi.AuthUser(ctx)
	vm, _ := h.usersvc.Get(ctx, user.ID)
	h.render.HTML(w, r, "user/profile.tmpl", map[string]any{
		"Item": vm,
	})
}

func (h *userHandler) Data(w http.ResponseWriter, r *http.Request) {
	vars := r.URL.Query()
	t := vars.Get("type")
	if t == "xmselect" {
		res, err := h.usersvc.XmSelect(r.Context())
		if err != nil {
			h.render.JSONERR(w, err.Error())
			return
		}

		h.render.JSON(w, res)
		return
	}

	h.render.JSON(w, nil)
}

func (h *userHandler) Login(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	switch r.Method {
	case http.MethodGet:
		if h.mi.IsAuth(ctx) && h.mi.RefreshToken(ctx) {
			http.Redirect(w, r, "/home.html", http.StatusFound)
			return
		}
		_ = h.mi.Destroy(ctx)
		h.render.HTML(w, r, "oauth/login.tmpl", nil)
	case http.MethodPost:
		defer r.Body.Close()
		var req form.Login
		if err := binding.Form.Bind(r, &req); err != nil {
			e := binding.ValidatorErrors(err)
			h.render.JSONERR(w, e)
			return
		}

		if !h.captchasvc.Verify(req.CaptchaID, req.Captcha, true) {
			h.render.JSONERR(w, "验证码错误")
			return
		}

		req = req.SetAttributes(r)
		err := h.usersvc.Login(ctx, &req)
		if err != nil {
			log.Println(err)
			h.render.JSONERR(w, err.Error())
			return
		}

		h.render.JSONOK(w, "login successfully")
	default:
		http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
	}
}

func (h *userHandler) Logout(w http.ResponseWriter, r *http.Request) {
	h.mi.Destroy(r.Context())
	http.Redirect(w, r, "/", http.StatusFound)
}