From 8a2f1145f254b2dde12370e15237d3cf56e8a859 Mon Sep 17 00:00:00 2001 From: Fuwn Date: Thu, 6 May 2021 19:25:08 -0700 Subject: fix(random): no range in one image languages --- src/routes.rs | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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(), -- cgit v1.2.3