Приём: добавление загрузки по .torrent-файлу

Принимаем .torrent как загруженные байты — через файл-пикер в веб-форме и Telegram-документ, наряду с magnet. Файл несёт полные метаданные: работает там, где magnet не резолвится (закрытые трекеры, без DHT), и даёт максимум контекста для распознавания без сети. - internal/torrent: парсер поверх anacrolix/torrent/metainfo — инфохэш(и) (v1 SHA1 исходных байтов info; v2 BEP52 при наличии) + Context() из имени, дерева файлов, размера, трекеров. Извлечение файлов панико-безопасно (недоверенный вход). - Персистентность байтов: таблица-спутник download_torrent (миграция 0009); пишется в транзакции создания загрузки, только на ветке создания (не при дедупе). Байты живут весь срок строки — нужны для повторного добавления при retry. - ingest: Request.TorrentData/TorrentName, диспетч парсера; source_ref — человекочитаемый референс (имя раздачи/файла), не адрес добавления. - worker: общий sourceAddParts ветвит по source_type в ОБОИХ add-путях — processCatched и Retry (torrent добавляется файлом, не magnet-хешем). - Транспорты: multipart-форма с файл-пикером (деградация без JS) и приём Telegram-документа (скачивание с редактированием токена из ошибок — секрет не в логи; обработка до ветки pending/текста). Разработка по OpenSpec (SDD): change torrent-file-ingest, два чекпоинта ревью (дизайн до кода, код до архива) сабагентами; дельты влиты в спеки, change архивирован. Ручная проверка на живом qBittorrent (7.3) — за деплоем. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
2026-07-08 10:52:31 +03:00
parent 90fd8640ed
commit 14d615a7c2
38 changed files with 2519 additions and 100 deletions
@@ -0,0 +1,120 @@
+package worker
+
+import (
+	"bytes"
+	"context"
+	"crypto/sha1"
+	"encoding/hex"
+	"testing"
+
+	"github.com/anacrolix/torrent/bencode"
+	"github.com/anacrolix/torrent/metainfo"
+
+	"git.vakhrushev.me/av/jellybit/internal/store"
+)
+
+// buildTorrent — валидные байты .torrent и их v1-инфохэш.
+func buildTorrent(t *testing.T, name string) (data []byte, infohash string) {
+	t.Helper()
+	info := metainfo.Info{Name: name, Length: 2048, PieceLength: 1024, Pieces: make([]byte, 40)}
+	infoBytes, err := bencode.Marshal(info)
+	if err != nil {
+		t.Fatalf("marshal info: %v", err)
+	}
+	sum := sha1.Sum(infoBytes)
+	mi := metainfo.MetaInfo{InfoBytes: infoBytes, Announce: "http://t/ann"}
+	var buf bytes.Buffer
+	if err := mi.Write(&buf); err != nil {
+		t.Fatalf("write metainfo: %v", err)
+	}
+	return buf.Bytes(), hex.EncodeToString(sum[:])
+}
+
+// catchedTorrentStore — пойманная torrent-загрузка с сохранёнными байтами.
+func catchedTorrentStore(id string, data []byte, infohash string) *fakeStore {
+	return &fakeStore{
+		downloads: map[string]*store.Download{
+			id: {
+				ID:         id,
+				State:      store.StateCatched,
+				SourceType: store.SourceTorrent,
+				SourceRef:  "Some.Release.Name", // человекочитаемый референс, не URL
+				Infohashes: hashesOf(id, infohash),
+				CreatedAt:  nowStr,
+			},
+		},
+		torrents: map[string][]byte{id: data},
+	}
+}
+
+// Пойманная torrent-загрузка добавляется в qBittorrent ФАЙЛОМ (Torrents), не URL.
+func TestProcessCatchedTorrentAddsAsFile(t *testing.T) {
+	data, ih := buildTorrent(t, "Dune.mkv")
+	st := catchedTorrentStore("1", data, ih)
+	qb := &fakeQbt{}
+	w := newTestWorker(st, qb)
+	w.SetNamer(&fakeNamer{name: "Дюна (2024)"})
+
+	w.processCatched(context.Background())
+
+	if len(qb.added) != 1 {
+		t.Fatalf("qbt.Add calls = %d, want 1", len(qb.added))
+	}
+	add := qb.added[0]
+	if len(add.URLs) != 0 {
+		t.Errorf("torrent не должен добавляться по URLs: %v", add.URLs)
+	}
+	if len(add.Torrents) != 1 || !bytes.Equal(add.Torrents[0], data) {
+		t.Errorf("байты .torrent не переданы в Add.Torrents")
+	}
+	if add.Rename != "Дюна (2024)" {
+		t.Errorf("rename = %q", add.Rename)
+	}
+	if st.downloads["1"].State != store.StateDownloading {
+		t.Errorf("state = %q, want downloading", st.downloads["1"].State)
+	}
+}
+
+// Retry torrent-загрузки без живой раздачи добавляет её ФАЙЛОМ (регрессия
+// BLOCKER 1: раньше add гейтился на magnet и torrent активировался без раздачи).
+func TestRetryTorrentAddsAsFile(t *testing.T) {
+	data, ih := buildTorrent(t, "Fargo.mkv")
+	st := catchedTorrentStore("1", data, ih)
+	st.downloads["1"].State = store.StateFailed
+	qb := &fakeQbt{} // раздачи в qBittorrent нет → должен добавить заново
+	w := newTestWorker(st, qb)
+
+	if err := w.Retry(context.Background(), "1"); err != nil {
+		t.Fatalf("Retry: %v", err)
+	}
+	if len(qb.added) != 1 {
+		t.Fatalf("ожидалось повторное добавление, got %d", len(qb.added))
+	}
+	if len(qb.added[0].Torrents) != 1 || len(qb.added[0].URLs) != 0 {
+		t.Errorf("retry torrent должен добавляться файлом, add = %+v", qb.added[0])
+	}
+	if st.downloads["1"].State != store.StateDownloading {
+		t.Errorf("state = %q, want downloading", st.downloads["1"].State)
+	}
+}
+
+// Retry torrent-загрузки без сохранённых байтов — откат активации (не оставляем
+// «качающуюся» задачу без раздачи).
+func TestRetryTorrentMissingBytesRollsBack(t *testing.T) {
+	_, ih := buildTorrent(t, "X.mkv")
+	st := catchedTorrentStore("1", nil, ih) // байтов нет
+	delete(st.torrents, "1")
+	st.downloads["1"].State = store.StateFailed
+	qb := &fakeQbt{}
+	w := newTestWorker(st, qb)
+
+	if err := w.Retry(context.Background(), "1"); err == nil {
+		t.Fatal("ожидалась ошибка (нет байтов torrent)")
+	}
+	if len(qb.added) != 0 {
+		t.Errorf("add не должен вызываться без байтов")
+	}
+	if st.downloads["1"].State != store.StateFailed {
+		t.Errorf("state = %q, want failed (откат активации)", st.downloads["1"].State)
+	}
+}