add arbitrary corpus support

Cargo.toml

@@ -12,3 +12,4 @@ crossterm = "0.27.0"
 ratatui = { version = "0.26.0", features = ["all-widgets"] }
 unicode-segmentation = "1.11.0"
 unicode-normalization = "0.1.22"
+rand = "0.8.5"

src/corpus.rs

@@ -0,0 +1,57 @@
+use rand::seq::IteratorRandom;
+pub enum SamplingMethod {
+    Top,
+    Random,
+}
+
+pub struct Corpus {
+    pub words: Vec<&'static str>,
+}
+
+impl Corpus {
+    pub fn sample(&self, n: usize, method: SamplingMethod) -> Corpus {
+        Corpus {
+            words: match method {
+                SamplingMethod::Top => self.words.iter().take(n).cloned().collect(),
+                SamplingMethod::Random => {
+                    let mut rng = rand::thread_rng();
+                    self.words.iter().cloned().choose_multiple(&mut rng, n)
+                }
+            },
+        }
+    }
+
+    pub fn english_200_words() -> Corpus {
+        Corpus {
+            words: vec![
+                "the", "be", "of", "and", "a", "to", "in", "he", "have", "it", "that", "for",
+                "they", "I", "with", "as", "not", "on", "she", "at", "by", "this", "we", "you",
+                "do", "but", "from", "or", "which", "one", "would", "all", "will", "there", "say",
+                "who", "make", "when", "can", "more", "if", "no", "man", "out", "other", "so",
+                "what", "time", "up", "go", "about", "than", "into", "could", "state", "only",
+                "new", "year", "some", "take", "come", "these", "know", "see", "use", "get",
+                "like", "then", "first", "any", "work", "now", "may", "such", "give", "over",
+                "think", "most", "even", "find", "day", "also", "after", "way", "many", "must",
+                "look", "before", "great", "back", "through", "long", "where", "much", "should",
+                "well", "people", "down", "own", "just", "because", "good", "each", "those",
+                "feel", "seem", "how", "high", "too", "place", "little", "world", "very", "still",
+                "nation", "hand", "old", "life", "tell", "write", "become", "here", "show",
+                "house", "both", "between", "need", "mean", "call", "develop", "under", "last",
+                "right", "move", "thing", "general", "school", "never", "same", "another", "begin",
+                "while", "number", "part", "turn", "real", "leave", "might", "want", "point",
+                "form", "off", "child", "few", "small", "since", "against", "ask", "late", "home",
+                "interest", "large", "person", "end", "open", "public", "follow", "during",
+                "present", "without", "again", "hold", "govern", "around", "possible", "head",
+                "consider", "word", "program", "problem", "however", "lead", "system", "set",
+                "order", "eye", "plan", "run", "keep", "face", "fact", "group", "play", "stand",
+                "increase", "early", "course", "change", "help", "line",
+            ],
+        }
+    }
+}
+
+impl std::fmt::Display for Corpus {
+    fn fmt(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(formatter, "{}", self.words.join(" "))
+    }
+}

src/main.rs

@@ -1,3 +1,5 @@
+mod corpus;
+
 pub mod action;
 pub mod model;
 pub mod terminal;

src/model.rs

@@ -1,4 +1,4 @@
-use crate::action::Action;
+use crate::{action::Action, corpus::Corpus, corpus::SamplingMethod};
 
 use unicode_normalization::UnicodeNormalization;
 use unicode_segmentation::UnicodeSegmentation;
@@ -31,7 +31,13 @@ impl Model {
     }
 
     fn next_test(&mut self) {
-        self.current_test = Test::new("What the fuck did you just fucking say about me, you little bitch? I'll have you know I graduated top of my class in the Navy Seals, and I've been involved in numerous secret raids on Al-Quaeda, and I have over 300 confirmed kills. I am trained in gorilla warfare and I'm the top sniper in the entire US armed forces. You are nothing to me but just another target. I will wipe you the fuck out with precision the likes of which has never been seen before on this Earth, mark my fucking words. You think you can get away with saying that shit to me over the Internet? Think again, fucker. As we speak I am contacting my secret network of spies across the USA and your IP is being traced right now so you better prepare for the storm, maggot. The storm that wipes out the pathetic little thing you call your life. You're fucking dead, kid. I can be anywhere, anytime, and I can kill you in over seven hundred ways, and that's just with my bare hands. Not only am I extensively trained in unarmed combat, but I have access to the entire arsenal of the United States Marine Corps and I will use it to its full extent to wipe your miserable ass off the face of the continent, you little shit. If only you could have known what unholy retribution your little 'clever' comment was about to bring down upon you, maybe you would have held your fucking tongue. But you couldn't, you didn't, and now you're paying the price, you goddamn idiot. I will shit fury all over you and you will drown in it. You're fucking dead, kiddo");
+        self.current_test = Test::new(
+            self.config
+                .corpus
+                .sample(10, SamplingMethod::Random)
+                .to_string()
+                .as_str(),
+        );
     }
 }
 
@@ -41,21 +47,23 @@ impl Default for Model {
             frame_statistics: FrameStatistics::default(),
             config: Config::default(),
             should_quit: false,
-            current_test: Test::new("Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum."),
+            current_test: Test::new(""),
         }
     }
 }
 
 pub struct Config {
     pub show_frame_statistics: bool,
     pub show_live_typing_statistics: bool,
+    pub corpus: Corpus,
 }
 
 impl Default for Config {
     fn default() -> Self {
         Config {
             show_frame_statistics: true,
             show_live_typing_statistics: true,
+            corpus: Corpus::english_200_words(),
         }
     }
 }