package service

import (
	"fmt"
	"strings"

	"ems.agt/features/udm_user/model"
	"ems.agt/lib/core/redis"
)

// phoneImsiList 获取所有imsi
func phoneImsiList() map[string]string {
	phoneKeys := redis.GetKeys("1*********")
	phoneAndImsiArr := make(map[string]string, 0)
	for _, phone := range phoneKeys {
		imsi := redis.Get(phone)
		phoneAndImsiArr[phone] = imsi
	}
	return phoneAndImsiArr
}

// redisUdmAuthUserList UDM鉴权用户
func redisUdmAuthUserList() []model.UdmAuthUser {
	phoneAndImsiArr := phoneImsiList()
	user := []model.UdmAuthUser{}
	for phone, imsi := range phoneAndImsiArr {
		key := fmt.Sprintf("ausf:%s", imsi)
		m := redis.GetHash(key)

		status := "0"
		if _, ok := m["auth_success"]; ok {
			status = "1"
		}
		a := model.UdmAuthUser{
			Msisdn:    phone,
			Imsi:      imsi,
			Amf:       m["amf"],
			Status:    status,
			Ki:        m["ki"],
			AlgoIndex: m["algo"],
			Opc:       m["opc"],
		}
		user = append(user, a)
	}
	return user
}

// redisUdmSubUserList UDM签约用户
func redisUdmSubUserList() []model.UdmSubUser {
	phoneAndImsiArr := phoneImsiList()
	user := []model.UdmSubUser{}
	for phone, imsi := range phoneAndImsiArr {
		key := fmt.Sprintf("udm-sd:%s", imsi)
		m := redis.GetHash(key)

		a := model.UdmSubUser{
			Imsi:    imsi,
			Msisdn:  m["gpsi"],
			SubData: m["sm-dat"],
		}

		if a.Msisdn == "" {
			a.Msisdn = phone
		}

		if v, ok := m["am-dat"]; ok {
			arr := strings.Split(v, ",")
			a.SubUeAMBRTemp = arr[0]
			a.SubSNSSAITemp = arr[1]
			a.Rai = arr[2]
			a.ForbiddenAreasTemp = arr[3]
			a.ServiceAreaRestrictTemp = arr[4]
			a.CnType = arr[5]
		}
		if v, ok := m["eps-dat"]; ok {
			arr := strings.Split(v, ",")
			a.Eps = arr[0]
		}

		user = append(user, a)
	}
	return user
}