Auth: SSO settings foundations (#77724)

* 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>
2023-11-08 10:50:01 +01:00
parent 8a46dc39d0
commit 5285e9503b
24 changed files with 1436 additions and 3 deletions
@@ -0,0 +1,126 @@
+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
+}