init

state/store.go

@@ -0,0 +1,76 @@
+package state
+
+import (
+	"encoding/json"
+	"fmt"
+	"os"
+	"path/filepath"
+)
+
+// Store 封装小说输出目录，提供所有状态读写操作。
+type Store struct {
+	dir string
+}
+
+// NewStore 创建状态管理器，dir 为小说输出根目录。
+func NewStore(dir string) *Store {
+	return &Store{dir: dir}
+}
+
+// Dir 返回输出根目录。
+func (s *Store) Dir() string { return s.dir }
+
+// Init 创建所需的子目录结构。
+func (s *Store) Init() error {
+	dirs := []string{"chapters", "summaries", "drafts", "reviews", "meta"}
+	for _, d := range dirs {
+		if err := os.MkdirAll(filepath.Join(s.dir, d), 0o755); err != nil {
+			return fmt.Errorf("create dir %s: %w", d, err)
+		}
+	}
+	return nil
+}
+
+func (s *Store) path(rel string) string {
+	return filepath.Join(s.dir, rel)
+}
+
+func (s *Store) readFile(rel string) ([]byte, error) {
+	return os.ReadFile(s.path(rel))
+}
+
+func (s *Store) writeFile(rel string, data []byte) error {
+	p := s.path(rel)
+	if err := os.MkdirAll(filepath.Dir(p), 0o755); err != nil {
+		return err
+	}
+	return os.WriteFile(p, data, 0o644)
+}
+
+func (s *Store) readJSON(rel string, v any) error {
+	data, err := s.readFile(rel)
+	if err != nil {
+		return err
+	}
+	return json.Unmarshal(data, v)
+}
+
+func (s *Store) writeJSON(rel string, v any) error {
+	data, err := json.MarshalIndent(v, "", "  ")
+	if err != nil {
+		return err
+	}
+	return s.writeFile(rel, data)
+}
+
+func (s *Store) writeMarkdown(rel string, content string) error {
+	return s.writeFile(rel, []byte(content))
+}
+
+func (s *Store) removeFile(rel string) error {
+	err := os.Remove(s.path(rel))
+	if os.IsNotExist(err) {
+		return nil
+	}
+	return err
+}