Merge pull request #61 from jrswab/f-search-with-numbers

Allow Numbers in Search

Author: jrswab

trie/trie.go

@@ -159,13 +159,13 @@ func removeNonAlpha(fileName string) string {
 	s = strings.ReplaceAll(s, ".org", "")
 
 	// Step 3: Keep only letter characters
-	letters := []rune{}
+	uniRune := []rune{}
 	for _, r := range s {
-		if unicode.IsLetter(r) {
-			letters = append(letters, r)
+		if unicode.IsLetter(r) || unicode.IsDigit(r) {
+			uniRune = append(uniRune, r)
 		}
 	}
-	s = string(letters)
+	s = string(uniRune)
 
 	// NFC = Normalization Form Canonical Composition
 	// This ensures that characters like "é" are treated consistently

trie/trie_test.go

@@ -26,7 +26,7 @@ func TestInit(t *testing.T) {
 		"basic files without aliases": {
 			files: map[string]string{
 				"test-one.md": "# Test 1\nSome content",
-				"test two.md": "# Test 2\nMore content",
+				"test 2.md":   "# Test 2\nMore content",
 			},
 			wantErr: false,
 			checkFn: func(t *testing.T, tr *trie.Trie) {
@@ -35,9 +35,9 @@ func TestInit(t *testing.T) {
 					t.Errorf("Expected to find test-one.md in trie, got %v", results)
 				}
 
-				results = tr.Search("test2")
-				if !slices.Contains(results, "test-one.md") {
-					t.Errorf("Expected to find \"test two.md\" in trie, got %v", results)
+				results = tr.Search("test 2")
+				if !slices.Contains(results, "test 2.md") {
+					t.Errorf("Expected to find \"test 2.md\" in trie, got %v", results)
 				}
 			},
 		},