admin-go/internal/app/system/logic/tenant/tenant.go

/*
* @desc:用户处理
* @company:云南奇讯科技有限公司
* @Author: yixiaohu<yxh669@qq.com>
* @Date:   2022/9/23 15:08
 */

package tenant

import (
	"context"
	"database/sql"
	"errors"
	"fmt"

	"gitea.com/red-future/common/beans"
	"gitea.com/red-future/common/utils"
	"github.com/gogf/gf/v2/container/gset"
	"github.com/gogf/gf/v2/database/gdb"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/util/gconv"
	"github.com/gogf/gf/v2/util/grand"
	"github.com/tiger1103/gfast/v3/api/v1/system"
	commonService "github.com/tiger1103/gfast/v3/internal/app/common/service"
	"github.com/tiger1103/gfast/v3/internal/app/system/consts"
	"github.com/tiger1103/gfast/v3/internal/app/system/dao"
	"github.com/tiger1103/gfast/v3/internal/app/system/model/do"
	"github.com/tiger1103/gfast/v3/internal/app/system/model/entity"
	"github.com/tiger1103/gfast/v3/internal/app/system/service"
	"github.com/tiger1103/gfast/v3/library/libUtils"
	"github.com/tiger1103/gfast/v3/library/liberr"
)

func init() {
	service.RegisterTenant(New())
}

type sTenant struct {
}

func New() *sTenant {
	return &sTenant{}
}

func (s *sTenant) GetTenantListSearch(ctx context.Context, req *system.TenantListReq) (res *system.TenantListRes, err error) {
	res = new(system.TenantListRes)
	err = g.Try(ctx, func(ctx context.Context) {
		res.ImgAddressPrefix, err = utils.GetFileAddressPrefix(ctx)
		liberr.ErrIsNil(ctx, err, "获取文件地址前缀失败")
		model := dao.TenantDao.Ctx(ctx)
		// 根据toke获取用户id
		userId := service.Context().GetUserId(ctx)
		getUserById, err := service.SysUser().GetUserById(ctx, userId)
		liberr.ErrIsNil(ctx, err, "获取用户信息失败")
		roleIds, err := service.SysUser().GetAdminRoleIds(ctx, userId)
		// 根据用户角色查询对应数据（9代理，10普通）
		for _, v := range roleIds {
			if v == consts.SalesAgentId || v == consts.SiteAdminId {
				model = model.Where(dao.TenantDao.Columns().TenantSource, gconv.Int64(getUserById.TenantId))
			}
		}
		if !g.IsEmpty(req.TenantName) {
			model = model.Where(dao.TenantDao.Columns().TenantName+" like ? ", "%"+req.TenantName+"%")
		}
		if !g.IsEmpty(req.TenantType) {
			model = model.Where(dao.TenantDao.Columns().TenantType, gconv.Int(req.TenantType))
		}
		if !g.IsEmpty(req.CityCode) {
			model = model.Where(dao.TenantDao.Columns().CityCode, gconv.String(req.CityCode))
		}
		res.Total, err = model.Count()
		liberr.ErrIsNil(ctx, err, "获取租户数据失败")
		if req.PageNum == 0 {
			req.PageNum = 1
		}
		res.CurrentPage = req.PageNum
		if req.PageSize == 0 {
			req.PageSize = consts.PageSize
		}
		err = model.Page(res.CurrentPage, req.PageSize).Order(dao.TenantDao.Columns().CreatedAt + " desc").Scan(&res.List)
		liberr.ErrIsNil(ctx, err, "获取数据失败")
		var list []map[string]interface{}
		err = utils.Struct(res.List, &list)
		liberr.ErrIsNil(ctx, err, "转换数据失败")
		// 2. 提取id
		userIds := make([]int, 0, len(list))
		cityIds := make([]string, 0, len(list))
		for _, obj := range list {
			// 提取id并转换（JSON数字是float64）
			idFloat, ok := obj["adminBy"].(float64)
			if !ok {
				fmt.Println("警告：adminBy字段非数字")
				continue
			}
			cityFloat, ok := obj["cityCode"].(string)
			if !ok {
				fmt.Println("警告：cityCode字段非字符串")
				continue
			}
			userIds = append(userIds, int(idFloat))
			cityIds = append(cityIds, cityFloat)
		}

		userList, err := service.SysUser().GetUsers(ctx, userIds)
		liberr.ErrIsNil(ctx, err, "获取租户信息失败")
		areaDicList, err := service.AreaDict().GetAreaDict(ctx)
		liberr.ErrIsNil(ctx, err, "获取地区字典失败")
		for _, list := range res.List {
			for _, areas := range areaDicList {
				if fmt.Sprintf("%d", areas.Id) == list.CityCode {
					list.CityName = areas.CityName
					list.CityMergerName = areas.MergerName
				}
			}
			for _, users := range userList {
				if users.Id == list.AdminBy {
					list.UserName = users.UserName
					list.Mobile = users.Mobile
					list.UserNickname = users.UserNickname
				}
			}
		}
	})
	return
}

