package service

import (
	"fmt"
	"strings"

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

// 实例化服务层 ServiceUdmUser 结构体
var NewServiceUdmUser = &ServiceUdmUser{}

// ServiceUdmUser UDM 服务层处理
type ServiceUdmUser struct{}

// phoneImsiList 获取所有imsi
func (r *ServiceUdmUser) 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
}

// ReadAusfList UDM鉴权用户
func (r *ServiceUdmUser) ReadAusfList() []model.UdmAusfUser {
	phoneAndImsiArr := r.phoneImsiList()
	user := []model.UdmAusfUser{}
	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.UdmAusfUser{
			Phone:     phone,
			Imsi:      imsi,
			Amf:       m["amf"],
			Status:    status,
			Ki:        m["ki"],
			AlgoIndex: m["algo"],
			Opc:       m["opc"],
		}
		user = append(user, a)
	}

	fmt.Println(len(user))
	return user
}

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

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

		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)
	}

	fmt.Println(len(user))
	return user
}