mirror of
https://github.com/grafana/grafana.git
synced 2025-02-20 11:48:34 -06: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.
67 lines
1.4 KiB
Go
67 lines
1.4 KiB
Go
package sqlstore
|
|
|
|
import (
|
|
"github.com/go-xorm/xorm"
|
|
"github.com/torkelo/grafana-pro/pkg/bus"
|
|
m "github.com/torkelo/grafana-pro/pkg/models"
|
|
"time"
|
|
)
|
|
|
|
func init() {
|
|
bus.AddHandler("sql", GetTokens)
|
|
bus.AddHandler("sql", AddToken)
|
|
bus.AddHandler("sql", UpdateToken)
|
|
bus.AddHandler("sql", DeleteToken)
|
|
}
|
|
|
|
func GetTokens(query *m.GetTokensQuery) error {
|
|
sess := x.Limit(100, 0).Where("account_id=?", query.AccountId).Asc("name")
|
|
|
|
query.Result = make([]*m.Token, 0)
|
|
return sess.Find(&query.Result)
|
|
}
|
|
|
|
func DeleteToken(cmd *m.DeleteTokenCommand) error {
|
|
return inTransaction(func(sess *xorm.Session) error {
|
|
var rawSql = "DELETE FROM token WHERE id=? and account_id=?"
|
|
_, err := sess.Exec(rawSql, cmd.Id, cmd.AccountId)
|
|
return err
|
|
})
|
|
}
|
|
|
|
func AddToken(cmd *m.AddTokenCommand) error {
|
|
|
|
return inTransaction(func(sess *xorm.Session) error {
|
|
t := m.Token{
|
|
AccountId: cmd.AccountId,
|
|
Name: cmd.Name,
|
|
Role: cmd.Role,
|
|
Token: cmd.Token,
|
|
Created: time.Now(),
|
|
Updated: time.Now(),
|
|
}
|
|
|
|
if _, err := sess.Insert(&t); err != nil {
|
|
return err
|
|
}
|
|
cmd.Result = &t
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func UpdateToken(cmd *m.UpdateTokenCommand) error {
|
|
|
|
return inTransaction(func(sess *xorm.Session) error {
|
|
t := m.Token{
|
|
Id: cmd.Id,
|
|
AccountId: cmd.AccountId,
|
|
Name: cmd.Name,
|
|
Role: cmd.Role,
|
|
Updated: time.Now(),
|
|
}
|
|
|
|
_, err := sess.Where("id=? and account_id=?", t.Id, t.AccountId).Update(&t)
|
|
return err
|
|
})
|
|
}
|