userCenter/entity/user.go

package entity

import (
	"encoding/json"
	"fmt"
	"strconv"
	"strings"
	"sync"
	"time"

	MC "app.yhyue.com/moapp/jybase/common"
	"app.yhyue.com/moapp/jybase/redis"
)

const (
	IsNewFreeTimeCell       = 1637830020
	BigMemberUserPowerTable = "bigmember_service_user"
	RedisCode               = "newother"
	RedisMenuKey            = "jy_workdesktopmenu_%s_%s_%s"
	UserPowerRedisKey       = "jy_userpowerredis_%s_%d_%s"
	UserEntIdKey            = "jy_userentid_%s_%d_%s"
	UserRegisterTime        = "jy_userregistertime_%s"
	EntnicheWaitEmpower     = "entniche_wait_empower"
	EntnichePower           = "entniche_power"
)

type UserInfo struct {
	Capitals    map[string]int
	Permissions map[string]int
	Lock        *sync.Mutex
}

var (
	CapitalRetention = "capital_retention"
	//UserCapitals     = map[string]map[string]int{}
	//UserPermissions  = map[string]map[string]int{}
	//UserLock         = map[string]*sync.Mutex{}
	OverallLock    = &sync.Mutex{}
	UserInfoMap    = map[string]*UserInfo{}
	UserRolePowers = map[string][]string{}
)

/*
*待调整  调整为存redis
*测试用例放的地方不对 待调整
 */
