Improve search

Closes #47

Author: lpil

src/packages/text_search.gleam

@@ -27,6 +27,8 @@ pub fn insert(
     False ->
       name
       |> string.append(" ")
+      |> string.append(string.replace(name, "_", " "))
+      |> string.append(" ")
       |> string.append(description)
       |> stem_words
       |> list.try_each(fn(word) { ethos.insert(index.table, word, name) })
@@ -48,6 +50,7 @@ pub fn lookup(
   phrase: String,
 ) -> Result(List(String), Error) {
   stem_words(phrase)
+  |> list.flat_map(expand_search_term)
   |> list.try_map(ethos.get(index.table, _))
   |> result.map(fn(names) {
     names
@@ -62,6 +65,16 @@ pub fn lookup(
   |> result.replace_error(error.EtsTableError)
 }
 
+/// Some words have common misspellings or associated words so we add those to
+/// the search to get all appropriate results.
+fn expand_search_term(term: String) -> List(String) {
+  case term {
+    "postgres" | "postgresql" -> ["postgres", "postgresql"]
+    "regex" | "regexp" -> ["regex", "regexp"]
+    term -> [term]
+  }
+}
+
 fn remove(index: TextSearchIndex, name: String) -> Result(Nil, Error) {
   ethos.delete_value(index.table, name)
   |> result.replace_error(error.EtsTableError)

test/packages/text_search_test.gleam

@@ -78,3 +78,23 @@ pub fn ignored_test() {
   |> should.be_ok
   |> should.equal(["clean_bson"])
 }
+
+pub fn word_in_title_test() {
+  let index = text_search.new()
+  let assert Ok(_) = text_search.insert(index, "gleam_regexp", "")
+
+  text_search.lookup(index, "regexp")
+  |> should.be_ok
+  |> should.equal(["gleam_regexp"])
+}
+
+// regex also searches for regexp
+pub fn extra_regex_test() {
+  let index = text_search.new()
+  let assert Ok(_) = text_search.insert(index, "gleam_regexp", "")
+  let assert Ok(_) = text_search.insert(index, "third_party_regex", "")
+
+  text_search.lookup(index, "regex")
+  |> should.be_ok
+  |> should.equal(["gleam_regexp", "third_party_regex"])
+}