// Copyright 2023 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn . //go:build plus package setting import ( "encoding/json" "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils" "github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb" "github.com/TeaOSLab/EdgeCommon/pkg/userconfigs" "github.com/iwind/TeaGo/actions" "regexp" ) type SmsTestAction struct { actionutils.ParentAction } func (this *SmsTestAction) Init() { this.Nav("", "", "") } func (this *SmsTestAction) RunGet(params struct{}) { this.Show() } func (this *SmsTestAction) RunPost(params struct { ConfigJSON []byte ToMobile string Body string Code string Must *actions.Must CSRF *actionutils.CSRF }) { if len(params.ConfigJSON) == 0 { this.Fail("读取不到短信配置") return } var config = userconfigs.NewSMSSenderConfig() err := json.Unmarshal(params.ConfigJSON, config) if err != nil { this.Fail("解析短信配置出错:" + err.Error()) return } var paramsJSON []byte switch config.Type { case userconfigs.SMSSenderWebHook: if config.WebHookParams == nil { this.Fail("请配置HTTP接口相关参数") return } if len(config.WebHookParams.URL) == 0 { this.Fail("请输入短信HTTP接口URL地址") return } if !regexp.MustCompile(`^(http|https)://`).MatchString(config.WebHookParams.URL) { this.Fail("短信WebHook URL地址必须以http://或https://开头") return } paramsJSON, err = json.Marshal(config.WebHookParams) if err != nil { this.ErrorPage(err) return } case userconfigs.SMSSenderAliyunSMS: if config.AliyunSMSParams == nil { this.Fail("请配置阿里云短信相关参数") return } if len(config.AliyunSMSParams.Sign) == 0 { this.Fail("请输入签名名称") return } if len(config.AliyunSMSParams.TemplateCode) == 0 { this.Fail("请输入模板CODE") return } if len(config.AliyunSMSParams.CodeVarName) == 0 { this.Fail("请输入模板中验证码变量名称") return } if len(config.AliyunSMSParams.AccessKeyId) == 0 { this.Fail("请输入AccessKey ID") return } if len(config.AliyunSMSParams.AccessKeySecret) == 0 { this.Fail("请输入AccessKey Secret") return } paramsJSON, err = json.Marshal(config.AliyunSMSParams) if err != nil { this.ErrorPage(err) return } case userconfigs.SMSSenderTencentSMS: if config.TencentSMSParams == nil { this.Fail("请配置腾讯云短信相关参数") return } var digitsReg = regexp.MustCompile(`^\d+$`) if len(config.TencentSMSParams.SDKAppId) == 0 { this.Fail("请输入SDK应用ID") return } if !digitsReg.MatchString(config.TencentSMSParams.SDKAppId) { this.Fail("SDK应用ID是一组数字,请重新输入") return } if len(config.TencentSMSParams.Sign) == 0 { this.Fail("请输入签名内容") return } if len(config.TencentSMSParams.TemplateId) == 0 { this.Fail("请输入模板ID") return } if !digitsReg.MatchString(config.TencentSMSParams.TemplateId) { this.Fail("正文模板ID是一组数字,请重新输入") return } if len(config.TencentSMSParams.AccessKeyId) == 0 { this.Fail("请输入密钥SecretId") return } if len(config.TencentSMSParams.AccessKeySecret) == 0 { this.Fail("请输入密钥SecretKey") return } paramsJSON, err = json.Marshal(config.TencentSMSParams) if err != nil { this.ErrorPage(err) return } default: this.Fail("不支持渠道'" + config.Type + "'") } params.Must. Field("toMobile", params.ToMobile). Require("请输入收信人手机号"). Mobile("请输入正确的收信人手机号"). Field("body", params.Body). Require("请输入测试内容") // 发送测试 resp, err := this.RPC().SMSSenderRPC().SendSMS(this.AdminContext(), &pb.SendSMSRequest{ Mobile: params.ToMobile, Body: params.Body, Code: params.Code, Type: config.Type, ParamsJSON: paramsJSON, }) if err != nil { this.Fail("发送失败:" + err.Error()) return } if !resp.IsOk { this.Fail("发送失败,结果:" + resp.Result) return } this.Success() }