scrunkly

markov.js

@@ -253,7 +253,7 @@ const catgirlTable = generateMarkov(catgirlNonsense.split('\n'), 2)
 const keysmashTable = generateMarkov(keysmash, 1);
 const scrunklyTable = generateMarkov(scrunks, 2);
 
-console.log(JSON.stringify(magic(catgirlTable)))
+console.log(JSON.stringify(magic(scrunklyTable)))
 
 // console.log(generateFromArray(catgirlTable, 100, 'ny'));
 //

src/lib.rs

@@ -5,7 +5,7 @@ pub fn add(left: usize, right: usize) -> usize {
 #[cfg(test)]
 mod tests {
 
-    use crate::keysmash;
+    use crate::{keysmash, scrunkly};
 
     use super::catgirl_nonsense;
 
@@ -27,16 +27,31 @@ mod tests {
     fn keysmash_test() {
         // Use a stable-algorithm RNG with fixed seed
         let mut rng = rand_pcg::Pcg32::new(0xcafef00dd15ea5e5, 0xa02bdbf7bb3c0a7);
-        let mut state_machine = keysmash::StateMachine::A;
+        let (mut state_machine, result) = keysmash::StateMachine::new_random(&mut rng);
-        let mut result = String::from("a");
+        let mut result = String::from(result);
         for _ in 0..100 {
             let (new_state, generated) = state_machine.generate(&mut rng);
             result.push(generated);
             state_machine = new_state;
         }
-        assert_eq!(&result, "ajhfhiurgjfgajhnghgadfghkfghiurgjeghnhgjalkjfhgnhrgjhnhiuradfdbahrgbafhg;djkafgjhjrajfjdfghfhdfgajgad");
+        assert_eq!(&result, "ra;ajgjdhdfkgldskadfghkfghiurgjeghnhgjalkjfhgnhrgjhnhiuradfdbahrgbafhg;djkafgjhjrajfjdfghfhdfgajgadka");
+    }
+
+    #[test]
+    fn scrunkly_test() {
+        // Use a stable-algorithm RNG with fixed seed
+        let mut rng = rand_pcg::Pcg32::new(0xcafef00dd15ea5e5, 0xa02bdbf7bb3c0a7);
+        let mut state_machine = scrunkly::StateMachine::Aw;
+        let mut result = String::from("aw");
+        for _ in 0..100 {
+            let (new_state, generated) = state_machine.generate(&mut rng);
+            result.push(generated);
+            state_machine = new_state;
+        }
+        assert_eq!(&result, "aw sproinkly,,,,scringle limtle cutest the like scrinkly double scrimblookimty... ittle adornale boink");
     }
 }
 
 mod catgirl_nonsense;
 mod keysmash;
+mod scrunkly;

uwurandom-proc-macros/src/lib.rs

@@ -12,8 +12,15 @@ pub fn gen_fsm(item: TokenStream) -> TokenStream {
     let input: Vec<MarkovArr> = serde_json::from_str(&format!("[{}]", item)).unwrap();
     let mut match_arms = quote!();
     let mut variants = quote!();
-    for state in input.iter() {
+    let mut random_selection_match_arms = quote!();
+    for (idx, state) in input.iter().enumerate() {
         let name = to_ident(&state.name);
+        let name_lit = &state.name;
+        let idx = idx as u32;
+        random_selection_match_arms = quote!(
+            #random_selection_match_arms
+            #idx => (Self::#name, #name_lit),
+        );
         variants = quote!(
             #variants
             #name,
@@ -47,6 +54,7 @@ pub fn gen_fsm(item: TokenStream) -> TokenStream {
             },
         );
     }
+    let variant_count = input.len() as u32;
     quote!(
         #[derive(Debug, Clone, Copy)]
         pub enum StateMachine {
@@ -58,11 +66,26 @@ pub fn gen_fsm(item: TokenStream) -> TokenStream {
                     #match_arms
                 }
             }
+            pub fn new_random(mut rng: impl ::rand_core::RngCore) -> (Self, &'static str) {
+                match rng.next_u32() % #variant_count {
+                    #random_selection_match_arms
+                    _ => unreachable!()
+                }
+            }
         }
     )
     .into()
 }
 
 fn to_ident(name: &str) -> Ident {
-    Ident::new(&name.replace(';', " semicolon").to_case(Case::Pascal), Span::call_site().into())
+    Ident::new(
+        &name
+            .replace(';', " semicolon") // Sanitize ident-unsafe characters
+            .replace('!', " exclamation")
+            .replace(',', " comma")
+            .replace('.', " period") // I'm not calling it a full stop
+            .replace(' ', " space")
+            .to_case(Case::Pascal),
+        Span::call_site().into(),
+    )
 }