mirror of
https://github.com/Jinnrry/PMail.git
synced 2025-02-20 11:43:09 +08:00
110 lines
3.0 KiB
Go
110 lines
3.0 KiB
Go
package controllers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/Jinnrry/pmail/db"
|
|
"github.com/Jinnrry/pmail/dto"
|
|
"github.com/Jinnrry/pmail/dto/response"
|
|
"github.com/Jinnrry/pmail/i18n"
|
|
"github.com/Jinnrry/pmail/models"
|
|
"github.com/Jinnrry/pmail/services/rule"
|
|
"github.com/Jinnrry/pmail/utils/address"
|
|
"github.com/Jinnrry/pmail/utils/array"
|
|
"github.com/Jinnrry/pmail/utils/context"
|
|
"github.com/Jinnrry/pmail/utils/errors"
|
|
log "github.com/sirupsen/logrus"
|
|
"io"
|
|
"net/http"
|
|
)
|
|
|
|
func GetRule(ctx *context.Context, w http.ResponseWriter, req *http.Request) {
|
|
res := rule.GetAllRules(ctx, ctx.UserID)
|
|
response.NewSuccessResponse(res).FPrint(w)
|
|
}
|
|
|
|
func UpsertRule(ctx *context.Context, w http.ResponseWriter, req *http.Request) {
|
|
|
|
requestBody, err := io.ReadAll(req.Body)
|
|
if err != nil {
|
|
log.WithContext(ctx).Errorf("ReadError:%v", err)
|
|
return
|
|
}
|
|
|
|
var data *dto.Rule
|
|
err = json.Unmarshal(requestBody, &data)
|
|
if err != nil {
|
|
response.NewErrorResponse(response.ParamsError, "params error", err).FPrint(w)
|
|
return
|
|
}
|
|
|
|
if data.Action == dto.FORWARD && !address.IsValidEmailAddress(data.Params) {
|
|
|
|
response.NewErrorResponse(response.ParamsError, "ParamsError error", i18n.GetText(ctx.Lang, "invalid_email_address")).FPrint(w)
|
|
return
|
|
}
|
|
|
|
for _, r := range data.Rules {
|
|
if !array.InArray(r.Field, []string{"From", "Subject", "To", "Cc", "Text", "Html", "Content"}) {
|
|
response.NewErrorResponse(response.ParamsError, "ParamsError error", "params error! Rule Field Error!").FPrint(w)
|
|
return
|
|
}
|
|
}
|
|
|
|
err = save(ctx, data.Encode())
|
|
if err != nil {
|
|
response.NewErrorResponse(response.ServerError, "server error", err).FPrint(w)
|
|
return
|
|
}
|
|
response.NewSuccessResponse("succ").FPrint(w)
|
|
}
|
|
|
|
func save(ctx *context.Context, p *models.Rule) error {
|
|
|
|
if p.Id > 0 {
|
|
_, err := db.Instance.Exec(db.WithContext(ctx, "update rule set name=? ,value = ? ,action = ?,params = ?,sort = ? where id = ?"), p.Name, p.Value, p.Action, p.Params, p.Sort, p.Id)
|
|
if err != nil {
|
|
return errors.Wrap(err)
|
|
}
|
|
return nil
|
|
} else {
|
|
_, err := db.Instance.Exec(db.WithContext(ctx, "insert into rule (name,value,user_id,action,params,sort) values (?,?,?,?,?,?)"), p.Name, p.Value, ctx.UserID, p.Action, p.Params, p.Sort)
|
|
if err != nil {
|
|
return errors.Wrap(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
}
|
|
|
|
type delRuleReq struct {
|
|
Id int `json:"id"`
|
|
}
|
|
|
|
func DelRule(ctx *context.Context, w http.ResponseWriter, req *http.Request) {
|
|
requestBody, err := io.ReadAll(req.Body)
|
|
if err != nil {
|
|
log.WithContext(ctx).Errorf("ReadError:%v", err)
|
|
return
|
|
}
|
|
|
|
var data delRuleReq
|
|
err = json.Unmarshal(requestBody, &data)
|
|
if err != nil {
|
|
response.NewErrorResponse(response.ParamsError, "params error", err).FPrint(w)
|
|
return
|
|
}
|
|
|
|
if data.Id <= 0 {
|
|
response.NewErrorResponse(response.ParamsError, "params error", "id is empty").FPrint(w)
|
|
return
|
|
}
|
|
|
|
_, err = db.Instance.Exec(db.WithContext(ctx, "delete from rule where id =? and user_id =?"), data.Id, ctx.UserID)
|
|
if err != nil {
|
|
response.NewErrorResponse(response.ServerError, "unknown error", err).FPrint(w)
|
|
return
|
|
}
|
|
|
|
response.NewSuccessResponse("succ").FPrint(w)
|
|
}
|