Fix send mail (#13312)

* Fix send mail

* Fix send mail

* Update modules/private/mail.go

Co-authored-by: techknowlogick <matti@mdranta.net>
This commit is contained in:
Lunny Xiao 2020-10-27 00:42:27 +08:00 committed by GitHub
parent dbebc6b0e3
commit 38d11eea58
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 36 additions and 6 deletions

View file

@ -9,10 +9,13 @@ import (
"net/http" "net/http"
"code.gitea.io/gitea/modules/private" "code.gitea.io/gitea/modules/private"
"code.gitea.io/gitea/modules/setting"
"github.com/urfave/cli" "github.com/urfave/cli"
) )
func runSendMail(c *cli.Context) error { func runSendMail(c *cli.Context) error {
setting.NewContext()
if err := argsSet(c, "title"); err != nil { if err := argsSet(c, "title"); err != nil {
return err return err
} }
@ -38,11 +41,11 @@ func runSendMail(c *cli.Context) error {
status, message := private.SendEmail(subject, body, nil) status, message := private.SendEmail(subject, body, nil)
if status != http.StatusOK { if status != http.StatusOK {
fmt.Printf("error: %s", message) fmt.Printf("error: %s\n", message)
return nil return nil
} }
fmt.Printf("Succseded: %s", message) fmt.Printf("Success: %s\n", message)
return nil return nil
} }

View file

@ -49,5 +49,10 @@ func SendEmail(subject, message string, to []string) (int, string) {
return http.StatusInternalServerError, fmt.Sprintf("Response body error: %v", err.Error()) return http.StatusInternalServerError, fmt.Sprintf("Response body error: %v", err.Error())
} }
return http.StatusOK, fmt.Sprintf("Was sent %s from %d", body, len(to)) var users = fmt.Sprintf("%d", len(to))
if len(to) == 0 {
users = "all"
}
return http.StatusOK, fmt.Sprintf("Sent %s email(s) to %s users", body, users)
} }

View file

@ -5,6 +5,7 @@
package private package private
import ( import (
"encoding/json"
"fmt" "fmt"
"net/http" "net/http"
"strconv" "strconv"
@ -12,6 +13,7 @@ import (
"code.gitea.io/gitea/models" "code.gitea.io/gitea/models"
"code.gitea.io/gitea/modules/log" "code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/private" "code.gitea.io/gitea/modules/private"
"code.gitea.io/gitea/modules/setting"
"code.gitea.io/gitea/services/mailer" "code.gitea.io/gitea/services/mailer"
"gitea.com/macaron/macaron" "gitea.com/macaron/macaron"
) )
@ -19,7 +21,25 @@ import (
// SendEmail pushes messages to mail queue // SendEmail pushes messages to mail queue
// //
// It doesn't wait before each message will be processed // It doesn't wait before each message will be processed
func SendEmail(ctx *macaron.Context, mail private.Email) { func SendEmail(ctx *macaron.Context) {
if setting.MailService == nil {
ctx.JSON(http.StatusInternalServerError, map[string]interface{}{
"err": "Mail service is not enabled.",
})
return
}
var mail private.Email
rd := ctx.Req.Body().ReadCloser()
defer rd.Close()
if err := json.NewDecoder(rd).Decode(&mail); err != nil {
log.Error("%v", err)
ctx.JSON(http.StatusInternalServerError, map[string]interface{}{
"err": err,
})
return
}
var emails []string var emails []string
if len(mail.To) > 0 { if len(mail.To) > 0 {
for _, uname := range mail.To { for _, uname := range mail.To {
@ -33,13 +53,15 @@ func SendEmail(ctx *macaron.Context, mail private.Email) {
return return
} }
if user != nil { if user != nil && len(user.Email) > 0 {
emails = append(emails, user.Email) emails = append(emails, user.Email)
} }
} }
} else { } else {
err := models.IterateUser(func(user *models.User) error { err := models.IterateUser(func(user *models.User) error {
if len(user.Email) > 0 {
emails = append(emails, user.Email) emails = append(emails, user.Email)
}
return nil return nil
}) })
if err != nil { if err != nil {