From 5dda54ecd49a43e79a52ded2a0fc0f1017689197 Mon Sep 17 00:00:00 2001
From: Strubbelkopp <nils130902@gmail.com>
Date: Fri, 25 Jul 2025 12:20:17 +0200
Subject: Add optional argument to select a specific style

---
 src/mikuart.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'src/mikuart.rs')

diff --git a/src/mikuart.rs b/src/mikuart.rs
index 5ba1849..6d7b5c9 100644
--- a/src/mikuart.rs
+++ b/src/mikuart.rs
@@ -3,7 +3,7 @@
 
 use rand::Rng;
 
-pub fn get_miku_art() -> Vec<&'static str> {
+pub fn get_miku_art(style: i32) -> Vec<&'static str> {
     let mut rng = rand::rng();
     let styles = vec![
         //MARK: small art
@@ -504,6 +504,6 @@ pub fn get_miku_art() -> Vec<&'static str> {
         ],
     ];
 
-    let selected_style = rng.random_range(0..styles.len());
+    let selected_style = if style < 0 { rng.random_range(0..styles.len()) } else { style as usize };
     styles[selected_style].clone()
 }
-- 
cgit v1.3-2-g0d8e