//用户权限 初始化
func (m *WorkDesktopMenu) AutoUserPowerInfo() map[string]int {
	/*
	* 商机管理--》新版商机管理vs老版商机管理
	* 大会员--》bigmember_service
	* 超级订阅--》新版超级订阅vs老版超级订阅
	* 免费用户--》新免费用户
	 */
	var UserPowerMap = map[string]int{}
	//redis newother 查询是否存在用户功能信息
	userPowerRedisKey := fmt.Sprintf(UserPowerRedisKey, m.AppId, time.Now().Day(), m.UserId)
	bytes, err := redis.GetBytes(RedisCode, userPowerRedisKey)
	if err == nil && len(*bytes) > 0 {
		if err = json.Unmarshal(*bytes, &UserPowerMap); err == nil {
			return UserPowerMap
		}
	}
	//查询用户信息
	data, ok := Mgo.FindById("user", m.UserId, `{"s_phone":1,"s_m_phone":1,i_member_status":1,“i_member_endtime”:1,"s_member_mainid":1,"i_member_sub_status":1,"i_vip_status":1,"l_vip_endtime":1,"o_vipjy":1,"o_jy":1,"l_registedate":1}`)
	if ok && *data != nil && len(*data) > 0 {
		var (
			isFree = true
			//第一版领域化权限判断  ；第二版：超级订阅||大会员 到期时间超过90天
			//商机管理服务 前提是超级订阅、大会员、医械通
			domainBool           = false
			registerTime   int64 = 0
			entnicheRoleId       = 0                     //企业级服务 企业管理员
			entnicheIsNew        = -1                    //是否是新版商机管理
			powerSource          = 0                     //商机管理服务
			memberStatus         = 0                     //大会员
			serviceIds           = map[string][]string{} //大会员服务id集合
		)
		phone, _ := MC.If((*data)["s_phone"] != nil, (*data)["s_phone"], (*data)["s_m_phone"]).(string)
		if phone != "" {
			//大会员
			memberStatus = MC.IntAll((*data)["i_member_status"])
			if memberStatus > 0 || ConfigJson.BigMemberOff {
				mainUserId := m.UserId
				if memberStatus > 0 {
					registerTime = MC.Int64All((*data)["i_member_endtime"])
					domainBool = true
					isFree = false
					UserPowerMap["0"] = memberStatus
					UserPowerMap[strconv.Itoa(50+memberStatus)] = 1
					serviceIds["M"] = append(serviceIds["M"], "0", strconv.Itoa(50+memberStatus))
					//是否是子账号 而且 子账号被启用
					if (*data)["s_member_mainid"] != nil && MC.ObjToString((*data)["s_member_mainid"]) != "" && MC.IntAllDef((*data)["i_member_sub_status"], 0) > 0 {
						mainUserId = MC.ObjToString((*data)["s_member_mainid"])
					}
				}
				//大会员用户购买的服务
				serviceList := Mysql.Find(BigMemberUserPowerTable, map[string]interface{}{"s_userid": mainUserId, "i_status": 0}, "DISTINCT(s_serviceid),i_frequency", "", -1, -1)
				if serviceList != nil && len(*serviceList) != 0 {
					for _, sv := range *serviceList {
						UserPowerMap[MC.ObjToString(sv["s_serviceid"])] = MC.If(MC.IntAll(sv["i_frequency"]) > 0, MC.IntAll(sv["i_frequency"]), memberStatus).(int)
						serviceIds["M"] = append(serviceIds["M"], MC.ObjToString(sv["s_serviceid"]))
					}
				}
			}
			//VIP用户
			if vipStatus := MC.IntAll((*data)["i_vip_status"]); vipStatus > 0 {
				if registerTime < MC.Int64All((*data)["l_vip_endtime"]) {
					registerTime = MC.Int64All((*data)["l_vip_endtime"])
				}
				domainBool = true
				isFree = false
				UserPowerMap["200"] = vipStatus
				serviceIds["V"] = append(serviceIds["V"], "200")
				if vipSet := MC.ObjToMap((*data)["o_vipjy"]); vipSet != nil {
					if buySet := MC.ObjToMap((*vipSet)["o_buyset"]); buySet != nil {
						//vip 升级用户
						if MC.IntAll((*buySet)["upgrade"]) > 0 {
							UserPowerMap["201"] = vipStatus
							serviceIds["V"] = append(serviceIds["V"], "201")
						}
					}
				}
			}
			//当前企业id
			if m.EntId > 0 {
				UserPowerMap["702"] = 1
				entNicheInfos := Mysql.SelectBySql(`SELECT i.status,i.isNew,i.power_source,r.role_id,u.power,i.model FROM (entniche_user u LEFT JOIN entniche_user_role r ON r.user_id = u.id)  LEFT JOIN entniche_info i ON u.ent_id=i.id WHERE u.phone = ? and i.id = ? `, phone, m.EntId)
				//商机管理用户信息判断
				if entNicheInfos != nil && len(*entNicheInfos) > 0 {
					entNicheInfo := (*entNicheInfos)[0]
					//1:企业管理员 2：部门管理员
					entnicheRoleId = MC.IntAll(entNicheInfo["role_id"])
					//商机管理
					//status>0 商机管理付费用户
					if MC.IntAll(entNicheInfo["status"]) == 1 {
						//是否是新版商机管理
						entnicheIsNew = MC.IntAll(entNicheInfo["isNew"])
						//管理员entnicheRoleId>0 并不一定 power==1，即使power==0 也有权限
						//员工被分配后power==1否则没有权限
						if MC.IntAll(entNicheInfo["power"]) == 1 || entnicheRoleId > 0 {
							//商机管理服务 P259需求
							//有商机管理服务 不会再有商机管理订阅菜单
							//商机管理服务，上次陈老师说叫“企业管理服务”，不然和商机管理产品容易混淆-- 杨蘭 2022/12/14
							powerSource = MC.IntAll(entNicheInfo["power_source"])
							switch powerSource {
							case 0:
								isFree = false
								switch entnicheIsNew {
								case 1: //新版商机管理
									UserPowerMap["110"] = 1
									switch MC.IntAll(entNicheInfo["role_id"]) {
									case 2: //部门管理员
										UserPowerMap["111"] = 1
									case 1: //企业管理员
										UserPowerMap["112"] = 1
									}
								case 0: //老版商机管理
									UserPowerMap["100"] = 1
									switch MC.IntAll(entNicheInfo["role_id"]) {
									case 2: //部门管理员
										UserPowerMap["101"] = 1
									case 1: //企业管理员
										UserPowerMap["102"] = 1
									}
									//老版商机管理： model：1-统一订阅，2-个人订阅
									if MC.IntAll(entNicheInfo["model"]) == 1 {
										UserPowerMap["103"] = 1
									} else if MC.IntAll(entNicheInfo["model"]) == 2 {
										UserPowerMap["104"] = 1
									}
								}
							case 1:
								//客户管理服务（商机管理服务） [前提：大会员、超级订阅、医械通用户]---免费用户也可以用 （需求调整来自刘苗：产品已确认）
								UserPowerMap["600"] = 1
								switch MC.IntAll(entNicheInfo["role_id"]) {
								case 2: //部门管理员
									UserPowerMap["602"] = 1
								case 1: //企业管理员
									UserPowerMap["601"] = 1
								}
							}
						}
					}
				}
			}
			//免费用户
			if isFree {
				UserPowerMap["300"] = 1
				freeSet := MC.ObjToMap((*data)["o_jy"])
				if MC.IntAll((*freeSet)["i_newfree"]) > 0 || IsNewFreeTimeCell < MC.IntAll((*data)["l_registedate"]) {
					//新免费用户
					UserPowerMap["301"] = 1
				}
			}
			//广东移动DICT
			if Mysql.CountBySql(`select count(*) from privatedata where phone = ?`, phone) > 0 {
				UserPowerMap["400"] = 1
			}
			//资源中台获取权限
			powerList := GetResources(m.AppId, m.NewUserId, m.EntId, m.EntUserId)
			if len(powerList) > 0 {
				for _, plv := range powerList {
					UserPowerMap[plv] = 1
				}
			}
			//人员行为统计菜单-1、购买大会员且创建了企业的管理员；2、新版商机管理管理员；3、商机管理服务管理员
			if entnicheRoleId > 0 && memberStatus > 0 {
				UserPowerMap["50"] = 1
			}
			//必须是企业管理员-&-购买了企业级应用服务
			c := Mysql.CountBySql(fmt.Sprintf(`SELECT COUNT(id) FROM %s WHERE ent_id = ? AND TO_DAYS(end_time) >= TO_DAYS(NOW()) AND (LOCATE('VIP订阅',product_type) OR LOCATE('大会员',product_type))`, EntnicheWaitEmpower), m.EntId)
			//企业级服务 权限管理（只有企业管理员有权限）
			if entnicheRoleId == 1 && c > 0 {
				UserPowerMap["700"] = 1
			}
			//企业级服务 企业订阅--
			//1、存在未到期的购买主体为“企业”切购买产品为大会员或者超级订阅的部门管理员或企业管理员；c > 0
			//2、存在未到期的老版或者新版商机管理的企业管理员或部门管理员(非商机管理服务)。entnicheIsNew > -1 && powerSource == 0
			if (c > 0 || entnicheIsNew > -1 && powerSource == 0) && entnicheRoleId > 0 {
				UserPowerMap["701"] = 1
			}
			//领域化产品权限
			//第一版：必须是大会员或者超级订阅用户 且留资 留资表：capital_retention；source = 'medical_domain',未留资提示留资信息
			//第二版：调资源中台rpc获取用户是否有使用领域化产品的权限 无权限则去购买
			//需求调整：
			//1：是否是大会员或者超级订阅用户 否：提示购买到超级订阅购买页； 是>-2
			//2：判断用户是否留资  否：提示用去留资；是：>-3
			//3：资源中台获取用户权限码判断是否有权限 否：提示用户去联系客服
			if domainBool {
				userRegisterTimeKey := fmt.Sprintf(UserRegisterTime, m.UserId)
				redis.Put(RedisCode, userRegisterTimeKey, strconv.Itoa(int(registerTime)), int(registerTime))
				UserPowerMap["500"] = 1
			}
			//当前用户是否被分配权限
			if m.EntUserId > 0 {
				entnichePowers := Mysql.SelectBySql(fmt.Sprintf(`SELECT a.product_type,a.ent_id FROM %s a LEFT JOIN %s b ON a.id = b.wait_empower_id WHERE b.status = 1 AND b.ent_user_id = ?`, EntnicheWaitEmpower, EntnichePower), m.EntUserId)
				if entnichePowers != nil && len(*entnichePowers) > 0 {
					for _, entnichePower := range *entnichePowers {
						//排除非当前企业分配的权限
						if MC.Int64All(entnichePower["ent_id"]) != m.EntId {
							productType := MC.ObjToString(entnichePower["product_type"])
							if strings.Contains(productType, "大会员") && len(serviceIds["M"]) > 0 {
								for _, v := range serviceIds["M"] {
									UserPowerMap[v] = 0
								}
							}
							if strings.Contains(productType, "VIP订阅") && len(serviceIds["V"]) > 0 {
								for _, v := range serviceIds["V"] {
									UserPowerMap[v] = 0
								}
							}
						}

					}
				}
			}
		}
		if UserPowerMap != nil {
			bytes, err := json.Marshal(UserPowerMap)
			if err == nil && len(bytes) > 0 {
				redis.PutBytes(RedisCode, userPowerRedisKey, &bytes, ConfigJson.InternalTime)
				if m.EntId > 0 {
					//商机管理用户entId 缓存
					userEntIdKey := fmt.Sprintf(UserEntIdKey, m.AppId, time.Now().Day(), m.UserId)
					redis.Put(RedisCode, userEntIdKey, m.EntId, ConfigJson.InternalTime)
				}
			}
		}
	}
	return UserPowerMap
}

