aboutsummaryrefslogtreecommitdiff
path: root/internal/mailer/mail.go
diff options
context:
space:
mode:
authorUnknwon <u@gogs.io>2019-10-24 01:51:46 -0700
committerGitHub <noreply@github.com>2019-10-24 01:51:46 -0700
commit01c8df01ec0608f1f25b2f1444adabb98fa5ee8a (patch)
treef8a7e5dd8d2a8c51e1ce2cabb9d33571a93314dd /internal/mailer/mail.go
parent613139e7bef81d3573e7988a47eb6765f3de347a (diff)
internal: move packages under this directory (#5836)
* Rename pkg -> internal * Rename routes -> route * Move route -> internal/route * Rename models -> db * Move db -> internal/db * Fix route2 -> route * Move cmd -> internal/cmd * Bump version
Diffstat (limited to 'internal/mailer/mail.go')
-rw-r--r--internal/mailer/mail.go204
1 files changed, 204 insertions, 0 deletions
diff --git a/internal/mailer/mail.go b/internal/mailer/mail.go
new file mode 100644
index 00000000..2022e240
--- /dev/null
+++ b/internal/mailer/mail.go
@@ -0,0 +1,204 @@
+// Copyright 2016 The Gogs 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 mailer
+
+import (
+ "fmt"
+ "html/template"
+
+ log "gopkg.in/clog.v1"
+ "gopkg.in/gomail.v2"
+ "gopkg.in/macaron.v1"
+
+ "gogs.io/gogs/internal/markup"
+ "gogs.io/gogs/internal/setting"
+)
+
+const (
+ MAIL_AUTH_ACTIVATE = "auth/activate"
+ MAIL_AUTH_ACTIVATE_EMAIL = "auth/activate_email"
+ MAIL_AUTH_RESET_PASSWORD = "auth/reset_passwd"
+ MAIL_AUTH_REGISTER_NOTIFY = "auth/register_notify"
+
+ MAIL_ISSUE_COMMENT = "issue/comment"
+ MAIL_ISSUE_MENTION = "issue/mention"
+
+ MAIL_NOTIFY_COLLABORATOR = "notify/collaborator"
+)
+
+type MailRender interface {
+ HTMLString(string, interface{}, ...macaron.HTMLOptions) (string, error)
+}
+
+var mailRender MailRender
+
+func InitMailRender(dir, appendDir string, funcMap []template.FuncMap) {
+ opt := &macaron.RenderOptions{
+ Directory: dir,
+ AppendDirectories: []string{appendDir},
+ Funcs: funcMap,
+ Extensions: []string{".tmpl", ".html"},
+ }
+ ts := macaron.NewTemplateSet()
+ ts.Set(macaron.DEFAULT_TPL_SET_NAME, opt)
+
+ mailRender = &macaron.TplRender{
+ TemplateSet: ts,
+ Opt: opt,
+ }
+}
+
+func SendTestMail(email string) error {
+ return gomail.Send(&Sender{}, NewMessage([]string{email}, "Gogs Test Email!", "Gogs Test Email!").Message)
+}
+
+/*
+ Setup interfaces of used methods in mail to avoid cycle import.
+*/
+
+type User interface {
+ ID() int64
+ DisplayName() string
+ Email() string
+ GenerateActivateCode() string
+ GenerateEmailActivateCode(string) string
+}
+
+type Repository interface {
+ FullName() string
+ HTMLURL() string
+ ComposeMetas() map[string]string
+}
+
+type Issue interface {
+ MailSubject() string
+ Content() string
+ HTMLURL() string
+}
+
+func SendUserMail(c *macaron.Context, u User, tpl, code, subject, info string) {
+ data := map[string]interface{}{
+ "Username": u.DisplayName(),
+ "ActiveCodeLives": setting.Service.ActiveCodeLives / 60,
+ "ResetPwdCodeLives": setting.Service.ResetPwdCodeLives / 60,
+ "Code": code,
+ }
+ body, err := mailRender.HTMLString(string(tpl), data)
+ if err != nil {
+ log.Error(2, "HTMLString: %v", err)
+ return
+ }
+
+ msg := NewMessage([]string{u.Email()}, subject, body)
+ msg.Info = fmt.Sprintf("UID: %d, %s", u.ID(), info)
+
+ Send(msg)
+}
+
+func SendActivateAccountMail(c *macaron.Context, u User) {
+ SendUserMail(c, u, MAIL_AUTH_ACTIVATE, u.GenerateActivateCode(), c.Tr("mail.activate_account"), "activate account")
+}
+
+func SendResetPasswordMail(c *macaron.Context, u User) {
+ SendUserMail(c, u, MAIL_AUTH_RESET_PASSWORD, u.GenerateActivateCode(), c.Tr("mail.reset_password"), "reset password")
+}
+
+// SendActivateAccountMail sends confirmation email.
+func SendActivateEmailMail(c *macaron.Context, u User, email string) {
+ data := map[string]interface{}{
+ "Username": u.DisplayName(),
+ "ActiveCodeLives": setting.Service.ActiveCodeLives / 60,
+ "Code": u.GenerateEmailActivateCode(email),
+ "Email": email,
+ }
+ body, err := mailRender.HTMLString(string(MAIL_AUTH_ACTIVATE_EMAIL), data)
+ if err != nil {
+ log.Error(3, "HTMLString: %v", err)
+ return
+ }
+
+ msg := NewMessage([]string{email}, c.Tr("mail.activate_email"), body)
+ msg.Info = fmt.Sprintf("UID: %d, activate email", u.ID())
+
+ Send(msg)
+}
+
+// SendRegisterNotifyMail triggers a notify e-mail by admin created a account.
+func SendRegisterNotifyMail(c *macaron.Context, u User) {
+ data := map[string]interface{}{
+ "Username": u.DisplayName(),
+ }
+ body, err := mailRender.HTMLString(string(MAIL_AUTH_REGISTER_NOTIFY), data)
+ if err != nil {
+ log.Error(3, "HTMLString: %v", err)
+ return
+ }
+
+ msg := NewMessage([]string{u.Email()}, c.Tr("mail.register_notify"), body)
+ msg.Info = fmt.Sprintf("UID: %d, registration notify", u.ID())
+
+ Send(msg)
+}
+
+// SendCollaboratorMail sends mail notification to new collaborator.
+func SendCollaboratorMail(u, doer User, repo Repository) {
+ subject := fmt.Sprintf("%s added you to %s", doer.DisplayName(), repo.FullName())
+
+ data := map[string]interface{}{
+ "Subject": subject,
+ "RepoName": repo.FullName(),
+ "Link": repo.HTMLURL(),
+ }
+ body, err := mailRender.HTMLString(string(MAIL_NOTIFY_COLLABORATOR), data)
+ if err != nil {
+ log.Error(3, "HTMLString: %v", err)
+ return
+ }
+
+ msg := NewMessage([]string{u.Email()}, subject, body)
+ msg.Info = fmt.Sprintf("UID: %d, add collaborator", u.ID())
+
+ Send(msg)
+}
+
+func composeTplData(subject, body, link string) map[string]interface{} {
+ data := make(map[string]interface{}, 10)
+ data["Subject"] = subject
+ data["Body"] = body
+ data["Link"] = link
+ return data
+}
+
+func composeIssueMessage(issue Issue, repo Repository, doer User, tplName string, tos []string, info string) *Message {
+ subject := issue.MailSubject()
+ body := string(markup.Markdown([]byte(issue.Content()), repo.HTMLURL(), repo.ComposeMetas()))
+ data := composeTplData(subject, body, issue.HTMLURL())
+ data["Doer"] = doer
+ content, err := mailRender.HTMLString(tplName, data)
+ if err != nil {
+ log.Error(3, "HTMLString (%s): %v", tplName, err)
+ }
+ from := gomail.NewMessage().FormatAddress(setting.MailService.FromEmail, doer.DisplayName())
+ msg := NewMessageFrom(tos, from, subject, content)
+ msg.Info = fmt.Sprintf("Subject: %s, %s", subject, info)
+ return msg
+}
+
+// SendIssueCommentMail composes and sends issue comment emails to target receivers.
+func SendIssueCommentMail(issue Issue, repo Repository, doer User, tos []string) {
+ if len(tos) == 0 {
+ return
+ }
+
+ Send(composeIssueMessage(issue, repo, doer, MAIL_ISSUE_COMMENT, tos, "issue comment"))
+}
+
+// SendIssueMentionMail composes and sends issue mention emails to target receivers.
+func SendIssueMentionMail(issue Issue, repo Repository, doer User, tos []string) {
+ if len(tos) == 0 {
+ return
+ }
+ Send(composeIssueMessage(issue, repo, doer, MAIL_ISSUE_MENTION, tos, "issue mention"))
+}