feat(api): surface series for IMDb id searches

crates/arr-api/src/search.rs

@@ -224,14 +224,23 @@ pub async fn search(
     let results = search_tmdb(&tmdb, kind, input).await?;
     if matches!(kind, SearchInputKind::ImdbId) {
         for result in &results {
-            let TmdbResult::Movie(movie) = result else {
-                continue;
-            };
-            let tmdb_id = i64::from(movie.tmdb_id);
-            if let Some(found) = sqlx::query_as!(Movie, r#"SELECT id AS "id!: i64", tmdb_id AS "tmdb_id!: i64", title AS "title!: String", year, original_language, root_id AS "root_id!: i64", wanted AS "wanted!: bool", overrides AS "overrides!: serde_json::Value", state AS "state!: String", blocked AS "blocked!: bool", search_attempts AS "search_attempts!: i64", last_searched_at, (SELECT f.waiver FROM media_files f WHERE f.owner_kind = 'movie' AND f.owner_id = movies.id AND f.waiver IS NOT NULL ORDER BY f.id LIMIT 1) AS "waiver?: serde_json::Value" FROM movies WHERE tmdb_id = ?"#, tmdb_id)
-                .fetch_optional(database.pool()).await?
-            {
-                library.push(LibraryResult::Movie(found));
+            match result {
+                TmdbResult::Movie(movie) => {
+                    let tmdb_id = i64::from(movie.tmdb_id);
+                    if let Some(found) = sqlx::query_as!(Movie, r#"SELECT id AS "id!: i64", tmdb_id AS "tmdb_id!: i64", title AS "title!: String", year, original_language, root_id AS "root_id!: i64", wanted AS "wanted!: bool", overrides AS "overrides!: serde_json::Value", state AS "state!: String", blocked AS "blocked!: bool", search_attempts AS "search_attempts!: i64", last_searched_at, (SELECT f.waiver FROM media_files f WHERE f.owner_kind = 'movie' AND f.owner_id = movies.id AND f.waiver IS NOT NULL ORDER BY f.id LIMIT 1) AS "waiver?: serde_json::Value" FROM movies WHERE tmdb_id = ?"#, tmdb_id)
+                        .fetch_optional(database.pool()).await?
+                    {
+                        library.push(LibraryResult::Movie(found));
+                    }
+                }
+                TmdbResult::Series(series) => {
+                    let tmdb_id = i64::from(series.tmdb_id);
+                    if let Some(found) = sqlx::query_as!(LibrarySeries, r#"SELECT s.id AS "id!: i64", s.tmdb_id AS "tmdb_id!: i64", s.title AS "title!: String", s.year, s.original_language, s.root_id AS "root_id!: i64", s.blocked AS "blocked!: bool" FROM series s WHERE s.tmdb_id = ?"#, tmdb_id)
+                        .fetch_optional(database.pool()).await?
+                    {
+                        library.push(LibraryResult::Series(found));
+                    }
+                }
             }
         }
     }
@@ -316,14 +325,24 @@ async fn search_tmdb(
             }
             matches
         }
-        SearchInputKind::ImdbId => match tmdb.find_movie_by_imdb(input).await {
-            Ok(movies) => movies
+        SearchInputKind::ImdbId => {
+            let found = tmdb
+                .find_by_imdb(input)
+                .await
+                .map_err(|error| upstream_error(&error))?;
+            let mut results: Vec<TmdbResult> = found
+                .movies
                 .into_iter()
                 .map(|movie| TmdbResult::Movie(movie.into()))
-                .collect(),
-            Err(arr_meta::Error::NotFound { .. }) => Vec::new(),
-            Err(error) => return Err(upstream_error(&error)),
-        },
+                .collect();
+            results.extend(
+                found
+                    .series
+                    .into_iter()
+                    .map(|series| TmdbResult::Series(series.into())),
+            );
+            results
+        }
         SearchInputKind::Text => {
             let mut results: Vec<TmdbResult> = tmdb
                 .search_movies(input, None)
@@ -950,6 +969,52 @@ mod tests {
         assert_eq!(response["tmdb"][0]["title"], "1917");
     }
 
+    /// §9.2: a pasted `tt` id resolves a series exactly as a TMDB id does —
+    /// a hit on TMDB and, when tracked, in the library set too.
+    #[tokio::test]
+    async fn an_imdb_id_resolves_to_a_series() {
+        let tmdb = MockServer::start().await;
+        let prowlarr = MockServer::start().await;
+        Mock::given(method("GET"))
+            .and(path("/find/tt7614372"))
+            .and(query_param("external_source", "imdb_id"))
+            .respond_with(ResponseTemplate::new(200).set_body_json(serde_json::json!({
+                "movie_results": [],
+                "tv_results": [{
+                    "id": 82_728, "name": "Bluey", "original_name": "Bluey",
+                    "original_language": "en", "first_air_date": "2018-10-01"
+                }]
+            })))
+            .mount(&tmdb)
+            .await;
+        let (_dir, state, base) = application(&tmdb, &prowlarr).await;
+        let pool = state.database().expect("database").pool();
+        let root_id: i64 =
+            sqlx::query_scalar("SELECT id FROM roots WHERE kind = 'tv' AND audience = 'main'")
+                .fetch_one(pool)
+                .await
+                .expect("TV root");
+        sqlx::query(
+            "INSERT INTO series (tmdb_id, title, year, original_language, root_id) VALUES (82728, 'Bluey', 2018, 'en', ?)",
+        )
+        .bind(root_id)
+        .execute(pool)
+        .await
+        .expect("series");
+
+        let response: serde_json::Value = reqwest::get(format!("{base}/api/search?q=tt7614372"))
+            .await
+            .expect("search")
+            .json()
+            .await
+            .expect("json");
+        assert_eq!(response["kind"], "imdb_id");
+        assert_eq!(response["library"][0]["kind"], "series");
+        assert_eq!(response["library"][0]["tmdb_id"], 82_728);
+        assert_eq!(response["tmdb"][0]["kind"], "series");
+        assert_eq!(response["tmdb"][0]["title"], "Bluey");
+    }
+
     #[tokio::test]
     async fn manual_releases_are_classified_and_name_rejection_rules() {
         let tmdb = MockServer::start().await;