71 lines
1.8 KiB
Go
71 lines
1.8 KiB
Go
package state
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"strings"
|
||
|
||
"github.com/voocel/ainovel-cli/domain"
|
||
)
|
||
|
||
// SaveRelationships 全量写入 relationship_state.json + relationship_state.md。
|
||
func (s *Store) SaveRelationships(entries []domain.RelationshipEntry) error {
|
||
if err := s.writeJSON("relationship_state.json", entries); err != nil {
|
||
return err
|
||
}
|
||
return s.writeMarkdown("relationship_state.md", renderRelationships(entries))
|
||
}
|
||
|
||
// LoadRelationships 读取人物关系状态。
|
||
func (s *Store) LoadRelationships() ([]domain.RelationshipEntry, error) {
|
||
var entries []domain.RelationshipEntry
|
||
if err := s.readJSON("relationship_state.json", &entries); err != nil {
|
||
if os.IsNotExist(err) {
|
||
return nil, nil
|
||
}
|
||
return nil, err
|
||
}
|
||
return entries, nil
|
||
}
|
||
|
||
// UpdateRelationships 合并关系变化。相同人物对的关系会被更新为最新值。
|
||
func (s *Store) UpdateRelationships(changes []domain.RelationshipEntry) error {
|
||
existing, err := s.LoadRelationships()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
// 用 pair key 索引
|
||
idx := make(map[string]int, len(existing))
|
||
for i, e := range existing {
|
||
idx[pairKey(e.CharacterA, e.CharacterB)] = i
|
||
}
|
||
for _, c := range changes {
|
||
key := pairKey(c.CharacterA, c.CharacterB)
|
||
if i, ok := idx[key]; ok {
|
||
existing[i].Relation = c.Relation
|
||
existing[i].Chapter = c.Chapter
|
||
} else {
|
||
idx[key] = len(existing)
|
||
existing = append(existing, c)
|
||
}
|
||
}
|
||
return s.SaveRelationships(existing)
|
||
}
|
||
|
||
func pairKey(a, b string) string {
|
||
if a > b {
|
||
a, b = b, a
|
||
}
|
||
return a + "|" + b
|
||
}
|
||
|
||
func renderRelationships(entries []domain.RelationshipEntry) string {
|
||
var b strings.Builder
|
||
b.WriteString("# 人物关系\n\n")
|
||
for _, e := range entries {
|
||
fmt.Fprintf(&b, "- **%s ↔ %s**：%s（第 %d 章）\n",
|
||
e.CharacterA, e.CharacterB, e.Relation, e.Chapter)
|
||
}
|
||
return b.String()
|
||
}
|