diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/routes.rs | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/routes.rs b/src/routes.rs index 8c831ac..5ec12d6 100644 --- a/src/routes.rs +++ b/src/routes.rs @@ -70,7 +70,11 @@ pub async fn random() -> HttpResponse { let random_language = &filtered_languages[thread_rng().gen_range(0..filtered_languages.len() - 1)]; let filtered_images = filter_images_by_language(random_language).await; - let random_image = &filtered_images[thread_rng().gen_range(0..filtered_images.len() - 1)]; + let random_image = if filtered_images.len() == 1 { + &filtered_images[0] + } else { + &filtered_images[thread_rng().gen_range(0..filtered_images.len() - 1)] + }; HttpResponse::Ok().json(SenpyRandom { language: random_language.clone(), |