feat(GoPsiContextVariableProvider): map related types to text #89

Update the GoPsiContextVariableProvider to map related types to their text representation. This change affects both GoFunctionOrMethodDeclaration and GoFile cases.
phodal · Sep 14, 2024 · cce91ed · cce91ed
1 parent 33216d7
commit cce91ed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...shire-go/src/main/kotlin/com/phodal/shirelang/go/variable/GoPsiContextVariableProvider.kt b/...shire-go/src/main/kotlin/com/phodal/shirelang/go/variable/GoPsiContextVariableProvider.kt
@@ -56,7 +56,7 @@ class GoPsiContextVariableProvider : PsiContextVariableProvider {
             PsiContextVariable.RELATED_CLASSES -> {
                 when (underTestElement) {
                     is GoFunctionOrMethodDeclaration -> {
-                        GoPsiUtil.findRelatedTypes(underTestElement)
+                        GoPsiUtil.findRelatedTypes(underTestElement).map { it.text}
                     }
 
                     is GoFile -> {
@@ -65,7 +65,7 @@ class GoPsiContextVariableProvider : PsiContextVariableProvider {
 
                         (functions + methods).flatMap {
                             GoPsiUtil.findRelatedTypes(it)
-                        }
+                        }.map { it.text}
                     }
 
                     else -> emptyList()