mirror of
https://github.com/grafana/grafana.git
synced 2024-12-02 05:29:42 -06:00
113 lines
2.6 KiB
Go
113 lines
2.6 KiB
Go
package sqlstore
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
)
|
|
|
|
func init() {
|
|
bus.AddHandler("sql", GetPreferences)
|
|
bus.AddHandler("sql", GetPreferencesWithDefaults)
|
|
bus.AddHandler("sql", SavePreferences)
|
|
}
|
|
|
|
func GetPreferencesWithDefaults(query *m.GetPreferencesWithDefaultsQuery) error {
|
|
params := make([]interface{}, 0)
|
|
filter := ""
|
|
if len(query.User.Teams) > 0 {
|
|
filter = "(org_id=? AND team_id IN (?" + strings.Repeat(",?", len(query.User.Teams)-1) + ")) OR "
|
|
params = append(params, query.User.OrgId)
|
|
for _, v := range query.User.Teams {
|
|
params = append(params, v)
|
|
}
|
|
}
|
|
filter += "(org_id=? AND user_id=? AND team_id=0) OR (org_id=? AND team_id=0 AND user_id=0)"
|
|
params = append(params, query.User.OrgId)
|
|
params = append(params, query.User.UserId)
|
|
params = append(params, query.User.OrgId)
|
|
prefs := make([]*m.Preferences, 0)
|
|
err := x.Where(filter, params...).
|
|
OrderBy("user_id ASC, team_id ASC").
|
|
Find(&prefs)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
res := &m.Preferences{
|
|
Theme: setting.DefaultTheme,
|
|
Timezone: "browser",
|
|
HomeDashboardId: 0,
|
|
}
|
|
|
|
for _, p := range prefs {
|
|
if p.Theme != "" {
|
|
res.Theme = p.Theme
|
|
}
|
|
if p.Timezone != "" {
|
|
res.Timezone = p.Timezone
|
|
}
|
|
if p.HomeDashboardId != 0 {
|
|
res.HomeDashboardId = p.HomeDashboardId
|
|
}
|
|
}
|
|
|
|
query.Result = res
|
|
return nil
|
|
}
|
|
|
|
func GetPreferences(query *m.GetPreferencesQuery) error {
|
|
var prefs m.Preferences
|
|
exists, err := x.Where("org_id=? AND user_id=? AND team_id=?", query.OrgId, query.UserId, query.TeamId).Get(&prefs)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if exists {
|
|
query.Result = &prefs
|
|
} else {
|
|
query.Result = new(m.Preferences)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func SavePreferences(cmd *m.SavePreferencesCommand) error {
|
|
return inTransaction(func(sess *DBSession) error {
|
|
|
|
var prefs m.Preferences
|
|
exists, err := sess.Where("org_id=? AND user_id=? AND team_id=?", cmd.OrgId, cmd.UserId, cmd.TeamId).Get(&prefs)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !exists {
|
|
prefs = m.Preferences{
|
|
UserId: cmd.UserId,
|
|
OrgId: cmd.OrgId,
|
|
TeamId: cmd.TeamId,
|
|
HomeDashboardId: cmd.HomeDashboardId,
|
|
Timezone: cmd.Timezone,
|
|
Theme: cmd.Theme,
|
|
Created: time.Now(),
|
|
Updated: time.Now(),
|
|
}
|
|
_, err = sess.Insert(&prefs)
|
|
return err
|
|
}
|
|
prefs.HomeDashboardId = cmd.HomeDashboardId
|
|
prefs.Timezone = cmd.Timezone
|
|
prefs.Theme = cmd.Theme
|
|
prefs.Updated = time.Now()
|
|
prefs.Version += 1
|
|
_, err = sess.ID(prefs.Id).AllCols().Update(&prefs)
|
|
return err
|
|
})
|
|
}
|