Добавил ручную перепривязку

2026-06-15 07:42:50 +03:00
parent 093211c9c7
commit 16a82572e7
10 changed files with 206 additions and 10 deletions
@@ -21,10 +21,11 @@ import (
 const (
 	ovrMediaType    = "media_type"
 	ovrIgnoredFiles = "ignored_files"
-	ovrProvider     = "provider"    // выбранная база ("none" = без базы)
-	ovrProviderID   = "provider_id" // id в выбранной базе
-	ovrTitle        = "title"       // запиненное каноническое название
-	ovrYear         = "year"        // запиненный год
+	ovrProvider     = "provider"     // выбранная база ("none" = без базы)
+	ovrProviderID   = "provider_id"  // id в выбранной базе
+	ovrTitle        = "title"        // запиненное каноническое название
+	ovrYear         = "year"         // запиненный год
+	ovrForceReview  = "force_review" // ручная перепривязка: не авто-раскладывать
 )

 // recognizePending распознаёт завершённые загрузки и перезапускает те, что
@@ -178,9 +179,13 @@ func (w *Worker) finishRecognition(ctx context.Context, id int64, res recognize.
 	}

 	// Авто-раскладка при подтверждённом матче и чистой валидации (Ф4);
-	// иначе — review. Раскладчик может быть не сконфигурирован.
-	if res.Decision.Auto && w.layouter != nil {
-		plan := applyOverrides(res.Plan, w.overridesOrNil(ctx, id))
+	// иначе — review. Раскладчик может быть не сконфигурирован. При ручной
+	// перепривязке (force_review) авто-раскладку не делаем — нужно явное
+	// подтверждение человеком.
+	overrides := w.overridesOrNil(ctx, id)
+	forceReview := overrides[ovrForceReview] == "1"
+	if res.Decision.Auto && !forceReview && w.layouter != nil {
+		plan := applyOverrides(res.Plan, overrides)
 		w.transition(ctx, *d, store.StateLinking, "", "")
 		if err := w.linkPlan(ctx, d, plan, tag, savePath); err != nil {
 			w.log.Warn("recognize: auto-apply failed, left for review",
@@ -281,6 +286,49 @@ func (w *Worker) linkPlan(ctx context.Context, d *store.Download, plan recognize
 	return nil
 }

+// Relink повторно привязывает откатанную задачу (reverted): возвращает её на
+// распознавание, и поллинг-цикл перезапустит recognize. Авто-раскладку при
+// этом не делаем — ручная перепривязка всегда проходит через ревью с
+// подтверждением (force_review). Источник (раздача в qBittorrent) для этого
+// должен быть на месте.
+func (w *Worker) Relink(ctx context.Context, id int64) error {
+	w.mu.Lock()
+	defer w.mu.Unlock()
+
+	d, err := w.store.GetDownload(ctx, id)
+	if err != nil {
+		return fmt.Errorf("relink: %w", err)
+	}
+	if d.State != store.StateReverted {
+		return fmt.Errorf("relink: download %d is in state %s (expected reverted)", id, d.State)
+	}
+	if !d.Infohash.Valid {
+		return fmt.Errorf("relink: download %d has no infohash", id)
+	}
+	// Раздача должна ещё быть в qBittorrent — без неё распознавать нечего.
+	if _, ok, terr := w.torrentByInfohash(ctx, d.Infohash.String); terr != nil {
+		return fmt.Errorf("relink: %w", terr)
+	} else if !ok {
+		return fmt.Errorf("relink: торрент не найден в qBittorrent")
+	}
+	// Вернуть задачу в активную обработку можно, только если другой активной
+	// задачи на этот infohash нет (partial unique index по idempotency_key).
+	active, err := w.store.FindActiveByInfohash(ctx, d.Infohash.String)
+	if err != nil {
+		return fmt.Errorf("relink: %w", err)
+	}
+	if active != nil {
+		return fmt.Errorf("relink: для этого торрента уже есть активная задача #%d", active.ID)
+	}
+	// Ручная перепривязка — всегда с подтверждением, без авто-раскладки.
+	if err := w.store.SetOverride(ctx, id, ovrForceReview, "1"); err != nil {
+		return fmt.Errorf("relink: %w", err)
+	}
+	w.transition(ctx, *d, store.StateRecognizing, "", "")
+	w.log.Info("relink: re-recognizing reverted download", "download_id", id)
+	return nil
+}
+
 // Refine добавляет подсказку и отправляет задачу на перераспознавание.
 func (w *Worker) Refine(ctx context.Context, id int64, hint string) error {
 	hint = strings.TrimSpace(hint)
@@ -97,6 +97,80 @@ func TestScanner_FiresOnDone(t *testing.T) {
 	}
 }

+func revertedDownload(id int64) *store.Download {
+	d := completedDownload(id)
+	d.State = store.StateReverted
+	return d
+}
+
+func TestRelink_RevertedToRecognizing(t *testing.T) {
+	st := newMemStore()
+	st.put(revertedDownload(1))
+	qb := &fakeQbt{torrents: []qbt.Torrent{{Hash: ihTest, Name: "Show", SavePath: "/d"}}}
+	w := testWorkerWith(st, qb, &fakeRecognizer{result: seriesResult()}, nil)
+
+	if err := w.Relink(context.Background(), 1); err != nil {
+		t.Fatalf("Relink: %v", err)
+	}
+	if st.downloads[1].State != store.StateRecognizing {
+		t.Fatalf("state = %q, want recognizing", st.downloads[1].State)
+	}
+	if st.overrides[1][ovrForceReview] != "1" {
+		t.Errorf("force_review override = %q, want 1", st.overrides[1][ovrForceReview])
+	}
+}
+
+func TestRelink_RejectsNonReverted(t *testing.T) {
+	st := newMemStore()
+	st.put(completedDownload(1)) // не reverted
+	qb := &fakeQbt{torrents: []qbt.Torrent{{Hash: ihTest}}}
+	w := testWorkerWith(st, qb, &fakeRecognizer{}, nil)
+
+	if err := w.Relink(context.Background(), 1); err == nil {
+		t.Fatal("ожидали ошибку для не-reverted задачи, получили nil")
+	}
+}
+
+func TestRelink_TorrentMissing(t *testing.T) {
+	st := newMemStore()
+	st.put(revertedDownload(1))
+	qb := &fakeQbt{torrents: nil} // раздачи в qBittorrent нет
+	w := testWorkerWith(st, qb, &fakeRecognizer{}, nil)
+
+	if err := w.Relink(context.Background(), 1); err == nil {
+		t.Fatal("ожидали ошибку при отсутствии торрента, получили nil")
+	}
+	if st.downloads[1].State != store.StateReverted {
+		t.Errorf("state = %q, want reverted (без изменений)", st.downloads[1].State)
+	}
+}
+
+// TestRelink_ForceReviewSkipsAuto проверяет, что после перепривязки даже
+// уверенный матч не уходит в авто-раскладку, а ждёт подтверждения в review.
+func TestRelink_ForceReviewSkipsAuto(t *testing.T) {
+	f := newApplyFixture(t, seriesResult().Plan)
+	// Готовим состояние «как после Relink»: reverted, force_review выставлен.
+	f.st.downloads[1].State = store.StateReverted
+	_ = f.st.SetOverride(context.Background(), 1, ovrForceReview, "1")
+
+	auto := seriesResult()
+	auto.Decision.Auto = true
+	auto.Match = &recognize.Match{Provider: "tvdb", ProviderID: "42"}
+	f.w.recognizer = &fakeRecognizer{result: auto}
+
+	if err := f.w.Relink(context.Background(), 1); err != nil {
+		t.Fatalf("Relink: %v", err)
+	}
+	f.w.recognizeOne(context.Background(), 1)
+
+	if f.st.downloads[1].State != store.StateReview {
+		t.Fatalf("state = %q, want review (авто-раскладка не должна сработать)", f.st.downloads[1].State)
+	}
+	if len(f.st.links) != 0 {
+		t.Errorf("file_links = %d, want 0 (ничего не линковали)", len(f.st.links))
+	}
+}
+
 // memStore — полноценный in-memory store для тестов Ф3.
 type memStore struct {
 	downloads  map[int64]*store.Download
@@ -138,6 +212,16 @@ func (m *memStore) ExistsByInfohash(_ context.Context, infohash string) (bool, e
 	return false, nil
 }

+func (m *memStore) FindActiveByInfohash(_ context.Context, infohash string) (*store.Download, error) {
+	for _, d := range m.downloads {
+		if d.Infohash.Valid && d.Infohash.String == infohash && !d.State.IsTerminal() {
+			cp := *d
+			return &cp, nil
+		}
+	}
+	return nil, nil
+}
+
 func (m *memStore) CreateDownload(_ context.Context, d *store.Download) (int64, error) {
 	id := int64(len(m.downloads) + 1)
 	cp := *d
@@ -33,6 +33,7 @@ type Store interface {

 	// Discovery (усыновление раздач по категории/тегу).
 	ExistsByInfohash(ctx context.Context, infohash string) (bool, error)
+	FindActiveByInfohash(ctx context.Context, infohash string) (*store.Download, error)
 	CreateDownload(ctx context.Context, d *store.Download) (int64, error)

 	// Ф3: распознавание, ревью, раскладка.
@@ -60,6 +60,16 @@ func (f *fakeStore) ExistsByInfohash(_ context.Context, infohash string) (bool,
 	return false, nil
 }

+func (f *fakeStore) FindActiveByInfohash(_ context.Context, infohash string) (*store.Download, error) {
+	for _, d := range f.downloads {
+		if d.Infohash.Valid && d.Infohash.String == infohash && !d.State.IsTerminal() {
+			cp := *d
+			return &cp, nil
+		}
+	}
+	return nil, nil
+}
+
 func (f *fakeStore) CreateDownload(_ context.Context, d *store.Download) (int64, error) {
 	id := int64(len(f.downloads) + 1)
 	cp := *d