More work on email and notification infra #1456

pkg/services/notifications/codes.go

@@ -0,0 +1,69 @@
+package notifications
+
+import (
+	"crypto/sha1"
+	"encoding/hex"
+	"fmt"
+	"time"
+
+	"github.com/Unknwon/com"
+	"github.com/grafana/grafana/pkg/setting"
+)
+
+// create a time limit code
+// code format: 12 length date time string + 6 minutes string + 40 sha1 encoded string
+func CreateTimeLimitCode(data string, minutes int, startInf interface{}) string {
+	format := "200601021504"
+
+	var start, end time.Time
+	var startStr, endStr string
+
+	if startInf == nil {
+		// Use now time create code
+		start = time.Now()
+		startStr = start.Format(format)
+	} else {
+		// use start string create code
+		startStr = startInf.(string)
+		start, _ = time.ParseInLocation(format, startStr, time.Local)
+		startStr = start.Format(format)
+	}
+
+	end = start.Add(time.Minute * time.Duration(minutes))
+	endStr = end.Format(format)
+
+	// create sha1 encode string
+	sh := sha1.New()
+	sh.Write([]byte(data + setting.SecretKey + startStr + endStr + com.ToStr(minutes)))
+	encoded := hex.EncodeToString(sh.Sum(nil))
+
+	code := fmt.Sprintf("%s%06d%s", startStr, minutes, encoded)
+	return code
+}
+
+// verify time limit code
+func VerifyTimeLimitCode(data string, minutes int, code string) bool {
+	if len(code) <= 18 {
+		return false
+	}
+
+	// split code
+	start := code[:12]
+	lives := code[12:18]
+	if d, err := com.StrTo(lives).Int(); err == nil {
+		minutes = d
+	}
+
+	// right active code
+	retCode := CreateTimeLimitCode(data, minutes, start)
+	if retCode == code && minutes > 0 {
+		// check time is expired or not
+		before, _ := time.ParseInLocation("200601021504", start, time.Local)
+		now := time.Now()
+		if before.Add(time.Minute*time.Duration(minutes)).Unix() > now.Unix() {
+			return true
+		}
+	}
+
+	return false
+}

pkg/services/notifications/email.go

@@ -5,60 +5,23 @@ import (
 	"github.com/grafana/grafana/pkg/setting"
 )
 
