Merge pull request #2 from Strubbelkopp/main

Add optional argument to select a specific style

1 files changed, 13 insertions, 2 deletions

diff --git a/src/mikuart.rs b/src/mikuart.rs
index 5ba1849..44915b6 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,17 @@ pub fn get_miku_art() -> Vec<&'static str> {
         ],
     ];
 
-    let selected_style = rng.random_range(0..styles.len());
+    if style > 0 && style as usize >= styles.len() {
+        panic!(
+            "Style {} does not exist! Chooese one below {}",
+            style,
+            styles.len()
+        );
+    }
+    let selected_style = if style < 0 {
+        rng.random_range(0..styles.len())
+    } else {
+        style as usize
+    };
     styles[selected_style].clone()
 }