feat: support activity types

Author: xepozz

src/main/kotlin/com/github/xepozz/temporal/languages/php/TemporalTypeProvider.kt

@@ -1,34 +1,97 @@
 package com.github.xepozz.temporal.languages.php
 
+import com.intellij.openapi.project.DumbService
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiElement
+import com.intellij.psi.util.CachedValueProvider
+import com.intellij.psi.util.CachedValuesManager
+import com.jetbrains.php.lang.psi.elements.Method
+import com.jetbrains.php.lang.psi.elements.MethodReference
 import com.jetbrains.php.lang.psi.elements.PhpNamedElement
+import com.jetbrains.php.lang.psi.elements.PhpReference
+import com.jetbrains.php.lang.psi.elements.PhpYield
 import com.jetbrains.php.lang.psi.resolve.types.PhpType
 import com.jetbrains.php.lang.psi.resolve.types.PhpTypeProvider4
+import com.jetbrains.php.lang.psi.resolve.types.PhpTypeSignatureKey
 
 class TemporalTypeProvider : PhpTypeProvider4 {
     override fun getKey() = '\uBEB1'
-    override fun getType(p0: PsiElement?): PhpType? {
-        TODO("Not yet implemented")
+
+    override fun getType(element: PsiElement) = when {
+        DumbService.isDumb(element.project) -> null
+        element is MethodReference -> getMethodUsageType(element)
+        element is PhpYield -> getYieldType(element)
+        else -> null
+    }
+
+    private fun getMethodUsageType(element: MethodReference): PhpType? {
+        val targetMethods = CachedValuesManager.getCachedValue(element) {
+            CachedValueProvider.Result.create(
+                element.resolveGlobal(true)
+                    .mapNotNull { it as? Method }
+                    .mapNotNull {
+                        val containingClass = it.containingClass ?: return@mapNotNull null
+
+                        return@mapNotNull when {
+                            containingClass.hasAttribute(TemporalClasses.ACTIVITY) -> it
+                            containingClass.hasAttribute(TemporalClasses.WORKFLOW) -> it
+                            else -> null
+                        }
+                    },
+                element.containingFile,
+            )
+        }
+        if (targetMethods.isEmpty()) return null
+
+        val newType = PhpType.PhpTypeBuilder()
+
+        targetMethods.map { method ->
+            val signed = PhpTypeSignatureKey.getSignature(method)
+            println("signed: $signed")
+            val parametrized = PhpType.createParametrized(
+                PhpTypeSignatureKey.CLASS.sign(UtilClasses.REACT_PROMISE),
+                signed
+            )
+            println("parametrized $parametrized ${parametrized.typesWithParametrisedParts} ${parametrized.isComplete}")
+            newType.add(parametrized)
+        }
+        return newType.build()
+    }
+
+    private fun getYieldType(element: PhpYield): PhpType? {
+        val argument = element.argument as? PhpReference ?: return null
+
+        val newType = PhpType.PhpTypeBuilder()
+        val signatureTypes = PhpType().add(argument.signature.substring(2))
+            .typesWithParametrisedParts
+            .mapNotNull {
+                val lBraceIndex = it.indexOf("<")
+                val rBraceIndex = it.indexOf(">")
+
+                when {
+                    rBraceIndex > lBraceIndex -> it.substring(lBraceIndex + 1, rBraceIndex)
+                    else -> null
+                }
+            }
+
+        signatureTypes.forEach { newType.add(it) }
+
+        return newType.build()
     }
 
     override fun complete(
-        p0: String?,
-        p1: Project?
+        signature: String,
+        project: Project,
     ): PhpType? {
-        TODO("Not yet implemented")
+        return null
     }
 
     override fun getBySignature(
-        p0: String?,
-        p1: Set<String?>?,
-        p2: Int,
-        p3: Project?
-    ): Collection<PhpNamedElement?>? {
-        TODO("Not yet implemented")
+        expression: String,
+        visited: Set<String>,
+        depth: Int,
+        project: Project,
+    ): Collection<PhpNamedElement>? {
+        return null
     }
-
-    override fun interceptsNativeSignature() = true
-
-
 }

src/main/resources/META-INF/language-php.xml

@@ -1,6 +1,7 @@
 <idea-plugin>
     <extensions defaultExtensionNs="com.jetbrains.php">
         <typeProvider4
+                order="first"
                 implementation="com.github.xepozz.temporal.languages.php.TemporalTypeProvider"/>
     </extensions>
 </idea-plugin>