Имя: восстановление display_name после распознавания + гард пустого входа

Голый magnet без dn/контекста заставлял namer звать LLM на пустом входе,
и модель галлюцинировала мусорное имя («Unknown»), которое писалось и в
display_name, и в rename qBittorrent, а заодно ломало UI-фолбэк на
распознанное название. Верное каноническое имя, вычисляемое позже при
распознавании, никуда не переливалось.

- naming: гард пустого входа в DeriveName (нет контекста и подсказки → ""
  без вызова LLM) + детерминированный форматтер FormatTitleYear.
- qbt: операция RenameTorrent (переименование существующей раздачи).
- store: SetDisplayName — обновление имени постфактум без гарда состояния.
- worker: refreshDisplayNameLocked/RefreshDisplayName — перелив канонического
  имени (эффективный план) в display_name + best-effort rename раздачи по
  реальному t.Hash; авто-триггер при подтверждении матча (choose/manual add).
- web-ui: кнопка «Обновить имя» на странице загрузки (htmx-своп заголовка,
  деградация без JS), видимая при наличии распознавания (вкл. done/orphaned).

Спека: дельты ingest/review/web-ui влиты в openspec/specs; change
refresh-display-name заархивирован.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
av
2026-07-10 16:51:04 +03:00
co-authored by Claude Opus 4.8
parent e2ea1840c9
commit 0c9421f4c1
26 changed files with 1179 additions and 21 deletions
+69
View File
@@ -15,6 +15,7 @@ import (
"git.vakhrushev.me/av/jellybit/internal/layout"
"git.vakhrushev.me/av/jellybit/internal/logctx"
"git.vakhrushev.me/av/jellybit/internal/metadata"
"git.vakhrushev.me/av/jellybit/internal/naming"
"git.vakhrushev.me/av/jellybit/internal/qbt"
"git.vakhrushev.me/av/jellybit/internal/recognize"
"git.vakhrushev.me/av/jellybit/internal/store"
@@ -794,6 +795,13 @@ func (w *Worker) chooseCandidateLocked(ctx context.Context, id string, d *store.
}
logctx.From(w.scoped(ctx, capReview, id, d.PrimaryInfohash())).Info("review candidate chosen",
"provider", cand.Provider, "provider_id", cand.ProviderID)
// Подтверждённый матч — переливаем каноническое имя в display_name и в ярлык
// раздачи (best-effort, косметика). Сбой обновления имени не должен ронять
// выбор кандидата: логируем и продолжаем.
if err := w.refreshDisplayNameLocked(ctx, id); err != nil {
logctx.From(w.scoped(ctx, capReview, id, d.PrimaryInfohash())).
Warn("display name refresh after candidate choice failed", "error", err)
}
return nil
}
@@ -1064,6 +1072,67 @@ func (w *Worker) effectivePlan(ctx context.Context, id string) (plan recognize.P
return applyOverrides(plan, overrides), prov, pid, nil
}
// RefreshDisplayName — внешняя точка входа обновления отображаемого имени
// (ручная кнопка): берёт w.mu и делегирует refreshDisplayNameLocked.
func (w *Worker) RefreshDisplayName(ctx context.Context, id string) (err error) {
defer func() { w.logCmd(ctx, "refresh_name", id, err) }()
w.mu.Lock()
defer w.mu.Unlock()
return w.refreshDisplayNameLocked(ctx, id)
}
// refreshDisplayNameLocked переливает уже вычисленное каноническое имя
// (эффективный план с учётом пинов) в download.display_name и в имя раздачи
// qBittorrent — без нового вызова LLM. Косметика: не влияет на пути/раскладку.
// Best-effort к qBittorrent: недоступность/отсутствие раздачи не проваливает
// операцию (display_name пишется в любом случае). Вызывается под w.mu.
func (w *Worker) refreshDisplayNameLocked(ctx context.Context, id string) error {
d, err := w.store.GetDownload(ctx, id)
if err != nil {
return fmt.Errorf("refresh display name: %w", err)
}
if d == nil {
return fmt.Errorf("refresh display name: download %s: %w", id, store.ErrNotFound)
}
cctx := w.scoped(ctx, capIngest, id, d.PrimaryInfohash())
rec, err := w.store.GetCurrentRecognition(cctx, id)
if err != nil {
return fmt.Errorf("refresh display name: %w", err)
}
if rec == nil || !rec.Plan.Valid {
return nil // распознавания ещё нет — обновлять нечего (no-op)
}
plan, _, _, err := w.effectivePlan(cctx, id)
if err != nil {
return fmt.Errorf("refresh display name: %w", err)
}
name := naming.FormatTitleYear(plan.Title, plan.Year)
if name == "" {
return nil // пустое распознанное название — no-op
}
if err := w.store.SetDisplayName(cctx, id, name); err != nil {
return fmt.Errorf("refresh display name: %w", err)
}
logctx.From(cctx).Info("display name refreshed", "display_name", name)
// Переименование раздачи — best-effort по её реальному Torrent.Hash (для
// гибрид/v2 он может не совпасть с нашим primary infohash). Отсутствие
// раздачи (удалена) — штатный no-op: display_name уже обновлён.
t, ok, err := w.torrentByInfohash(cctx, d.HashList())
if err != nil {
logctx.From(cctx).Warn("refresh display name: qbittorrent lookup failed", "error", err)
return nil
}
if !ok {
return nil
}
if err := w.qbt.RenameTorrent(cctx, t.Hash, name); err != nil {
logctx.From(cctx).Warn("refresh display name: qbittorrent rename failed", "error", err)
}
return nil
}
// Коды причины (error_code) ухода задачи в review при раскладке (linkPlan) —
// корреляционный ключ шага, на котором раскладка остановилась. Свод в одном
// месте (как errCode* в worker.go); человекочитаемый текст кладётся в error_msg.
+143
View File
@@ -463,6 +463,13 @@ func (m *memStore) PromoteCatched(_ context.Context, id, displayName string) err
return nil
}
func (m *memStore) SetDisplayName(_ context.Context, id, name string) error {
if d, ok := m.downloads[id]; ok {
d.DisplayName = name
}
return nil
}
func (m *memStore) SetSourceMissCount(_ context.Context, id string, n int) error {
if d, ok := m.downloads[id]; ok {
d.SourceMissCount = n
@@ -1621,6 +1628,142 @@ func TestChooseCandidate_PinsOverrides(t *testing.T) {
}
}
// refreshSetup: загрузка в review с распознаванием (movie «Догадка»/2000),
// display_name="Unknown", кандидатом cand и раздачей в qBit по ihTest — для
// проверок перелива имени после подтверждения матча.
func refreshSetup(t *testing.T, cand store.MetadataCandidate) (*Worker, *memStore, *fakeQbt) {
t.Helper()
st := newMemStore()
d := completedDownload("1")
d.State = store.StateReview
d.DisplayName = "Unknown"
st.put(d)
planJSON, _ := json.Marshal(recognize.Plan{Type: recognize.MediaMovie, Title: "Догадка", Year: 2000})
st.recs = append(st.recs, &store.Recognition{
ID: "1", DownloadID: "1", IsCurrent: true, Plan: store.NullString(string(planJSON)),
Provider: store.NullString("none"),
})
cand.RecognitionID = "1"
_ = st.CreateCandidates(context.Background(), []store.MetadataCandidate{cand})
// Ключ раздачи в qBit (Hash) НАМЕРЕННО отличается от нашего инфохэша: matched
// по InfohashV1. Так тест ловит, что rename идёт по реальному t.Hash (мотив
// D3: для гибрид/v2 t.Hash ≠ PrimaryInfohash), а не по сырому инфохэшу.
qb := &fakeQbt{torrents: []qbt.Torrent{{Hash: qbtHash, InfohashV1: ihTest, Name: "Unknown"}}}
w := testWorkerWith(st, qb, &fakeRecognizer{}, nil)
return w, st, qb
}
// qbtHash — внутренний ключ раздачи в qBittorrent, отличный от инфохэша загрузки.
const qbtHash = "qbt-internal-key"
func TestChooseCandidate_RefreshesDisplayName(t *testing.T) {
w, st, qb := refreshSetup(t, store.MetadataCandidate{
Provider: "tvdb", ProviderID: "343558",
Title: store.NullString("Harold and the Purple Crayon"), Year: sql.NullInt64{Int64: 2024, Valid: true},
})
if err := w.ChooseCandidate(context.Background(), "1", st.candidates[0].ID); err != nil {
t.Fatalf("ChooseCandidate: %v", err)
}
const want = "Harold and the Purple Crayon (2024)"
if got := st.downloads["1"].DisplayName; got != want {
t.Errorf("display_name = %q, want %q", got, want)
}
// rename адресован по реальному t.Hash (qbtHash), а НЕ по инфохэшу загрузки.
if len(qb.renamed) != 1 || qb.renamed[0].hash != qbtHash || qb.renamed[0].name != want {
t.Errorf("renamed = %+v, want hash=%s name=%q", qb.renamed, qbtHash, want)
}
if qbtHash == ihTest {
t.Fatal("тест бессмыслен: qbtHash совпал с инфохэшем загрузки")
}
}
func TestChooseCandidate_RefreshDisplayName_QbitErrorTolerated(t *testing.T) {
w, st, qb := refreshSetup(t, store.MetadataCandidate{
Provider: "tvdb", ProviderID: "343558",
Title: store.NullString("Harold and the Purple Crayon"), Year: sql.NullInt64{Int64: 2024, Valid: true},
})
qb.renameErr = errors.New("qbit down")
// Best-effort: сбой переименования раздачи не должен ронять выбор кандидата.
if err := w.ChooseCandidate(context.Background(), "1", st.candidates[0].ID); err != nil {
t.Fatalf("ChooseCandidate must tolerate qbit rename failure: %v", err)
}
if got := st.downloads["1"].DisplayName; got != "Harold and the Purple Crayon (2024)" {
t.Errorf("display_name = %q (должен быть обновлён несмотря на сбой qBit)", got)
}
}
func TestChooseCandidate_RefreshDisplayName_TorrentMissing(t *testing.T) {
// Раздача удалена (нет в qBit) — display_name всё равно обновлён, rename нет.
w, st, qb := refreshSetup(t, store.MetadataCandidate{
Provider: "tvdb", ProviderID: "343558",
Title: store.NullString("Harold and the Purple Crayon"), Year: sql.NullInt64{Int64: 2024, Valid: true},
})
qb.torrents = nil // torrentByInfohash не найдёт раздачу → best-effort no-op
if err := w.ChooseCandidate(context.Background(), "1", st.candidates[0].ID); err != nil {
t.Fatalf("ChooseCandidate: %v", err)
}
if got := st.downloads["1"].DisplayName; got != "Harold and the Purple Crayon (2024)" {
t.Errorf("display_name = %q (должен быть обновлён и без раздачи)", got)
}
if len(qb.renamed) != 0 {
t.Errorf("renamed = %+v, want none (раздачи нет)", qb.renamed)
}
}
func TestChooseCandidate_RefreshDisplayName_TorrentsLookupError(t *testing.T) {
// Листинг qBit упал — display_name обновлён, rename не зовём, команда цела.
w, st, qb := refreshSetup(t, store.MetadataCandidate{
Provider: "tvdb", ProviderID: "343558",
Title: store.NullString("Harold and the Purple Crayon"), Year: sql.NullInt64{Int64: 2024, Valid: true},
})
qb.torrentsErr = errors.New("qbit down")
if err := w.ChooseCandidate(context.Background(), "1", st.candidates[0].ID); err != nil {
t.Fatalf("ChooseCandidate must tolerate qbit lookup failure: %v", err)
}
if got := st.downloads["1"].DisplayName; got != "Harold and the Purple Crayon (2024)" {
t.Errorf("display_name = %q (должен быть обновлён несмотря на сбой листинга)", got)
}
if len(qb.renamed) != 0 {
t.Errorf("renamed = %+v, want none (листинг упал)", qb.renamed)
}
}
func TestAddManualSource_RefreshesDisplayName(t *testing.T) {
// Ручное добавление источника — тоже подтверждение матча → refresh.
// Кандидат без title/year → имя берётся из распознавания (fallback).
w, st, qb := refreshSetup(t, store.MetadataCandidate{Provider: "tvdb", ProviderID: "269613"})
if err := w.AddManualSource(context.Background(), "1", "tmdb", "603"); err != nil {
t.Fatalf("AddManualSource: %v", err)
}
if got := st.downloads["1"].DisplayName; got != "Догадка (2000)" {
t.Errorf("display_name = %q, want fallback «Догадка (2000)»", got)
}
if len(qb.renamed) != 1 || qb.renamed[0].name != "Догадка (2000)" {
t.Errorf("renamed = %+v", qb.renamed)
}
}
func TestRefreshDisplayName_NoRecognitionNoop(t *testing.T) {
// Нет распознавания — обновлять нечего: display_name не трогаем, rename нет.
st := newMemStore()
d := completedDownload("1")
d.State = store.StateDownloading
d.DisplayName = ""
st.put(d)
qb := &fakeQbt{torrents: []qbt.Torrent{{Hash: ihTest, Name: "raw"}}}
w := testWorkerWith(st, qb, &fakeRecognizer{}, nil)
if err := w.RefreshDisplayName(context.Background(), "1"); err != nil {
t.Fatalf("RefreshDisplayName: %v", err)
}
if got := st.downloads["1"].DisplayName; got != "" {
t.Errorf("display_name = %q, want empty (no-op)", got)
}
if len(qb.renamed) != 0 {
t.Errorf("renamed = %+v, want none", qb.renamed)
}
}
func TestChooseCandidate_RejectsForeign(t *testing.T) {
w, _ := reviewWithCandidate(t, store.MetadataCandidate{Provider: "tvdb", ProviderID: "1"})
if err := w.ChooseCandidate(context.Background(), "1", "999"); err == nil {
+5
View File
@@ -49,6 +49,9 @@ type Store interface {
// PromoteCatched атомарно переводит catched → downloading с записью имени
// (гард state='catched' — ре-валидация после сетевых вызовов вне блокировки).
PromoteCatched(ctx context.Context, id, displayName string) error
// SetDisplayName обновляет отображаемое имя постфактум (перелив канонического
// имени после распознавания) — без гарда состояния, FSM не двигает.
SetDisplayName(ctx context.Context, id, name string) error
SetSourceMissCount(ctx context.Context, id string, n int) error
SetSourceAddedAt(ctx context.Context, id string, t time.Time) error
// SetRetriedAt проставляет время ручного retry — сброс базиса отсчёта
@@ -94,6 +97,8 @@ type QBittorrent interface {
Add(ctx context.Context, ar qbt.AddRequest) error
Files(ctx context.Context, hash string) ([]qbt.File, error)
Delete(ctx context.Context, hashes []string, deleteFiles bool) error
// RenameTorrent задаёт имя уже добавленной раздачи по её ключу (Torrent.Hash).
RenameTorrent(ctx context.Context, hash, name string) error
}
// Recognizer — распознаватель (recognize.Recognizer).
+34 -6
View File
@@ -191,6 +191,15 @@ func (f *fakeStore) PromoteCatched(_ context.Context, id, displayName string) er
return nil
}
func (f *fakeStore) SetDisplayName(_ context.Context, id, name string) error {
d, ok := f.downloads[id]
if !ok {
return fmt.Errorf("download %s not found", id)
}
d.DisplayName = name
return nil
}
func (f *fakeStore) SetSourceMissCount(_ context.Context, id string, n int) error {
d, ok := f.downloads[id]
if !ok {
@@ -258,12 +267,20 @@ func (f *fakeStore) GetCandidate(_ context.Context, _ string) (*store.MetadataCa
func (f *fakeStore) SetCandidateChosen(_ context.Context, _, _ string) error { return nil }
type fakeQbt struct {
torrents []qbt.Torrent
added []qbt.AddRequest
addErr error
files []qbt.File
deleted [][]string // хеши каждого вызова Delete
deleteErr error
torrents []qbt.Torrent
torrentsErr error
added []qbt.AddRequest
addErr error
files []qbt.File
deleted [][]string // хеши каждого вызова Delete
deleteErr error
renamed []renameCall // каждый вызов RenameTorrent (hash, name)
renameErr error
}
type renameCall struct {
hash string
name string
}
// Torrents имитирует /torrents/info: пустая категория — все торренты, иначе
@@ -271,6 +288,9 @@ type fakeQbt struct {
// регрессии: раздача, усыновлённая по тегу, имеет чужую категорию и не должна
// теряться при поиске по infohash.
func (f *fakeQbt) Torrents(_ context.Context, category string) ([]qbt.Torrent, error) {
if f.torrentsErr != nil {
return nil, f.torrentsErr
}
if category == "" {
return f.torrents, nil
}
@@ -303,6 +323,14 @@ func (f *fakeQbt) Delete(_ context.Context, hashes []string, _ bool) error {
return nil
}
func (f *fakeQbt) RenameTorrent(_ context.Context, hash, name string) error {
if f.renameErr != nil {
return f.renameErr
}
f.renamed = append(f.renamed, renameCall{hash, name})
return nil
}
func newTestWorker(st *fakeStore, qb *fakeQbt) *Worker {
w := New(st, qb, nil, nil, Config{
Category: "jellybit",