package ldap_lib

import (
	"errors"
	"github.com/go-ldap/ldap/v3"
	"github.com/zeromicro/go-zero/core/logx"
	"strconv"
	"strings"
)

type GetLdapUserInfoRsp struct {
	UserId   int64  `json:"userId"`
	UserDN   string `json:"user_dn"`
	UserName string `json:"user_name"`          //用户名
	Password string `json:"password"`           //密码
	Email    string `json:"email"`              //邮箱
	Mobile   string `json:"mobile"`             //手机号
	Avatar   string `json:"avatar"`             //头像地址
	Status   int64  `json:"status,options=0|1"` //状态 1正常0离职
}

func (l *Ldap) GetLdapUserInfo(userDN string) (*GetLdapUserInfoRsp, error) {
	res, err := l.Search(userDN, ldap.ScopeWholeSubtree, "(&(objectClass=posixAccount)(objectClass=inetOrgPerson))", nil, nil)
	if err != nil {
		return nil, err
	}
	if len(res.Entries) != 1 {
		return nil, errors.New("查询到不到用户信息")
	}
	user := &GetLdapUserInfoRsp{}
	for _, entry := range res.Entries {
		if entry.DN != userDN {
			continue
		}
		user.UserDN = entry.DN
		for _, attr := range entry.Attributes {
			switch attr.Name {
			case "uidNumber": //用户id
				if len(attr.Values) == 0 {
					continue
				}
				user.UserId, err = strconv.ParseInt(attr.Values[0], 10, 64)
				if err != nil {
					logx.Error(err)
					return nil, errors.New("用户id转数字失败")
				}
			case "sn": //用户真名
				user.UserName = strings.Join(attr.Values, "")
			case "mail": //邮箱
				user.Email = strings.Join(attr.Values, "")
			case "mobile": //手机号
				user.Mobile = strings.Join(attr.Values, "")
			case "postalAddress": //头像
				user.Avatar = strings.Join(attr.Values, "")
			case "userPassword": //密码
				user.Password = strings.Join(attr.Values, ",")
			case "postalCode": //状态
				if len(attr.Values) == 0 {
					continue
				}
				user.Status, err = strconv.ParseInt(attr.Values[0], 10, 64)
				if err != nil {
					return nil, errors.New("用户状态转数字失败")
				}
			}
		}
		break
	}
	if user.UserId == 0 {
		return nil, errors.New("查询到的不是用户信息!!!")
	}
	return user, nil
}