func (s *sTenant) Add(ctx context.Context, req *system.TenantAddReq) (err error) {
	err = g.DB().Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
		err = g.Try(ctx, func(ctx context.Context) {
			// 根据token获取用户id
			userId := service.Context().GetUserId(ctx)
			var getUserInfo *beans.User
			getUserInfo, err = utils.GetUserInfo(ctx)
			var userTenantId uint64
			if err == nil && !g.IsEmpty(getUserInfo.TenantId) {
				userTenantId = gconv.Uint64(getUserInfo.TenantId)
			}
			if !g.IsEmpty(userId) {
				var roleIds []uint
				roleIds, err = service.SysUser().GetAdminRoleIds(ctx, userId)
				liberr.ErrIsNil(ctx, err, "获取用户角色失败")
				// 如果是代理和普通用户只能添加普通类型（1普通类型，2代理类型）
				for _, v := range roleIds {
					if v == consts.SalesAgentId || v == consts.SiteAdminId {
						req.TenantType = consts.TenantTypeSite
						if v == consts.SalesAgentId {
							// 验证代理区域是否在当前代理的管辖范围内
							tenantEntity := new(entity.Tenant)
							err = dao.TenantDao.Ctx(ctx).TX(tx).Where(dao.TenantDao.Columns().Id, userTenantId).Scan(tenantEntity)
							liberr.ErrIsNil(ctx, err, "获取租户数据失败")
							var count int
							count, err = dao.AreaDictDao.Ctx(ctx).TX(tx).Where("("+dao.AreaDictDao.Columns().Id+"=? OR "+dao.AreaDictDao.Columns().ParentId+"=?) AND "+dao.AreaDictDao.Columns().Id+"=?", tenantEntity.CityCode, tenantEntity.CityCode, req.CityCode).Count()
							liberr.ErrIsNil(ctx, err, "验证代理区域失败")
							if count == 0 {
								liberr.ErrIsNil(ctx, errors.New("所选城市不在当前代理的管辖范围内"))
							}
						} else {
							if req.TenantType == consts.TenantTypeAgent {
								liberr.ErrIsNil(ctx, errors.New("代理商不能添加代理商"))
							}
						}
						break
					}
				}
			} else {
				// 如果是普通用户只能添加普通类型（1普通类型，2代理类型）
				req.TenantType = consts.TenantTypeSite
			}
			if req.TenantType == consts.TenantTypeAgent {
				// 验证代理区域是否已有代理商
				var count int
				count, err = dao.TenantDao.Ctx(ctx).TX(tx).Where(dao.TenantDao.Columns().CityCode, req.CityCode).Count()
				liberr.ErrIsNil(ctx, err, "获取租户数据失败")
				if count > 0 {
					liberr.ErrIsNil(ctx, errors.New("该城市已有代理商"))
				}
			}
			var tenantId int64
			tenantId, err = dao.TenantDao.Ctx(ctx).TX(tx).InsertAndGetId(do.Tenant{
				CreateBy:        userId,
				UpdateBy:        userId,
				TenantName:      req.TenantName,
				TenantType:      req.TenantType,
				CityCode:        req.CityCode,
				BusinessLicense: req.BusinessLicense,
				TenantSource:    userTenantId,
			})
			liberr.ErrIsNil(ctx, err, "添加租户失败")
			var RoleIds = []int64{consts.SiteAdminId}
			if !g.IsNil(req.TenantType) && req.TenantType == consts.TenantTypeAgent {
				RoleIds = []int64{consts.SalesAgentId}
			}

			// 在同一事务中添加部门
			var deptId int64
			var deptInsertId sql.Result
			deptInsertId, err = dao.SysDept.Ctx(ctx).TX(tx).Insert(do.SysDept{
				ParentId:  0,
				DeptName:  req.TenantName,
				OrderNum:  0,
				Leader:    req.UserNickname,
				Phone:     req.Mobile,
				Email:     req.Mobile + "@qq.com",
				Status:    1,
				CreatedBy: userId,
				TenantId:  uint64(tenantId),
			})
			liberr.ErrIsNil(ctx, err, "添加部门失败")
			deptId, err = deptInsertId.LastInsertId()
			liberr.ErrIsNil(ctx, err, "获取部门ID失败")

			// 在同一事务中添加用户
			err = service.SysUser().UserNameOrMobileExists(ctx, req.UserName, req.Mobile)
			liberr.ErrIsNil(ctx, err)
			userSalt := grand.S(10)
			userPassword := libUtils.EncryptPassword(req.UserPassword, userSalt)
			var adminUserId int64
			adminUserId, err = dao.SysUser.Ctx(ctx).TX(tx).InsertAndGetId(do.SysUser{
				UserName:     req.UserName,
				Mobile:       req.Mobile,
				UserNickname: req.UserNickname,
				UserPassword: userPassword,
				UserSalt:     userSalt,
				UserStatus:   1,
				DeptId:       uint64(deptId),
				IsAdmin:      1,
				TenantId:     uint64(tenantId),
			})
			liberr.ErrIsNil(ctx, err, "添加用户失败")
			// 添加用户角色信息（使用Casbin）
			enforcer, e := commonService.CasbinEnforcer(ctx)
			liberr.ErrIsNil(ctx, e, "获取权限引擎失败")
			for _, v := range RoleIds {
				_, e = enforcer.AddGroupingPolicy(fmt.Sprintf("u_%d", adminUserId), gconv.String(v))
				liberr.ErrIsNil(ctx, e, "设置用户权限失败")
			}
			// 更新租户的AdminBy字段
			_, err = dao.TenantDao.Ctx(ctx).TX(tx).WherePri(tenantId).Update(do.Tenant{
				AdminBy: adminUserId,
			})
			liberr.ErrIsNil(ctx, err, "修改租户信息失败")
		})
		return err
	})
	return
}

