diff options
Diffstat (limited to 'modules/middleware')
-rw-r--r-- | modules/middleware/auth.go | 39 | ||||
-rw-r--r-- | modules/middleware/context.go | 8 | ||||
-rw-r--r-- | modules/middleware/repo.go | 42 |
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] |