131 lines
3.8 KiB
Go
131 lines
3.8 KiB
Go
package controller
|
|
|
|
import (
|
|
"electricity_bill_calc/cache"
|
|
"electricity_bill_calc/exceptions"
|
|
"electricity_bill_calc/logger"
|
|
"electricity_bill_calc/model"
|
|
"electricity_bill_calc/repository"
|
|
"electricity_bill_calc/response"
|
|
"electricity_bill_calc/security"
|
|
"electricity_bill_calc/service"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
var userLog = logger.Named("Handler", "User")
|
|
|
|
func InitializeUserHandlers(router *fiber.App) {
|
|
router.Delete("/login", security.MustAuthenticated, doLogout)
|
|
router.Post("/login", doLogin)
|
|
router.Get("/account", security.OPSAuthorize, searchUsers)
|
|
router.Get("/expiration", security.EnterpriseAuthorize, getAccountExpiration)
|
|
}
|
|
|
|
type _LoginForm struct {
|
|
Username string `json:"uname"`
|
|
Password string `json:"upass"`
|
|
Type int16 `json:"type"`
|
|
}
|
|
|
|
func doLogin(c *fiber.Ctx) error {
|
|
result := response.NewResult(c)
|
|
loginData := new(_LoginForm)
|
|
if err := c.BodyParser(loginData); err != nil {
|
|
userLog.Error("表单解析失败！", zap.Error(err))
|
|
return result.Error(http.StatusInternalServerError, "表单解析失败。")
|
|
}
|
|
var (
|
|
session *model.Session
|
|
err error
|
|
)
|
|
userLog.Info("有用户请求登录。", zap.String("username", loginData.Username), zap.Int16("type", loginData.Type))
|
|
if loginData.Type == model.USER_TYPE_ENT {
|
|
session, err = service.UserService.ProcessEnterpriseUserLogin(loginData.Username, loginData.Password)
|
|
} else {
|
|
session, err = service.UserService.ProcessManagementUserLogin(loginData.Username, loginData.Password)
|
|
}
|
|
if err != nil {
|
|
if authError, ok := err.(*exceptions.AuthenticationError); ok {
|
|
if authError.NeedReset {
|
|
return result.LoginNeedReset()
|
|
}
|
|
return result.Error(int(authError.Code), authError.Message)
|
|
} else {
|
|
userLog.Error("用户登录请求处理失败！", zap.Error(err))
|
|
return result.Error(http.StatusInternalServerError, err.Error())
|
|
}
|
|
}
|
|
return result.LoginSuccess(session)
|
|
}
|
|
|
|
func doLogout(c *fiber.Ctx) error {
|
|
result := response.NewResult(c)
|
|
session, err := _retreiveSession(c)
|
|
if err != nil {
|
|
return result.Success("用户会话已结束。")
|
|
}
|
|
_, err = cache.ClearSession(session.Token)
|
|
if err != nil {
|
|
userLog.Error("用户登出处理失败！", zap.Error(err))
|
|
return result.Error(http.StatusInternalServerError, err.Error())
|
|
}
|
|
return result.Success("用户已成功登出系统。")
|
|
}
|
|
|
|
func searchUsers(c *fiber.Ctx) error {
|
|
result := response.NewResult(c)
|
|
requestPage, err := strconv.Atoi(c.Query("page", "1"))
|
|
if err != nil {
|
|
return result.NotAccept("查询参数[page]格式不正确。")
|
|
}
|
|
requestKeyword := c.Query("keyword")
|
|
requestUserType, err := strconv.Atoi(c.Query("type", "-1"))
|
|
if err != nil {
|
|
return result.NotAccept("查询参数[type]格式不正确。")
|
|
}
|
|
var requestUserStat *bool
|
|
state, err := strconv.ParseBool(c.Query("state"))
|
|
if err != nil {
|
|
requestUserStat = nil
|
|
} else {
|
|
requestUserStat = &state
|
|
}
|
|
users, total, err := repository.UserRepository.FindUser(
|
|
&requestKeyword,
|
|
int16(requestUserType),
|
|
requestUserStat,
|
|
uint(requestPage),
|
|
)
|
|
if err != nil {
|
|
return result.NotFound(err.Error())
|
|
}
|
|
return result.Json(
|
|
http.StatusOK,
|
|
"已取得符合条件的用户集合。",
|
|
response.NewPagedResponse(requestPage, total).ToMap(),
|
|
fiber.Map{"accounts": users},
|
|
)
|
|
}
|
|
|
|
func getAccountExpiration(c *fiber.Ctx) error {
|
|
result := response.NewResult(c)
|
|
session, err := _retreiveSession(c)
|
|
if err != nil {
|
|
userLog.Error("未找到有效的用户会话。", zap.Error(err))
|
|
return result.Error(http.StatusInternalServerError, err.Error())
|
|
}
|
|
userDetail, err := repository.UserRepository.FindUserDetailById(session.Uid)
|
|
if err != nil {
|
|
return result.NotFound("未找到指定的用户档案")
|
|
}
|
|
return result.Json(
|
|
http.StatusOK,
|
|
"已经取得用户的服务期限信息",
|
|
fiber.Map{"expiration": userDetail.ServiceExpiration.Format("2006-01-02")},
|
|
)
|
|
}
|