mirror of
https://github.com/truewhile/MeBox.git
synced 2026-10-05 21:06:38 +08:00
fix subscription download dedup scope
This commit is contained in:
@@ -49,7 +49,7 @@ func (d *DownloadService) AddDownloadWithMeta(ctx context.Context, userID, urlSt
|
|||||||
if !req.meta.AllowExistingLibrary && d.localMediaAlreadyExists(ctx, req.title) {
|
if !req.meta.AllowExistingLibrary && d.localMediaAlreadyExists(ctx, req.title) {
|
||||||
return nil, ErrMediaAlreadyInLibrary
|
return nil, ErrMediaAlreadyInLibrary
|
||||||
}
|
}
|
||||||
if existing, ok := d.findExistingDownloadTask(ctx, req.title, strings.TrimSpace(req.meta.SubscriptionID) != ""); ok {
|
if existing, ok := d.findExistingDownloadTask(ctx, req); ok {
|
||||||
return existing, ErrDownloadAlreadyExists
|
return existing, ErrDownloadAlreadyExists
|
||||||
}
|
}
|
||||||
_ = d.ReloadConfig(ctx)
|
_ = d.ReloadConfig(ctx)
|
||||||
@@ -228,8 +228,8 @@ func localMediaRowSeasonEpisode(row model.Media) (int, int) {
|
|||||||
return rowSeason, rowEpisode
|
return rowSeason, rowEpisode
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *DownloadService) findExistingDownloadTask(ctx context.Context, title string, allowDeletedReadd bool) (*model.DownloadTask, bool) {
|
func (d *DownloadService) findExistingDownloadTask(ctx context.Context, req downloadAddRequest) (*model.DownloadTask, bool) {
|
||||||
key := downloadTaskIdentityKey(title)
|
key := downloadTaskIdentityKey(req.title)
|
||||||
if key == "" || d == nil || d.repo == nil || d.repo.Download == nil {
|
if key == "" || d == nil || d.repo == nil || d.repo.Download == nil {
|
||||||
return nil, false
|
return nil, false
|
||||||
}
|
}
|
||||||
@@ -237,22 +237,43 @@ func (d *DownloadService) findExistingDownloadTask(ctx context.Context, title st
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, false
|
return nil, false
|
||||||
}
|
}
|
||||||
|
subscriptionID := strings.TrimSpace(req.meta.SubscriptionID)
|
||||||
for i := range rows {
|
for i := range rows {
|
||||||
if allowDeletedReadd {
|
if subscriptionID != "" {
|
||||||
if !downloadTaskBlocksReadd(rows[i].Status) {
|
if !downloadTaskBlocksReadd(rows[i].Status) {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
if !downloadTaskInSubscriptionScope(rows[i], req) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
} else if !downloadTaskBlocksDuplicate(rows[i].Status) {
|
} else if !downloadTaskBlocksDuplicate(rows[i].Status) {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
current := downloadTaskIdentityKey(rows[i].Title)
|
current := downloadTaskIdentityKey(rows[i].Title)
|
||||||
if downloadTitleCoversRequest(rows[i].Title, title) || current == key {
|
if downloadTitleCoversRequest(rows[i].Title, req.title) || current == key {
|
||||||
return &rows[i], true
|
return &rows[i], true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return nil, false
|
return nil, false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func downloadTaskInSubscriptionScope(row model.DownloadTask, req downloadAddRequest) bool {
|
||||||
|
subscriptionID := strings.TrimSpace(req.meta.SubscriptionID)
|
||||||
|
if subscriptionID == "" {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
rowSubscriptionID := strings.TrimSpace(row.SubscriptionID)
|
||||||
|
if rowSubscriptionID != "" {
|
||||||
|
return rowSubscriptionID == subscriptionID
|
||||||
|
}
|
||||||
|
rowSavePath := strings.TrimSpace(row.SavePath)
|
||||||
|
requestSavePath := strings.TrimSpace(req.savePath)
|
||||||
|
if rowSavePath == "" || requestSavePath == "" {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return sameOrChildPath(rowSavePath, requestSavePath) || sameOrChildPath(requestSavePath, rowSavePath)
|
||||||
|
}
|
||||||
|
|
||||||
func (d *DownloadService) torrentExistsByIdentity(ctx context.Context, title string) bool {
|
func (d *DownloadService) torrentExistsByIdentity(ctx context.Context, title string) bool {
|
||||||
query := downloadTaskIdentityKey(title)
|
query := downloadTaskIdentityKey(title)
|
||||||
if query == "" {
|
if query == "" {
|
||||||
|
|||||||
@@ -429,6 +429,69 @@ func TestAddDownloadWithMetaDoesNotDedupRangeAgainstSingleEpisodeTask(t *testing
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestAddDownloadWithMetaScopesSubscriptionDedupBySubscriptionOrSavePath(t *testing.T) {
|
||||||
|
var addCalls int32
|
||||||
|
qb := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
switch r.URL.Path {
|
||||||
|
case "/api/v2/auth/login":
|
||||||
|
_, _ = w.Write([]byte("Ok."))
|
||||||
|
case "/api/v2/torrents/info":
|
||||||
|
http.Error(w, "temporary list unavailable", http.StatusInternalServerError)
|
||||||
|
case "/api/v2/torrents/add":
|
||||||
|
atomic.AddInt32(&addCalls, 1)
|
||||||
|
_, _ = w.Write([]byte("Ok."))
|
||||||
|
default:
|
||||||
|
http.NotFound(w, r)
|
||||||
|
}
|
||||||
|
}))
|
||||||
|
defer qb.Close()
|
||||||
|
|
||||||
|
db := newServiceTestDB(t, &model.DownloadTask{}, &model.DownloadClient{}, &model.Setting{})
|
||||||
|
repos := repository.New(db)
|
||||||
|
configureTestDefaultQB(t, repos, qb.URL)
|
||||||
|
for _, existing := range []model.DownloadTask{
|
||||||
|
{
|
||||||
|
UserID: "u1",
|
||||||
|
SubscriptionID: "other-subscription",
|
||||||
|
Source: "qbittorrent",
|
||||||
|
URL: "https://pt.example/download?id=old-sub",
|
||||||
|
Title: "Archives The Nanyang Mystery 2026 S01E07-S01E08 2160p WEB-DL",
|
||||||
|
SavePath: "/downloads/other",
|
||||||
|
Status: "completed",
|
||||||
|
Progress: 1,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
UserID: "u1",
|
||||||
|
Source: "qbittorrent",
|
||||||
|
URL: "https://pt.example/download?id=old-manual",
|
||||||
|
Title: "Archives The Nanyang Mystery 2026 S01E09-S01E10 2160p WEB-DL",
|
||||||
|
SavePath: "/downloads/archive",
|
||||||
|
Status: "completed",
|
||||||
|
Progress: 1,
|
||||||
|
},
|
||||||
|
} {
|
||||||
|
row := existing
|
||||||
|
if err := repos.Download.Create(t.Context(), &row); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), nil)
|
||||||
|
task, err := svc.AddDownloadWithMeta(t.Context(), "u1", "magnet:?xt=urn:btih:efefefefefefefefefefefefefefefefefefefef&dn=Archives+The+Nanyang+Mystery+2026+S01E07-S01E08", "/downloads/tv", DownloadTaskMeta{
|
||||||
|
SubscriptionID: "current-subscription",
|
||||||
|
Title: "Archives The Nanyang Mystery 2026 S01E07-S01E08 2160p WEB-DL",
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("AddDownloadWithMeta returned %v, want queued because old task is outside current subscription scope", err)
|
||||||
|
}
|
||||||
|
if task == nil {
|
||||||
|
t.Fatal("task = nil, want queued task")
|
||||||
|
}
|
||||||
|
if got := atomic.LoadInt32(&addCalls); got != 1 {
|
||||||
|
t.Fatalf("qb add calls = %d, want 1", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestAddDownloadWithMetaAutoClassifiesSavePathAndQBitCategory(t *testing.T) {
|
func TestAddDownloadWithMetaAutoClassifiesSavePathAndQBitCategory(t *testing.T) {
|
||||||
var addCalls int32
|
var addCalls int32
|
||||||
var gotSavePath string
|
var gotSavePath string
|
||||||
|
|||||||
Reference in New Issue
Block a user