add story service

This commit is contained in:
2026-03-07 04:34:58 +07:00
parent 9e0a19d25a
commit 645f6a7246
5 changed files with 80 additions and 46 deletions
+20 -44
View File
@@ -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
}