-type SendEmailCommand struct {
-	To      []string
-	From    string
-	Subject string
-	Body    string
-	Type    string
-	Massive bool
-	Info    string
-}
-
-type SendResetPasswordEmailCommand struct {
-	Email string
-}
-
-// create mail content
-func (m *SendEmailCommand) Content() string {
-	// set mail type
-	contentType := "text/plain; charset=UTF-8"
-	if m.Type == "html" {
-		contentType = "text/html; charset=UTF-8"
-	}
-
-	// create mail content
-	content := "From: " + m.From + "\r\nSubject: " + m.Subject + "\r\nContent-Type: " + contentType + "\r\n\r\n" + m.Body
-	return content
-}
-
-// Create html mail command
-func NewSendEmailCommand(To []string, From, Subject, Body string) SendEmailCommand {
-	return SendEmailCommand{
-		To:      To,
-		From:    From,
-		Subject: Subject,
-		Body:    Body,
-		Type:    "html",
-	}
+// Create New mail message use MailFrom and MailUser
+func newMailMessageFrom(To []string, from, subject, body string) m.SendEmailCommand {
+	return m.NewSendEmailCommand(To, from, subject, body)
 }
 
 // Create New mail message use MailFrom and MailUser
-func NewMailMessageFrom(To []string, from, subject, body string) SendEmailCommand {
-	return NewSendEmailCommand(To, from, subject, body)
+func newMailMessage(To string, subject, body string) m.SendEmailCommand {
+	return newMailMessageFrom([]string{To}, setting.Smtp.FromAddress, subject, body)
 }
 
-// Create New mail message use MailFrom and MailUser
-func NewMailMessage(To string, subject, body string) SendEmailCommand {
-	return NewMailMessageFrom([]string{To}, setting.Smtp.FromAddress, subject, body)
-}
-
-func GetMailTmplData(u *m.User) map[interface{}]interface{} {
+func getMailTmplData(u *m.User) map[interface{}]interface{} {
 	data := make(map[interface{}]interface{}, 10)
 	data["AppUrl"] = setting.AppUrl
 	data["BuildVersion"] = setting.BuildVersion
 	data["BuildStamp"] = setting.BuildStamp
 	data["BuildCommit"] = setting.BuildCommit
+	data["Subject"] = map[string]interface{}{}
 	if u != nil {
 		data["User"] = u
 	}

pkg/services/notifications/notifications.go

@@ -1,11 +1,77 @@
 package notifications
 
-import "github.com/grafana/grafana/pkg/bus"
+import (
+	"bytes"
+	"encoding/hex"
+	"errors"
+	"html/template"
+	"path/filepath"
 
-func Init() {
+	"github.com/Unknwon/com"
+	"github.com/grafana/grafana/pkg/bus"
+	m "github.com/grafana/grafana/pkg/models"
+	"github.com/grafana/grafana/pkg/setting"
+	"github.com/grafana/grafana/pkg/util"
+)
+
+var mailTemplates *template.Template
+var tmplResetPassword = "reset_password.html"
+
+func Init() error {
 	bus.AddHandler("email", sendResetPasswordEmail)
-}
 
-func sendResetPasswordEmail(cmd *SendResetPasswordEmailCommand) error {
+	mailTemplates = template.New("name")
+	mailTemplates.Funcs(template.FuncMap{
+		"Subject": subjectTemplateFunc,
+	})
+
+	templatePattern := filepath.Join(setting.StaticRootPath, "emails/*.html")
+	_, err := mailTemplates.ParseGlob(templatePattern)
+	if err != nil {
+		return err
+	}
+
+	if !util.IsEmail(setting.Smtp.FromAddress) {
+		return errors.New("Invalid email address for smpt from_adress config")
+	}
+
 	return nil
 }
+
+var dispatchMail = func(cmd *m.SendEmailCommand) error {
+	return bus.Dispatch(cmd)
+}
+
+func subjectTemplateFunc(obj map[string]interface{}, value string) string {
+	obj["value"] = value
+	return ""
+}
+
+func sendResetPasswordEmail(cmd *m.SendResetPasswordEmailCommand) error {
+	var buffer bytes.Buffer
+
+	var data = getMailTmplData(nil)
+	code := CreateUserActiveCode(cmd.User, nil)
+	data["Code"] = code
+
+	mailTemplates.ExecuteTemplate(&buffer, tmplResetPassword, data)
+
+	dispatchMail(&m.SendEmailCommand{
+		To:      []string{cmd.User.Email},
+		From:    setting.Smtp.FromAddress,
+		Subject: data["Subject"].(map[string]interface{})["value"].(string),
+		Body:    buffer.String(),
+	})
+
+	return nil
+}
+
+func CreateUserActiveCode(u *m.User, startInf interface{}) string {
+	minutes := setting.EmailCodeValidMinutes
+	data := com.ToStr(u.Id) + u.Email + u.Login + u.Password + u.Rands
+	code := CreateTimeLimitCode(data, minutes, startInf)
+
+	// add tail hex username
+	code += hex.EncodeToString([]byte(u.Login))
+	return code
+}

pkg/services/notifications/notifications_test.go

@@ -0,0 +1,37 @@
+package notifications
+
+import (
+	"testing"
+
+	"github.com/grafana/grafana/pkg/bus"
+	m "github.com/grafana/grafana/pkg/models"
+	"github.com/grafana/grafana/pkg/setting"
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+func TestNotifications(t *testing.T) {
+
+	Convey("Given the notifications service", t, func() {
+		bus.ClearBusHandlers()
+
+		setting.StaticRootPath = "../../../public/"
+		setting.Smtp.FromAddress = "from@address.com"
+
+		err := Init()
+		So(err, ShouldBeNil)
+
+		var sentMail *m.SendEmailCommand
+		dispatchMail = func(cmd *m.SendEmailCommand) error {
+			sentMail = cmd
+			return nil
+		}
+
+		Convey("When sending reset email password", func() {
+			sendResetPasswordEmail(&m.SendResetPasswordEmailCommand{User: &m.User{Email: "asd@asd.com"}})
+			So(sentMail.Body, ShouldContainSubstring, "h2")
+			So(sentMail.Subject, ShouldEqual, "Welcome to Grafana")
+			So(sentMail.Body, ShouldNotContainSubstring, "Subject")
+		})
+	})
+
+}