EdgeAdmin/internal/web/actions/default/users/user.go

package users

import (
	"encoding/json"
	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/users/userutils"
	"github.com/TeaOSLab/EdgeCommon/pkg/iplibrary"
	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
	"github.com/iwind/TeaGo/maps"
)

type UserAction struct {
	actionutils.ParentAction
}

func (this *UserAction) Init() {
	this.Nav("", "", "index")
}

func (this *UserAction) RunGet(params struct {
	UserId int64
}) {
	err := userutils.InitUser(this.Parent(), params.UserId)
	if err != nil {
		if err == userutils.ErrUserNotFound {
			this.RedirectURL("/users")
			return
		}

		this.ErrorPage(err)
		return
	}

	userResp, err := this.RPC().UserRPC().FindEnabledUser(this.AdminContext(), &pb.FindEnabledUserRequest{UserId: params.UserId})
	if err != nil {
		this.ErrorPage(err)
		return
	}
	var user = userResp.User
	if user == nil {
		this.NotFound("user", params.UserId)
		return
	}

	var clusterMap maps.Map = nil
	if user.NodeCluster != nil {
		clusterMap = maps.Map{
			"id":   user.NodeCluster.Id,
			"name": user.NodeCluster.Name,
		}
	}

	// AccessKey数量
	countAccessKeyResp, err := this.RPC().UserAccessKeyRPC().CountAllEnabledUserAccessKeys(this.AdminContext(), &pb.CountAllEnabledUserAccessKeysRequest{UserId: params.UserId})
	if err != nil {
		this.ErrorPage(err)
		return
	}
	countAccessKeys := countAccessKeyResp.Count

	// IP地址
	var registeredRegion = ""
	if len(user.RegisteredIP) > 0 {
		var ipRegion = iplibrary.LookupIP(user.RegisteredIP)
		if ipRegion != nil && ipRegion.IsOk() {
			registeredRegion = ipRegion.Summary()
		}
	}

	// 是否有实名认证
	hasNewIndividualIdentity, hasNewEnterpriseIdentity, identityTag, err := userutils.CheckUserIdentity(this.RPC(), this.AdminContext(), params.UserId)
	if err != nil {
		this.ErrorPage(err)
		return
	}

	// OTP
	this.Data["otp"] = nil
	if user.OtpLogin != nil && user.OtpLogin.IsOn {
		loginParams := maps.Map{}
		err = json.Unmarshal(user.OtpLogin.ParamsJSON, &loginParams)
		if err != nil {
			this.ErrorPage(err)
			return
		}
		this.Data["otp"] = maps.Map{
			"isOn":   true,
			"params": loginParams,
		}
	}

	this.Data["user"] = maps.Map{
		"id":               user.Id,
		"username":         user.Username,
		"fullname":         user.Fullname,
		"email":            user.Email,
		"verifiedEmail":    user.VerifiedEmail,
		"tel":              user.Tel,
		"remark":           user.Remark,
		"mobile":           user.Mobile,
		"isOn":             user.IsOn,
		"cluster":          clusterMap,
		"countAccessKeys":  countAccessKeys,
		"isRejected":       user.IsRejected,
		"rejectReason":     user.RejectReason,
		"isVerified":       user.IsVerified,
		"registeredIP":     user.RegisteredIP,
		"registeredRegion": registeredRegion,
		"bandwidthAlgo":    user.BandwidthAlgo,

		// 实名认证
		"hasNewIndividualIdentity": hasNewIndividualIdentity,
		"hasNewEnterpriseIdentity": hasNewEnterpriseIdentity,
		"identityTag":              identityTag,
	}

	this.Show()
}