70 lines
1.3 KiB
Go
70 lines
1.3 KiB
Go
|
package middleware
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/drone/drone/shared/model"
|
||
|
|
"github.com/zenazn/goji/web"
|
||
|
|
)
|
||
|
|
|
||
|
|
// UserToC sets the User in the current
|
||
|
|
// web context.
|
||
|
|
func UserToC(c *web.C, user *model.User) {
|
||
|
|
c.Env["user"] = user
|
||
|
|
}
|
||
|
|
|
||
|
|
// RepoToC sets the User in the current
|
||
|
|
// web context.
|
||
|
|
func RepoToC(c *web.C, repo *model.Repo) {
|
||
|
|
c.Env["repo"] = repo
|
||
|
|
}
|
||
|
|
|
||
|
|
// RoleToC sets the User in the current
|
||
|
|
// web context.
|
||
|
|
func RoleToC(c *web.C, role *model.Perm) {
|
||
|
|
c.Env["role"] = role
|
||
|
|
}
|
||
|
|
|
||
|
|
// ToUser returns the User from the current
|
||
|
|
// request context. If the User does not exist
|
||
|
|
// a nil value is returned.
|
||
|
|
func ToUser(c *web.C) *model.User {
|
||
|
|
var v = c.Env["user"]
|
||
|
|
if v == nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
u, ok := v.(*model.User)
|
||
|
|
if !ok {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return u
|
||
|
|
}
|
||
|
|
|
||
|
|
// ToRepo returns the Repo from the current
|
||
|
|
// request context. If the Repo does not exist
|
||
|
|
// a nil value is returned.
|
||
|
|
func ToRepo(c *web.C) *model.Repo {
|
||
|
|
var v = c.Env["repo"]
|
||
|
|
if v == nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
r, ok := v.(*model.Repo)
|
||
|
|
if !ok {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return r
|
||
|
|
}
|
||
|
|
|
||
|
|
// ToRole returns the Role from the current
|
||
|
|
// request context. If the Role does not exist
|
||
|
|
// a nil value is returned.
|
||
|
|
func ToRole(c *web.C) *model.Perm {
|
||
|
|
var v = c.Env["role"]
|
||
|
|
if v == nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
p, ok := v.(*model.Perm)
|
||
|
|
if !ok {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return p
|
||
|
|
}
|