package users

import (
	"encoding/json"
	teaconst "github.com/TeaOSLab/EdgeAdmin/internal/const"
	"github.com/TeaOSLab/EdgeAdmin/internal/utils/numberutils"
	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
	"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
	"github.com/TeaOSLab/EdgeCommon/pkg/systemconfigs"
	"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
	"github.com/iwind/TeaGo/actions"
	"github.com/iwind/TeaGo/maps"
	"github.com/xlzd/gotp"
)

type CreatePopupAction struct {
	actionutils.ParentAction
}

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

func (this *CreatePopupAction) RunGet(params struct{}) {
	// 检查是否启用了 HTTPDNS 功能（全局用户注册设置中）
	var hasHTTPDNSFeature = false

	resp, err := this.RPC().SysSettingRPC().ReadSysSetting(this.AdminContext(), &pb.ReadSysSettingRequest{Code: systemconfigs.SettingCodeUserRegisterConfig})
	if err == nil && len(resp.ValueJSON) > 0 {
		var config = userconfigs.DefaultUserRegisterConfig()
		if json.Unmarshal(resp.ValueJSON, config) == nil {
			hasHTTPDNSFeature = config.HTTPDNSIsOn
		}
	}
	this.Data["hasHTTPDNSFeature"] = hasHTTPDNSFeature
	this.Data["httpdnsClusterId"] = 0

	// 加载所有 HTTPDNS 集群
	var httpdnsClusters = []maps.Map{}
	if hasHTTPDNSFeature {
		httpdnsClusterResp, err := this.RPC().HTTPDNSClusterRPC().FindAllHTTPDNSClusters(this.AdminContext(), &pb.FindAllHTTPDNSClustersRequest{})
		if err == nil {
			for _, c := range httpdnsClusterResp.GetClusters() {
				httpdnsClusters = append(httpdnsClusters, maps.Map{
					"id":   c.GetId(),
					"name": c.GetName(),
				})
			}
		}
	}
	this.Data["httpdnsClusters"] = httpdnsClusters

	this.Show()
}

