JavaScriptIdentifierReference only from JavaScriptIdentifierExpression

ris58h · Jun 20, 2023 · 99d8ab2 · 99d8ab2
1 parent 452be5d
commit 99d8ab2
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/main/kotlin/ris58h/webcalm/javascript/psi/JavaScriptIdentifier.kt b/src/main/kotlin/ris58h/webcalm/javascript/psi/JavaScriptIdentifier.kt
@@ -15,8 +15,9 @@ class JavaScriptIdentifier(node: ASTNode) : ASTWrapperPsiElement(node), PsiNameI
         if (introducesName() || parent is JavaScriptIdentifierOwner) return null
         val rangeInElement = TextRange(0, node.textLength)
         return when (parent) {
-            is JavaScriptBreakStatement, is JavaScriptContinueStatement -> return JavaScriptLabelReference(this, rangeInElement)
-            else -> JavaScriptIdentifierReference(this, rangeInElement)
+            is JavaScriptBreakStatement, is JavaScriptContinueStatement -> JavaScriptLabelReference(this, rangeInElement)
+            is JavaScriptIdentifierExpression -> JavaScriptIdentifierReference(this, rangeInElement)
+            else -> null
         }
     }
 
@@ -96,7 +97,8 @@ class JavaScriptIdentifier(node: ASTNode) : ASTWrapperPsiElement(node), PsiNameI
     }
 
     private fun useScopeForVariableDeclaration(variableDeclaration: JavaScriptVariableDeclaration): LocalSearchScope? {
-        val variableDeclarationList = PsiTreeUtil.getParentOfType(variableDeclaration, JavaScriptVariableDeclarationList::class.java)
+        val variableDeclarationList =
+            PsiTreeUtil.getParentOfType(variableDeclaration, JavaScriptVariableDeclarationList::class.java)
         val variableDeclarationListParent = variableDeclarationList?.parent
         if (variableDeclarationListParent is JavaScriptIterationStatement) {
             return LocalSearchScope(variableDeclarationListParent)