mirror of
https://github.com/grafana/grafana.git
synced 2026-08-09 04:38:16 -05:00
126 lines
2.7 KiB
Go
126 lines
2.7 KiB
Go
package middleware
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/Unknwon/macaron"
|
|
"github.com/macaron-contrib/session"
|
|
|
|
"github.com/torkelo/grafana-pro/pkg/bus"
|
|
"github.com/torkelo/grafana-pro/pkg/log"
|
|
m "github.com/torkelo/grafana-pro/pkg/models"
|
|
"github.com/torkelo/grafana-pro/pkg/setting"
|
|
)
|
|
|
|
type Context struct {
|
|
*macaron.Context
|
|
Session session.Store
|
|
|
|
IsSignedIn bool
|
|
IsAdmin bool
|
|
|
|
Account *m.Account
|
|
UserAccount *m.Account
|
|
}
|
|
|
|
func (c *Context) GetAccountId() int64 {
|
|
return c.Account.Id
|
|
}
|
|
|
|
func GetContextHandler() macaron.Handler {
|
|
return func(c *macaron.Context, sess session.Store) {
|
|
ctx := &Context{
|
|
Context: c,
|
|
Session: sess,
|
|
}
|
|
|
|
// try get account id from request
|
|
if accountId, err := getRequestAccountId(ctx); err == nil {
|
|
// fetch user
|
|
userQuery := m.GetAccountByIdQuery{Id: accountId}
|
|
if err := bus.Dispatch(&userQuery); err != nil {
|
|
log.Error(3, "Failed to get user by id, %v, %v", accountId, err)
|
|
} else {
|
|
// fetch using account
|
|
ctx.UserAccount = userQuery.Result
|
|
usingQuery := m.GetAccountByIdQuery{Id: ctx.UserAccount.UsingAccountId}
|
|
if err := bus.Dispatch(&usingQuery); err != nil {
|
|
log.Error(3, "Faild to get account's using account, account: %v, usingAccountId: %v, err:%v", accountId, ctx.UserAccount.Id, err)
|
|
} else {
|
|
ctx.Account = usingQuery.Result
|
|
}
|
|
}
|
|
}
|
|
|
|
if ctx.Account != nil {
|
|
ctx.IsSignedIn = true
|
|
ctx.IsAdmin = ctx.Account.IsAdmin
|
|
}
|
|
|
|
c.Map(ctx)
|
|
}
|
|
}
|
|
|
|
// Handle handles and logs error by given status.
|
|
func (ctx *Context) Handle(status int, title string, err error) {
|
|
if err != nil {
|
|
log.Error(4, "%s: %v", title, err)
|
|
if macaron.Env != macaron.PROD {
|
|
ctx.Data["ErrorMsg"] = err
|
|
}
|
|
}
|
|
|
|
switch status {
|
|
case 404:
|
|
ctx.Data["Title"] = "Page Not Found"
|
|
case 500:
|
|
ctx.Data["Title"] = "Internal Server Error"
|
|
}
|
|
|
|
ctx.HTML(status, strconv.Itoa(status))
|
|
}
|
|
|
|
func (ctx *Context) JsonOK(message string) {
|
|
resp := make(map[string]interface{})
|
|
|
|
resp["message"] = message
|
|
|
|
ctx.JSON(200, resp)
|
|
}
|
|
|
|
func (ctx *Context) IsApiRequest() bool {
|
|
return strings.HasPrefix(ctx.Req.URL.Path, "/api")
|
|
}
|
|
|
|
func (ctx *Context) JsonApiErr(status int, message string, err error) {
|
|
resp := make(map[string]interface{})
|
|
|
|
if err != nil {
|
|
log.Error(4, "%s: %v", message, err)
|
|
if setting.Env != setting.PROD {
|
|
resp["error"] = err.Error()
|
|
}
|
|
}
|
|
|
|
switch status {
|
|
case 404:
|
|
resp["message"] = "Not Found"
|
|
case 500:
|
|
resp["message"] = "Internal Server Error"
|
|
}
|
|
|
|
if message != "" {
|
|
resp["message"] = message
|
|
}
|
|
|
|
ctx.JSON(status, resp)
|
|
}
|
|
|
|
func (ctx *Context) JsonBody(model interface{}) bool {
|
|
b, _ := ctx.Req.Body().Bytes()
|
|
err := json.Unmarshal(b, &model)
|
|
return err == nil
|
|
}
|