Fix search tags in split and grouped display modes (#1412)

1 files changed, 10 insertions, 1 deletions

diff --git a/ext/js/display/display-generator.js b/ext/js/display/display-generator.js
index 1b3908fe..d02e2d9c 100644
--- a/ext/js/display/display-generator.js
+++ b/ext/js/display/display-generator.js
@@ -70,7 +70,8 @@ class DisplayGenerator {
 
         const uniqueExpressions = new Set();
         const uniqueReadings = new Set();
-        for (const {expression, reading} of expressions) {
+        for (let {expression, reading} of expressions) {
+            if (reading.length === 0) { reading = expression; }
             uniqueExpressions.add(expression);
             uniqueReadings.add(reading);
         }
@@ -92,6 +93,14 @@ class DisplayGenerator {
         this._appendMultiple(pitchesContainer, this._createPitches.bind(this), pitches);
         this._appendMultiple(termTagsContainer, this._createTermTag.bind(this), termTags, expressions.length);
 
+        for (const expression of uniqueExpressions) {
+            termTagsContainer.appendChild(this._createSearchTag(expression));
+        }
+        for (const reading of uniqueReadings) {
+            if (uniqueExpressions.has(reading)) { continue; }
+            termTagsContainer.appendChild(this._createSearchTag(reading));
+        }
+
         // Add definitions
         const dictionaryTag = this._createDictionaryTag(null);
         for (let i = 0, ii = definitions.length; i < ii; ++i) {