Add getPartOfSpeechFlags

yomidevs · Jan 28, 2024 · 2870c7b · 2870c7b
1 parent 293f878
commit 2870c7b
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/ext/js/language/language-transformer.js b/ext/js/language/language-transformer.js
@@ -21,6 +21,8 @@ export class LanguageTransformer {
         this._nextFlagIndex = 0;
         /** @type {import('language-transformer-internal').Transform[]} */
         this._transforms = [];
+        /** @type {Map<string, number>} */
+        this._partOfSpeechToFlagsMap = new Map();
     }
 
     /**
@@ -58,6 +60,23 @@ export class LanguageTransformer {
         for (const transform of transforms2) {
             this._transforms.push(transform);
         }
+
+        for (const [type, condition] of conditionEntries) {
+            const flags = conditionFlagsMap.get(type);
+            if (typeof flags === 'undefined') { continue; } // This case should never happen
+            for (const partOfSpeech of condition.partsOfSpeech) {
+                this._partOfSpeechToFlagsMap.set(partOfSpeech, this.getPartOfSpeechFlags(partOfSpeech) | flags);
+            }
+        }
+    }
+
+    /**
+     * @param {string} partOfSpeech
+     * @returns {number}
+     */
+    getPartOfSpeechFlags(partOfSpeech) {
+        const partOfSpeechFlags = this._partOfSpeechToFlagsMap.get(partOfSpeech);
+        return typeof partOfSpeechFlags !== 'undefined' ? partOfSpeechFlags : 0;
     }
 
     /**