gitroast/routers/user/user.go

458 lines
12 KiB
Go
Raw Normal View History

2014-02-17 23:38:50 +00:00
// 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 user
import (
2014-03-22 21:59:22 +00:00
"net/url"
2014-03-19 08:48:45 +00:00
"strings"
2014-02-17 23:38:50 +00:00
"github.com/gogits/gogs/models"
2014-03-06 07:21:44 +00:00
"github.com/gogits/gogs/modules/auth"
"github.com/gogits/gogs/modules/base"
2014-03-19 08:48:45 +00:00
"github.com/gogits/gogs/modules/log"
2014-03-19 14:46:48 +00:00
"github.com/gogits/gogs/modules/mailer"
"github.com/gogits/gogs/modules/middleware"
2014-05-26 00:11:25 +00:00
"github.com/gogits/gogs/modules/setting"
2014-02-17 23:38:50 +00:00
)
2014-06-23 03:11:12 +00:00
const (
SIGNIN base.TplName = "user/signin"
SIGNUP base.TplName = "user/signup"
DELETE base.TplName = "user/delete"
ACTIVATE base.TplName = "user/activate"
FORGOT_PASSWORD base.TplName = "user/forgot_passwd"
RESET_PASSWORD base.TplName = "user/reset_passwd"
)
2014-04-10 20:36:50 +00:00
func SignIn(ctx *middleware.Context) {
2014-03-15 13:17:16 +00:00
ctx.Data["Title"] = "Log In"
2014-03-06 16:42:14 +00:00
2014-04-13 22:12:07 +00:00
if _, ok := ctx.Session.Get("socialId").(int64); ok {
ctx.Data["IsSocialLogin"] = true
2014-06-23 03:11:12 +00:00
ctx.HTML(200, SIGNIN)
2014-04-13 22:12:07 +00:00
return
}
2014-05-31 21:15:04 +00:00
if setting.OauthService != nil {
ctx.Data["OauthEnabled"] = true
ctx.Data["OauthService"] = setting.OauthService
}
2014-04-10 20:36:50 +00:00
// Check auto-login.
2014-06-06 02:07:35 +00:00
uname := ctx.GetCookie(setting.CookieUserName)
if len(uname) == 0 {
2014-06-23 03:11:12 +00:00
ctx.HTML(200, SIGNIN)
2014-04-10 20:36:50 +00:00
return
}
2014-03-22 20:40:09 +00:00
2014-04-10 20:36:50 +00:00
isSucceed := false
defer func() {
if !isSucceed {
2014-06-06 02:07:35 +00:00
log.Trace("user.SignIn(auto-login cookie cleared): %s", uname)
2014-05-26 00:11:25 +00:00
ctx.SetCookie(setting.CookieUserName, "", -1)
ctx.SetCookie(setting.CookieRememberName, "", -1)
2014-04-11 17:01:30 +00:00
return
2014-03-22 20:40:09 +00:00
}
2014-04-10 20:36:50 +00:00
}()
2014-03-22 20:40:09 +00:00
2014-06-06 02:07:35 +00:00
user, err := models.GetUserByName(uname)
2014-04-10 20:36:50 +00:00
if err != nil {
2014-05-05 20:21:43 +00:00
ctx.Handle(500, "user.SignIn(GetUserByName)", err)
2014-04-10 20:36:50 +00:00
return
}
2014-03-22 20:40:09 +00:00
2014-04-10 20:36:50 +00:00
secret := base.EncodeMd5(user.Rands + user.Passwd)
2014-05-26 00:11:25 +00:00
value, _ := ctx.GetSecureCookie(secret, setting.CookieRememberName)
2014-04-10 20:36:50 +00:00
if value != user.Name {
2014-06-23 03:11:12 +00:00
ctx.HTML(200, SIGNIN)
2014-03-06 16:42:14 +00:00
return
}
2014-04-10 20:36:50 +00:00
isSucceed = true
2014-04-11 17:01:30 +00:00
2014-04-10 20:36:50 +00:00
ctx.Session.Set("userId", user.Id)
ctx.Session.Set("userName", user.Name)
if redirectTo, _ := url.QueryUnescape(ctx.GetCookie("redirect_to")); len(redirectTo) > 0 {
ctx.SetCookie("redirect_to", "", -1)
ctx.Redirect(redirectTo)
return
}
ctx.Redirect("/")
}
func SignInPost(ctx *middleware.Context, form auth.LogInForm) {
ctx.Data["Title"] = "Log In"
2014-04-13 22:12:07 +00:00
sid, isOauth := ctx.Session.Get("socialId").(int64)
if isOauth {
ctx.Data["IsSocialLogin"] = true
2014-05-26 00:11:25 +00:00
} else if setting.OauthService != nil {
2014-04-10 20:36:50 +00:00
ctx.Data["OauthEnabled"] = true
2014-05-26 00:11:25 +00:00
ctx.Data["OauthService"] = setting.OauthService
2014-04-10 20:36:50 +00:00
}
if ctx.HasError() {
2014-06-23 03:11:12 +00:00
ctx.HTML(200, SIGNIN)
2014-03-06 16:42:14 +00:00
return
}
2014-06-06 02:07:35 +00:00
user, err := models.UserSignIn(form.UserName, form.Password)
2014-03-06 16:42:14 +00:00
if err != nil {
2014-03-22 12:49:53 +00:00
if err == models.ErrUserNotExist {
log.Trace("%s Log in failed: %s", ctx.Req.RequestURI, form.UserName)
2014-06-23 03:11:12 +00:00
ctx.RenderWithErr("Username or password is not correct", SIGNIN, &form)
return
}
2014-03-06 16:42:14 +00:00
2014-06-06 02:07:35 +00:00
ctx.Handle(500, "user.SignInPost(UserSignIn)", err)
2014-03-06 16:42:14 +00:00
return
}
2014-03-06 16:42:14 +00:00
2014-05-05 20:21:43 +00:00
if form.Remember {
2014-03-22 20:40:09 +00:00
secret := base.EncodeMd5(user.Rands + user.Passwd)
2014-05-26 00:11:25 +00:00
days := 86400 * setting.LogInRememberDays
ctx.SetCookie(setting.CookieUserName, user.Name, days)
ctx.SetSecureCookie(secret, setting.CookieRememberName, user.Name, days)
2014-03-22 20:40:09 +00:00
}
2014-04-13 22:12:07 +00:00
// Bind with social account.
if isOauth {
2014-04-11 17:01:30 +00:00
if err = models.BindUserOauth2(user.Id, sid); err != nil {
2014-04-13 22:12:07 +00:00
if err == models.ErrOauth2RecordNotExist {
ctx.Handle(404, "user.SignInPost(GetOauth2ById)", err)
} else {
ctx.Handle(500, "user.SignInPost(GetOauth2ById)", err)
}
return
2014-04-11 17:01:30 +00:00
}
ctx.Session.Delete("socialId")
2014-04-13 22:12:07 +00:00
log.Trace("%s OAuth binded: %s -> %d", ctx.Req.RequestURI, form.UserName, sid)
2014-04-11 17:01:30 +00:00
}
2014-04-13 22:12:07 +00:00
2014-03-15 14:34:33 +00:00
ctx.Session.Set("userId", user.Id)
ctx.Session.Set("userName", user.Name)
2014-04-10 20:36:50 +00:00
if redirectTo, _ := url.QueryUnescape(ctx.GetCookie("redirect_to")); len(redirectTo) > 0 {
2014-03-22 21:59:22 +00:00
ctx.SetCookie("redirect_to", "", -1)
ctx.Redirect(redirectTo)
2014-04-10 20:36:50 +00:00
return
2014-03-22 21:59:22 +00:00
}
2014-04-10 20:36:50 +00:00
ctx.Redirect("/")
2014-02-17 23:38:50 +00:00
}
2014-03-19 13:57:55 +00:00
func SignOut(ctx *middleware.Context) {
ctx.Session.Delete("userId")
ctx.Session.Delete("userName")
2014-04-11 17:01:30 +00:00
ctx.Session.Delete("socialId")
2014-04-12 01:42:09 +00:00
ctx.Session.Delete("socialName")
ctx.Session.Delete("socialEmail")
2014-05-26 00:11:25 +00:00
ctx.SetCookie(setting.CookieUserName, "", -1)
ctx.SetCookie(setting.CookieRememberName, "", -1)
2014-03-19 13:57:55 +00:00
ctx.Redirect("/")
2014-03-06 18:18:19 +00:00
}
2014-04-10 20:36:50 +00:00
func SignUp(ctx *middleware.Context) {
2014-03-15 14:34:33 +00:00
ctx.Data["Title"] = "Sign Up"
ctx.Data["PageIsSignUp"] = true
2014-03-06 07:21:44 +00:00
2014-05-26 00:11:25 +00:00
if setting.Service.DisableRegistration {
ctx.Data["DisableRegistration"] = true
2014-06-23 03:11:12 +00:00
ctx.HTML(200, SIGNUP)
2014-03-21 05:09:22 +00:00
return
}
2014-04-13 22:12:07 +00:00
if sid, ok := ctx.Session.Get("socialId").(int64); ok {
oauthSignUp(ctx, sid)
return
}
2014-06-23 03:11:12 +00:00
ctx.HTML(200, SIGNUP)
2014-04-13 22:12:07 +00:00
}
func oauthSignUp(ctx *middleware.Context, sid int64) {
ctx.Data["Title"] = "OAuth Sign Up"
ctx.Data["PageIsSignUp"] = true
if _, err := models.GetOauth2ById(sid); err != nil {
if err == models.ErrOauth2RecordNotExist {
ctx.Handle(404, "user.oauthSignUp(GetOauth2ById)", err)
} else {
ctx.Handle(500, "user.oauthSignUp(GetOauth2ById)", err)
}
return
}
ctx.Data["IsSocialLogin"] = true
ctx.Data["username"] = strings.Replace(ctx.Session.Get("socialName").(string), " ", "", -1)
ctx.Data["email"] = ctx.Session.Get("socialEmail")
log.Trace("user.oauthSignUp(social ID): %v", ctx.Session.Get("socialId"))
2014-06-23 03:11:12 +00:00
ctx.HTML(200, SIGNUP)
2014-04-10 20:36:50 +00:00
}
func SignUpPost(ctx *middleware.Context, form auth.RegisterForm) {
ctx.Data["Title"] = "Sign Up"
ctx.Data["PageIsSignUp"] = true
2014-05-26 00:11:25 +00:00
if setting.Service.DisableRegistration {
2014-04-10 20:36:50 +00:00
ctx.Handle(403, "user.SignUpPost", nil)
2014-02-17 23:38:50 +00:00
return
}
2014-04-13 22:12:07 +00:00
sid, isOauth := ctx.Session.Get("socialId").(int64)
if isOauth {
ctx.Data["IsSocialLogin"] = true
}
2014-05-05 20:21:43 +00:00
if ctx.HasError() {
2014-06-23 03:11:12 +00:00
ctx.HTML(200, SIGNUP)
2014-05-05 20:21:43 +00:00
return
}
2014-03-06 16:10:35 +00:00
if form.Password != form.RetypePasswd {
2014-03-15 14:34:33 +00:00
ctx.Data["Err_Password"] = true
ctx.Data["Err_RetypePasswd"] = true
2014-06-23 03:11:12 +00:00
ctx.RenderWithErr("Password and re-type password are not same.", SIGNUP, &form)
2014-03-06 07:21:44 +00:00
return
2014-02-18 22:31:16 +00:00
}
2014-03-04 00:03:08 +00:00
2014-03-06 07:21:44 +00:00
u := &models.User{
2014-03-19 11:21:23 +00:00
Name: form.UserName,
Email: form.Email,
Passwd: form.Password,
2014-05-26 00:11:25 +00:00
IsActive: !setting.Service.RegisterEmailConfirm || isOauth,
2014-02-18 22:31:16 +00:00
}
2014-03-06 07:21:44 +00:00
2014-03-19 12:27:27 +00:00
var err error
2014-06-25 04:44:48 +00:00
if u, err = models.CreateUser(u); err != nil {
2014-03-20 15:41:24 +00:00
switch err {
case models.ErrUserAlreadyExist:
2014-06-06 02:07:35 +00:00
ctx.Data["Err_UserName"] = true
2014-06-23 03:11:12 +00:00
ctx.RenderWithErr("Username has been already taken", SIGNUP, &form)
2014-03-20 15:41:24 +00:00
case models.ErrEmailAlreadyUsed:
2014-06-06 02:07:35 +00:00
ctx.Data["Err_Email"] = true
2014-06-23 03:11:12 +00:00
ctx.RenderWithErr("E-mail address has been already used", SIGNUP, &form)
2014-03-20 15:41:24 +00:00
case models.ErrUserNameIllegal:
2014-06-25 04:44:48 +00:00
ctx.Data["Err_UserName"] = true
2014-06-23 03:11:12 +00:00
ctx.RenderWithErr(models.ErrRepoNameIllegal.Error(), SIGNUP, &form)
2014-03-06 16:10:35 +00:00
default:
2014-06-25 04:44:48 +00:00
ctx.Handle(500, "user.SignUpPost(CreateUser)", err)
2014-03-06 07:21:44 +00:00
}
2014-02-18 22:31:16 +00:00
return
}
2014-06-25 04:44:48 +00:00
log.Trace("%s User created: %s", ctx.Req.RequestURI, u.Name)
2014-04-13 22:12:07 +00:00
// Bind social account.
if isOauth {
if err = models.BindUserOauth2(u.Id, sid); err != nil {
ctx.Handle(500, "user.SignUp(BindUserOauth2)", err)
return
}
2014-04-11 17:01:30 +00:00
ctx.Session.Delete("socialId")
2014-04-13 22:12:07 +00:00
log.Trace("%s OAuth binded: %s -> %d", ctx.Req.RequestURI, form.UserName, sid)
2014-04-11 17:01:30 +00:00
}
2014-03-19 12:27:27 +00:00
2014-04-13 22:12:07 +00:00
// Send confirmation e-mail, no need for social account.
2014-05-26 00:11:25 +00:00
if !isOauth && setting.Service.RegisterEmailConfirm && u.Id > 1 {
2014-03-19 14:46:48 +00:00
mailer.SendRegisterMail(ctx.Render, u)
ctx.Data["IsSendRegisterMail"] = true
ctx.Data["Email"] = u.Email
2014-05-26 00:11:25 +00:00
ctx.Data["Hours"] = setting.Service.ActiveCodeLives / 60
2014-04-19 01:17:43 +00:00
ctx.HTML(200, "user/activate")
2014-03-21 14:09:57 +00:00
if err = ctx.Cache.Put("MailResendLimit_"+u.LowerName, u.LowerName, 180); err != nil {
log.Error("Set cache(MailResendLimit) fail: %v", err)
}
2014-03-19 14:46:48 +00:00
return
2014-03-19 12:27:27 +00:00
}
2014-06-06 02:07:35 +00:00
2014-03-19 13:57:55 +00:00
ctx.Redirect("/user/login")
2014-02-17 23:38:50 +00:00
}
2014-02-19 18:13:02 +00:00
2014-03-15 14:34:33 +00:00
func Delete(ctx *middleware.Context) {
ctx.Data["Title"] = "Delete Account"
2014-03-18 22:31:54 +00:00
ctx.Data["PageIsUserSetting"] = true
ctx.Data["IsUserPageSettingDelete"] = true
2014-06-23 03:11:12 +00:00
ctx.HTML(200, DELETE)
2014-04-10 20:36:50 +00:00
}
2014-04-10 20:36:50 +00:00
func DeletePost(ctx *middleware.Context) {
ctx.Data["Title"] = "Delete Account"
ctx.Data["PageIsUserSetting"] = true
ctx.Data["IsUserPageSettingDelete"] = true
2014-02-20 02:45:43 +00:00
2014-04-10 20:36:50 +00:00
tmpUser := models.User{
Passwd: ctx.Query("password"),
Salt: ctx.User.Salt,
}
2014-03-16 14:35:25 +00:00
tmpUser.EncodePasswd()
2014-04-10 20:36:50 +00:00
if tmpUser.Passwd != ctx.User.Passwd {
ctx.Flash.Error("Password is not correct. Make sure you are owner of this account.")
} else {
if err := models.DeleteUser(ctx.User); err != nil {
switch err {
case models.ErrUserOwnRepos:
2014-04-10 20:36:50 +00:00
ctx.Flash.Error("Your account still have ownership of repository, you have to delete or transfer them first.")
default:
2014-06-28 04:40:07 +00:00
ctx.Handle(500, "user.DeletePost(DeleteUser)", err)
return
}
} else {
2014-03-19 13:57:55 +00:00
ctx.Redirect("/")
2014-03-11 15:40:47 +00:00
return
}
}
2014-04-10 20:36:50 +00:00
ctx.Redirect("/user/delete")
2014-02-19 18:13:02 +00:00
}
2014-03-19 13:24:02 +00:00
func Activate(ctx *middleware.Context) {
code := ctx.Query("code")
if len(code) == 0 {
ctx.Data["IsActivatePage"] = true
2014-03-20 07:24:17 +00:00
if ctx.User.IsActive {
2014-03-23 05:12:55 +00:00
ctx.Handle(404, "user.Activate", nil)
2014-03-20 07:24:17 +00:00
return
}
2014-03-19 13:24:02 +00:00
// Resend confirmation e-mail.
2014-05-26 00:11:25 +00:00
if setting.Service.RegisterEmailConfirm {
2014-03-21 14:09:57 +00:00
if ctx.Cache.IsExist("MailResendLimit_" + ctx.User.LowerName) {
ctx.Data["ResendLimited"] = true
} else {
2014-05-26 00:11:25 +00:00
ctx.Data["Hours"] = setting.Service.ActiveCodeLives / 60
2014-03-21 14:09:57 +00:00
mailer.SendActiveMail(ctx.Render, ctx.User)
2014-04-10 01:42:25 +00:00
if err := ctx.Cache.Put("MailResendLimit_"+ctx.User.LowerName, ctx.User.LowerName, 180); err != nil {
log.Error("Set cache(MailResendLimit) fail: %v", err)
}
2014-03-21 14:09:57 +00:00
}
2014-03-19 13:24:02 +00:00
} else {
ctx.Data["ServiceNotEnabled"] = true
}
2014-06-23 03:11:12 +00:00
ctx.HTML(200, ACTIVATE)
2014-03-19 13:24:02 +00:00
return
}
2014-03-19 16:50:44 +00:00
// Verify code.
if user := models.VerifyUserActiveCode(code); user != nil {
user.IsActive = true
user.Rands = models.GetUserSalt()
2014-04-05 16:32:34 +00:00
if err := models.UpdateUser(user); err != nil {
ctx.Handle(404, "user.Activate", err)
return
}
2014-03-20 01:05:48 +00:00
2014-04-05 16:32:34 +00:00
log.Trace("%s User activated: %s", ctx.Req.RequestURI, user.Name)
2014-03-20 01:05:48 +00:00
2014-03-19 16:50:44 +00:00
ctx.Session.Set("userId", user.Id)
ctx.Session.Set("userName", user.Name)
2014-03-22 21:59:22 +00:00
ctx.Redirect("/")
2014-03-19 16:50:44 +00:00
return
}
ctx.Data["IsActivateFailed"] = true
2014-06-23 03:11:12 +00:00
ctx.HTML(200, ACTIVATE)
2014-03-19 13:24:02 +00:00
}
2014-04-05 16:32:34 +00:00
func ForgotPasswd(ctx *middleware.Context) {
ctx.Data["Title"] = "Forgot Password"
2014-05-26 00:11:25 +00:00
if setting.MailService == nil {
2014-04-05 16:32:34 +00:00
ctx.Data["IsResetDisable"] = true
2014-06-23 03:11:12 +00:00
ctx.HTML(200, FORGOT_PASSWORD)
2014-04-05 16:32:34 +00:00
return
}
ctx.Data["IsResetRequest"] = true
2014-06-23 03:11:12 +00:00
ctx.HTML(200, FORGOT_PASSWORD)
2014-04-10 20:36:50 +00:00
}
func ForgotPasswdPost(ctx *middleware.Context) {
ctx.Data["Title"] = "Forgot Password"
2014-05-26 00:11:25 +00:00
if setting.MailService == nil {
2014-04-10 20:36:50 +00:00
ctx.Handle(403, "user.ForgotPasswdPost", nil)
2014-04-05 16:32:34 +00:00
return
}
2014-04-10 20:36:50 +00:00
ctx.Data["IsResetRequest"] = true
2014-04-05 16:32:34 +00:00
email := ctx.Query("email")
u, err := models.GetUserByEmail(email)
if err != nil {
if err == models.ErrUserNotExist {
ctx.RenderWithErr("This e-mail address does not associate to any account.", "user/forgot_passwd", nil)
} else {
2014-04-10 20:36:50 +00:00
ctx.Handle(500, "user.ResetPasswd(check existence)", err)
2014-04-05 16:32:34 +00:00
}
return
}
2014-04-10 01:42:25 +00:00
if ctx.Cache.IsExist("MailResendLimit_" + u.LowerName) {
ctx.Data["ResendLimited"] = true
2014-06-23 03:11:12 +00:00
ctx.HTML(200, FORGOT_PASSWORD)
2014-04-10 01:42:25 +00:00
return
}
2014-04-05 16:32:34 +00:00
mailer.SendResetPasswdMail(ctx.Render, u)
2014-04-10 01:42:25 +00:00
if err = ctx.Cache.Put("MailResendLimit_"+u.LowerName, u.LowerName, 180); err != nil {
log.Error("Set cache(MailResendLimit) fail: %v", err)
}
2014-04-05 16:32:34 +00:00
ctx.Data["Email"] = email
2014-05-26 00:11:25 +00:00
ctx.Data["Hours"] = setting.Service.ActiveCodeLives / 60
2014-04-05 16:32:34 +00:00
ctx.Data["IsResetSent"] = true
2014-06-23 03:11:12 +00:00
ctx.HTML(200, FORGOT_PASSWORD)
2014-04-05 16:32:34 +00:00
}
func ResetPasswd(ctx *middleware.Context) {
2014-04-10 20:36:50 +00:00
ctx.Data["Title"] = "Reset Password"
2014-04-05 16:32:34 +00:00
code := ctx.Query("code")
if len(code) == 0 {
ctx.Error(404)
return
}
ctx.Data["Code"] = code
2014-04-10 20:36:50 +00:00
ctx.Data["IsResetForm"] = true
2014-06-23 03:11:12 +00:00
ctx.HTML(200, RESET_PASSWORD)
2014-04-10 20:36:50 +00:00
}
func ResetPasswdPost(ctx *middleware.Context) {
ctx.Data["Title"] = "Reset Password"
code := ctx.Query("code")
if len(code) == 0 {
ctx.Error(404)
2014-04-05 16:32:34 +00:00
return
}
2014-04-10 20:36:50 +00:00
ctx.Data["Code"] = code
2014-04-05 16:32:34 +00:00
if u := models.VerifyUserActiveCode(code); u != nil {
// Validate password length.
passwd := ctx.Query("passwd")
if len(passwd) < 6 || len(passwd) > 30 {
ctx.Data["IsResetForm"] = true
ctx.RenderWithErr("Password length should be in 6 and 30.", "user/reset_passwd", nil)
return
}
u.Passwd = passwd
u.Rands = models.GetUserSalt()
2014-04-06 20:10:57 +00:00
u.Salt = models.GetUserSalt()
u.EncodePasswd()
2014-04-05 16:32:34 +00:00
if err := models.UpdateUser(u); err != nil {
2014-04-10 20:36:50 +00:00
ctx.Handle(500, "user.ResetPasswd(UpdateUser)", err)
2014-04-05 16:32:34 +00:00
return
}
log.Trace("%s User password reset: %s", ctx.Req.RequestURI, u.Name)
ctx.Redirect("/user/login")
return
}
ctx.Data["IsResetFailed"] = true
2014-06-23 03:11:12 +00:00
ctx.HTML(200, RESET_PASSWORD)
2014-04-05 16:32:34 +00:00
}