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"],
			SmfSel: m["smf-sel"],
			SmData: m["sm-dat"], // 1-000001&cmnet&ims&3gnet
		}

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

		// def_ambr,def_nssai,0,def_arfb,def_sar,3,1,12000,1,1000,0,1,-
		if v, ok := m["am-dat"]; ok {
			arr := strings.Split(v, ",")
			a.Ambr = arr[0]
			a.Nssai = arr[1]
			a.Rat = arr[2]
			a.Arfb = arr[3]
			a.Sar = arr[4]
			a.Cn = arr[5]
		}
		// 1,64,24,65,def_eps,1,2,010200000000,-
		if v, ok := m["eps-dat"]; ok {
			a.EpsDat = v
			arr := strings.Split(v, ",")
			a.EpsFlag = arr[0]
			a.EpsOdb = arr[1]
			a.HplmnOdb = arr[2]
			a.Ard = arr[3]
			a.Epstpl = arr[4]
			a.ContextId = arr[5]
			a.ApnContext = arr[7]
		}

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