aboutsummaryrefslogtreecommitdiff
path: root/modules/middleware
diff options
context:
space:
mode:
Diffstat (limited to 'modules/middleware')
-rw-r--r--modules/middleware/auth.go39
-rw-r--r--modules/middleware/context.go8
-rw-r--r--modules/middleware/repo.go42
3 files changed, 80 insertions, 9 deletions
diff --git a/modules/middleware/auth.go b/modules/middleware/auth.go
index 2a02d276..db643ccf 100644
--- a/modules/middleware/auth.go
+++ b/modules/middleware/auth.go
@@ -5,12 +5,16 @@
package middleware
import (
+ "fmt"
"net/url"
"github.com/Unknwon/macaron"
"github.com/macaron-contrib/csrf"
+ "github.com/gogits/gogs/models"
"github.com/gogits/gogs/modules/auth"
+ "github.com/gogits/gogs/modules/base"
+ "github.com/gogits/gogs/modules/log"
"github.com/gogits/gogs/modules/setting"
)
@@ -21,6 +25,41 @@ type ToggleOptions struct {
DisableCsrf bool
}
+// AutoSignIn reads cookie and try to auto-login.
+func AutoSignIn(ctx *Context) (bool, error) {
+ uname := ctx.GetCookie(setting.CookieUserName)
+ if len(uname) == 0 {
+ return false, nil
+ }
+
+ isSucceed := false
+ defer func() {
+ if !isSucceed {
+ log.Trace("auto-login cookie cleared: %s", uname)
+ ctx.SetCookie(setting.CookieUserName, "", -1, setting.AppSubUrl)
+ ctx.SetCookie(setting.CookieRememberName, "", -1, setting.AppSubUrl)
+ }
+ }()
+
+ u, err := models.GetUserByName(uname)
+ if err != nil {
+ if !models.IsErrUserNotExist(err) {
+ return false, fmt.Errorf("GetUserByName: %v", err)
+ }
+ return false, nil
+ }
+
+ if val, _ := ctx.GetSuperSecureCookie(
+ base.EncodeMd5(u.Rands+u.Passwd), setting.CookieRememberName); val != u.Name {
+ return false, nil
+ }
+
+ isSucceed = true
+ ctx.Session.Set("uid", u.Id)
+ ctx.Session.Set("uname", u.Name)
+ return true, nil
+}
+
func Toggle(options *ToggleOptions) macaron.Handler {
return func(ctx *Context) {
// Cannot view any page before installation.
diff --git a/modules/middleware/context.go b/modules/middleware/context.go
index 2995d578..d4bc8f03 100644
--- a/modules/middleware/context.go
+++ b/modules/middleware/context.go
@@ -197,6 +197,14 @@ func Contexter() macaron.Handler {
ctx.Data["PageStartTime"] = time.Now()
+ // Check auto-signin.
+ if sess.Get("uid") == nil {
+ if _, err := AutoSignIn(ctx); err != nil {
+ ctx.Handle(500, "AutoSignIn", err)
+ return
+ }
+ }
+
// Get user from session if logined.
ctx.User, ctx.IsBasicAuth = auth.SignedInUser(ctx.Req.Request, ctx.Session)
diff --git a/modules/middleware/repo.go b/modules/middleware/repo.go
index 12164632..c4c53c03 100644
--- a/modules/middleware/repo.go
+++ b/modules/middleware/repo.go
@@ -257,7 +257,7 @@ func RepoAssignment(redirect bool, args ...bool) macaron.Handler {
ctx.Data["HasAccess"] = true
if repo.IsMirror {
- ctx.Repo.Mirror, err = models.GetMirror(repo.Id)
+ ctx.Repo.Mirror, err = models.GetMirror(repo.ID)
if err != nil {
ctx.Handle(500, "GetMirror", err)
return
@@ -291,10 +291,34 @@ func RepoAssignment(redirect bool, args ...bool) macaron.Handler {
ctx.Data["Tags"] = tags
ctx.Repo.Repository.NumTags = len(tags)
- // Non-fork repository will not return error in this method.
- if err = repo.GetForkRepo(); err != nil {
- ctx.Handle(500, "GetForkRepo", err)
- return
+ if repo.IsFork {
+ // Non-fork repository will not return error in this method.
+ if err = repo.GetBaseRepo(); err != nil {
+ ctx.Handle(500, "GetBaseRepo", err)
+ return
+ } else if repo.BaseRepo.GetOwner(); err != nil {
+ ctx.Handle(500, "BaseRepo.GetOwner", err)
+ return
+ }
+
+ bsaeRepo := repo.BaseRepo
+ baseGitRepo, err := git.OpenRepository(models.RepoPath(bsaeRepo.Owner.Name, bsaeRepo.Name))
+ if err != nil {
+ ctx.Handle(500, "OpenRepository", err)
+ return
+ }
+ if len(bsaeRepo.DefaultBranch) > 0 && baseGitRepo.IsBranchExist(bsaeRepo.DefaultBranch) {
+ ctx.Data["BaseDefaultBranch"] = bsaeRepo.DefaultBranch
+ } else {
+ baseBranches, err := baseGitRepo.GetBranches()
+ if err != nil {
+ ctx.Handle(500, "GetBranches", err)
+ return
+ }
+ if len(baseBranches) > 0 {
+ ctx.Data["BaseDefaultBranch"] = baseBranches[0]
+ }
+ }
}
ctx.Data["Title"] = u.Name + "/" + repo.Name
@@ -327,8 +351,8 @@ func RepoAssignment(redirect bool, args ...bool) macaron.Handler {
}
if ctx.IsSigned {
- ctx.Data["IsWatchingRepo"] = models.IsWatching(ctx.User.Id, repo.Id)
- ctx.Data["IsStaringRepo"] = models.IsStaring(ctx.User.Id, repo.Id)
+ ctx.Data["IsWatchingRepo"] = models.IsWatching(ctx.User.Id, repo.ID)
+ ctx.Data["IsStaringRepo"] = models.IsStaring(ctx.User.Id, repo.ID)
}
ctx.Data["TagName"] = ctx.Repo.TagName
@@ -342,8 +366,8 @@ func RepoAssignment(redirect bool, args ...bool) macaron.Handler {
// If not branch selected, try default one.
// If default branch doesn't exists, fall back to some other branch.
- if ctx.Repo.BranchName == "" {
- if ctx.Repo.Repository.DefaultBranch != "" && gitRepo.IsBranchExist(ctx.Repo.Repository.DefaultBranch) {
+ if len(ctx.Repo.BranchName) == 0 {
+ if len(ctx.Repo.Repository.DefaultBranch) > 0 && gitRepo.IsBranchExist(ctx.Repo.Repository.DefaultBranch) {
ctx.Repo.BranchName = ctx.Repo.Repository.DefaultBranch
} else if len(brs) > 0 {
ctx.Repo.BranchName = brs[0]