add punctuation

internal/app/app.go

@@ -32,14 +32,15 @@ type AppModel struct {
 
 // Number of words to use per each test
 const NUM_WORDS = 20
+const PUNC_CHANCE = 0.2
 
 func NewAppModel(init_state AppState) tea.Model {
 	text := ""
 	line_len := 0
 	lines := 0
 	for i := 0; i < NUM_WORDS; i++ {
 		// Pull random word from data
-		word := data.GetWord() + " "
+		word := data.GetWord(PUNC_CHANCE) + " "
 
 		// Manually insert newlines
 		if line_len+len(word) >= styles.APP_WIDTH-4 {

internal/data/data.go

@@ -11,13 +11,22 @@ import (
 var words_english_json []byte
 var words_english []string
 
+var punctuation string = ".,?!;:"
+
 func init() {
 	if err := json.Unmarshal(words_english_json, &words_english); err != nil {
 		log.Fatal("data/init:", err)
 	}
 }
 
 // Return random word from dataset
-func GetWord() string {
-	return words_english[rand.Intn(len(words_english))]
+func GetWord(punctuation_chance float64) string {
+	word := words_english[rand.Intn(len(words_english))]
+
+	if rand.Float64() < punctuation_chance {
+		p := punctuation[rand.Intn(len(punctuation))]
+		word += string(p)
+	}
+
+	return word
 }