examples: fix word_counter

examples/word_counter/README.md

@@ -1,24 +1,25 @@
 ```
 usage: word_counter [text_file]
 using cinderella.txt
+a => 25
 able => 2
-afterwards => 1
 after => 1
-against => 2
+afterwards => 1
 again => 10
-allowed => 2
+against => 2
-allow => 1
 all => 12
+allow => 1
+allowed => 2
 along => 1
 also => 2
 always => 2
+an => 4
 and => 140
 anew => 1
 anger => 1
 another => 2
 answered => 1
-anyone => 2
 any => 1
-an => 4
+anyone => 2
 ...
 ```

examples/word_counter/word_counter.v

@@ -18,12 +18,8 @@ fn main() {
 		return
 	}
 	mut m := map[string]int
-	for word in contents.to_lower().split(' ') {
+	for word in extract_words(contents) {
-		key := filter_word(word)
+		m[word] = m[word] + 1 // TODO m[key]++
-		if key == '' {
-			continue
-		}
-		m[key] = m[key] + 1// TODO m[key]++
 	}
 	// Sort the keys
 	mut keys := m.keys() 
@@ -35,6 +31,30 @@ fn main() {
 	}
 }
 
+// Creates an array of words from a given string
+fn extract_words(contents string) []string {
+	mut splitted := []string
+	for space_splitted in contents.to_lower().split(' ') {
+		if space_splitted.contains('\n') {
+			splitted << space_splitted.split('\n')
+		}
+		else {
+			splitted << space_splitted
+		}
+	}
+
+	mut results := []string
+	for s in splitted {
+		result := filter_word(s)
+		if result == '' {
+			continue
+		}
+		results << result
+	}
+
+	return results
+}
+
 // Removes punctuation
 fn filter_word(word string) string {
 	if word == '' || word == ' ' {