//clear One
func ClearUserPowerFunc(userId, appId string) {
	userPowerRedisKey := fmt.Sprintf(UserPowerRedisKey, appId, time.Now().Day(), userId)
	redis.Del(RedisCode, userPowerRedisKey)
	for _, v := range []string{"PC", "APP", "WX"} {
		RedisMenuKey := fmt.Sprintf(RedisMenuKey, appId, v, userId)
		redis.Del(RedisCode, RedisMenuKey)
	}
}

//用户角色权限初始化
func UserRolePowerInit(strs []string) {
	if len(strs) > 0 {
		jyUserRoleData := BaseMysql.SelectBySql(`SELECT id,name FROM jyfunction WHERE status = 1`)
		if jyUserRoleData != nil && len(*jyUserRoleData) > 0 {
			for _, jv := range *jyUserRoleData {
				for _, v := range strs {
					if strings.Contains(MC.ObjToString(jv["name"]), v) {
						UserRolePowers[v] = append(UserRolePowers[v], strconv.Itoa(MC.IntAll(jv["id"])))
					}
				}
			}
		}
	}
}

/*
1、不符合可以留资申请开通权限的用户
	1、免费用户
	2、超级订阅用户、大会员用户到期时间是否超过3个月
	（1、2：弹窗-医械通上线啦）
2、符合可以留资申请开通的用户
	1、未留资
	（1：弹窗-完善基本信息）
	2、已留资未开通
	3、已留资且开通
	（2、3：弹窗-恭喜留资成功）
*/
/*
1、判断是否满足超级订阅用户或大会员用户 且到期时间超过90天（配置）；
	1.1、否：权限=0；提示上线啦
	1.2、是：是否留资
		1.2.1、否：权限=0；提示留资
		1.2.2、是：是否开通权限
			1.2.2.1、否：权限=0；提示恭喜留资成功，客服联系
			1.2.2.2、是：权限=1
*/
/*存在超级订阅 或 大会员到期  而医械通未到期的情况*/

