Имя: восстановление 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:
@@ -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.
|
||||
|
||||
@@ -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 {
|
||||
|
||||
@@ -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).
|
||||
|
||||
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user