forked from Shiloh/githaven
131 lines
2.9 KiB
Go
131 lines
2.9 KiB
Go
// 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 auth
|
|
|
|
import (
|
|
"net/http"
|
|
"reflect"
|
|
|
|
"github.com/codegangsta/martini"
|
|
"github.com/martini-contrib/render"
|
|
"github.com/martini-contrib/sessions"
|
|
|
|
"github.com/gogits/binding"
|
|
|
|
"github.com/gogits/gogs/models"
|
|
"github.com/gogits/gogs/modules/base"
|
|
"github.com/gogits/gogs/modules/log"
|
|
)
|
|
|
|
func SignedInId(session sessions.Session) int64 {
|
|
userId := session.Get("userId")
|
|
if userId == nil {
|
|
return 0
|
|
}
|
|
if s, ok := userId.(int64); ok {
|
|
if _, err := models.GetUserById(s); err != nil {
|
|
return 0
|
|
}
|
|
return s
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func SignedInName(session sessions.Session) string {
|
|
userName := session.Get("userName")
|
|
if userName == nil {
|
|
return ""
|
|
}
|
|
if s, ok := userName.(string); ok {
|
|
return s
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func SignedInUser(session sessions.Session) *models.User {
|
|
id := SignedInId(session)
|
|
if id <= 0 {
|
|
return nil
|
|
}
|
|
|
|
user, err := models.GetUserById(id)
|
|
if err != nil {
|
|
log.Error("user.SignedInUser: %v", err)
|
|
return nil
|
|
}
|
|
return user
|
|
}
|
|
|
|
func IsSignedIn(session sessions.Session) bool {
|
|
return SignedInId(session) > 0
|
|
}
|
|
|
|
// SignInRequire checks user status from session.
|
|
// It will assign correspoding values to
|
|
// template data map if user has signed in.
|
|
func SignInRequire(redirect bool) martini.Handler {
|
|
return func(r render.Render, data base.TmplData, session sessions.Session) {
|
|
if !IsSignedIn(session) {
|
|
if redirect {
|
|
r.Redirect("/")
|
|
}
|
|
return
|
|
}
|
|
|
|
data["IsSigned"] = true
|
|
data["SignedUserId"] = SignedInId(session)
|
|
data["SignedUserName"] = SignedInName(session)
|
|
data["SignedAvatar"] = SignedInUser(session).Avatar
|
|
}
|
|
}
|
|
|
|
func SignOutRequire() martini.Handler {
|
|
return func(r render.Render, session sessions.Session) {
|
|
if IsSignedIn(session) {
|
|
r.Redirect("/")
|
|
}
|
|
}
|
|
}
|
|
|
|
type FeedsForm struct {
|
|
UserId int64 `form:"userid" binding:"Required"`
|
|
Offset int64 `form:"offset"`
|
|
}
|
|
|
|
type UpdateProfileForm struct {
|
|
Email string `form:"email" binding:"Required;Email;MaxSize(50)"`
|
|
Website string `form:"website" binding:"AlphaDash;MaxSize(50)"`
|
|
Location string `form:"location" binding:"MaxSize(50)"`
|
|
Avatar string `form:"avatar" binding:"Required;Email;MaxSize(50)"`
|
|
}
|
|
|
|
func (f *UpdateProfileForm) Name(field string) string {
|
|
names := map[string]string{
|
|
"Email": "Email address",
|
|
"Website": "Website",
|
|
"Location": "Location",
|
|
"Avatar": "Gravatar Email",
|
|
}
|
|
return names[field]
|
|
}
|
|
|
|
func (f *UpdateProfileForm) Validate(errors *binding.Errors, req *http.Request, context martini.Context) {
|
|
if req.Method == "GET" || errors.Count() == 0 {
|
|
return
|
|
}
|
|
|
|
data := context.Get(reflect.TypeOf(base.TmplData{})).Interface().(base.TmplData)
|
|
data["HasError"] = true
|
|
|
|
if len(errors.Overall) > 0 {
|
|
for _, err := range errors.Overall {
|
|
log.Error("UpdateProfileForm.Validate: %v", err)
|
|
}
|
|
return
|
|
}
|
|
|
|
validate(errors, data, f)
|
|
}
|