48 lines
1.7 KiB
Go
48 lines
1.7 KiB
Go
package controller
|
|
|
|
import (
|
|
"electricity_bill_calc/exceptions"
|
|
"electricity_bill_calc/model"
|
|
"electricity_bill_calc/repository"
|
|
"electricity_bill_calc/response"
|
|
"net/http"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func _retreiveSession(c *fiber.Ctx) (*model.Session, error) {
|
|
session := c.Locals("session")
|
|
if session == nil {
|
|
return nil, exceptions.NewUnauthorizedError("用户会话不存在")
|
|
}
|
|
userSession, ok := session.(*model.Session)
|
|
if !ok {
|
|
return nil, exceptions.NewUnauthorizedError("用户会话格式不正确，需要重新登录")
|
|
}
|
|
return userSession, nil
|
|
}
|
|
|
|
// 检查当前用户是否拥有指定园区，在判断完成之后直接产生响应
|
|
func checkParkBelongs(parkId string, logger *zap.Logger, c *fiber.Ctx, result *response.Result) (bool, error) {
|
|
session := c.Locals("session")
|
|
if session == nil {
|
|
logger.Error("用户会话不存在。")
|
|
return false, result.Unauthorized("用户会话不存在。")
|
|
}
|
|
userSession, ok := session.(*model.Session)
|
|
if !ok {
|
|
return false, result.Unauthorized("用户会话格式不正确，需要重新登录")
|
|
}
|
|
ok, err := repository.ParkRepository.IsParkBelongs(parkId, userSession.Uid)
|
|
switch {
|
|
case err != nil:
|
|
logger.Error("无法判断园区是否隶属于当前用户。", zap.String("park id", parkId), zap.String("user id", userSession.Uid), zap.Error(err))
|
|
return false, result.Error(http.StatusInternalServerError, err.Error())
|
|
case err == nil && !ok:
|
|
logger.Error("用户试图访问不属于自己的园区。", zap.String("park id", parkId), zap.String("user id", userSession.Uid))
|
|
return false, result.Forbidden("您无权访问该园区。")
|
|
}
|
|
return true, nil
|
|
}
|