package httpapi import ( "context" "errors" "net/http" "net/url" "strconv" "strings" "git.vakhrushev.me/av/jellybit/internal/ident" "git.vakhrushev.me/av/jellybit/internal/store" "git.vakhrushev.me/av/jellybit/internal/worker" ) // Reviewer — операции ревью и раскладки (worker.Worker). type Reviewer interface { ReviewData(ctx context.Context, id string) (*worker.ReviewData, error) Apply(ctx context.Context, id string) error Refine(ctx context.Context, id string, hint string) error IgnoreFile(ctx context.Context, id string, src string) error Defer(ctx context.Context, id string) error Undo(ctx context.Context, id string) error Relink(ctx context.Context, id string) error Rerecognize(ctx context.Context, id string) error ChooseCandidate(ctx context.Context, id, candidateID string) error SetProviderID(ctx context.Context, id string, provider, providerID string) error AddManualSource(ctx context.Context, id, provider, providerID string) error ClearProvider(ctx context.Context, id string) error } // --- Представление страницы ревью --- type reviewView struct { ID string Source string Context string State string Error string // из ?err= StateError string // error_msg загрузки (напр. причина коллизии) MediaType string IsSeries bool Title string OriginalTitle string Year int SeasonSummary string // сводка сезонов для сериала (пусто для фильма) Provider string ProviderID string MatchURL string // ссылка на подтверждённую запись метабазы (пусто — текстом) Confidence string Reasons []string Hints []string Files []fileRow HasPlan bool HasLinks bool // есть хотя бы один целевой путь → можно применять NoBase bool // выбрано «без базы» Sources []sourceView // единый список источников совпадения BlockError string // ошибка выбора внутри блока (htmx); не путать с Error (?err=) } // sourceView — строка единого списка источников на экране ревью: нейронка или // кандидат базы. Инфо и предпросмотр раскладки показываются для активного // источника из верхнеуровневых полей reviewView, поэтому per-source превью // строка не несёт. type sourceView struct { Kind string // "neural" | "candidate" CandidateID string Provider string ProviderID string Title string Year int MatchURL string Active bool } func (s *server) handleReview(w http.ResponseWriter, r *http.Request) { id, err := pathID(r) if err != nil { // Невалидный id = несуществующая сущность; в БД не ходим. http.Error(w, "задача не найдена", http.StatusNotFound) return } rd, err := s.deps.Reviewer.ReviewData(r.Context(), id) if err != nil { if errors.Is(err, store.ErrNotFound) { http.Error(w, "задача не найдена", http.StatusNotFound) return } s.deps.Logger.Error("review data", "id", id, "error", err) http.Error(w, "внутренняя ошибка", http.StatusInternalServerError) return } s.render(w, "review.html", buildReviewView(id, rd, r.URL.Query().Get("err"))) } // buildReviewView собирает представление страницы ревью из доменных данных. // Общий для полной страницы (handleReview) и htmx-свопа блока источника // (reviewBlockAction); errMsg — верхний баннер из ?err= (пусто на htmx-пути, // там ошибка идёт в BlockError). func buildReviewView(id string, rd *worker.ReviewData, errMsg string) reviewView { view := reviewView{ ID: id, Source: shorten(rd.Download.SourceRef, 80), Context: rd.Download.Context, State: string(rd.Download.State), Error: errMsg, StateError: rd.Download.ErrorMsg.String, Hints: rd.Hints, } if rec := rd.Recognition; rec != nil { view.MediaType = string(rd.Plan.Type) view.IsSeries = rd.Plan.Type == "series" view.Title = rd.Plan.Title view.OriginalTitle = rd.Plan.OriginalTitle view.Year = rd.Plan.Year if view.IsSeries { view.SeasonSummary = seasonSummary(rd.Plan) } view.Reasons = rec.ReasonList() switch rd.Provider { case "", "none": view.NoBase = rd.Provider == "none" default: view.Provider = rd.Provider view.ProviderID = rd.ProviderID view.MatchURL = matchURL(rd, string(rd.Plan.Type)) } if rec.Confidence.Valid { view.Confidence = strconv.FormatFloat(rec.Confidence.Float64, 'f', 2, 64) } view.Files = buildFileRows(rd.Plan, rd.Preview) view.HasPlan = len(rd.Plan.Files) > 0 view.HasLinks = len(rd.Preview) > 0 for _, src := range rd.Sources { sv := sourceView{ Kind: string(src.Kind), CandidateID: src.CandidateID, Provider: src.Provider, ProviderID: src.ProviderID, Title: src.Title, Year: src.Year, Active: src.Active, } if src.Kind == worker.SourceCandidate { sv.MatchURL = sourceMatchURL(src) } view.Sources = append(view.Sources, sv) } } return view } // --- Действия ревью (POST → redirect) --- func (s *server) handleApply(w http.ResponseWriter, r *http.Request) { id, err := pathID(r) if err != nil { redirectErr(w, r, "некорректный id") return } if err := s.deps.Reviewer.Apply(r.Context(), id); err != nil { // Тонкий транспорт: ошибку воркера переводим в ответ, не логируя // повторно (доменный слой уже залогировал реальный сбой). redirectReview(w, r, id, userErr(r, err, id)) return } http.Redirect(w, r, "/", http.StatusSeeOther) } func (s *server) handleRefine(w http.ResponseWriter, r *http.Request) { s.reviewAction(w, r, func(ctx context.Context, id string) error { _ = r.ParseForm() return s.deps.Reviewer.Refine(ctx, id, r.PostForm.Get("hint")) }) } func (s *server) handleRerecognize(w http.ResponseWriter, r *http.Request) { s.reviewAction(w, r, func(ctx context.Context, id string) error { return s.deps.Reviewer.Rerecognize(ctx, id) }) } func (s *server) handleIgnore(w http.ResponseWriter, r *http.Request) { s.reviewAction(w, r, func(ctx context.Context, id string) error { _ = r.ParseForm() return s.deps.Reviewer.IgnoreFile(ctx, id, r.PostForm.Get("src")) }) } func (s *server) handleChooseCandidate(w http.ResponseWriter, r *http.Request) { s.reviewBlockAction(w, r, func(ctx context.Context, id string) error { _ = r.ParseForm() // Входная граница: id кандидата из формы валидируется как ULID. candidateID, err := ident.Parse(r.PostForm.Get("candidate_id")) if err != nil { return errInvalidCandidate } return s.deps.Reviewer.ChooseCandidate(ctx, id, candidateID) }) } func (s *server) handleSetProvider(w http.ResponseWriter, r *http.Request) { s.reviewAction(w, r, func(ctx context.Context, id string) error { _ = r.ParseForm() return s.deps.Reviewer.SetProviderID(ctx, id, r.PostForm.Get("provider"), r.PostForm.Get("provider_id")) }) } func (s *server) handleNoBase(w http.ResponseWriter, r *http.Request) { s.reviewBlockAction(w, r, func(ctx context.Context, id string) error { return s.deps.Reviewer.ClearProvider(ctx, id) }) } // handleAddSource добавляет источник вручную по id или URL записи метабазы и // выбирает его. Разбор ввода — на входной границе транспорта. func (s *server) handleAddSource(w http.ResponseWriter, r *http.Request) { s.reviewBlockAction(w, r, func(ctx context.Context, id string) error { _ = r.ParseForm() provider, providerID, err := parseManualSource(r.PostForm.Get("provider"), r.PostForm.Get("provider_id")) if err != nil { return err } return s.deps.Reviewer.AddManualSource(ctx, id, provider, providerID) }) } var errInvalidCandidate = errors.New("некорректный id кандидата") var errManualSource = errors.New("не удалось разобрать id или URL записи (для TVDB — числовой id)") // parseManualSource разбирает ручной ввод источника: либо id (с выбранным в // форме провайдером), либо URL записи метабазы (провайдер и id из URL). func parseManualSource(provider, raw string) (string, string, error) { raw = strings.TrimSpace(raw) if raw == "" { return "", "", errManualSource } if looksLikeURL(raw) { // URL без схемы (`themoviedb.org/tv/1`) url.Parse кладёт в Path, а не в // Host → достраиваем схему, иначе валидный копипаст отвергнется. u := raw if !strings.Contains(u, "://") { u = "https://" + u } p, id, ok := parseProviderURL(u) if !ok { return "", "", errManualSource } return p, id, nil } return strings.ToLower(strings.TrimSpace(provider)), raw, nil } func looksLikeURL(s string) bool { return strings.Contains(s, "://") || strings.HasPrefix(s, "www.") || strings.Contains(s, ".org/") || strings.Contains(s, ".com/") } // parseProviderURL — обратная к providerURL: URL записи → (provider, id). // TMDB/IMDb извлекаются из URL; TVDB — только dereferrer с числовым id (URL // сайта thetvdb.com/series/{slug} числового id не содержит → не распознаём). func parseProviderURL(raw string) (provider, id string, ok bool) { u, err := url.Parse(raw) if err != nil || u.Host == "" { return "", "", false } host := strings.ToLower(u.Host) parts := strings.Split(strings.Trim(u.Path, "/"), "/") switch { case strings.Contains(host, "themoviedb.org"): if len(parts) >= 2 && (parts[0] == "movie" || parts[0] == "tv") { if d := leadingDigits(parts[1]); d != "" { return "tmdb", d, true } } case strings.Contains(host, "imdb.com"): if len(parts) >= 2 && parts[0] == "title" && strings.HasPrefix(parts[1], "tt") { return "imdb", parts[1], true } case strings.Contains(host, "thetvdb.com"): if len(parts) >= 3 && parts[0] == "dereferrer" { if d := leadingDigits(parts[2]); d != "" { return "tvdb", d, true } } } return "", "", false } // leadingDigits возвращает ведущие цифры строки (TMDB-URL вида // `/movie/60622-fargo` → «60622»). func leadingDigits(s string) string { i := 0 for i < len(s) && s[i] >= '0' && s[i] <= '9' { i++ } return s[:i] } // sourceMatchURL — ссылка на запись источника-кандидата: URL кандидата, если // есть, иначе канонический URL из provider/id (обратный порядок к matchURL). func sourceMatchURL(src worker.SourceOption) string { if src.URL != "" { return src.URL } return providerURL(src.Provider, src.ProviderID, src.Type) } func (s *server) handleDefer(w http.ResponseWriter, r *http.Request) { id, err := pathID(r) if err != nil { redirectErr(w, r, "некорректный id") return } if err := s.deps.Reviewer.Defer(r.Context(), id); err != nil { redirectReview(w, r, id, userErr(r, err, id)) return } http.Redirect(w, r, "/", http.StatusSeeOther) } func (s *server) handleUndo(w http.ResponseWriter, r *http.Request) { id, err := pathID(r) if err != nil { redirectErr(w, r, "некорректный id") return } if err := s.deps.Reviewer.Undo(r.Context(), id); err != nil { redirectErr(w, r, userErr(r, err, id)) return } http.Redirect(w, r, "/", http.StatusSeeOther) } // handleRelink повторно привязывает откатанную задачу: перезапускает // распознавание, задача пройдёт recognizing → review для подтверждения. func (s *server) handleRelink(w http.ResponseWriter, r *http.Request) { id, err := pathID(r) if err != nil { redirectErr(w, r, "некорректный id") return } if err := s.deps.Reviewer.Relink(r.Context(), id); err != nil { redirectErr(w, r, userErr(r, err, id)) return } http.Redirect(w, r, "/", http.StatusSeeOther) } // reviewAction — общий помощник: выполнить действие и вернуться на страницу // ревью (с ошибкой в ?err при неудаче). func (s *server) reviewAction(w http.ResponseWriter, r *http.Request, fn func(context.Context, string) error) { id, err := pathID(r) if err != nil { redirectErr(w, r, "некорректный id") return } if err := fn(r.Context(), id); err != nil { // Тонкий транспорт: ошибку переводим в ?err= на странице ревью, не // логируя повторно (доменный слой/валидация — не дело транспорта). redirectReview(w, r, id, userErr(r, err, id)) return } redirectReview(w, r, id, "") } // isHTMX — запрос инициирован htmx (ждёт партиал, а не полную страницу). func isHTMX(r *http.Request) bool { return r.Header.Get("HX-Request") == "true" } // reviewBlockAction — помощник для действий выбора источника: выполнить // операцию и вернуть свежий блок источника. На htmx-запрос перечитывает // состояние и рендерит партиал `review_source_block` (ошибку кладёт в // BlockError, активный источник не меняется); без htmx деградирует до // PRG-редиректа, как reviewAction. func (s *server) reviewBlockAction(w http.ResponseWriter, r *http.Request, fn func(context.Context, string) error) { id, err := pathID(r) if err != nil { redirectErr(w, r, "некорректный id") return } actionErr := fn(r.Context(), id) if !isHTMX(r) { msg := "" if actionErr != nil { msg = userErr(r, actionErr, id) } redirectReview(w, r, id, msg) return } // htmx: перечитываем состояние (уже с новым активным источником при успехе) // и рендерим свежий партиал блока. rd, err := s.deps.Reviewer.ReviewData(r.Context(), id) if err != nil { s.deps.Logger.Error("review data", "id", id, "error", err) http.Error(w, "внутренняя ошибка", http.StatusInternalServerError) return } view := buildReviewView(id, rd, "") if actionErr != nil { view.BlockError = userErr(r, actionErr, id) } s.render(w, "review_source_block", view) } // matchURL выбирает ссылку на подтверждённую запись метабазы. Приоритет — URL // выбранного кандидата, но только если его provider+id совпадают с эффективными // (человек мог выбрать кандидата, затем вручную переопределить id — тогда // кандидат указывает на другую запись). Иначе строим канонический URL; если не // удаётся — возвращаем пусто (транспорт покажет матч текстом). func matchURL(rd *worker.ReviewData, mediaType string) string { for _, c := range rd.Candidates { if c.Chosen && c.Provider == rd.Provider && c.ProviderID == rd.ProviderID && c.URL.Valid && c.URL.String != "" { return c.URL.String } } return providerURL(rd.Provider, rd.ProviderID, mediaType) } // providerURL строит канонический URL записи метабазы с учётом типа медиа. // Пустой id или неизвестный провайдер → пусто. func providerURL(provider, id, mediaType string) string { if id == "" { return "" } switch provider { case "tmdb": kind := "movie" if mediaType == "series" { kind = "tv" } return "https://www.themoviedb.org/" + kind + "/" + id case "tvdb": kind := "series" if mediaType == "movie" { kind = "movie" } return "https://www.thetvdb.com/dereferrer/" + kind + "/" + id case "imdb": return "https://www.imdb.com/title/" + id default: return "" } } func redirectReview(w http.ResponseWriter, r *http.Request, id string, msg string) { u := "/review/" + id if msg != "" { u += "?err=" + url.QueryEscape(msg) } http.Redirect(w, r, u, http.StatusSeeOther) }