EdgeAdmin/internal/web/actions/default/admins/recipients/instances/createPopup.go

package instances

import (
	"encoding/json"
	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
	"github.com/TeaOSLab/EdgeCommon/pkg/monitorconfigs"
	"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{}) {
	this.Show()
}

func (this *CreatePopupAction) RunPost(params struct {
	Name      string
	MediaType string

	EmailSmtp     string
	EmailUsername string
	EmailPassword string
	EmailFrom     string

	WebHookURL          string
	WebHookMethod       string
	WebHookHeaderNames  []string
	WebHookHeaderValues []string
	WebHookContentType  string
	WebHookParamNames   []string
	WebHookParamValues  []string
	WebHookBody         string

	ScriptType      string
	ScriptPath      string
	ScriptLang      string
	ScriptCode      string
	ScriptCwd       string
	ScriptEnvNames  []string
	ScriptEnvValues []string

	DingTalkWebHookURL string

	QyWeixinCorporateId string
	QyWeixinAgentId     string
	QyWeixinAppSecret   string
	QyWeixinTextFormat  string

	QyWeixinRobotWebHookURL string
	QyWeixinRobotTextFormat string

	AliyunSmsSign              string
	AliyunSmsTemplateCode      string
	AliyunSmsTemplateVarNames  []string
	AliyunSmsTemplateVarValues []string
	AliyunSmsAccessKeyId       string
	AliyunSmsAccessKeySecret   string

	TelegramToken string

	RateMinutes int32
	RateCount   int32

	HashLife int32

	Description string

	Must *actions.Must
	CSRF *actionutils.CSRF
}) {
	params.Must.
		Field("name", params.Name).
		Require("请输入媒介名称").
		Field("mediaType", params.MediaType).
		Require("请选择媒介类型")

	options := maps.Map{}

	switch params.MediaType {
	case "email":
		params.Must.
			Field("emailSmtp", params.EmailSmtp).
			Require("请输入SMTP地址").
			Field("emailUsername", params.EmailUsername).
			Require("请输入邮箱账号").
			Field("emailPassword", params.EmailPassword).
			Require("请输入密码或授权码")

		options["smtp"] = params.EmailSmtp
		options["username"] = params.EmailUsername
		options["password"] = params.EmailPassword
		options["from"] = params.EmailFrom
	case "webHook":
		params.Must.
			Field("webHookURL", params.WebHookURL).
			Require("请输入URL地址").
			Match("(?i)^(http|https)://", "URL地址必须以http或https开头").
			Field("webHookMethod", params.WebHookMethod).
			Require("请选择请求方法")

		options["url"] = params.WebHookURL
		options["method"] = params.WebHookMethod
		options["contentType"] = params.WebHookContentType

		headers := []maps.Map{}
		if len(params.WebHookHeaderNames) > 0 {
			for index, name := range params.WebHookHeaderNames {
				if index < len(params.WebHookHeaderValues) {
					headers = append(headers, maps.Map{
						"name":  name,
						"value": params.WebHookHeaderValues[index],
					})
				}
			}
		}
		options["headers"] = headers

		if params.WebHookContentType == "params" {
			webHookParams := []maps.Map{}
			for index, name := range params.WebHookParamNames {
				if index < len(params.WebHookParamValues) {
					webHookParams = append(webHookParams, maps.Map{
						"name":  name,
						"value": params.WebHookParamValues[index],
					})
				}
			}
			options["params"] = webHookParams
		} else if params.WebHookContentType == "body" {
			options["body"] = params.WebHookBody
		}
	case "script":
		if params.ScriptType == "path" {
			params.Must.
				Field("scriptPath", params.ScriptPath).
				Require("请输入脚本路径")
		} else if params.ScriptType == "code" {
			params.Must.
				Field("scriptCode", params.ScriptCode).
				Require("请输入脚本代码")
		} else {
			params.Must.
				Field("scriptPath", params.ScriptPath).
				Require("请输入脚本路径")
		}

		options["scriptType"] = params.ScriptType
		options["path"] = params.ScriptPath
		options["scriptLang"] = params.ScriptLang
		options["script"] = params.ScriptCode
		options["cwd"] = params.ScriptCwd

		env := []maps.Map{}
		for index, envName := range params.ScriptEnvNames {
			if index < len(params.ScriptEnvValues) {
				env = append(env, maps.Map{
					"name":  envName,
					"value": params.ScriptEnvValues[index],
				})
			}
		}
		options["env"] = env
	case "dingTalk":
		params.Must.
			Field("dingTalkWebHookURL", params.DingTalkWebHookURL).
			Require("请输入Hook地址").
			Match("^https:", "Hook地址必须以https://开头")

		options["webHookURL"] = params.DingTalkWebHookURL
	case "qyWeixin":
		params.Must.
			Field("qyWeixinCorporateId", params.QyWeixinCorporateId).
			Require("请输入企业ID").
			Field("qyWeixinAgentId", params.QyWeixinAgentId).
			Require("请输入应用AgentId").
			Field("qyWeixinSecret", params.QyWeixinAppSecret).
			Require("请输入应用Secret")

		options["corporateId"] = params.QyWeixinCorporateId
		options["agentId"] = params.QyWeixinAgentId
		options["appSecret"] = params.QyWeixinAppSecret
		options["textFormat"] = params.QyWeixinTextFormat
	case "qyWeixinRobot":
		params.Must.
			Field("qyWeixinRobotWebHookURL", params.QyWeixinRobotWebHookURL).
			Require("请输入WebHook地址").
			Match("^https:", "WebHook地址必须以https://开头")

		options["webHookURL"] = params.QyWeixinRobotWebHookURL
		options["textFormat"] = params.QyWeixinRobotTextFormat
	case "aliyunSms":
		params.Must.
			Field("aliyunSmsSign", params.AliyunSmsSign).
			Require("请输入签名名称").
			Field("aliyunSmsTemplateCode", params.AliyunSmsTemplateCode).
			Require("请输入模板CODE").
			Field("aliyunSmsAccessKeyId", params.AliyunSmsAccessKeyId).
			Require("请输入AccessKey ID").
			Field("aliyunSmsAccessKeySecret", params.AliyunSmsAccessKeySecret).
			Require("请输入AccessKey Secret")

		options["sign"] = params.AliyunSmsSign
		options["templateCode"] = params.AliyunSmsTemplateCode
		options["accessKeyId"] = params.AliyunSmsAccessKeyId
		options["accessKeySecret"] = params.AliyunSmsAccessKeySecret

		variables := []maps.Map{}
		for index, name := range params.AliyunSmsTemplateVarNames {
			if index < len(params.AliyunSmsTemplateVarValues) {
				variables = append(variables, maps.Map{
					"name":  name,
					"value": params.AliyunSmsTemplateVarValues[index],
				})
			}
		}
		options["variables"] = variables
	case "telegram":
		params.Must.
			Field("telegramToken", params.TelegramToken).
			Require("请输入机器人Token")
		options["token"] = params.TelegramToken
	default:
		this.Fail("错误的媒介类型")
	}

	optionsJSON, err := json.Marshal(options)
	if err != nil {
		this.ErrorPage(err)
		return
	}

	var rateConfig = &monitorconfigs.RateConfig{
		Minutes: params.RateMinutes,
		Count:   params.RateCount,
	}
	rateJSON, err := json.Marshal(rateConfig)
	if err != nil {
		this.ErrorPage(err)
		return
	}

	resp, err := this.RPC().MessageMediaInstanceRPC().CreateMessageMediaInstance(this.AdminContext(), &pb.CreateMessageMediaInstanceRequest{
		Name:        params.Name,
		MediaType:   params.MediaType,
		ParamsJSON:  optionsJSON,
		Description: params.Description,
		RateJSON:    rateJSON,
		HashLife:    params.HashLife,
	})
	if err != nil {
		this.ErrorPage(err)
		return
	}

	defer this.CreateLogInfo("创建消息媒介 %d", resp.MessageMediaInstanceId)

	this.Success()
}