77 lines
1.6 KiB
Go
77 lines
1.6 KiB
Go
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
|
||
}
|