func (s *sTenant) Edit(ctx context.Context, req *system.TenantEditReq) (err error) {
	err = g.DB().Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
		err = g.Try(ctx, func(ctx context.Context) {
			userId := service.Context().GetUserId(ctx)
			tenant := do.Tenant{
				UpdateBy:        userId,
				TenantName:      req.TenantName,
				CityCode:        req.CityCode,
				BusinessLicense: req.BusinessLicense,
			}
			if !g.IsEmpty(req.TenantType) {
				tenant.TenantType = req.TenantType
			}
			_, err = dao.TenantDao.Ctx(ctx).TX(tx).WherePri(req.Id).Update(tenant)
			liberr.ErrIsNil(ctx, err, "修改租户信息失败")
		})
		return err
	})
	return
}

func (s *sTenant) GetTenantDetailsByIds(ctx context.Context, req *system.GetTenantDetailsByIdsReq) (res *system.GetTenantDetailsByIdsRes, err error) {
	res = new(system.GetTenantDetailsByIdsRes)
	if len(req.TenantIds) == 0 {
		return
	}
	idsSet := gset.NewIntSetFrom(gconv.Ints(req.TenantIds)).Slice()
	err = g.Try(ctx, func(ctx context.Context) {
		err = dao.TenantDao.Ctx(ctx).Where(dao.TenantDao.Columns().Id+" in(?)", idsSet).Order(dao.TenantDao.Columns().Id + " ASC").Scan(&res.List)
	})
	return
}

func (s *sTenant) GetTenantDetails(ctx context.Context, id uint64) (res *entity.Tenant, err error) {
	err = g.Try(ctx, func(ctx context.Context) {
		err = dao.TenantDao.Ctx(ctx).WherePri(id).Scan(&res)
	})
	return
}

func (s *sTenant) GetTenantAdminById(ctx context.Context, id uint64) (res []entity.Tenant, err error) {
	err = g.Try(ctx, func(ctx context.Context) {
		err = dao.TenantDao.Ctx(ctx).Where(dao.TenantDao.Columns().Id+" =(?) or "+dao.TenantDao.Columns().TenantSource+" =(?)", id, id).Order(dao.TenantDao.Columns().Id + " ASC").Scan(&res)
	})
	return
}

func (s *sTenant) GetTenantIdList(ctx context.Context, req *system.GetTenantListReq) (res []entity.Tenant, err error) {
	err = g.Try(ctx, func(ctx context.Context) {
		model := dao.TenantDao.Ctx(ctx)
		if !g.IsEmpty(req.TenantId) {
			model.Where(dao.TenantDao.Columns().Id, req.TenantId)
		}
		err = model.Scan(&res)
	})
	return
}