fix toc named block completions and refs

Author: patricklx

src/main/kotlin/com/emberjs/utils/EmberUtils.kt

@@ -6,9 +6,11 @@ import com.dmarcotte.handlebars.parsing.HbTokenTypes
 import com.dmarcotte.handlebars.psi.*
 import com.dmarcotte.handlebars.psi.impl.HbPathImpl
 import com.emberjs.cli.EmberCliFrameworkDetector
+import com.emberjs.gts.GjsLanguage
 import com.emberjs.gts.GtsElementTypes
 import com.emberjs.gts.GtsFile
 import com.emberjs.gts.GtsFileViewProvider
+import com.emberjs.gts.GtsLanguage
 import com.emberjs.hbs.*
 import com.emberjs.index.EmberNameIndex
 import com.emberjs.navigation.EmberGotoRelatedProvider
@@ -844,6 +846,12 @@ class EmberUtils {
                         }
                     }
                 }
+
+                if (cls is TypeScriptVariable) {
+                    jsTemplate = f.containingFile.viewProvider.findElementAt(cls.children[1].textOffset, JavaScriptSupportLoader.TYPESCRIPT)
+                } else if (cls is JSVariable) {
+                    jsTemplate = f.containingFile.viewProvider.findElementAt(cls.children[1].textOffset, JavaScriptSupportLoader.ECMA_SCRIPT_6)
+                }
             }
 
             template = dir?.findFile("$name.hbs") ?: dir?.findFile("template.hbs") ?: dir?.findFile("template.ts") ?: dir?.findFile("template.js")
@@ -936,6 +944,13 @@ class EmberUtils {
                         }
                     }
                 }
+                if (cls is TypeScriptVariable) {
+                    if (cls.jsType.toString().startsWith("TOC<") || cls.jsType.toString().startsWith("TemplateOnlyComponent<")) {
+                        val arg = (cls.jsType as JSGenericTypeImpl).arguments[0]
+                        blocks = arg.asRecordType().properties.find { it.memberName == "Blocks" }
+                        args = arg.asRecordType().properties.find { it.memberName == "Args" }
+                    }
+                }
                 if (blocks != null) {
                     val yields = blocks?.jsType?.asRecordType()?.properties
                     yields?.forEach {