func (this *CreatePopupAction) RunPost(params struct {
	Username     string
	Pass1        string
	Pass2        string
	Fullname     string
	Mobile       string
	Tel          string
	Email        string
	Remark       string
	ClusterId        int64
	HttpdnsClusterId int64
	FeaturesType     string

	// OTP
	OtpOn bool

	Must *actions.Must
	CSRF *actionutils.CSRF
}) {
	var userId int64

	defer func() {
		this.CreateLogInfo(codes.User_LogCreateUser, userId)
	}()

	params.Must.
		Field("username", params.Username).
		Require("请输入用户名").
		Match(`^[a-zA-Z0-9_]+$`, "用户名中只能含有英文、数字和下划线")

	checkUsernameResp, err := this.RPC().UserRPC().CheckUserUsername(this.AdminContext(), &pb.CheckUserUsernameRequest{
		UserId:   0,
		Username: params.Username,
	})
	if err != nil {
		this.ErrorPage(err)
		return
	}
	if checkUsernameResp.Exists {
		this.FailField("username", "此用户名已经被占用，请换一个")
	}

	params.Must.
		Field("pass1", params.Pass1).
		Require("请输入密码").
		Field("pass2", params.Pass2).
		Require("请再次输入确认密码").
		Equal(params.Pass1, "两次输入的密码不一致")

	params.Must.
		Field("fullname", params.Fullname).
		Require("请输入全名")

	if params.ClusterId <= 0 {
		this.Fail("请选择关联集群")
	}

	if len(params.Mobile) > 0 {
		params.Must.
			Field("mobile", params.Mobile).
			Mobile("请输入正确的手机号")
	}
	if len(params.Email) > 0 {
		params.Must.
			Field("email", params.Email).
			Email("请输入正确的电子邮箱")
	}

	createResp, err := this.RPC().UserRPC().CreateUser(this.AdminContext(), &pb.CreateUserRequest{
		Username:      params.Username,
		Password:      params.Pass1,
		Fullname:      params.Fullname,
		Mobile:        params.Mobile,
		Tel:           params.Tel,
		Email:         params.Email,
		Remark:        params.Remark,
		Source:        "admin:" + numberutils.FormatInt64(this.AdminId()),
		NodeClusterId: params.ClusterId,
	})
	if err != nil {
		this.ErrorPage(err)
		return
	}

	userId = createResp.UserId

	// 如果表单选择了 HTTPDNS 关联集群，在这里予以保存
	if params.HttpdnsClusterId > 0 {
		httpdnsJSON, _ := json.Marshal([]int64{params.HttpdnsClusterId})
		_, err = this.RPC().UserRPC().UpdateUser(this.AdminContext(), &pb.UpdateUserRequest{
			UserId:                userId,
			Username:              params.Username,
			Password:              params.Pass1,
			Fullname:              params.Fullname,
			Mobile:                params.Mobile,
			Tel:                   params.Tel,
			Email:                 params.Email,
			Remark:                params.Remark,
			IsOn:                  true,
			NodeClusterId:         params.ClusterId,
			HttpdnsClusterIdsJSON: httpdnsJSON,
		})
		if err != nil {
			this.ErrorPage(err)
			return
		}
	}

	// 功能
	if teaconst.IsPlus {
		if params.FeaturesType == "default" {
			resp, err := this.RPC().SysSettingRPC().ReadSysSetting(this.AdminContext(), &pb.ReadSysSettingRequest{Code: systemconfigs.SettingCodeUserRegisterConfig})
			if err != nil {
				this.ErrorPage(err)
				return
			}

			var config = userconfigs.DefaultUserRegisterConfig()
			if len(resp.ValueJSON) > 0 {
				err = json.Unmarshal(resp.ValueJSON, config)
				if err != nil {
					this.ErrorPage(err)
					return
				}
				var featureCodes = config.Features

				_, err = this.RPC().UserRPC().UpdateUserFeatures(this.AdminContext(), &pb.UpdateUserFeaturesRequest{
					UserId:       userId,
					FeatureCodes: featureCodes,
				})
				if err != nil {
					this.ErrorPage(err)
					return
				}

				// 自动关联默认 HTTPDNS 集群 (如果没有在表单中选择的话)
				if config.HTTPDNSIsOn && params.HttpdnsClusterId <= 0 && len(config.HTTPDNSDefaultClusterIds) > 0 {
					httpdnsJSON, _ := json.Marshal(config.HTTPDNSDefaultClusterIds)
					_, err = this.RPC().UserRPC().UpdateUser(this.AdminContext(), &pb.UpdateUserRequest{
						UserId:                userId,
						Username:              params.Username,
						Password:              params.Pass1,
						Fullname:              params.Fullname,
						Mobile:                params.Mobile,
						Tel:                   params.Tel,
						Email:                 params.Email,
						Remark:                params.Remark,
						IsOn:                  true,
						NodeClusterId:         params.ClusterId,
						HttpdnsClusterIdsJSON: httpdnsJSON,
					})
					if err != nil {
						this.ErrorPage(err)
						return
					}
				}
			}
		} else if params.FeaturesType == "all" {
			featuresResp, err := this.RPC().UserRPC().FindAllUserFeatureDefinitions(this.AdminContext(), &pb.FindAllUserFeatureDefinitionsRequest{})
			if err != nil {
				this.ErrorPage(err)
				return
			}
			var featureCodes = []string{}
			for _, def := range featuresResp.Features {
				featureCodes = append(featureCodes, def.Code)
			}
			_, err = this.RPC().UserRPC().UpdateUserFeatures(this.AdminContext(), &pb.UpdateUserFeaturesRequest{
				UserId:       userId,
				FeatureCodes: featureCodes,
			})
			if err != nil {
				this.ErrorPage(err)
				return
			}
		}
	}

	// OTP
	if params.OtpOn {
		_, err = this.RPC().LoginRPC().UpdateLogin(this.AdminContext(), &pb.UpdateLoginRequest{Login: &pb.Login{
			Id:   0,
			Type: "otp",
			ParamsJSON: maps.Map{
				"secret": gotp.RandomSecret(16), // TODO 改成可以设置secret长度
			}.AsJSON(),
			IsOn:    true,
			AdminId: 0,
			UserId:  userId,
		}})
		if err != nil {
			this.ErrorPage(err)
			return
		}
	}

	this.Success()
}