add email server api
This commit is contained in:
liugq
parent
63cacee1a7
commit
a5e036d62a
|
|
@ -0,0 +1,36 @@
|
|||
/* Copyright © INFINI Ltd. All rights reserved.
|
||||
* Web: https://infinilabs.com
|
||||
* Email: hello#infini.ltd */
|
||||
|
||||
package model
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"infini.sh/framework/core/orm"
|
||||
)
|
||||
|
||||
type EmailServer struct {
|
||||
orm.ORMObjectBase
|
||||
Name string `json:"name" elastic_mapping:"name:{type:text}"`
|
||||
Host string `json:"host" elastic_mapping:"host:{type:keyword}"`
|
||||
Port int `json:"port" elastic_mapping:"port:{type:keyword}"`
|
||||
TLS bool `json:"tls" elastic_mapping:"tls:{type:keyword}"`
|
||||
Auth struct {
|
||||
Username string `json:"username" elastic_mapping:"username:{type:keyword}"`
|
||||
Password string `json:"password" elastic_mapping:"password:{type:keyword}"`
|
||||
} `json:"auth" elastic_mapping:"auth:{type:object}"`
|
||||
Enabled bool `json:"enabled" elastic_mapping:"enabled:{type:boolean}"`
|
||||
}
|
||||
|
||||
func (serv *EmailServer) Validate(requireName bool) error {
|
||||
if serv.Host == "" {
|
||||
return fmt.Errorf("host can not be empty")
|
||||
}
|
||||
if serv.Port <= 0 {
|
||||
return fmt.Errorf("invalid port [%d]", serv.Port)
|
||||
}
|
||||
if requireName && serv.Name == "" {
|
||||
return fmt.Errorf("name can not be empty")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
|
@ -0,0 +1,56 @@
|
|||
/* Copyright © INFINI Ltd. All rights reserved.
|
||||
* Web: https://infinilabs.com
|
||||
* Email: hello#infini.ltd */
|
||||
|
||||
package email
|
||||
|
||||
import (
|
||||
"infini.sh/console/model"
|
||||
"infini.sh/console/plugin/api/email/common"
|
||||
"infini.sh/framework/core/api"
|
||||
"infini.sh/framework/core/orm"
|
||||
"infini.sh/framework/core/util"
|
||||
log "src/github.com/cihub/seelog"
|
||||
)
|
||||
|
||||
type EmailAPI struct {
|
||||
api.Handler
|
||||
}
|
||||
func InitAPI() {
|
||||
email := EmailAPI{}
|
||||
api.HandleAPIMethod(api.POST, "/email/server/_test", email.testEmailServer)
|
||||
api.HandleAPIMethod(api.GET, "/email/server/:email_server_id", email.getEmailServer)
|
||||
api.HandleAPIMethod(api.POST, "/email/server", email.createEmailServer)
|
||||
api.HandleAPIMethod(api.PUT, "/email/server/:email_server_id", email.updateEmailServer)
|
||||
api.HandleAPIMethod(api.DELETE, "/email/server/:email_server_id", email.deleteEmailServer)
|
||||
api.HandleAPIMethod(api.GET, "/email/server/_search", email.searchEmailServer)
|
||||
}
|
||||
|
||||
func InitEmailServer() error {
|
||||
q := orm.Query{
|
||||
Size: 10,
|
||||
}
|
||||
q.Conds = orm.And(orm.Eq("enabled", true))
|
||||
err, result := orm.Search(model.EmailServer{}, &q )
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if len(result.Result) == 0 {
|
||||
return nil
|
||||
}
|
||||
for _, row := range result.Result {
|
||||
emailServer := model.EmailServer{}
|
||||
buf := util.MustToJSONBytes(row)
|
||||
util.MustFromJSONBytes(buf, &emailServer)
|
||||
err = emailServer.Validate(false)
|
||||
if err != nil {
|
||||
log.Error(err)
|
||||
continue
|
||||
}
|
||||
err = common.StartEmailServer(&emailServer)
|
||||
if err != nil {
|
||||
log.Error(err)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
|
@ -0,0 +1,97 @@
|
|||
/* Copyright © INFINI Ltd. All rights reserved.
|
||||
* Web: https://infinilabs.com
|
||||
* Email: hello#infini.ltd */
|
||||
|
||||
package common
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"infini.sh/console/model"
|
||||
"infini.sh/framework/core/global"
|
||||
"infini.sh/framework/core/pipeline"
|
||||
"infini.sh/framework/lib/go-ucfg/yaml"
|
||||
pipeline2 "infini.sh/framework/modules/pipeline"
|
||||
)
|
||||
|
||||
func StartEmailServer(serv *model.EmailServer) error {
|
||||
pipeCfgStr := GeneratePipelineConfig(serv)
|
||||
cfg, err := yaml.NewConfig([]byte(pipeCfgStr))
|
||||
if err != nil {
|
||||
return fmt.Errorf("new config error: %w", err)
|
||||
}
|
||||
pipeCfg := pipeline.PipelineConfigV2{}
|
||||
err = cfg.Unpack(&pipeCfg)
|
||||
if err != nil {
|
||||
return fmt.Errorf("unpack pipeline config error: %w", err)
|
||||
}
|
||||
v := global.Lookup("pipeline_module")
|
||||
var (
|
||||
pipeM *pipeline2.PipeModule
|
||||
ok bool
|
||||
)
|
||||
if pipeM, ok = v.(*pipeline2.PipeModule); !ok {
|
||||
return fmt.Errorf("can not find pipeline module")
|
||||
}
|
||||
err = pipeM.CreatePipeline(pipeCfg, true)
|
||||
if err != nil {
|
||||
return fmt.Errorf("create email server pipeline error: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func StopEmailServer(serv *model.EmailServer) error {
|
||||
v := global.Lookup("pipeline_module")
|
||||
var (
|
||||
pipeM *pipeline2.PipeModule
|
||||
ok bool
|
||||
)
|
||||
if pipeM, ok = v.(*pipeline2.PipeModule); !ok {
|
||||
return fmt.Errorf("can not find pipeline module")
|
||||
}
|
||||
emailServerTaskID := getEmailServerTaskID(serv)
|
||||
exists := pipeM.StopTask(emailServerTaskID)
|
||||
if exists {
|
||||
pipeM.DeleteTask(emailServerTaskID)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func getEmailServerTaskID(serv *model.EmailServer) string {
|
||||
return fmt.Sprintf("send_email_%s", serv.ID)
|
||||
}
|
||||
|
||||
func GeneratePipelineConfig(serv *model.EmailServer) string {
|
||||
pipelineTpl := `name: %s
|
||||
auto_start: true
|
||||
keep_running: true
|
||||
retry_delay_in_ms: 5000
|
||||
processor:
|
||||
- consumer:
|
||||
consumer:
|
||||
fetch_max_messages: 1
|
||||
max_worker_size: 200
|
||||
num_of_slices: 1
|
||||
idle_timeout_in_seconds: 30
|
||||
queue_selector:
|
||||
keys:
|
||||
- %s
|
||||
processor:
|
||||
- smtp:
|
||||
idle_timeout_in_seconds: 1
|
||||
server:
|
||||
host: "%s"
|
||||
port: %d
|
||||
tls: %v
|
||||
auth:
|
||||
username: "%s"
|
||||
password: "%s"
|
||||
recipients:
|
||||
cc: []
|
||||
templates:
|
||||
raw:
|
||||
content_type: "text/plain"
|
||||
subject: "$[[subject]]"
|
||||
body: "$[[body]]"`
|
||||
pipelineCfg := fmt.Sprintf(pipelineTpl, getEmailServerTaskID(serv), serv.ID, serv.Host, serv.Port, serv.TLS, serv.Auth.Username, serv.Auth.Password)
|
||||
return pipelineCfg
|
||||
}
|
||||
|
|
@ -0,0 +1,255 @@
|
|||
/* Copyright © INFINI Ltd. All rights reserved.
|
||||
* Web: https://infinilabs.com
|
||||
* Email: hello#infini.ltd */
|
||||
|
||||
package email
|
||||
|
||||
import (
|
||||
"crypto/tls"
|
||||
"fmt"
|
||||
log "github.com/cihub/seelog"
|
||||
"infini.sh/console/model"
|
||||
"infini.sh/console/plugin/api/email/common"
|
||||
httprouter "infini.sh/framework/core/api/router"
|
||||
"infini.sh/framework/core/orm"
|
||||
"infini.sh/framework/core/util"
|
||||
"net/http"
|
||||
"src/github.com/gopkg.in/gomail.v2"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
func (h *EmailAPI) createEmailServer(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
||||
var obj = &model.EmailServer{}
|
||||
err := h.DecodeJSON(req, obj)
|
||||
if err != nil {
|
||||
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
||||
log.Error(err)
|
||||
return
|
||||
}
|
||||
q := util.MapStr{
|
||||
"size": 1,
|
||||
"query": util.MapStr{
|
||||
"bool": util.MapStr{
|
||||
"must": []util.MapStr{
|
||||
{
|
||||
"term": util.MapStr{
|
||||
"host": util.MapStr{
|
||||
"value": obj.Host,
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
"term": util.MapStr{
|
||||
"port": util.MapStr{
|
||||
"value": obj.Port,
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
query := orm.Query{
|
||||
RawQuery: util.MustToJSONBytes(q),
|
||||
}
|
||||
err, result := orm.Search(obj, &query)
|
||||
if err != nil {
|
||||
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
||||
log.Error(err)
|
||||
return
|
||||
}
|
||||
if len(result.Result) > 0 {
|
||||
h.WriteError(w, fmt.Sprintf("email server [%s:%d] already exists", obj.Host, obj.Port), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
err = orm.Create(nil, obj)
|
||||
if err != nil {
|
||||
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
||||
log.Error(err)
|
||||
return
|
||||
}
|
||||
if obj.Enabled {
|
||||
err = common.StartEmailServer(obj)
|
||||
if err != nil {
|
||||
log.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
h.WriteJSON(w, util.MapStr{
|
||||
"_id": obj.ID,
|
||||
"result": "created",
|
||||
}, 200)
|
||||
|
||||
}
|
||||
|
||||
func (h *EmailAPI) getEmailServer(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
||||
id := ps.MustGetParameter("email_server_id")
|
||||
|
||||
obj := model.EmailServer{}
|
||||
obj.ID = id
|
||||
|
||||
exists, err := orm.Get(&obj)
|
||||
if !exists || err != nil {
|
||||
h.WriteJSON(w, util.MapStr{
|
||||
"_id": id,
|
||||
"found": false,
|
||||
}, http.StatusNotFound)
|
||||
return
|
||||
}
|
||||
if err != nil {
|
||||
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
||||
log.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
h.WriteJSON(w, util.MapStr{
|
||||
"found": true,
|
||||
"_id": id,
|
||||
"_source": obj,
|
||||
}, 200)
|
||||
}
|
||||
|
||||
func (h *EmailAPI) updateEmailServer(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
||||
id := ps.MustGetParameter("email_server_id")
|
||||
obj := model.EmailServer{}
|
||||
|
||||
obj.ID = id
|
||||
exists, err := orm.Get(&obj)
|
||||
if !exists || err != nil {
|
||||
h.WriteJSON(w, util.MapStr{
|
||||
"_id": id,
|
||||
"result": "not_found",
|
||||
}, http.StatusNotFound)
|
||||
return
|
||||
}
|
||||
|
||||
id = obj.ID
|
||||
create := obj.Created
|
||||
obj = model.EmailServer{}
|
||||
err = h.DecodeJSON(req, &obj)
|
||||
if err != nil {
|
||||
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
||||
log.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
//protect
|
||||
obj.ID = id
|
||||
obj.Created = create
|
||||
err = orm.Update(nil, &obj)
|
||||
if err != nil {
|
||||
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
||||
log.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
h.WriteJSON(w, util.MapStr{
|
||||
"_id": obj.ID,
|
||||
"result": "updated",
|
||||
}, 200)
|
||||
}
|
||||
|
||||
func (h *EmailAPI) deleteEmailServer(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
||||
id := ps.MustGetParameter("email_server_id")
|
||||
|
||||
obj := model.EmailServer{}
|
||||
obj.ID = id
|
||||
|
||||
exists, err := orm.Get(&obj)
|
||||
if !exists || err != nil {
|
||||
h.WriteJSON(w, util.MapStr{
|
||||
"_id": id,
|
||||
"result": "not_found",
|
||||
}, http.StatusNotFound)
|
||||
return
|
||||
}
|
||||
//todo check whether referenced
|
||||
if obj.Enabled {
|
||||
err = common.StopEmailServer(&obj)
|
||||
if err != nil {
|
||||
log.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
err = orm.Delete(nil, &obj)
|
||||
if err != nil {
|
||||
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
||||
log.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
h.WriteJSON(w, util.MapStr{
|
||||
"_id": obj.ID,
|
||||
"result": "deleted",
|
||||
}, 200)
|
||||
}
|
||||
|
||||
func (h *EmailAPI) searchEmailServer(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
||||
|
||||
var (
|
||||
strSize = h.GetParameterOrDefault(req, "size", "20")
|
||||
strFrom = h.GetParameterOrDefault(req, "from", "0")
|
||||
)
|
||||
size, _ := strconv.Atoi(strSize)
|
||||
if size <= 0 {
|
||||
size = 20
|
||||
}
|
||||
from, _ := strconv.Atoi(strFrom)
|
||||
if from < 0 {
|
||||
from = 0
|
||||
}
|
||||
|
||||
q := orm.Query{
|
||||
From: from,
|
||||
Size: size,
|
||||
}
|
||||
q.Conds = orm.And(orm.Eq("enabled", true))
|
||||
|
||||
err, res := orm.Search(&model.EmailServer{}, &q)
|
||||
if err != nil {
|
||||
log.Error(err)
|
||||
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
h.Write(w, res.Raw)
|
||||
}
|
||||
|
||||
func (h *EmailAPI) testEmailServer(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
|
||||
reqBody := &struct {
|
||||
SendTo []string `json:"send_to"`
|
||||
model.EmailServer
|
||||
}{}
|
||||
err := h.DecodeJSON(req, reqBody)
|
||||
if err != nil {
|
||||
log.Error(err)
|
||||
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
if len(reqBody.SendTo) == 0 {
|
||||
h.WriteError(w, "receiver email address can not be empty", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
if err = reqBody.Validate(false); err != nil {
|
||||
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
message := gomail.NewMessage()
|
||||
message.SetHeader("From", reqBody.Auth.Username)
|
||||
message.SetHeader("To", reqBody.SendTo...)
|
||||
message.SetHeader("Subject", "test email")
|
||||
|
||||
message.SetBody("text/plain", "This is just a test email, do not reply!")
|
||||
d := gomail.NewDialerWithTimeout(reqBody.Host, reqBody.Port, reqBody.Auth.Username, reqBody.Auth.Password, 3*time.Second)
|
||||
d.TLSConfig = &tls.Config{InsecureSkipVerify: true}
|
||||
d.SSL = reqBody.TLS
|
||||
|
||||
err = d.DialAndSend(message)
|
||||
if err != nil {
|
||||
log.Error(err)
|
||||
h.WriteError(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
h.WriteAckOKJSON(w)
|
||||
}
|
||||
Loading…
Reference in New Issue