eta_api/services/data/data_manage_permission.go

package data

import (
	"eta/eta_api/models/data_manage"
	"eta/eta_api/utils"
	"fmt"
)

// SetEdbChartPermission
// @Description:  单独给资产（指标、图表、ETA表格）设置权限
// @author: Roc
// @datetime 2024-03-27 10:52:32
// @param source int
// @param subSource int
// @param userId int
// @param userList []int
// @param isSelectAll bool
// @param dataId []string
// @param noDataId []string
// @param keyword string
// @param classify string
// @return err error
// @return errMsg string
func SetEdbChartPermission(source, subSource, userId int, userList []int, isSelectAll bool, dataId, noDataId []string, keyword, classify string) (err error, errMsg string) {
	// TODO 消息通知

	// 如果勾选全部数据，那么
	if isSelectAll {
		// 找出不要的指标ID列表
		noDataIdMap := make(map[string]string, 0)
		for _, v := range noDataId {
			noDataIdMap[v] = v
		}

		// 需要转义的指标/图表ID列表
		dataId = make([]string, 0)
		// 获取所有指标/图表
		list, _, tmpErr := GetMoveEdbChartList(source, subSource, userId, keyword, classify, 0, 100000)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		for _, v := range list {
			if _, ok := noDataIdMap[v.DataId]; !ok {
				dataId = append(dataId, v.DataId)
			}
		}
	}

	if len(userList) <= 0 {
		// 说明是取消权限管控
	}

	switch source {
	case 3, 4: //ETA指标库、ETA预测指标
		//tmpList, tmpErr := data_manage.GetEdbInfoListGroupByUserId(dataId)
		//if tmpErr != nil {
		//	err = tmpErr
		//	return
		//}
		//if len(tmpList) > 0 {
		//}
		// 重新设置权限
		err = data_manage.SetPermissionEdbChartByEdbIdList(dataId, userList)
	//case 5: //图库
	//	tmpList, tmpErr := data_manage.GetChartInfoListGroupByUserId(dataId)
	//	if tmpErr != nil {
	//		err = tmpErr
	//		return
	//	}
	//	if len(tmpList) > 0 {
	//		for _, v := range tmpList {
	//			if v.SysUserId == newUserId {
	//				errMsg = "新创建人不可和原创建人一致"
	//				err = errors.New(errMsg)
	//				return
	//			}
	//		}
	//		// 修改创建人
	//		err = data_manage.ModifyChartInfoUserIdByCodeList(dataId, adminInfo.AdminId, adminInfo.RealName)
	//	}
	//case 6: // ETA表格
	//	tmpList, tmpErr := excel.GetNoContentExcelListGroupByUserId(dataId)
	//	if tmpErr != nil {
	//		err = tmpErr
	//		return
	//	}
	//	if len(tmpList) > 0 {
	//		for _, v := range tmpList {
	//			if v.SysUserId == newUserId {
	//				errMsg = "新创建人不可和原创建人一致"
	//				err = errors.New(errMsg)
	//				return
	//			}
	//		}
	//		// 修改创建人
	//		err = excel.ModifyExcelInfoUserIdByCodeList(dataId, adminInfo.AdminId, adminInfo.RealName)
	//	}
	default:
		return
	}

	return
}

// SetDataIsPermission
// @Description:  设置资产（指标、图表、ETA表格）分类是否涉密
// @author: Roc
// @datetime 2024-03-27 10:52:32
// @param source int
// @param subSource int
// @param dataId []string
// @return err error
// @return errMsg string
func SetDataIsPermission(source, subSource int, classifyIdList []int) (err error, errMsg string) {
	// TODO 消息通知

	switch source {
	case 3, 4: //ETA指标库、ETA预测指标
		//tmpList, tmpErr := data_manage.GetEdbInfoListGroupByUserId(dataId)
		//if tmpErr != nil {
		//	err = tmpErr
		//	return
		//}
		//if len(tmpList) > 0 {
		//}
		// 重新设置权限
		err = data_manage.SetIsPermissionEdbChartByEdbClassifyIdList(classifyIdList)
	//case 5: //图库

	//case 6: // ETA表格

	default:
		return
	}

	return
}

