ehentai-go/internal/store/store.go

// Package store 封装运行时生成目录(pre-download/、downloads/)的路径与基础文件操作。
package store

import (
	"encoding/json"
	"errors"
	"fmt"
	"io/fs"
	"os"
	"path/filepath"
	"strings"
)

// 默认相对项目根的目录名。Step 1 约定全部相对 cwd 解析。
const (
	PreDownloadDir = "pre-download"
	DownloadsDir   = "downloads"
)

// Gallery 是 pre-download/<name>.json 中保存的画廊元信息结构。
type Gallery struct {
	GalleryName string            `json:"gallery_name"` // 已清洗,可用作文件夹/文件名
	RawTitle    string            `json:"raw_title"`    // 页面原始标题(未清洗)
	SourceURL   string            `json:"source_url"`   // 用户粘贴的画廊完整 URL
	TotalPages  int               `json:"total_pages"`  // 总图片数
	FetchedAt   string            `json:"fetched_at"`   // RFC3339
	Pages       map[string]string `json:"pages"`        // "0001" -> 图片直链
}

// ErrGalleryNotFound 指定的画廊 json 不存在。
var ErrGalleryNotFound = errors.New("E_NOT_FOUND: 指定的画廊 json 不存在")

// LoadGallery 读取 pre-download/<name>.json 并返回解析后的 Gallery。
// 与 ListGalleries 的差异:这里是「单个 + 错误透传」,调用方能用 errors.Is 区分。
func LoadGallery(name string) (*Gallery, error) {
	data, err := os.ReadFile(GalleryJSONPath(name))
	if err != nil {
		if errors.Is(err, os.ErrNotExist) {
			return nil, ErrGalleryNotFound
		}
		return nil, fmt.Errorf("读取 %s 失败: %w", GalleryJSONPath(name), err)
	}
	var g Gallery
	if err := json.Unmarshal(data, &g); err != nil {
		return nil, fmt.Errorf("解析 %s 失败: %w", GalleryJSONPath(name), err)
	}
	return &g, nil
}

// EnsureDirs 确保运行时目录存在,并在日志中报告每条目录是「已存在」还是「新建」。
// 同时做一次存在性检测 (Stat),遇到普通文件占位 (同名文件) 时返回明确错误,
// 避免把目录创建失败误报为后续错误。
func EnsureDirs() (created []string, err error) {
	for _, dir := range []string{PreDownloadDir, DownloadsDir} {
		// 先检测:同名位置已经是普通文件而非目录,直接报错。
		if fi, statErr := os.Stat(dir); statErr == nil {
			if !fi.IsDir() {
				return nil, fmt.Errorf("%s 已存在但不是目录 (type=%s),请手动处理后重试", dir, modeType(fi))
			}
			// 已存在且是目录,无需创建。
			continue
		} else if !os.IsNotExist(statErr) {
			return nil, fmt.Errorf("stat %s: %w", dir, statErr)
		}
		// 不存在,创建。
		if mkErr := os.MkdirAll(dir, 0o755); mkErr != nil {
			return nil, fmt.Errorf("mkdir %s: %w", dir, mkErr)
		}
		created = append(created, dir)
	}
	return created, nil
}

// modeType 把 FileInfo 的 mode 翻译为可读字符串(用于错误信息)。
func modeType(fi os.FileInfo) string {
	m := fi.Mode()
	switch {
	case m.IsDir():
		return "dir"
	case m&os.ModeSymlink != 0:
		return "symlink"
	case m.IsRegular():
		return "regular-file"
	default:
		return m.String()
	}
}

// GalleryJSONPath 返回画廊 json 的绝对/相对路径。
func GalleryJSONPath(name string) string {
	return filepath.Join(PreDownloadDir, name+".json")
}

// GalleryDownloadDir 返回画廊图片输出目录。
func GalleryDownloadDir(name string) string {
	return filepath.Join(DownloadsDir, name)
}

// GalleryExists 判断画廊 json 是否已存在。
func GalleryExists(name string) (bool, error) {
	_, err := os.Stat(GalleryJSONPath(name))
	if err == nil {
		return true, nil
	}
	if errors.Is(err, fs.ErrNotExist) {
		return false, nil
	}
	return false, err
}

// SaveGallery 原子写画廊 json(先写 .tmp 再 rename),避免半截文件。
func SaveGallery(g *Gallery) error {
	if _, err := EnsureDirs(); err != nil {
		return err
	}
	final := GalleryJSONPath(g.GalleryName)
	tmp := final + ".tmp"

	data, err := json.MarshalIndent(g, "", "  ")
	if err != nil {
		return fmt.Errorf("marshal gallery: %w", err)
	}
	if err := os.WriteFile(tmp, data, 0o644); err != nil {
		return fmt.Errorf("write tmp: %w", err)
	}
	if err := os.Rename(tmp, final); err != nil {
		return fmt.Errorf("rename tmp -> final: %w", err)
	}
	return nil
}

// ListGalleries 列出 pre-download 下所有画廊 json 的概要信息。
// json 损坏的文件会被跳过(但不删除)。
func ListGalleries() ([]Gallery, error) {
	entries, err := os.ReadDir(PreDownloadDir)
	if err != nil {
		if errors.Is(err, fs.ErrNotExist) {
			return nil, nil
		}
		return nil, err
	}
	var out []Gallery
	for _, e := range entries {
		if e.IsDir() || !strings.HasSuffix(e.Name(), ".json") {
			continue
		}
		path := filepath.Join(PreDownloadDir, e.Name())
		data, err := os.ReadFile(path)
		if err != nil {
			continue
		}
		var g Gallery
		if err := json.Unmarshal(data, &g); err != nil {
			continue
		}
		out = append(out, g)
	}
	return out, nil
}