274 lines
7.6 KiB
Go
274 lines
7.6 KiB
Go
package core
|
||
|
||
// 下载目录模板引擎
|
||
//
|
||
// ── 与参考项目（EH）的差异 ──────────────────────────────────────
|
||
//
|
||
// EH 有 artist / group / series / category / lang / uploader / tag… 一整套
|
||
// 标签命名空间，所以占位符很多。4KHD 是纯图站，页面里**没有**这些元数据，
|
||
// 硬套只会全部落到「未知作者」兜底值，毫无意义。
|
||
//
|
||
// 所以这里只保留站点真实提供的字段：
|
||
//
|
||
// {title} 画廊标题
|
||
// {id} 画廊标识（域名 + 路径），唯一但很长，一般不用
|
||
// {slug} 路径里的名字，含 .html，如 myua-sweetbox-vol52.html
|
||
// {name} 同上但去掉 .html 后缀，如 myua-sweetbox-vol52
|
||
// {host} 站点域名，如 migmg.ssuu.uk
|
||
// {category} 分类数字，如 09
|
||
//
|
||
// 例：{title}、{host}/{title}、{category}/{name}
|
||
|
||
import (
|
||
"fmt"
|
||
"net"
|
||
"os"
|
||
"path/filepath"
|
||
"regexp"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
"unicode"
|
||
|
||
"hd4k-downloader/internal/hd4k"
|
||
)
|
||
|
||
// Fallback 是各占位符的兜底值
|
||
var Fallback = map[string]string{
|
||
"host": "unknown",
|
||
"category": "unknown",
|
||
"name": "unnamed",
|
||
"slug": "unnamed",
|
||
}
|
||
|
||
var (
|
||
// 路径保留字符 + 控制字符
|
||
reIllegal = regexp.MustCompile(`[\\/:*?"<>|]`)
|
||
reControl = regexp.MustCompile(`[\x00-\x1f\x7f]`)
|
||
reSpace = regexp.MustCompile(`\s+`)
|
||
// 目录分隔符 —— 只用来切「模板文本」本身。
|
||
// ⚠️ 千万别拿它去切占位符替换后的结果：那会把「值里的 / \ : * ? " < > 」
|
||
// 变成真的目录层级。标题里一个斜杠就多一层文件夹。
|
||
reSep = regexp.MustCompile(`[\\/]+`)
|
||
// 模板占位符：{key}
|
||
rePlaceholder = regexp.MustCompile(`\{([a-zA-Z0-9_]+)\}`)
|
||
)
|
||
|
||
// SanitizeSegment 清洗单个路径段：干掉路径分隔符和保留字符
|
||
func SanitizeSegment(name, fallback string) string {
|
||
s := reIllegal.ReplaceAllString(name, "_")
|
||
s = reControl.ReplaceAllString(s, "_")
|
||
s = reSpace.ReplaceAllString(s, " ")
|
||
s = strings.TrimSpace(s)
|
||
|
||
// "." / ".." 是危险路径
|
||
if s == "." || s == ".." {
|
||
s = "_"
|
||
}
|
||
// Windows 不允许段名以点或空格结尾
|
||
s = strings.TrimRight(s, ". ")
|
||
|
||
if s == "" {
|
||
s = fallback
|
||
}
|
||
// 超长截断（Windows 路径上限 260，留足余量）
|
||
if len([]rune(s)) > 110 {
|
||
s = string([]rune(s)[:110])
|
||
s = strings.TrimRight(s, ". ")
|
||
}
|
||
// Windows 保留设备名（CON/PRN/NUL/COM1…）不能做目录名
|
||
if isWindowsReservedName(s) {
|
||
s = "_" + s
|
||
}
|
||
return s
|
||
}
|
||
|
||
// buildDirVars 展开模板得到全部变量
|
||
func buildDirVars(g *hd4k.Gallery) map[string]string {
|
||
title := g.Title
|
||
if title == "" {
|
||
title = g.Slug
|
||
}
|
||
name := strings.TrimSuffix(g.Slug, ".html")
|
||
return map[string]string{
|
||
"id": g.ID,
|
||
"title": title,
|
||
"slug": FallbackIfEmpty(g.Slug, Fallback["slug"]),
|
||
"name": FallbackIfEmpty(name, Fallback["name"]),
|
||
"host": FallbackIfEmpty(g.Host, Fallback["host"]),
|
||
"category": FallbackIfEmpty(g.Meta.Category, Fallback["category"]),
|
||
}
|
||
}
|
||
|
||
// FallbackIfEmpty 空字符串才用兜底值
|
||
func FallbackIfEmpty(s, fb string) string {
|
||
if strings.TrimSpace(s) == "" {
|
||
return fb
|
||
}
|
||
return s
|
||
}
|
||
|
||
// BuildDir 用模板算出画廊的相对目录（相对 downloadRoot，逐段清洗）
|
||
//
|
||
// ⚠️ 分层只认「模板文本里写的分隔符」。
|
||
// 占位符替换出来的值一律当普通名字：里面的 \ / : * ? " < > |
|
||
// 全部清洗成 _（SanitizeSegment 干的活），绝不凭空多出一层目录。
|
||
func BuildDir(tpl string, g *hd4k.Gallery) string {
|
||
tpl = strings.TrimSpace(tpl)
|
||
if tpl == "" {
|
||
tpl = "{title}"
|
||
}
|
||
|
||
vars := buildDirVars(g)
|
||
|
||
expand := func(m string) string {
|
||
key := strings.ToLower(m[1 : len(m)-1])
|
||
if v, ok := vars[key]; ok {
|
||
return v
|
||
}
|
||
return m // 未知占位符原样保留，避免用户以为拼错了却静默消失
|
||
}
|
||
|
||
// 先按模板里的分隔符切段，再逐段展开占位符
|
||
segs := []string{}
|
||
for _, raw := range reSep.Split(tpl, -1) {
|
||
if strings.TrimSpace(raw) == "" {
|
||
continue
|
||
}
|
||
fb := "未命名"
|
||
if len(segs) == 0 {
|
||
fb = FallbackIfEmpty(g.Slug, "gallery")
|
||
}
|
||
s := SanitizeSegment(rePlaceholder.ReplaceAllStringFunc(raw, expand), fb)
|
||
if s != "" {
|
||
segs = append(segs, s)
|
||
}
|
||
}
|
||
|
||
if len(segs) == 0 {
|
||
return SanitizeSegment(g.Title, g.Slug)
|
||
}
|
||
return strings.Join(segs, string(filepath.Separator))
|
||
}
|
||
|
||
// ListPlaceholders 给设置界面提示用
|
||
func ListPlaceholders() []string {
|
||
return []string{"{title}", "{name}", "{slug}", "{host}", "{category}", "{id}"}
|
||
}
|
||
|
||
// SanitizeFilename 清洗图片文件名（不含目录分隔）
|
||
func SanitizeFilename(name, fallback string) string {
|
||
s := reIllegal.ReplaceAllString(name, "_")
|
||
s = reControl.ReplaceAllString(s, "_")
|
||
s = strings.TrimSpace(s)
|
||
if s == "" || s == "." || s == ".." {
|
||
s = fallback
|
||
}
|
||
if len([]rune(s)) > 150 {
|
||
s = string([]rune(s)[:150])
|
||
}
|
||
return s
|
||
}
|
||
|
||
// DefaultDownloadRoot 默认下载目录 ~/Downloads/4KHD
|
||
func DefaultDownloadRoot() string {
|
||
home, err := os.UserHomeDir()
|
||
if err != nil {
|
||
return "4KHD"
|
||
}
|
||
return filepath.Join(home, "Downloads", "4KHD")
|
||
}
|
||
|
||
// DetectLocalProxy 探测本机常见代理端口（mihomo/clash/v2ray 等）。
|
||
// 真正等连接结果，连不上就试下一个端口。
|
||
func DetectLocalProxy(ports []int) string {
|
||
for _, p := range ports {
|
||
addr := fmt.Sprintf("127.0.0.1:%d", p)
|
||
conn, err := net.DialTimeout("tcp", addr, 250*time.Millisecond)
|
||
if err == nil {
|
||
conn.Close()
|
||
return "http://127.0.0.1:" + strconv.Itoa(p)
|
||
}
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// isWindowsReservedName Windows 保留设备名，目录名撞上会失败
|
||
func isWindowsReservedName(s string) bool {
|
||
base := strings.ToUpper(strings.SplitN(s, ".", 2)[0])
|
||
switch base {
|
||
case "CON", "PRN", "AUX", "NUL",
|
||
"COM1", "COM2", "COM3", "COM4", "COM5", "COM6", "COM7", "COM8", "COM9",
|
||
"LPT1", "LPT2", "LPT3", "LPT4", "LPT5", "LPT6", "LPT7", "LPT8", "LPT9":
|
||
return true
|
||
}
|
||
return false
|
||
}
|
||
|
||
// validSegmentName 段名合法性检查（给测试用）
|
||
func validSegmentName(s string) bool {
|
||
if s == "" || s == "." || s == ".." {
|
||
return false
|
||
}
|
||
if strings.ContainsAny(s, `\/:*?"<>|`) {
|
||
return false
|
||
}
|
||
for _, r := range s {
|
||
if unicode.IsControl(r) {
|
||
return false
|
||
}
|
||
}
|
||
if isWindowsReservedName(s) {
|
||
return false
|
||
}
|
||
return true
|
||
}
|
||
|
||
// MinImageBytes 清理时的大小门槛：小于这个字节数的图视为占位图/损坏图，删掉。
|
||
//
|
||
// 10KB 取自实测：4KHD 正文图最小也在 100KB 以上，而占位图/
|
||
// 损坏图/CDN 错误页伪装的图通常只有几 KB。
|
||
const MinImageBytes = 10 * 1024
|
||
|
||
// imageExts 只认这些是图片，避免误删清单或别的东西
|
||
var imageExts = map[string]bool{
|
||
".webp": true, ".jpg": true, ".jpeg": true, ".png": true,
|
||
".gif": true, ".bmp": true, ".avif": true, ".jfif": true,
|
||
}
|
||
|
||
// SweepSmallImages 递归扫下载目录，删掉小于 minBytes 的图片文件。
|
||
//
|
||
// 返回删掉的文件数和释放的字节数。
|
||
//
|
||
// 只动图片扩展名的文件；清单（.hd4k.json）和 .part 临时文件一律不碰 ——
|
||
// 清单要留着续传，.part 是未完成的下载（可能正好小于门槛但其实没下完）。
|
||
func SweepSmallImages(root string, minBytes int64) (int, int64, error) {
|
||
if minBytes <= 0 {
|
||
minBytes = MinImageBytes
|
||
}
|
||
removed, freed := 0, int64(0)
|
||
|
||
err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
|
||
if err != nil {
|
||
// 单个文件出错不该中断整个清扫
|
||
return nil
|
||
}
|
||
if info.IsDir() {
|
||
return nil
|
||
}
|
||
ext := strings.ToLower(filepath.Ext(path))
|
||
if !imageExts[ext] {
|
||
return nil
|
||
}
|
||
if info.Size() >= minBytes {
|
||
return nil
|
||
}
|
||
if rmErr := os.Remove(path); rmErr == nil {
|
||
removed++
|
||
freed += info.Size()
|
||
}
|
||
return nil
|
||
})
|
||
return removed, freed, err
|
||
}
|