refactor: modularize downloader architecture

webdav.go

@@ -0,0 +1,76 @@
+package main
+
+import (
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+func (a *App) upload(t *Task) {
+	if a.storage == nil || a.cfg.WebDAVURL == "" {
+		t.Status, t.Progress = "done", 100
+		a.db.Save(t)
+		return
+	}
+	a.db.Model(t).Updates(map[string]any{"status": "uploading"})
+	remote := strings.TrimRight(a.cfg.WebDAVPath, "/") + "/" + safe(t.Artist) + "/" + safe(t.Album) + "/" + safe(t.Title) + "." + safe(t.Format)
+	t.RemotePath = remote
+	if err := a.storage.MkdirAll(filepath.ToSlash(filepath.Dir(remote))); err != nil {
+		a.fail(t, err)
+		return
+	}
+	if fi, err := os.Stat(t.LocalPath); err == nil {
+		if same, _ := a.storage.ExistsSameSize(remote, fi.Size()); same {
+			t.Status, t.Progress = "done", 100
+			a.db.Save(t)
+			a.cleanupLocalFile(t)
+			return
+		}
+	}
+	if err := a.storage.Upload(t.LocalPath, remote); err != nil {
+		a.fail(t, err)
+		return
+	}
+	if a.cfg.WebDAVVersionPolicy == "replace_lower" {
+		a.cleanupLowerQualityVersions(t)
+	}
+	t.Status, t.Progress = "done", 100
+	a.db.Save(t)
+	a.cleanupLocalFile(t)
+}
+
+func (a *App) cleanupLocalFile(t *Task) {
+	if a.cfg.LocalFilePolicy != "delete_after_upload" || t.LocalPath == "" {
+		return
+	}
+	if err := os.Remove(t.LocalPath); err != nil && !os.IsNotExist(err) {
+		t.Error = "WebDAV 上传成功，但删除本地文件失败: " + err.Error()
+		a.db.Save(t)
+	}
+}
+
+func (a *App) cleanupLowerQualityVersions(t *Task) {
+	levels := map[string]int{"standard": 1, "exhigh": 2, "sq": 3, "lossless": 4, "hires": 5, "jyeffect": 6, "sky": 7, "jymaster": 8, "atmos": 9, "atmos_plus": 10, "master": 11}
+	current := qualityRank(levels, t.ActualQuality, t.ResolvedQuality, t.Quality)
+	if current == 0 {
+		return
+	}
+	var old []Task
+	a.db.Where("artist = ? AND title = ? AND album = ? AND status = ? AND id <> ?", t.Artist, t.Title, t.Album, "done", t.ID).Find(&old)
+	for _, candidate := range old {
+		q := qualityRank(levels, candidate.ActualQuality, candidate.ResolvedQuality, candidate.Quality)
+		if q == 0 || q >= current || candidate.RemotePath == "" || candidate.RemotePath == t.RemotePath {
+			continue
+		}
+		_ = a.storage.Remove(candidate.RemotePath)
+	}
+}
+
+func qualityRank(levels map[string]int, values ...string) int {
+	for _, q := range values {
+		if n := levels[q]; n > 0 {
+			return n
+		}
+	}
+	return 0
+}