EdgeAdmin/internal/web/actions/default/servers/certs/acme/users/createPopup.go

package users

import (
	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
	"github.com/iwind/TeaGo/actions"
	"github.com/iwind/TeaGo/maps"
)

type CreatePopupAction struct {
	actionutils.ParentAction
}

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

func (this *CreatePopupAction) RunGet(params struct{}) {
	// 服务商
	providersResp, err := this.RPC().ACMEProviderRPC().FindAllACMEProviders(this.AdminContext(), &pb.FindAllACMEProvidersRequest{})
	if err != nil {
		this.ErrorPage(err)
		return
	}
	var providerMaps = []maps.Map{}
	for _, provider := range providersResp.AcmeProviders {
		providerMaps = append(providerMaps, maps.Map{
			"code":       provider.Code,
			"name":       provider.Name,
			"requireEAB": provider.RequireEAB,
		})
	}
	this.Data["providers"] = providerMaps

	this.Show()
}

func (this *CreatePopupAction) RunPost(params struct {
	Email        string
	ProviderCode string
	AccountId    int64
	Description  string

	Must *actions.Must
	CSRF *actionutils.CSRF
}) {
	params.Must.
		Field("email", params.Email).
		Require("请输入邮箱").
		Email("请输入正确的邮箱格式").
		Field("providerCode", params.ProviderCode).
		Require("请选择所属服务商")

	providerResp, err := this.RPC().ACMEProviderRPC().FindACMEProviderWithCode(this.AdminContext(), &pb.FindACMEProviderWithCodeRequest{
		AcmeProviderCode: params.ProviderCode,
	})
	if err != nil {
		this.ErrorPage(err)
		return
	}
	if providerResp.AcmeProvider == nil {
		this.Fail("找不到要选择的证书")
	}
	if providerResp.AcmeProvider.RequireEAB {
		if params.AccountId <= 0 {
			this.Fail("此服务商要求必须选择或创建服务商账号")
		}

		// 同一个账号只能有一个用户
		countResp, err := this.RPC().ACMEUserRPC().
			CountACMEUsers(this.AdminContext(), &pb.CountAcmeUsersRequest{
				AcmeProviderAccountId: params.AccountId,
			})
		if err != nil {
			this.ErrorPage(err)
			return
		}
		if countResp.Count > 0 {
			this.Fail("此服务商账号已被别的用户使用，请换成别的账号")
		}
	}

	createResp, err := this.RPC().ACMEUserRPC().CreateACMEUser(this.AdminContext(), &pb.CreateACMEUserRequest{
		Email:                 params.Email,
		Description:           params.Description,
		AcmeProviderCode:      params.ProviderCode,
		AcmeProviderAccountId: params.AccountId,
	})
	if err != nil {
		this.ErrorPage(err)
		return
	}

	// 返回数据
	this.Data["acmeUser"] = maps.Map{
		"id":           createResp.AcmeUserId,
		"description":  params.Description,
		"email":        params.Email,
		"providerCode": params.ProviderCode,
	}

	// 日志
	defer this.CreateLogInfo("创建ACME用户 %d", createResp.AcmeUserId)

	this.Success()
}