floraachy
/
console
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

generate email server service with saving password to keystore

This commit is contained in:
liugq 2023-08-02 20:35:13 +08:00
parent cf085fbd1a
commit a5fee144d9
1 changed files with 20 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
plugin/api/email/common/pipeline.go
View File

@ -5,13 +5,15 @@
package common package common
import ( import (
"fmt"
"gopkg.in/yaml.v2"
"infini.sh/console/model" "infini.sh/console/model"
"infini.sh/framework/core/global" "infini.sh/framework/core/global"
"infini.sh/framework/core/keystore"
"infini.sh/framework/core/orm" "infini.sh/framework/core/orm"
"infini.sh/framework/core/util" "infini.sh/framework/core/util"
"os" "os"
"path" "path"
"gopkg.in/yaml.v2"
) )
const emailServerConfigFile = "send_email.yml" const emailServerConfigFile = "send_email.yml"
@ -43,7 +45,10 @@ func RefreshEmailServer() error {
emailServer.Auth = &auth emailServer.Auth = &auth
servers = append(servers, emailServer) servers = append(servers, emailServer)
} }
pipeCfgStr := GeneratePipelineConfig(servers) pipeCfgStr, err := GeneratePipelineConfig(servers)
if err != nil {
return err
}
cfgDir := global.Env().GetConfigDir() cfgDir := global.Env().GetConfigDir()
sendEmailCfgFile := path.Join(cfgDir, emailServerConfigFile) sendEmailCfgFile := path.Join(cfgDir, emailServerConfigFile)
_, err = util.FilePutContent(sendEmailCfgFile, pipeCfgStr) _, err = util.FilePutContent(sendEmailCfgFile, pipeCfgStr)
@ -65,13 +70,21 @@ func CheckEmailPipelineExists() bool {
return util.FilesExists(sendEmailCfgFile) return util.FilesExists(sendEmailCfgFile)
} }
func getEmailPasswordKey(srv model.EmailServer) string{
return fmt.Sprintf("%s_password", srv.ID)
}
func GeneratePipelineConfig(servers []model.EmailServer) string { func GeneratePipelineConfig(servers []model.EmailServer) (string, error) {
if len(servers) == 0 { if len(servers) == 0 {
return "" return "", nil
} }
smtpServers := map[string]util.MapStr{} smtpServers := map[string]util.MapStr{}
for _, srv := range servers { for _, srv := range servers {
key := getEmailPasswordKey(srv)
err := keystore.SetValue(key, []byte(srv.Auth.Password))
if err != nil {
return "", err
}
smtpServers[srv.ID] = util.MapStr{ smtpServers[srv.ID] = util.MapStr{
"server": util.MapStr{ "server": util.MapStr{
"host": srv.Host, "host": srv.Host,
@ -80,7 +93,7 @@ func GeneratePipelineConfig(servers []model.EmailServer) string {
}, },
"auth": util.MapStr{ "auth": util.MapStr{
"username": srv.Auth.Username, "username": srv.Auth.Username,
"password": srv.Auth.Password, "password": fmt.Sprintf("$[[keystore.%s]]", key),
}, },
} }
} }
@ -128,7 +141,7 @@ func GeneratePipelineConfig(servers []model.EmailServer) string {
buf, err := yaml.Marshal(pipelineCfg) buf, err := yaml.Marshal(pipelineCfg)
if err != nil { if err != nil {
panic(err) return "", err
} }
return string(buf) return string(buf), nil
} }