generated from VLADIMIR/template
add story service
This commit is contained in:
@@ -2,73 +2,49 @@ package story_service
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"evening_detective/internal/modules/cleaner"
|
||||
"evening_detective/internal/modules/formatter"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"regexp"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type StoryService struct {
|
||||
cleaner cleaner.ICleaner
|
||||
formatter formatter.IFormatter
|
||||
filepath string
|
||||
story *Story
|
||||
cleaner cleaner.ICleaner
|
||||
formatter formatter.IFormatter
|
||||
story *Story
|
||||
storyStorage IStoryStorage
|
||||
}
|
||||
|
||||
func NewStoryService(
|
||||
cleaner cleaner.ICleaner,
|
||||
formatter formatter.IFormatter,
|
||||
filepath string,
|
||||
storyStorage IStoryStorage,
|
||||
) (*StoryService, error) {
|
||||
s := &StoryService{
|
||||
cleaner: cleaner,
|
||||
formatter: formatter,
|
||||
filepath: filepath,
|
||||
cleaner: cleaner,
|
||||
formatter: formatter,
|
||||
storyStorage: storyStorage,
|
||||
}
|
||||
if err := s.Load(); err != nil {
|
||||
story, err := s.storyStorage.Load(context.Background())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
s.story = story
|
||||
return s, nil
|
||||
}
|
||||
|
||||
func (s *StoryService) Load() error {
|
||||
data, err := os.ReadFile(s.filepath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("story file %s not found", s.filepath)
|
||||
func (s *StoryService) Update(ctx context.Context) error {
|
||||
if err := s.storyStorage.Save(ctx, s.story); err != nil {
|
||||
return err
|
||||
}
|
||||
log.Printf("load story from: %s", s.filepath)
|
||||
story := &Story{}
|
||||
if err := json.Unmarshal(data, story); err != nil {
|
||||
story, err := s.storyStorage.Load(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
s.story = story
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *StoryService) Save() error {
|
||||
story := s.story
|
||||
data, err := json.Marshal(story)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if err := os.WriteFile(s.filepath, data, 0x777); err != nil {
|
||||
return err
|
||||
}
|
||||
log.Printf("save story to: %s", s.filepath)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *StoryService) Update() error {
|
||||
if err := s.Save(); err != nil {
|
||||
return err
|
||||
}
|
||||
return s.Load()
|
||||
}
|
||||
|
||||
func (s *StoryService) GetPlace(code string) *Place {
|
||||
if strings.HasPrefix(code, "[") || strings.HasSuffix(code, "]") {
|
||||
return &Place{
|
||||
@@ -107,7 +83,7 @@ func (s *StoryService) GetPlace(code string) *Place {
|
||||
}
|
||||
}
|
||||
|
||||
func (s *StoryService) UpdatePlace(code string, node *GraphNode) error {
|
||||
func (s *StoryService) UpdatePlace(ctx context.Context, code string, node *GraphNode) error {
|
||||
if code != "" && node.Code == "" {
|
||||
for i := range s.story.Places {
|
||||
if s.story.Places[i].Code == code {
|
||||
@@ -115,7 +91,7 @@ func (s *StoryService) UpdatePlace(code string, node *GraphNode) error {
|
||||
break
|
||||
}
|
||||
}
|
||||
s.Update()
|
||||
s.Update(ctx)
|
||||
return nil
|
||||
}
|
||||
nodeApplications := make([]*Application, 0, len(node.Applications))
|
||||
@@ -137,7 +113,7 @@ func (s *StoryService) UpdatePlace(code string, node *GraphNode) error {
|
||||
Applications: nodeApplications,
|
||||
},
|
||||
)
|
||||
s.Update()
|
||||
s.Update(ctx)
|
||||
return nil
|
||||
}
|
||||
if code == "" || node.Code == "" {
|
||||
@@ -170,7 +146,7 @@ func (s *StoryService) UpdatePlace(code string, node *GraphNode) error {
|
||||
}
|
||||
}
|
||||
}
|
||||
s.Update()
|
||||
s.Update(ctx)
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user