becomer/likedream
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐛 http代理关闭 代理用户鉴权忽略admin

Browse Source
This commit is contained in:
sugarBecomer
2025-11-03 14:30:40 +08:00
parent 75931dfde6
commit c9aeb90209
2 changed files with 55 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/application/httpclient/cpdaily.go
+3 -2
View File
@@ -9,5 +9,6 @@ import (
var Cpdaily = resty.New().
SetCookieJar(nil).
SetTimeout(30 * time.Second).
SetDoNotParseResponse(true).
SetProxy("http://127.0.0.1:9000")
SetDoNotParseResponse(true)
//SetProxy("http://127.0.0.1:9000")
internal/service/agent/agent_service.go
+52 -41
View File
@@ -22,7 +22,9 @@ func NewV1() *AgentServiceV1 {
func (s *AgentServiceV1) List(ctx context.Context, req *v1.AgentListReq) *utils.Response[[]v1.AgentListRes] {
query := db.Global.SysUser.Query()
query.Where(sysuser.HasParentWith(sysuser.IDEQ(utils.GetUserInfoFromCtx(ctx).ID)))
if !utils.IsAdminFromCtx(ctx) {
query.Where(sysuser.HasParentWith(sysuser.IDEQ(utils.GetUserInfoFromCtx(ctx).ID)))
}
if req.Email != "" {
query.Where(sysuser.EmailContains(req.Email))
}
@@ -109,13 +111,15 @@ func (s *AgentServiceV1) Add(ctx context.Context, req *v1.AgentAddReq) *utils.Re
}
func (s *AgentServiceV1) Ban(ctx context.Context, req *v1.AgentBanReq) *utils.Response[any] {
// 参数校验
currentUser := utils.GetUserInfoFromCtx(ctx)
user := db.Global.SysUser.Query().Where(sysuser.IDEQ(req.ID)).OnlyX(ctx)
// 判断是否是自己的下级代理
parent := user.QueryParent().OnlyX(ctx)
if parent.ID != currentUser.ID {
return utils.FailParam("非法的下级代理参数")
if !utils.IsAdminFromCtx(ctx) {
// 参数校验
currentUser := utils.GetUserInfoFromCtx(ctx)
// 判断是否是自己的下级代理
parent := user.QueryParent().OnlyX(ctx)
if parent.ID != currentUser.ID {
return utils.FailParam("非法的下级代理参数")
}
}
db.Global.SysUser.UpdateOne(user).SetBlock(req.Block).ExecX(ctx)
return utils.SuccessWithMsg("操作成功")
@@ -123,33 +127,37 @@ func (s *AgentServiceV1) Ban(ctx context.Context, req *v1.AgentBanReq) *utils.Re
func (s *AgentServiceV1) Recharge(ctx context.Context, req *v1.AgentRechargeReq) *utils.Response[any] {
// 参数校验
currentUser := db.Global.SysUser.Query().Where(sysuser.ID(utils.GetUserInfoFromCtx(ctx).ID)).WithAgent().WithRoles().OnlyX(ctx)
user := db.Global.SysUser.Query().Where(sysuser.IDEQ(req.ID)).WithAgent().OnlyX(ctx)
// 判断是否是自己的下级代理
parent := user.QueryParent().WithAgent().FirstX(ctx)
if parent == nil || parent.ID != currentUser.ID {
return utils.FailParam("非法的下级代理参数")
}
tx, err := db.Global.Tx(ctx)
utils.OkOrPanic(err, func(err error) error {
return oops.
WithContext(ctx).
With("创建事务失败").
Wrap(err)
})
if !utils.HasRole(currentUser.Edges.Roles, cst.KeyRoleAdmin) {
parent.Balance = utils.CalculateBalance(parent.Balance, parent.Edges.Agent.Rate, req.Amount, user.Edges.Agent.Rate)
if parent.Balance.Compare(decimal.NewFromInt(0)) < 0 {
return utils.FailParam("余额不足")
defer tx.Rollback()
if !utils.IsAdminFromCtx(ctx) {
currentUser := db.Global.SysUser.Query().Where(sysuser.ID(utils.GetUserInfoFromCtx(ctx).ID)).WithAgent().WithRoles().OnlyX(ctx)
// 判断是否是自己的下级代理
parent := user.QueryParent().WithAgent().FirstX(ctx)
if parent == nil || parent.ID != currentUser.ID {
return utils.FailParam("非法的下级代理参数")
}
utils.OkOrPanic(err, func(err error) error {
return oops.
WithContext(ctx).
With("创建事务失败").
Wrap(err)
})
if !utils.HasRole(currentUser.Edges.Roles, cst.KeyRoleAdmin) {
parent.Balance = utils.CalculateBalance(parent.Balance, parent.Edges.Agent.Rate, req.Amount, user.Edges.Agent.Rate)
if parent.Balance.Compare(decimal.NewFromInt(0)) < 0 {
return utils.FailParam("余额不足")
}
}
tx.SysUser.UpdateOne(parent).SetBalance(parent.Balance).SaveX(ctx)
utils.OkOrPanic(err, func(err error) error {
return oops.
WithContext(ctx).
With("更新用户余额失败").
Wrap(err)
})
}
tx.SysUser.UpdateOne(parent).SetBalance(parent.Balance).SaveX(ctx)
utils.OkOrPanic(err, func(err error) error {
return oops.
WithContext(ctx).
With("更新用户余额失败").
Wrap(err)
})
tx.SysUser.UpdateOne(user).SetBalance(user.Balance.Add(req.Amount)).SaveX(ctx)
utils.OkOrPanic(err, func(err error) error {
return oops.
@@ -170,18 +178,21 @@ func (s *AgentServiceV1) AgentPutInfoReq(ctx context.Context, req *v1.AgentPutIn
With("创建事务失败").
Wrap(err)
})
currentUser := tx.SysUser.Query().Where(sysuser.ID(utils.GetUserInfoFromCtx(ctx).ID)).WithAgent().WithRoles().OnlyX(ctx)
defer tx.Rollback()
user := tx.SysUser.Query().Where(sysuser.ID(req.ID)).WithParent().WithAgent().FirstX(ctx)
if user.Edges.Parent == nil || user.Edges.Parent.ID != currentUser.ID {
return utils.FailParam("非法的下级代理参数")
}
// 2. 如果没有代理权限
if !utils.HasRole(currentUser.Edges.Roles, cst.KeyRoleAdmin) && !utils.HasRole(currentUser.Edges.Roles, cst.KeyRoleAgent) {
return utils.FailParam("代理权限不足")
}
// 3. 检测代理费率或level是否低于自己
if currentUser.Edges.Agent != nil && (req.Level > currentUser.Edges.Agent.Level || req.Rate.LessThan(currentUser.Edges.Agent.Rate)) {
return utils.FailParam("代理等级或费率低于自己")
if !utils.IsAdminFromCtx(ctx) {
currentUser := tx.SysUser.Query().Where(sysuser.ID(utils.GetUserInfoFromCtx(ctx).ID)).WithAgent().WithRoles().OnlyX(ctx)
if user.Edges.Parent == nil || user.Edges.Parent.ID != currentUser.ID {
return utils.FailParam("非法的下级代理参数")
}
// 2. 如果没有代理权限
if !utils.HasRole(currentUser.Edges.Roles, cst.KeyRoleAdmin) && !utils.HasRole(currentUser.Edges.Roles, cst.KeyRoleAgent) {
return utils.FailParam("代理权限不足")
}
// 3. 检测代理费率或level是否低于自己
if currentUser.Edges.Agent != nil && (req.Level > currentUser.Edges.Agent.Level || req.Rate.LessThan(currentUser.Edges.Agent.Rate)) {
return utils.FailParam("代理等级或费率低于自己")
}
}
// 5. 检测是否存在代理信息，
// 将用户余额先计算为比例为1的用户余额