Add global settings (#40)

database/settings.go

@@ -2,84 +2,34 @@ package database
 
 import (
 	"context"
-	"errors"
 
-	"github.com/arangodb/go-driver"
-
-	"github.com/SecurityBrewery/catalyst/database/busdb"
 	"github.com/SecurityBrewery/catalyst/generated/model"
 )
 
-func toUserDataResponse(key string, doc *model.UserData) *model.UserDataResponse {
-	return &model.UserDataResponse{
-		Email:      doc.Email,
-		ID:         key,
-		Image:      doc.Image,
-		Name:       doc.Name,
-		Timeformat: doc.Timeformat,
+func (db *Database) Settings(ctx context.Context) (*model.Settings, error) {
+	settings := &model.Settings{}
+	if _, err := db.settingsCollection.ReadDocument(ctx, "global", settings); err != nil {
+		return nil, err
 	}
+
+	return settings, nil
 }
 
-func (db *Database) UserDataCreate(ctx context.Context, id string, userdata *model.UserData) error {
-	if userdata == nil {
-		return errors.New("requires setting")
-	}
-	if id == "" {
-		return errors.New("requires username")
-	}
-
-	_, err := db.userdataCollection.CreateDocument(ctx, ctx, id, userdata)
-	return err
-}
-
-func (db *Database) UserDataGetOrCreate(ctx context.Context, id string, newUserData *model.UserData) (*model.UserDataResponse, error) {
-	setting, err := db.UserDataGet(ctx, id)
-	if err != nil {
-		return toUserDataResponse(id, newUserData), db.UserDataCreate(ctx, id, newUserData)
-	}
-	return setting, nil
-}
-
-func (db *Database) UserDataGet(ctx context.Context, id string) (*model.UserDataResponse, error) {
-	var doc model.UserData
-	meta, err := db.userdataCollection.ReadDocument(ctx, id, &doc)
+func (db *Database) SaveSettings(ctx context.Context, settings *model.Settings) (*model.Settings, error) {
+	exists, err := db.settingsCollection.DocumentExists(ctx, "global")
 	if err != nil {
 		return nil, err
 	}
 
-	return toUserDataResponse(meta.Key, &doc), err
-}
-
-func (db *Database) UserDataList(ctx context.Context) ([]*model.UserDataResponse, error) {
-	query := "FOR d IN @@collection SORT d.username ASC RETURN d"
-	cursor, _, err := db.Query(ctx, query, map[string]interface{}{"@collection": UserDataCollectionName}, busdb.ReadOperation)
-	if err != nil {
-		return nil, err
-	}
-	defer cursor.Close()
-	var docs []*model.UserDataResponse
-	for {
-		var doc model.UserData
-		meta, err := cursor.ReadDocument(ctx, &doc)
-		if driver.IsNoMoreDocuments(err) {
-			break
-		} else if err != nil {
+	if exists {
+		if _, err := db.settingsCollection.ReplaceDocument(ctx, "global", settings); err != nil {
+			return nil, err
+		}
+	} else {
+		if _, err := db.settingsCollection.CreateDocument(ctx, ctx, "global", settings); err != nil {
 			return nil, err
 		}
-		docs = append(docs, toUserDataResponse(meta.Key, &doc))
 	}
 
-	return docs, err
-}
-
-func (db *Database) UserDataUpdate(ctx context.Context, id string, userdata *model.UserData) (*model.UserDataResponse, error) {
-	var doc model.UserData
-	ctx = driver.WithReturnNew(ctx, &doc)
-
-	meta, err := db.userdataCollection.ReplaceDocument(ctx, id, userdata)
-	if err != nil {
-		return nil, err
-	}
-
-	return toUserDataResponse(meta.Key, &doc), nil
+	return settings, nil
 }