// SetEdbChartClassifyPermission
// @Description: 给用户设置涉密分类的权限
// @author: Roc
// @datetime 2024-03-28 14:22:30
// @param source int
// @param subSource int
// @param userList []int
// @param classifyIdList []int
// @return err error
// @return errMsg string
func SetEdbChartClassifyPermission(source, subSource int, userList []int, classifyIdList []int) (err error, errMsg string) {
	// TODO 消息通知

	if len(classifyIdList) <= 0 {
		// 说明是取消权限管控
	}

	switch source {
	case 3, 4: //ETA指标库、ETA预测指标
		//tmpList, tmpErr := data_manage.GetEdbInfoListGroupByUserId(dataId)
		//if tmpErr != nil {
		//	err = tmpErr
		//	return
		//}
		//if len(tmpList) > 0 {
		//}
		// 重新设置权限
		err = data_manage.SetPermissionEdbChartClassifyIdByClassifyIdList(classifyIdList, userList)
	//case 5: //图库
	//	tmpList, tmpErr := data_manage.GetChartInfoListGroupByUserId(dataId)
	//	if tmpErr != nil {
	//		err = tmpErr
	//		return
	//	}
	//	if len(tmpList) > 0 {
	//		for _, v := range tmpList {
	//			if v.SysUserId == newUserId {
	//				errMsg = "新创建人不可和原创建人一致"
	//				err = errors.New(errMsg)
	//				return
	//			}
	//		}
	//		// 修改创建人
	//		err = data_manage.ModifyChartInfoUserIdByCodeList(dataId, adminInfo.AdminId, adminInfo.RealName)
	//	}
	//case 6: // ETA表格
	//	tmpList, tmpErr := excel.GetNoContentExcelListGroupByUserId(dataId)
	//	if tmpErr != nil {
	//		err = tmpErr
	//		return
	//	}
	//	if len(tmpList) > 0 {
	//		for _, v := range tmpList {
	//			if v.SysUserId == newUserId {
	//				errMsg = "新创建人不可和原创建人一致"
	//				err = errors.New(errMsg)
	//				return
	//			}
	//		}
	//		// 修改创建人
	//		err = excel.ModifyExcelInfoUserIdByCodeList(dataId, adminInfo.AdminId, adminInfo.RealName)
	//	}
	default:
		return
	}

	return
}

// GetUserEdbAndClassifyPermissionList
// @Description: 根据用户获取已经授权指标ID列表和指标分类ID列表
// @author: Roc
// @datetime 2024-03-28 17:20:36
// @param userId int
// @param edbInfoId int 如果传递了edbInfoId，则只返回该edbInfoId的权限列表（其实也就是为了判断是否存在该指标权限了，目的是为了少获取数据）
// @param classifyId int  classifyId，则只返回该classifyId的权限列表（其实也就是为了判断是否存在该指标分类权限了，目的是为了少获取数据）
// @return edbIdList []int
// @return classifyIdList []int
// @return err error
func GetUserEdbAndClassifyPermissionList(userId, edbInfoId, classifyId int) (edbIdList, classifyIdList []int, err error) {
	edbIdList, err = data_manage.GetPermissionEdbIdList(userId, edbInfoId)
	if err != nil {
		fmt.Println("获取授权指标列表失败， err:", err)
		return
	}
	classifyIdList, err = data_manage.GetPermissionEdbClassifyIdList(userId, classifyId)
	if err != nil {
		fmt.Println("获取授权指标分类列表失败， err:", err)
		return
	}

	return
}

// CheckEdbPermission
// @Description: 检查EDB指标（含预测指标）权限（方法内部自己获取所有的指标和指标分类权限，不用额外传递）
// @author: Roc
// @datetime 2024-03-28 16:12:08
// @param edbIsJoinPermission int
// @param edbClassifyIsJoinPermission int
// @param edbInfoId int
// @param edbClassifyId int
// @return hasAuth bool
func CheckEdbPermission(edbIsJoinPermission, edbClassifyIsJoinPermission, userId, edbInfoId, edbClassifyId int) (hasAuth bool, err error) {
	edbIdList, classifyIdList, err := GetUserEdbAndClassifyPermissionList(userId, edbInfoId, edbClassifyId)
	if err != nil {
		fmt.Println("GetUserEdbAndClassifyPermissionList err:", err)
		return
	}

	hasAuth = CheckEdbPermissionByPermissionIdList(edbIsJoinPermission, edbClassifyIsJoinPermission, edbInfoId, edbClassifyId, edbIdList, classifyIdList)

	return
}

// CheckEdbPermissionByPermissionIdList
// @Description: 检查EDB指标（含预测指标）权限
// @author: Roc
// @datetime 2024-03-28 16:12:08
// @param edbIsJoinPermission int
// @param edbClassifyIsJoinPermission int
// @param edbInfoId int
// @param edbClassifyId int
// @param permissionEdbInfoIdList []int
// @param permissionEdbClassifyIdList []int
// @return hasAuth bool
func CheckEdbPermissionByPermissionIdList(edbIsJoinPermission, edbClassifyIsJoinPermission, edbInfoId, edbClassifyId int, permissionEdbInfoIdList, permissionEdbClassifyIdList []int) (hasAuth bool) {
	//hasAuth = true

	// 判断 分类是否纳入权限管控
	if edbClassifyIsJoinPermission == 1 {
		// 不属于已授权的分类，那么就无权限
		if !utils.InArrayByInt(permissionEdbClassifyIdList, edbClassifyId) {
			return false
		}
	}

	// 判断 指标是否纳入权限管控
	if edbIsJoinPermission == 1 {
		// 不属于已授权的指标，那么就无权限
		if !utils.InArrayByInt(permissionEdbInfoIdList, edbInfoId) {
			return false
		}
	}

	hasAuth = true

	return
}