mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
5285e9503b
* inital changes, db migration * changes * Implement basic GetAll, Delete * Add first batch of tests * Add more tests * Add service tests for GetForProvider, List * Update http_server.go + wire.go * Lint + update fixed role * Update CODEOWNERS * Change API init * Change roles, rename * Review with @kalleep * Revert a mistakenly changed part * Updates based on @dmihai 's feedback --------- Co-authored-by: Karl Persson <kalle.persson@grafana.com>
127 lines
3.0 KiB
Go
127 lines
3.0 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/grafana/grafana/pkg/infra/db"
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/services/ssosettings"
|
|
"github.com/grafana/grafana/pkg/services/ssosettings/models"
|
|
)
|
|
|
|
type SSOSettingsStore struct {
|
|
sqlStore db.DB
|
|
log log.Logger
|
|
}
|
|
|
|
func ProvideStore(sqlStore db.DB) *SSOSettingsStore {
|
|
return &SSOSettingsStore{
|
|
sqlStore: sqlStore,
|
|
log: log.New("ssosettings.store"),
|
|
}
|
|
}
|
|
|
|
var _ ssosettings.Store = (*SSOSettingsStore)(nil)
|
|
|
|
func (s *SSOSettingsStore) Get(ctx context.Context, provider string) (*models.SSOSetting, error) {
|
|
result := models.SSOSetting{Provider: provider}
|
|
err := s.sqlStore.WithDbSession(ctx, func(sess *db.Session) error {
|
|
var err error
|
|
sess.Table("sso_setting")
|
|
found, err := sess.Where("is_deleted = ?", s.sqlStore.GetDialect().BooleanStr(false)).Get(&result)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !found {
|
|
return ssosettings.ErrNotFound
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &result, nil
|
|
}
|
|
|
|
func (s *SSOSettingsStore) List(ctx context.Context) ([]*models.SSOSetting, error) {
|
|
result := make([]*models.SSOSetting, 0)
|
|
err := s.sqlStore.WithDbSession(ctx, func(sess *db.Session) error {
|
|
sess.Table("sso_setting")
|
|
err := sess.Where("is_deleted = ?", s.sqlStore.GetDialect().BooleanStr(false)).Find(&result)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return result, nil
|
|
}
|
|
|
|
func (s *SSOSettingsStore) Upsert(ctx context.Context, provider string, data map[string]interface{}) error {
|
|
err := s.sqlStore.WithDbSession(ctx, func(sess *db.Session) error {
|
|
var err error
|
|
found, err := sess.Where("provider = ? AND is_deleted = ?", provider, s.sqlStore.GetDialect().BooleanStr(false)).Exist(&models.SSOSetting{})
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if found {
|
|
_, err = sess.Where("provider = ? AND is_deleted = ?", provider, s.sqlStore.GetDialect().BooleanStr(false)).Update(&models.SSOSetting{
|
|
Settings: data,
|
|
Updated: time.Now().UTC(),
|
|
})
|
|
} else {
|
|
_, err = sess.Insert(&models.SSOSetting{
|
|
ID: uuid.New().String(),
|
|
Provider: provider,
|
|
Settings: data,
|
|
Created: time.Now().UTC(),
|
|
Updated: time.Now().UTC(),
|
|
})
|
|
}
|
|
|
|
return err
|
|
})
|
|
|
|
return err
|
|
}
|
|
|
|
func (s *SSOSettingsStore) Patch(ctx context.Context, provider string, data map[string]interface{}) error {
|
|
panic("not implemented") // TODO: Implement
|
|
}
|
|
|
|
func (s *SSOSettingsStore) Delete(ctx context.Context, provider string) error {
|
|
err := s.sqlStore.WithDbSession(ctx, func(sess *db.Session) error {
|
|
existing := new(models.SSOSetting)
|
|
found, err := sess.Where("provider = ? AND is_deleted = ?", provider, s.sqlStore.GetDialect().BooleanStr(false)).Get(existing)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !found {
|
|
return nil // nothing to delete
|
|
}
|
|
|
|
existing.Updated = time.Now().UTC()
|
|
existing.IsDeleted = true
|
|
|
|
_, err = sess.ID(existing.ID).MustCols("updated", "is_deleted").Update(existing)
|
|
return err
|
|
})
|
|
return err
|
|
}
|