add user row to request context
This commit is contained in:
@@ -60,6 +60,25 @@ type Controller struct {
|
||||
SStore *gormstore.Store
|
||||
}
|
||||
|
||||
func (c *Controller) GetSetting(key string) string {
|
||||
var setting db.Setting
|
||||
c.DB.Where("key = ?", key).First(&setting)
|
||||
return setting.Value
|
||||
}
|
||||
|
||||
func (c *Controller) SetSetting(key, value string) {
|
||||
c.DB.
|
||||
Where(db.Setting{Key: key}).
|
||||
Assign(db.Setting{Value: value}).
|
||||
FirstOrCreate(&db.Setting{})
|
||||
}
|
||||
|
||||
func (c *Controller) GetUserFromName(name string) *db.User {
|
||||
var user db.User
|
||||
c.DB.Where("name = ?", name).First(&user)
|
||||
return &user
|
||||
}
|
||||
|
||||
type templateData struct {
|
||||
Flashes []interface{}
|
||||
User *db.User
|
||||
@@ -151,7 +170,7 @@ func renderTemplate(w http.ResponseWriter, r *http.Request,
|
||||
}
|
||||
data.Flashes = session.Flashes()
|
||||
session.Save(r, w)
|
||||
user, ok := session.Values["user"].(*db.User)
|
||||
user, ok := r.Context().Value("user").(*db.User)
|
||||
if ok {
|
||||
data.User = user
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user