// CheckCapitalResources  是否需要留资 且权限验证--弹窗处理
//b 一级权限（超级订阅、大会员等）
//p 二级权限（请求资源中台：医械通等）
func CheckCapitalResources(menu *JYMenu, wd *WorkDesktopMenu, b, p bool) (title, content, confirmUrl, confirmText, appType, openType string, isShowCancel, usable bool) {
	OverallLock.Lock()
	userInfo := UserInfoMap[wd.NewUserId]
	if userInfo == nil {
		userInfo = &UserInfo{}
		userInfo.Lock = &sync.Mutex{}
		userInfo.Capitals = map[string]int{}
		userInfo.Permissions = map[string]int{}
		UserInfoMap[wd.NewUserId] = userInfo
	}
	OverallLock.Unlock()
	userInfo.Lock.Lock()
	defer userInfo.Lock.Unlock()
	var (
		capitalBool    = true //是否留资
		permissionBool = true //是否有功能权限
		//customPopup    = true  //自定义弹窗
		defaultPopup = false //默认弹窗
	)
	//大会员超级订阅用户 才有权限去验证是否有医疗权限
	//用户是否需要留资
	if menu.CapitalCode != "" {
		capitalBool = false
		for _, cv := range strings.Split(menu.CapitalCode, ",") {
			if userInfo.Capitals[cv] == 0 {
				if c := BaseMysql.CountBySql(`SELECT COUNT(id) FROM `+CapitalRetention+` WHERE  source = ? AND user_id = ? AND appid = ?`, cv, wd.NewUserId, wd.AppId); c > 0 {
					userInfo.Capitals[cv] = 1
				} else {
					userInfo.Capitals[cv] = -1
				}
			}
			if userInfo.Capitals[cv] < 0 {
				//customPopup = false
				//留资弹窗信息
				title = menu.CapitalInfo.Title
				content = menu.CapitalInfo.Content
				confirmUrl = menu.CapitalInfo.ConfirmUrl
				confirmText = menu.CapitalInfo.ConfirmText
				isShowCancel = menu.CapitalInfo.IsShowCancel
				appType = menu.CapitalInfo.AppType
				openType = menu.CapitalInfo.OpenType
			} else {
				capitalBool = true
				//无权限（有权限未留资|留资了未开通权限） --》 没有配置弹窗信息  --》自定义弹窗
				title = menu.AdditionalInfo.Title
				content = menu.AdditionalInfo.Content
				confirmUrl = menu.AdditionalInfo.ConfirmUrl
				confirmText = menu.AdditionalInfo.ConfirmText
				isShowCancel = menu.AdditionalInfo.IsShowCancel
				appType = menu.AdditionalInfo.AppType
				openType = menu.AdditionalInfo.OpenType
				break
			}
		}
	}
	//无权限（医械通-：1：免费用户；2：超级订阅、大会员到期时间不在规定范围内）
	if b {
		//资源中台--- 无权限弹窗信息为默认信息（先留资）
		if menu.PermissionCode != "" {
			permissionBool = false
			if (menu.CapitalCode != "" && capitalBool) || menu.CapitalCode == "" {
				if p {
					title = ""
					permissionBool = true
					//customPopup = false
				}
			}
		}
		//有剑鱼用户权限 查看资源中台功能权限
		//医械通---用户没有资源中台功能权限 也没有留资 查看
		if !permissionBool {
			switch menu.PowerIds {
			case "500":
				userRegisterTime, _ := strconv.ParseInt(redis.GetStr(RedisCode, fmt.Sprintf(UserRegisterTime, wd.UserId)), 10, 64)
				//超级订阅 大会员到期时间 是否 大于 90天
				if userRegisterTime-time.Now().Unix() < ConfigJson.MedicalFieldTimespan {
					//即使是超级订阅或大会员 也没有权限
					defaultPopup = true
				}
			}
		}
	}
	//
	usable = func() bool {
		switch menu.Authority {
		case 0:
			return b && permissionBool && capitalBool
		case 1:
			return b || permissionBool || capitalBool
		case 2:
			return b || permissionBool && capitalBool
		case 3:
			return b && permissionBool || capitalBool
		case 4:
			return b && capitalBool || permissionBool
		default:
			return false
		}
	}()
	//无权限--免费用户||不符合条件的付费用户
	if !b || defaultPopup {
		//配置弹窗信息
		if ConfigJson.DefaultPopup[menu.PowerIds].Title != "" {
			//customPopup = false
			title = ConfigJson.DefaultPopup[menu.PowerIds].Title
			content = ConfigJson.DefaultPopup[menu.PowerIds].Content
			confirmUrl = ConfigJson.DefaultPopup[menu.PowerIds].ConfirmUrl
			confirmText = ConfigJson.DefaultPopup[menu.PowerIds].ConfirmText
			isShowCancel = ConfigJson.DefaultPopup[menu.PowerIds].IsShowCancel
			appType = ConfigJson.DefaultPopup[menu.PowerIds].AppType
			openType = ConfigJson.DefaultPopup[menu.PowerIds].OpenType
		}
	}
	return
}