feat: create custom element for block name

Author: xepozz

src/main/kotlin/com/github/xepozz/php_opcodes_language/language/parser/PHPOp.bnf

@@ -33,7 +33,15 @@ phpOpFile ::= item_*
 private item_ ::= block | COMMENT | EOL
 
 block ::= block_name COLON statements
-block_name ::= var_name | string_literal | IDENTIFIER COLON COLON IDENTIFIER | IDENTIFIER
+block_name ::= var_name | string_literal | method_name | function_name
+{
+    implements=["com.intellij.psi.NavigatablePsiElement" "com.intellij.psi.PsiNamedElement" "com.intellij.psi.PsiLiteralValue"]
+    extends="com.github.xepozz.php_opcodes_language.language.psi.impl.PHPOpBlockNameBaseImpl"
+    methods=[isFunction isClass isClassMethod isMain]
+}
+
+private method_name ::= IDENTIFIER COLON COLON IDENTIFIER
+private function_name ::= IDENTIFIER
 
 statements ::= (statement | COMMENT | EOL)+
 statement ::= NUMBER (assignment_instruction | instruction) | live_range

src/main/kotlin/com/github/xepozz/php_opcodes_language/language/psi/impl/PHPOpBlockNameBaseImpl.kt

@@ -0,0 +1,35 @@
+package com.github.xepozz.php_opcodes_language.language.psi.impl
+
+import com.github.xepozz.php_opcodes_language.language.psi.PHPOpBlock
+import com.github.xepozz.php_opcodes_language.language.psi.PHPOpBlockName
+import com.intellij.icons.AllIcons
+import com.intellij.ide.projectView.PresentationData
+import com.intellij.lang.ASTNode
+import com.intellij.openapi.util.NlsSafe
+import com.intellij.psi.PsiElement
+import com.intellij.psi.search.LocalSearchScope
+import com.intellij.psi.search.SearchScope
+import com.jetbrains.php.lang.psi.PhpPsiUtil
+
+abstract class PHPOpBlockNameBaseImpl : PHPOpBlockName, PHPOpElementImpl {
+    constructor(node: ASTNode) : super(node)
+
+    override fun getValue(): Any? {
+        return text
+    }
+
+    override fun getName(): String = text
+
+    override fun setName(name: @NlsSafe String): PsiElement? {
+        TODO("Not yet implemented")
+    }
+
+    override fun getPresentation() = PresentationData(text, null, getIcon(0), null)
+
+    override fun getIcon(flags: Int) = AllIcons.Nodes.Property
+
+    override fun getUseScope(): SearchScope {
+        val block = PhpPsiUtil.getParentOfClass(this, PHPOpBlock::class.java) ?: return super.getUseScope()
+        return LocalSearchScope(block)
+    }
+}

src/main/kotlin/com/github/xepozz/php_opcodes_language/language/psi/impl/PHPOpParameterBaseImpl.kt

@@ -44,8 +44,8 @@ abstract class PHPOpParameterBaseImpl : PHPOpParameter, PHPOpElementImpl {
 
     override fun isReferenceTo(psiElement: PsiElement) = when (psiElement) {
         !is PHPOpParameter -> false
-        else -> when (true) {
-            !psiElement.isVariable -> false
+        else -> when {
+            !this.isVariable || !psiElement.isVariable -> false
             else -> this.text == psiElement.text
         }
     }

src/main/kotlin/com/github/xepozz/php_opcodes_language/language/psi/impl/PHPOpPsiImplUtil.kt

@@ -1,11 +1,43 @@
 package com.github.xepozz.php_opcodes_language.language.psi.impl
 
+import com.github.xepozz.php_opcodes_language.language.psi.PHPOpBlockName
 import com.github.xepozz.php_opcodes_language.language.psi.PHPOpParameter
+import com.github.xepozz.php_opcodes_language.language.psi.PHPOpTypes
+import com.intellij.lang.tree.util.children
 
 class PHPOpPsiImplUtil {
     companion object {
         @JvmStatic
-        fun isVariable(element: PHPOpParameter): Boolean = element.text.startsWith("$")
         fun isVariable(element: PHPOpParameter): Boolean = element.text.matches(Regex("[TV]\\d+"))
+
+        @JvmStatic
+        fun isFunction(element: PHPOpBlockName): Boolean = element.node.let {
+            val children = it.children().toList()
+            children.size == 1
+                    && children[0].elementType == PHPOpTypes.IDENTIFIER
+                    && !children[0].text.contains("\\")
+        }
+
+        // todo: check for class in a different way
+        @JvmStatic
+        fun isClass(element: PHPOpBlockName): Boolean = element.node.let {
+            val children = it.children().toList()
+            children.size == 1
+                    && children[0].elementType == PHPOpTypes.IDENTIFIER
+                    && children[0].text.contains("\\")
+        }
+
+        @JvmStatic
+        fun isClassMethod(element: PHPOpBlockName): Boolean = element.node.let {
+            val children = it.children().toList()
+            children.size == 4
+                    && children[0].elementType == PHPOpTypes.IDENTIFIER
+                    && children[1].elementType == PHPOpTypes.COLON
+                    && children[2].elementType == PHPOpTypes.COLON
+                    && children[3].elementType == PHPOpTypes.IDENTIFIER
+        }
+
+        @JvmStatic
+        fun isMain(element: PHPOpBlockName): Boolean = element.text == $$"$_main"
     }
-}
+}