Fix ranking bug affecting compound words

CloudCannon · Dec 16, 2024 · 477a142 · 477a142
1 parent 1fdbac3
commit 477a142
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pagefind/integration_tests/weighting/compound-words-sum-to-a-full-weight.toolproof.yml b/pagefind/integration_tests/weighting/compound-words-sum-to-a-full-weight.toolproof.yml
@@ -27,4 +27,4 @@ steps:
   - step: In my browser, I evaluate {js}
     js: |-
       let val = await toolproof.querySelector("p");
-      toolproof.assert_eq(val.innerHTML, `weight:1/bal:512.14/loc:4`);
+      toolproof.assert_eq(val.innerHTML, `weight:1/bal:1024.29/loc:4`);
diff --git a/pagefind_web/src/search.rs b/pagefind_web/src/search.rs
@@ -336,10 +336,12 @@ impl SearchIndex {
                             // If the new word is weighted _lower_ than the working word,
                             // we want to use the lower value. (Lowest weight wins)
                             working_word.weight = next_word.weight;
+                            working_word.length_bonus = next_word.length_bonus;
                         } else if next_word.weight == working_word.weight {
                             // If the new word is weighted the same,
                             // we want to combine them to boost matching both halves of a compound word
                             working_word.weight += next_word.weight;
+                            working_word.length_bonus += next_word.length_bonus;
                         }
                         // We don't want to do anything if the new word is weighted higher
                         // (Lowest weight wins)