mirror of https://github.com/stashapp/stash.git
113 lines
2.5 KiB
Go
113 lines
2.5 KiB
Go
package group
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/stashapp/stash/pkg/models"
|
|
"github.com/stashapp/stash/pkg/sliceutil"
|
|
)
|
|
|
|
type SubGroupAlreadyInGroupError struct {
|
|
GroupIDs []int
|
|
}
|
|
|
|
func (e *SubGroupAlreadyInGroupError) Error() string {
|
|
return fmt.Sprintf("subgroups with IDs %v already in group", e.GroupIDs)
|
|
}
|
|
|
|
type ImageInput struct {
|
|
Image []byte
|
|
Set bool
|
|
}
|
|
|
|
func (s *Service) UpdatePartial(ctx context.Context, id int, updatedGroup models.GroupPartial, frontImage ImageInput, backImage ImageInput) (*models.Group, error) {
|
|
if err := s.validateUpdate(ctx, id, updatedGroup); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
r := s.Repository
|
|
|
|
group, err := r.UpdatePartial(ctx, id, updatedGroup)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// update image table
|
|
if frontImage.Set {
|
|
if err := r.UpdateFrontImage(ctx, id, frontImage.Image); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
if backImage.Set {
|
|
if err := r.UpdateBackImage(ctx, id, backImage.Image); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return group, nil
|
|
}
|
|
|
|
func (s *Service) AddSubGroups(ctx context.Context, groupID int, subGroups []models.GroupIDDescription, insertIndex *int) error {
|
|
// get the group
|
|
existing, err := s.Repository.Find(ctx, groupID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// ensure it exists
|
|
if existing == nil {
|
|
return models.ErrNotFound
|
|
}
|
|
|
|
// ensure the subgroups aren't already sub-groups of the group
|
|
subGroupIDs := sliceutil.Map(subGroups, func(sg models.GroupIDDescription) int {
|
|
return sg.GroupID
|
|
})
|
|
|
|
existingSubGroupIDs, err := s.Repository.FindSubGroupIDs(ctx, groupID, subGroupIDs)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(existingSubGroupIDs) > 0 {
|
|
return &SubGroupAlreadyInGroupError{
|
|
GroupIDs: existingSubGroupIDs,
|
|
}
|
|
}
|
|
|
|
// validate the hierarchy
|
|
d := &models.UpdateGroupDescriptions{
|
|
Groups: subGroups,
|
|
Mode: models.RelationshipUpdateModeAdd,
|
|
}
|
|
if err := s.validateUpdateGroupHierarchy(ctx, existing, nil, d); err != nil {
|
|
return err
|
|
}
|
|
|
|
// validate insert index
|
|
if insertIndex != nil && *insertIndex < 0 {
|
|
return ErrInvalidInsertIndex
|
|
}
|
|
|
|
// add the subgroups
|
|
return s.Repository.AddSubGroups(ctx, groupID, subGroups, insertIndex)
|
|
}
|
|
|
|
func (s *Service) RemoveSubGroups(ctx context.Context, groupID int, subGroupIDs []int) error {
|
|
// get the group
|
|
existing, err := s.Repository.Find(ctx, groupID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// ensure it exists
|
|
if existing == nil {
|
|
return models.ErrNotFound
|
|
}
|
|
|
|
// add the subgroups
|
|
return s.Repository.RemoveSubGroups(ctx, groupID, subGroupIDs)
|
|
}
|