Fix query by IMDB ID (#16)

mgdigital · Jun 19, 2023 · 54c5d80 · 54c5d80
1 parent 26facd8
commit 54c5d80
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sqlitedb/query.go b/sqlitedb/query.go
@@ -41,7 +41,11 @@ func Query(db *sql.DB, query *SearchQuery) ([]Record, error) {
 		expressions = append(expressions, goqu.C("cat").In(query.Categories))
 	}
 	if query.ImdbId.Valid {
-		expressions = append(expressions, goqu.C("imdb").Eq(query.ImdbId.String))
+		imdbId := query.ImdbId.String
+		if !strings.HasPrefix(imdbId, "tt") {
+			imdbId = "tt" + imdbId
+		}
+		expressions = append(expressions, goqu.C("imdb").Eq(imdbId))
 	}
 	if query.Season.Valid && query.Episode.Valid {
 		expressions = append(expressions, goqu.C("title").Like("%.S"+fmt.Sprintf("%02d", query.Season.Int64)+"E"+fmt.Sprintf("%02d", query.Episode.Int64)+".%"))