128 lines
3.4 KiB
Go
128 lines
3.4 KiB
Go
package system
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"management/internal/db/model/dto"
|
|
"management/internal/erpserver/biz"
|
|
"management/internal/erpserver/model/req"
|
|
"management/internal/global/know"
|
|
"management/internal/pkg/session"
|
|
"management/internal/pkg/tpl"
|
|
|
|
"github.com/zhang2092/browser"
|
|
)
|
|
|
|
type UserHandler interface {
|
|
Add(w http.ResponseWriter, r *http.Request)
|
|
Edit(w http.ResponseWriter, r *http.Request)
|
|
Save(w http.ResponseWriter, r *http.Request)
|
|
List(w http.ResponseWriter, r *http.Request)
|
|
|
|
UserExpansion
|
|
}
|
|
|
|
type UserExpansion interface {
|
|
Login(w http.ResponseWriter, r *http.Request)
|
|
Logout(w http.ResponseWriter, r *http.Request)
|
|
}
|
|
|
|
// userHandler 是 UserHandler 接口的实现.
|
|
type userHandler struct {
|
|
render tpl.Renderer
|
|
session session.ISession
|
|
biz biz.IBiz
|
|
}
|
|
|
|
// 确保 userHandler 实现了 UserHandler 接口.
|
|
var _ UserHandler = (*userHandler)(nil)
|
|
|
|
func NewUserHandler(render tpl.Renderer, session session.ISession, biz biz.IBiz) *userHandler {
|
|
return &userHandler{
|
|
render: render,
|
|
session: session,
|
|
biz: biz,
|
|
}
|
|
}
|
|
|
|
func (h *userHandler) Add(w http.ResponseWriter, r *http.Request) {}
|
|
|
|
func (h *userHandler) Edit(w http.ResponseWriter, r *http.Request) {}
|
|
|
|
func (h *userHandler) Save(w http.ResponseWriter, r *http.Request) {}
|
|
|
|
func (h *userHandler) List(w http.ResponseWriter, r *http.Request) {}
|
|
|
|
func (h *userHandler) Login(w http.ResponseWriter, r *http.Request) {
|
|
ctx := r.Context()
|
|
if r.Method == http.MethodGet {
|
|
var user dto.AuthorizeUser
|
|
u := h.session.GetBytes(ctx, know.StoreName)
|
|
if err := json.Unmarshal(u, &user); err == nil {
|
|
// 判断租户是否一致, 一致则刷新令牌，跳转到首页
|
|
if err := h.session.RenewToken(ctx); err == nil {
|
|
h.session.Put(ctx, know.StoreName, u)
|
|
http.Redirect(w, r, "/home.html", http.StatusFound)
|
|
return
|
|
}
|
|
}
|
|
|
|
h.session.Destroy(ctx)
|
|
h.render.HTML(w, r, "oauth/login.tmpl", nil)
|
|
return
|
|
} else if r.Method == http.MethodPost {
|
|
req := &req.Login{
|
|
Email: strings.TrimSpace(r.PostFormValue("email")),
|
|
Password: strings.TrimSpace(r.PostFormValue("password")),
|
|
CaptchaID: strings.TrimSpace(r.PostFormValue("captcha_id")),
|
|
Captcha: strings.TrimSpace(r.PostFormValue("captcha")),
|
|
Ip: r.RemoteAddr,
|
|
Referrer: r.Header.Get("Referer"),
|
|
Url: r.URL.RequestURI(),
|
|
}
|
|
|
|
if len(req.Email) == 0 {
|
|
h.render.JSON(w, tpl.Response{Success: false, Message: "请填写邮箱"})
|
|
return
|
|
}
|
|
if len(req.Password) == 0 {
|
|
h.render.JSON(w, tpl.Response{Success: false, Message: "请填写密码"})
|
|
return
|
|
}
|
|
if len(req.Captcha) == 0 {
|
|
h.render.JSON(w, tpl.Response{Success: false, Message: "请填写验证码"})
|
|
return
|
|
}
|
|
if !h.biz.CommonV1().CaptchaBiz().Verify(req.CaptchaID, req.Captcha, true) {
|
|
h.render.JSON(w, tpl.Response{Success: false, Message: "验证码错误"})
|
|
return
|
|
}
|
|
|
|
br, err := browser.NewBrowser(r.Header.Get("User-Agent"))
|
|
if err != nil {
|
|
h.render.JSONERR(w, "平台信息获取错误")
|
|
return
|
|
}
|
|
|
|
req.Os = br.Platform().Name()
|
|
req.Browser = br.Name()
|
|
err = h.biz.SystemV1().UserBiz().Login(ctx, req)
|
|
if err != nil {
|
|
h.render.JSONERR(w, err.Error())
|
|
return
|
|
}
|
|
|
|
h.render.JSONOK(w, "login successful")
|
|
return
|
|
}
|
|
|
|
http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
|
|
}
|
|
|
|
func (h *userHandler) Logout(w http.ResponseWriter, r *http.Request) {
|
|
h.session.Destroy(r.Context())
|
|
http.Redirect(w, r, "/", http.StatusFound)
|
|
}
|