mirror of
https://github.com/SecurityBrewery/catalyst.git
synced 2025-12-06 15:22:47 +01:00
60 lines
1.7 KiB
Go
60 lines
1.7 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/arangodb/go-driver"
|
|
|
|
"github.com/SecurityBrewery/catalyst/database"
|
|
"github.com/SecurityBrewery/catalyst/database/busdb"
|
|
"github.com/SecurityBrewery/catalyst/generated/model"
|
|
)
|
|
|
|
func userDataResponseID(userData *model.UserDataResponse) []driver.DocumentID {
|
|
if userData == nil {
|
|
return nil
|
|
}
|
|
|
|
return userDataID(userData.ID)
|
|
}
|
|
|
|
func userDataID(id string) []driver.DocumentID {
|
|
return []driver.DocumentID{driver.DocumentID(fmt.Sprintf("%s/%s", database.UserDataCollectionName, id))}
|
|
}
|
|
|
|
func (s *Service) ListUserData(ctx context.Context) (doc []*model.UserDataResponse, err error) {
|
|
return s.database.UserDataList(ctx)
|
|
}
|
|
|
|
func (s *Service) GetUserData(ctx context.Context, id string) (*model.UserDataResponse, error) {
|
|
return s.database.UserDataGet(ctx, id)
|
|
}
|
|
|
|
func (s *Service) UpdateUserData(ctx context.Context, id string, data *model.UserData) (doc *model.UserDataResponse, err error) {
|
|
defer s.publishRequest(ctx, err, "CreateUser", userDataResponseID(doc))
|
|
|
|
return s.database.UserDataUpdate(ctx, id, data)
|
|
}
|
|
|
|
func (s *Service) CurrentUserData(ctx context.Context) (doc *model.UserDataResponse, err error) {
|
|
user, ok := busdb.UserFromContext(ctx)
|
|
if !ok {
|
|
return nil, errors.New("no user in context")
|
|
}
|
|
|
|
return s.database.UserDataGet(ctx, user.ID)
|
|
}
|
|
|
|
func (s *Service) UpdateCurrentUserData(ctx context.Context, data *model.UserData) (doc *model.UserDataResponse, err error) {
|
|
user, ok := busdb.UserFromContext(ctx)
|
|
if !ok {
|
|
return nil, errors.New("no user in context")
|
|
}
|
|
|
|
defer s.publishRequest(ctx, err, "UpdateCurrentUserData", userDataResponseID(doc))
|
|
|
|
return s.database.UserDataUpdate(ctx, user.ID, data)
|
|
}
|