mirror of
				https://github.com/go-gitea/gitea
				synced 2025-11-04 05:18:25 +00:00 
			
		
		
		
	Add send register confirm mail
This commit is contained in:
		@@ -39,6 +39,7 @@ There are two ways to install Gogs:
 | 
			
		||||
 | 
			
		||||
## Acknowledgments
 | 
			
		||||
 | 
			
		||||
- Mail service is based on [WeTalk](https://github.com/beego/wetalk).
 | 
			
		||||
- Logo inspired by [martini](https://github.com/martini-contrib).
 | 
			
		||||
 | 
			
		||||
## Contributors
 | 
			
		||||
 
 | 
			
		||||
@@ -35,14 +35,17 @@ SECRET_KEY = !#@FDEWREWR&*(
 | 
			
		||||
ACTIVE_CODE_LIVE_MINUTES = 180
 | 
			
		||||
RESET_PASSWD_CODE_LIVE_MINUTES = 180
 | 
			
		||||
; User need to confirm e-mail for registration
 | 
			
		||||
REGISTER_EMAIL_CONFIRM = true
 | 
			
		||||
REGISTER_EMAIL_CONFIRM = false
 | 
			
		||||
 | 
			
		||||
[mailer]
 | 
			
		||||
ENABLED = false
 | 
			
		||||
; Name displayed in mail title
 | 
			
		||||
SUBJECT = %(APP_NAME)s
 | 
			
		||||
; Mail server
 | 
			
		||||
; Gmail: smtp.gmail.com:587
 | 
			
		||||
HOST = 
 | 
			
		||||
; Mail from address
 | 
			
		||||
FROM = 
 | 
			
		||||
; Mailer user name and password
 | 
			
		||||
USER = 
 | 
			
		||||
PASSWD = 
 | 
			
		||||
 
 | 
			
		||||
@@ -105,19 +105,19 @@ func GetUserSalt() string {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// RegisterUser creates record of a new user.
 | 
			
		||||
func RegisterUser(user *User) (err error) {
 | 
			
		||||
func RegisterUser(user *User) (*User, error) {
 | 
			
		||||
	isExist, err := IsUserExist(user.Name)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
		return nil, err
 | 
			
		||||
	} else if isExist {
 | 
			
		||||
		return ErrUserAlreadyExist
 | 
			
		||||
		return nil, ErrUserAlreadyExist
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	isExist, err = IsEmailUsed(user.Email)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
		return nil, err
 | 
			
		||||
	} else if isExist {
 | 
			
		||||
		return ErrEmailAlreadyUsed
 | 
			
		||||
		return nil, ErrEmailAlreadyUsed
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	user.LowerName = strings.ToLower(user.Name)
 | 
			
		||||
@@ -126,22 +126,17 @@ func RegisterUser(user *User) (err error) {
 | 
			
		||||
	user.Expired = time.Now().Add(3 * 24 * time.Hour)
 | 
			
		||||
	user.Rands = GetUserSalt()
 | 
			
		||||
	if err = user.EncodePasswd(); err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
		return nil, err
 | 
			
		||||
	} else if _, err = orm.Insert(user); err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
		return nil, err
 | 
			
		||||
	} else if err = os.MkdirAll(UserPath(user.Name), os.ModePerm); err != nil {
 | 
			
		||||
		if _, err := orm.Id(user.Id).Delete(&User{}); err != nil {
 | 
			
		||||
			return errors.New(fmt.Sprintf(
 | 
			
		||||
			return nil, errors.New(fmt.Sprintf(
 | 
			
		||||
				"both create userpath %s and delete table record faild: %v", user.Name, err))
 | 
			
		||||
		}
 | 
			
		||||
		return err
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Send confirmation e-mail.
 | 
			
		||||
	if base.Service.RegisterEmailConfitm {
 | 
			
		||||
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
	return user, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// UpdateUser updates user's information.
 | 
			
		||||
 
 | 
			
		||||
@@ -16,7 +16,7 @@ import (
 | 
			
		||||
// create a time limit code for user active
 | 
			
		||||
func CreateUserActiveCode(user *models.User, startInf interface{}) string {
 | 
			
		||||
	hours := base.Service.ActiveCodeLives / 60
 | 
			
		||||
	data := fmt.Sprintf("%d", user.Id) + user.Email + user.LowerName + user.Passwd + user.Rands
 | 
			
		||||
	data := base.ToStr(user.Id) + user.Email + user.LowerName + user.Passwd + user.Rands
 | 
			
		||||
	code := base.CreateTimeLimitCode(data, hours, startInf)
 | 
			
		||||
 | 
			
		||||
	// add tail hex username
 | 
			
		||||
@@ -32,11 +32,10 @@ func SendRegisterMail(user *models.User) {
 | 
			
		||||
	data := mailer.GetMailTmplData(user)
 | 
			
		||||
	data["Code"] = code
 | 
			
		||||
	body := base.RenderTemplate("mail/auth/register_success.html", data)
 | 
			
		||||
	_, _, _ = code, subject, body
 | 
			
		||||
 | 
			
		||||
	// msg := mailer.NewMailMessage([]string{user.Email}, subject, body)
 | 
			
		||||
	// msg.Info = fmt.Sprintf("UID: %d, send register mail", user.Id)
 | 
			
		||||
	msg := mailer.NewMailMessage([]string{user.Email}, subject, body)
 | 
			
		||||
	msg.Info = fmt.Sprintf("UID: %d, send register mail", user.Id)
 | 
			
		||||
 | 
			
		||||
	// // async send mail
 | 
			
		||||
	// mailer.SendAsync(msg)
 | 
			
		||||
	// async send mail
 | 
			
		||||
	mailer.SendAsync(msg)
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -37,7 +37,7 @@ var (
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var Service struct {
 | 
			
		||||
	RegisterEmailConfitm bool
 | 
			
		||||
	RegisterEmailConfirm bool
 | 
			
		||||
	ActiveCodeLives      int
 | 
			
		||||
	ResetPwdCodeLives    int
 | 
			
		||||
}
 | 
			
		||||
@@ -138,7 +138,7 @@ func newRegisterService() {
 | 
			
		||||
		log.Warn("Register Service: Mail Service is not enabled")
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	Service.RegisterEmailConfitm = true
 | 
			
		||||
	Service.RegisterEmailConfirm = true
 | 
			
		||||
	log.Info("Register Service Enabled")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -13,6 +13,7 @@ import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"math"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
@@ -59,13 +60,14 @@ func CreateTimeLimitCode(data string, minutes int, startInf interface{}) string
 | 
			
		||||
 | 
			
		||||
	// create sha1 encode string
 | 
			
		||||
	sh := sha1.New()
 | 
			
		||||
	sh.Write([]byte(data + SecretKey + startStr + endStr + fmt.Sprintf("%d", minutes)))
 | 
			
		||||
	sh.Write([]byte(data + SecretKey + startStr + endStr + ToStr(minutes)))
 | 
			
		||||
	encoded := hex.EncodeToString(sh.Sum(nil))
 | 
			
		||||
 | 
			
		||||
	code := fmt.Sprintf("%s%06d%s", startStr, minutes, encoded)
 | 
			
		||||
	return code
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// TODO:
 | 
			
		||||
func RenderTemplate(TplNames string, Data map[interface{}]interface{}) string {
 | 
			
		||||
	// if beego.RunMode == "dev" {
 | 
			
		||||
	// 	beego.BuildTemplate(beego.ViewsPath)
 | 
			
		||||
@@ -300,6 +302,57 @@ func DateFormat(t time.Time, format string) string {
 | 
			
		||||
	return t.Format(format)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type argInt []int
 | 
			
		||||
 | 
			
		||||
func (a argInt) Get(i int, args ...int) (r int) {
 | 
			
		||||
	if i >= 0 && i < len(a) {
 | 
			
		||||
		r = a[i]
 | 
			
		||||
	}
 | 
			
		||||
	if len(args) > 0 {
 | 
			
		||||
		r = args[0]
 | 
			
		||||
	}
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// convert any type to string
 | 
			
		||||
func ToStr(value interface{}, args ...int) (s string) {
 | 
			
		||||
	switch v := value.(type) {
 | 
			
		||||
	case bool:
 | 
			
		||||
		s = strconv.FormatBool(v)
 | 
			
		||||
	case float32:
 | 
			
		||||
		s = strconv.FormatFloat(float64(v), 'f', argInt(args).Get(0, -1), argInt(args).Get(1, 32))
 | 
			
		||||
	case float64:
 | 
			
		||||
		s = strconv.FormatFloat(v, 'f', argInt(args).Get(0, -1), argInt(args).Get(1, 64))
 | 
			
		||||
	case int:
 | 
			
		||||
		s = strconv.FormatInt(int64(v), argInt(args).Get(0, 10))
 | 
			
		||||
	case int8:
 | 
			
		||||
		s = strconv.FormatInt(int64(v), argInt(args).Get(0, 10))
 | 
			
		||||
	case int16:
 | 
			
		||||
		s = strconv.FormatInt(int64(v), argInt(args).Get(0, 10))
 | 
			
		||||
	case int32:
 | 
			
		||||
		s = strconv.FormatInt(int64(v), argInt(args).Get(0, 10))
 | 
			
		||||
	case int64:
 | 
			
		||||
		s = strconv.FormatInt(v, argInt(args).Get(0, 10))
 | 
			
		||||
	case uint:
 | 
			
		||||
		s = strconv.FormatUint(uint64(v), argInt(args).Get(0, 10))
 | 
			
		||||
	case uint8:
 | 
			
		||||
		s = strconv.FormatUint(uint64(v), argInt(args).Get(0, 10))
 | 
			
		||||
	case uint16:
 | 
			
		||||
		s = strconv.FormatUint(uint64(v), argInt(args).Get(0, 10))
 | 
			
		||||
	case uint32:
 | 
			
		||||
		s = strconv.FormatUint(uint64(v), argInt(args).Get(0, 10))
 | 
			
		||||
	case uint64:
 | 
			
		||||
		s = strconv.FormatUint(v, argInt(args).Get(0, 10))
 | 
			
		||||
	case string:
 | 
			
		||||
		s = v
 | 
			
		||||
	case []byte:
 | 
			
		||||
		s = string(v)
 | 
			
		||||
	default:
 | 
			
		||||
		s = fmt.Sprintf("%v", v)
 | 
			
		||||
	}
 | 
			
		||||
	return s
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Actioner interface {
 | 
			
		||||
	GetOpType() int
 | 
			
		||||
	GetActUserName() string
 | 
			
		||||
 
 | 
			
		||||
@@ -9,6 +9,13 @@ import (
 | 
			
		||||
	"github.com/gogits/gogs/modules/base"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Create New mail message use MailFrom and MailUser
 | 
			
		||||
func NewMailMessage(To []string, subject, body string) Message {
 | 
			
		||||
	msg := NewHtmlMessage(To, base.MailService.User, subject, body)
 | 
			
		||||
	msg.User = base.MailService.User
 | 
			
		||||
	return msg
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func GetMailTmplData(user *models.User) map[interface{}]interface{} {
 | 
			
		||||
	data := make(map[interface{}]interface{}, 10)
 | 
			
		||||
	data["AppName"] = base.AppName
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										112
									
								
								modules/mailer/mailer.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										112
									
								
								modules/mailer/mailer.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,112 @@
 | 
			
		||||
// Copyright 2014 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"
 | 
			
		||||
	"net/smtp"
 | 
			
		||||
	"strings"
 | 
			
		||||
 | 
			
		||||
	"github.com/gogits/gogs/modules/base"
 | 
			
		||||
	"github.com/gogits/gogs/modules/log"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Message struct {
 | 
			
		||||
	To      []string
 | 
			
		||||
	From    string
 | 
			
		||||
	Subject string
 | 
			
		||||
	Body    string
 | 
			
		||||
	User    string
 | 
			
		||||
	Type    string
 | 
			
		||||
	Massive bool
 | 
			
		||||
	Info    string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// create mail content
 | 
			
		||||
func (m Message) 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.User + "<" + m.From +
 | 
			
		||||
		">\r\nSubject: " + m.Subject + "\r\nContent-Type: " + contentType + "\r\n\r\n" + m.Body
 | 
			
		||||
	return content
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Direct Send mail message
 | 
			
		||||
func Send(msg Message) (int, error) {
 | 
			
		||||
	log.Trace("Sending mails to: %s", strings.Join(msg.To, "; "))
 | 
			
		||||
	host := strings.Split(base.MailService.Host, ":")
 | 
			
		||||
 | 
			
		||||
	// get message body
 | 
			
		||||
	content := msg.Content()
 | 
			
		||||
 | 
			
		||||
	auth := smtp.PlainAuth("", base.MailService.User, base.MailService.Passwd, host[0])
 | 
			
		||||
 | 
			
		||||
	if len(msg.To) == 0 {
 | 
			
		||||
		return 0, fmt.Errorf("empty receive emails")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if len(msg.Body) == 0 {
 | 
			
		||||
		return 0, fmt.Errorf("empty email body")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if msg.Massive {
 | 
			
		||||
		// send mail to multiple emails one by one
 | 
			
		||||
		num := 0
 | 
			
		||||
		for _, to := range msg.To {
 | 
			
		||||
			body := []byte("To: " + to + "\r\n" + content)
 | 
			
		||||
			err := smtp.SendMail(base.MailService.Host, auth, msg.From, []string{to}, body)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				return num, err
 | 
			
		||||
			}
 | 
			
		||||
			num++
 | 
			
		||||
		}
 | 
			
		||||
		return num, nil
 | 
			
		||||
	} else {
 | 
			
		||||
		body := []byte("To: " + strings.Join(msg.To, ";") + "\r\n" + content)
 | 
			
		||||
 | 
			
		||||
		// send to multiple emails in one message
 | 
			
		||||
		err := smtp.SendMail(base.MailService.Host, auth, msg.From, msg.To, body)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return 0, err
 | 
			
		||||
		} else {
 | 
			
		||||
			return 1, nil
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Async Send mail message
 | 
			
		||||
func SendAsync(msg Message) {
 | 
			
		||||
	// TODO may be need pools limit concurrent nums
 | 
			
		||||
	go func() {
 | 
			
		||||
		num, err := Send(msg)
 | 
			
		||||
		tos := strings.Join(msg.To, "; ")
 | 
			
		||||
		info := ""
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			if len(msg.Info) > 0 {
 | 
			
		||||
				info = ", info: " + msg.Info
 | 
			
		||||
			}
 | 
			
		||||
			// log failed
 | 
			
		||||
			log.Error(fmt.Sprintf("Async sent email %d succeed, not send emails: %s%s err: %s", num, tos, info, err))
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		log.Trace(fmt.Sprintf("Async sent email %d succeed, sent emails: %s%s", num, tos, info))
 | 
			
		||||
	}()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Create html mail message
 | 
			
		||||
func NewHtmlMessage(To []string, From, Subject, Body string) Message {
 | 
			
		||||
	return Message{
 | 
			
		||||
		To:      To,
 | 
			
		||||
		From:    From,
 | 
			
		||||
		Subject: Subject,
 | 
			
		||||
		Body:    Body,
 | 
			
		||||
		Type:    "html",
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
@@ -50,10 +50,10 @@ func SettingPost(ctx *middleware.Context) {
 | 
			
		||||
 | 
			
		||||
		if err := models.DeleteRepository(ctx.User.Id, ctx.Repo.Repository.Id, ctx.User.LowerName); err != nil {
 | 
			
		||||
			ctx.Handle(200, "repo.Delete", err)
 | 
			
		||||
			log.Trace("%s Repository deleted: %s/%s", ctx.Req.RequestURI, ctx.User.LowerName, ctx.Repo.Repository.LowerName)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	log.Trace("%s Repository deleted: %s/%s", ctx.Req.RequestURI, ctx.User.LowerName, ctx.Repo.Repository.LowerName)
 | 
			
		||||
	ctx.Render.Redirect("/", 302)
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -134,10 +134,11 @@ func SignUp(ctx *middleware.Context, form auth.RegisterForm) {
 | 
			
		||||
		Name:     form.UserName,
 | 
			
		||||
		Email:    form.Email,
 | 
			
		||||
		Passwd:   form.Password,
 | 
			
		||||
		IsActive: !base.Service.RegisterEmailConfitm,
 | 
			
		||||
		IsActive: !base.Service.RegisterEmailConfirm,
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if err := models.RegisterUser(u); err != nil {
 | 
			
		||||
	var err error
 | 
			
		||||
	if u, err = models.RegisterUser(u); err != nil {
 | 
			
		||||
		switch err.Error() {
 | 
			
		||||
		case models.ErrUserAlreadyExist.Error():
 | 
			
		||||
			ctx.RenderWithErr("Username has been already taken", "user/signup", &form)
 | 
			
		||||
@@ -150,6 +151,11 @@ func SignUp(ctx *middleware.Context, form auth.RegisterForm) {
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	log.Trace("%s User created: %s", ctx.Req.RequestURI, strings.ToLower(form.UserName))
 | 
			
		||||
 | 
			
		||||
	// Send confirmation e-mail.
 | 
			
		||||
	if base.Service.RegisterEmailConfirm {
 | 
			
		||||
		auth.SendRegisterMail(u)
 | 
			
		||||
	}
 | 
			
		||||
	ctx.Render.Redirect("/user/login")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user