Sendmail command (#13079)
* Add SendSync method Usefull to have when you need to be confident that message was sent. * Add sendmail command * add checks that if either title or content is empty then error out * Add a confirmation step * Add --force option to bypass confirm step * Move implementation of runSendMail to a different file * Add copyrighting comment * Make content optional Print waring if it's empty or haven't been set up. The warning will be skiped if there's a `--force` flag. * Fix import style Co-authored-by: 6543 <6543@obermui.de> * Use batch when getting all users IterateUsers uses batching by default. Signed-off-by: Maxim Zhiburt <zhiburt@gmail.com> * Send emails one by one instead of as one chunck Signed-off-by: Maxim Zhiburt <zhiburt@gmail.com> * Send messages concurantly Signed-off-by: Maxim Zhiburt <zhiburt@gmail.com> * Use SendAsync+Flush instead of SendSync Signed-off-by: Maxim Zhiburt <zhiburt@gmail.com> * Add timeout parameter to sendemail command Signed-off-by: Maxim Zhiburt <zhiburt@gmail.com> * Fix spelling mistake Signed-off-by: Maxim Zhiburt <zhiburt@gmail.com> * Update cmd/admin.go Co-authored-by: 6543 <6543@obermui.de> * Connect to a running Gitea instance * Fix mispelling * Add copyright comment Co-authored-by: 6543 <6543@obermui.de> Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com> Co-authored-by: techknowlogick <techknowlogick@gitea.io>tokarchuk/v1.17
parent
c5020cff3d
commit
a1952afc38
@ -0,0 +1,48 @@ |
|||||||
|
// Copyright 2020 The Gitea Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package cmd |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
"net/http" |
||||||
|
|
||||||
|
"code.gitea.io/gitea/modules/private" |
||||||
|
"github.com/urfave/cli" |
||||||
|
) |
||||||
|
|
||||||
|
func runSendMail(c *cli.Context) error { |
||||||
|
if err := argsSet(c, "title"); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
subject := c.String("title") |
||||||
|
confirmSkiped := c.Bool("force") |
||||||
|
body := c.String("content") |
||||||
|
|
||||||
|
if !confirmSkiped { |
||||||
|
if len(body) == 0 { |
||||||
|
fmt.Print("warning: Content is empty") |
||||||
|
} |
||||||
|
|
||||||
|
fmt.Print("Proceed with sending email? [Y/n] ") |
||||||
|
isConfirmed, err := confirm() |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} else if !isConfirmed { |
||||||
|
fmt.Println("The mail was not sent") |
||||||
|
return nil |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
status, message := private.SendEmail(subject, body, nil) |
||||||
|
if status != http.StatusOK { |
||||||
|
fmt.Printf("error: %s", message) |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
fmt.Printf("Succseded: %s", message) |
||||||
|
|
||||||
|
return nil |
||||||
|
} |
@ -0,0 +1,53 @@ |
|||||||
|
// Copyright 2020 The Gitea Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package private |
||||||
|
|
||||||
|
import ( |
||||||
|
"encoding/json" |
||||||
|
"fmt" |
||||||
|
"io/ioutil" |
||||||
|
"net/http" |
||||||
|
|
||||||
|
"code.gitea.io/gitea/modules/setting" |
||||||
|
) |
||||||
|
|
||||||
|
// Email structure holds a data for sending general emails
|
||||||
|
type Email struct { |
||||||
|
Subject string |
||||||
|
Message string |
||||||
|
To []string |
||||||
|
} |
||||||
|
|
||||||
|
// SendEmail calls the internal SendEmail function
|
||||||
|
//
|
||||||
|
// It accepts a list of usernames.
|
||||||
|
// If DB contains these users it will send the email to them.
|
||||||
|
//
|
||||||
|
// If to list == nil its supposed to send an email to every
|
||||||
|
// user present in DB
|
||||||
|
func SendEmail(subject, message string, to []string) (int, string) { |
||||||
|
reqURL := setting.LocalURL + "api/internal/mail/send" |
||||||
|
|
||||||
|
req := newInternalRequest(reqURL, "POST") |
||||||
|
req = req.Header("Content-Type", "application/json") |
||||||
|
jsonBytes, _ := json.Marshal(Email{ |
||||||
|
Subject: subject, |
||||||
|
Message: message, |
||||||
|
To: to, |
||||||
|
}) |
||||||
|
req.Body(jsonBytes) |
||||||
|
resp, err := req.Response() |
||||||
|
if err != nil { |
||||||
|
return http.StatusInternalServerError, fmt.Sprintf("Unable to contact gitea: %v", err.Error()) |
||||||
|
} |
||||||
|
defer resp.Body.Close() |
||||||
|
|
||||||
|
body, err := ioutil.ReadAll(resp.Body) |
||||||
|
if err != nil { |
||||||
|
return http.StatusInternalServerError, fmt.Sprintf("Response body error: %v", err.Error()) |
||||||
|
} |
||||||
|
|
||||||
|
return http.StatusOK, fmt.Sprintf("Was sent %s from %d", body, len(to)) |
||||||
|
} |
@ -0,0 +1,67 @@ |
|||||||
|
// Copyright 2020 The Gitea Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package private |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
"net/http" |
||||||
|
"strconv" |
||||||
|
|
||||||
|
"code.gitea.io/gitea/models" |
||||||
|
"code.gitea.io/gitea/modules/log" |
||||||
|
"code.gitea.io/gitea/modules/private" |
||||||
|
"code.gitea.io/gitea/services/mailer" |
||||||
|
"gitea.com/macaron/macaron" |
||||||
|
) |
||||||
|
|
||||||
|
// SendEmail pushes messages to mail queue
|
||||||
|
//
|
||||||
|
// It doesn't wait before each message will be processed
|
||||||
|
func SendEmail(ctx *macaron.Context, mail private.Email) { |
||||||
|
var emails []string |
||||||
|
if len(mail.To) > 0 { |
||||||
|
for _, uname := range mail.To { |
||||||
|
user, err := models.GetUserByName(uname) |
||||||
|
if err != nil { |
||||||
|
err := fmt.Sprintf("Failed to get user information: %v", err) |
||||||
|
log.Error(err) |
||||||
|
ctx.JSON(http.StatusInternalServerError, map[string]interface{}{ |
||||||
|
"err": err, |
||||||
|
}) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
if user != nil { |
||||||
|
emails = append(emails, user.Email) |
||||||
|
} |
||||||
|
} |
||||||
|
} else { |
||||||
|
err := models.IterateUser(func(user *models.User) error { |
||||||
|
emails = append(emails, user.Email) |
||||||
|
return nil |
||||||
|
}) |
||||||
|
if err != nil { |
||||||
|
err := fmt.Sprintf("Failed to find users: %v", err) |
||||||
|
log.Error(err) |
||||||
|
ctx.JSON(http.StatusInternalServerError, map[string]interface{}{ |
||||||
|
"err": err, |
||||||
|
}) |
||||||
|
return |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
sendEmail(ctx, mail.Subject, mail.Message, emails) |
||||||
|
} |
||||||
|
|
||||||
|
func sendEmail(ctx *macaron.Context, subject, message string, to []string) { |
||||||
|
for _, email := range to { |
||||||
|
msg := mailer.NewMessage([]string{email}, subject, message) |
||||||
|
mailer.SendAsync(msg) |
||||||
|
} |
||||||
|
|
||||||
|
wasSent := strconv.Itoa(len(to)) |
||||||
|
|
||||||
|
ctx.PlainText(http.StatusOK, []byte(wasSent)) |
||||||
|
} |
Loading…
Reference in new issue