add places operations

internal/services/scenarios_service/service.go

@@ -2,6 +2,8 @@ package scenarios_service
 
 import (
 	"context"
+	"errors"
+	"evening_detective_server/internal/modules/storytelling"
 	"evening_detective_server/internal/repos/scenarios_repo"
 )
 
@@ -67,3 +69,81 @@ func (s *ScenarioService) DeleteScenarioByID(
 ) error {
 	return s.scenariosRepo.DeleteScenarioByID(ctx, id)
 }
+
+func (s *ScenarioService) AddScenarioPlace(
+	ctx context.Context,
+	id int,
+	place *storytelling.Place,
+) error {
+	story, err := s.getStory(ctx, id)
+	if err != nil {
+		return err
+	}
+	story.Places = append(
+		story.Places,
+		place,
+	)
+	return s.updateStory(ctx, id, story)
+}
+
+func (s *ScenarioService) UpdateScenarioPlace(
+	ctx context.Context,
+	id int,
+	code string,
+	place *storytelling.Place,
+) error {
+	story, err := s.getStory(ctx, id)
+	if err != nil {
+		return err
+	}
+	for i := range story.Places {
+		if story.Places[i].Code == code {
+			story.Places[i] = place
+			break
+		}
+	}
+	return s.updateStory(ctx, id, story)
+}
+
+func (s *ScenarioService) DeleteScenarioPlace(
+	ctx context.Context,
+	id int,
+	code string,
+) error {
+	story, err := s.getStory(ctx, id)
+	if err != nil {
+		return err
+	}
+	for i := range story.Places {
+		if story.Places[i].Code == code {
+			story.Places = append(story.Places[:i], story.Places[i+1:]...)
+			break
+		}
+	}
+	return s.updateStory(ctx, id, story)
+}
+
+func (s *ScenarioService) getStory(ctx context.Context, id int) (*storytelling.Story, error) {
+	storyString, err := s.scenariosRepo.GetStoryByScenarioID(ctx, id)
+	if err != nil {
+		return nil, err
+	}
+	return mapStory(storyString)
+}
+
+func (s *ScenarioService) updateStory(ctx context.Context, id int, story *storytelling.Story) error {
+	mapCodes := map[string]struct{}{}
+	for _, place := range story.Places {
+		mapCodes[place.Code] = struct{}{}
+	}
+	if len(mapCodes) != len(story.Places) {
+		return errors.New("Такой код точки уже существует")
+	}
+
+	storyString, err := convertStory(story)
+	if err != nil {
+		return err
+	}
+
+	return s.scenariosRepo.UpdateStoryByScenarioID(ctx, id, storyString)
+}