mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
package sqlstore
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/go-xorm/xorm"
|
|
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
)
|
|
|
|
func init() {
|
|
bus.AddHandler("sql", AddOrgUser)
|
|
bus.AddHandler("sql", RemoveOrgUser)
|
|
bus.AddHandler("sql", GetOrgUsers)
|
|
}
|
|
|
|
func AddOrgUser(cmd *m.AddOrgUserCommand) error {
|
|
return inTransaction(func(sess *xorm.Session) error {
|
|
|
|
entity := m.OrgUser{
|
|
OrgId: cmd.OrgId,
|
|
UserId: cmd.UserId,
|
|
Role: cmd.Role,
|
|
Created: time.Now(),
|
|
Updated: time.Now(),
|
|
}
|
|
|
|
_, err := sess.Insert(&entity)
|
|
return err
|
|
})
|
|
}
|
|
|
|
func GetOrgUsers(query *m.GetOrgUsersQuery) error {
|
|
query.Result = make([]*m.OrgUserDTO, 0)
|
|
sess := x.Table("org_user")
|
|
sess.Join("INNER", "user", fmt.Sprintf("org_user.user_id=%s.id", x.Dialect().Quote("user")))
|
|
sess.Where("org_user.org_id=?", query.OrgId)
|
|
sess.Cols("org_user.org_id", "org_user.user_id", "user.email", "user.login", "org_user.role")
|
|
sess.Asc("user.email", "user.login")
|
|
|
|
err := sess.Find(&query.Result)
|
|
return err
|
|
}
|
|
|
|
func RemoveOrgUser(cmd *m.RemoveOrgUserCommand) error {
|
|
return inTransaction(func(sess *xorm.Session) error {
|
|
var rawSql = "DELETE FROM org_user WHERE org_id=? and user_id=?"
|
|
_, err := sess.Exec(rawSql, cmd.OrgId, cmd.UserId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// validate that there is an admin user left
|
|
res, err := sess.Query("SELECT 1 from org_user WHERE org_id=? and role='Admin'", cmd.OrgId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(res) == 0 {
|
|
return m.ErrLastOrgAdmin
|
|
}
|
|
|
|
return err
|
|
})
|
|
}
|