Add support for Go to Definition of a translation key

Author: mkevinosullivan

src/core/loaders/Loader.ts

@@ -201,15 +201,22 @@ export abstract class Loader extends Disposable {
     return str
   }
 
-  private treeNodeValueHasPluralizationKeys(value: Record<string, any>, locale: string) {
+  private treeNodeValueHasPluralizationKeys(value: Record<string, any>) {
     return value && isObject(value) && Object.keys(value).some(key => NESTED_PLURALIZATION_KEYS.includes(key))
   }
 
-  private firstPluralizationKeyValue(value: Record<string, any>, locale: string) {
+  private firstPluralizationKey(value: Record<string, any>) {
     if (!value || !isObject(value))
       return undefined
 
-    const firstPluralizationKey = Object.keys(value).find(key => NESTED_PLURALIZATION_KEYS.includes(key))
+    return Object.keys(value).find(key => NESTED_PLURALIZATION_KEYS.includes(key))
+  }
+
+  private firstPluralizationKeyValue(value: Record<string, any>) {
+    if (!value || !isObject(value))
+      return undefined
+
+    const firstPluralizationKey = this.firstPluralizationKey(value)
 
     return firstPluralizationKey ? (value as Record<string, any>)[firstPluralizationKey] : undefined
   }
@@ -227,8 +234,8 @@ export abstract class Loader extends Disposable {
       if (!value)
         return undefined
 
-      if (Config._keyStyle !== 'flat' && this.treeNodeValueHasPluralizationKeys(value, locale))
-        return this.stripAnnotationString(this.firstPluralizationKeyValue(value, locale), maxlength)
+      if (Config._keyStyle !== 'flat' && this.treeNodeValueHasPluralizationKeys(value))
+        return this.stripAnnotationString(this.firstPluralizationKeyValue(value), maxlength)
 
       let text = JSON
         .stringify(value, null, stringifySpace)
@@ -252,16 +259,28 @@ export abstract class Loader extends Disposable {
     return new LocaleNode({ keypath: key, shadow: true })
   }
 
-  getNodeByKey(key: string, shadow = false): LocaleNode | undefined {
+  getNodeByKey(key: string, shadow = false, locale?: string): LocaleNode | undefined {
     const node = resolveFlattenRoot(this.getTreeNodeByKey(key))
     if (!node && shadow)
       return this.getShadowNodeByKey(key)
     if (node && node.type !== 'tree')
       return node
+
+    const language = locale || Config.sourceLanguage
+    if (
+      node
+      && node.type === 'tree'
+      && Config._keyStyle !== 'flat'
+      && this.treeNodeValueHasPluralizationKeys(node.values[language])
+    ) {
+      const subkey = this.firstPluralizationKey(node.values[language])
+      if (subkey && node.children[subkey] && node.children[subkey].type === 'node')
+        return node.children[subkey] as LocaleNode
+    }
   }
 
-  getTranslationsByKey(key: string, shadow = true) {
-    const node = this.getNodeByKey(key, shadow)
+  getTranslationsByKey(key: string, shadow = true, locale?: string) {
+    const node = this.getNodeByKey(key, shadow, locale)
     if (!node)
       return {}
     if (shadow)
@@ -271,7 +290,7 @@ export abstract class Loader extends Disposable {
   }
 
   getRecordByKey(key: string, locale: string, shadow = false): LocaleRecord | undefined {
-    const trans = this.getTranslationsByKey(key, shadow)
+    const trans = this.getTranslationsByKey(key, shadow, locale)
     return trans[locale]
   }