mirror of
https://github.com/grafana/grafana.git
synced 2025-08-01 10:17:05 +08:00
7b17e38f5d
Added CRUD methods for Tokens. Extend Auth Handler to check for the presence of a Bearer Authorization header to authenticate against. If there is no header, or the token is not valid, the Auth Handler falls back to looking for a Session.
91 lines
1.8 KiB
Go
91 lines
1.8 KiB
Go
package api
|
|
|
|
import (
|
|
"github.com/torkelo/grafana-pro/pkg/bus"
|
|
"github.com/torkelo/grafana-pro/pkg/middleware"
|
|
m "github.com/torkelo/grafana-pro/pkg/models"
|
|
"github.com/torkelo/grafana-pro/pkg/util"
|
|
)
|
|
|
|
func GetTokens(c *middleware.Context) {
|
|
query := m.GetTokensQuery{AccountId: c.Account.Id}
|
|
err := bus.Dispatch(&query)
|
|
|
|
if err != nil {
|
|
c.JsonApiErr(500, "Failed to list tokens", err)
|
|
return
|
|
}
|
|
result := make([]*m.TokenDTO, len(query.Result))
|
|
for i, t := range query.Result {
|
|
result[i] = &m.TokenDTO{
|
|
Id: t.Id,
|
|
Name: t.Name,
|
|
Role: t.Role,
|
|
Token: t.Token,
|
|
}
|
|
}
|
|
c.JSON(200, result)
|
|
}
|
|
|
|
func DeleteToken(c *middleware.Context) {
|
|
id := c.ParamsInt64(":id")
|
|
|
|
cmd := &m.DeleteTokenCommand{Id: id, AccountId: c.UserAccount.Id}
|
|
|
|
err := bus.Dispatch(cmd)
|
|
if err != nil {
|
|
c.JsonApiErr(500, "Failed to delete token", err)
|
|
return
|
|
}
|
|
|
|
c.JsonOK("Token deleted")
|
|
}
|
|
|
|
func AddToken(c *middleware.Context) {
|
|
cmd := m.AddTokenCommand{}
|
|
|
|
if !c.JsonBody(&cmd) {
|
|
c.JsonApiErr(400, "Validation failed", nil)
|
|
return
|
|
}
|
|
|
|
if cmd.Role != m.ROLE_READ_WRITE && cmd.Role != m.ROLE_READ {
|
|
c.JsonApiErr(400, "Invalid role specified", nil)
|
|
return
|
|
}
|
|
|
|
cmd.AccountId = c.Account.Id
|
|
cmd.Token = util.GetRandomString(64)
|
|
|
|
if err := bus.Dispatch(&cmd); err != nil {
|
|
c.JsonApiErr(500, "Failed to add token", err)
|
|
return
|
|
}
|
|
result := &m.TokenDTO{
|
|
Id: cmd.Result.Id,
|
|
Name: cmd.Result.Name,
|
|
Role: cmd.Result.Role,
|
|
Token: cmd.Result.Token,
|
|
}
|
|
c.JSON(200, result)
|
|
}
|
|
|
|
func UpdateToken(c *middleware.Context) {
|
|
cmd := m.UpdateTokenCommand{}
|
|
|
|
if !c.JsonBody(&cmd) {
|
|
c.JsonApiErr(400, "Validation failed", nil)
|
|
return
|
|
}
|
|
|
|
cmd.AccountId = c.Account.Id
|
|
|
|
err := bus.Dispatch(&cmd)
|
|
if err != nil {
|
|
c.JsonApiErr(500, "Failed to update token", err)
|
|
return
|
|
}
|
|
|
|
c.JsonOK("Token updated")
|
|
}
|