make the wrapper's value a class reference

nextras · Feb 5, 2024 · 0e1a9c9 · 0e1a9c9
1 parent a1154c2
commit 0e1a9c9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/src/main/kotlin/org/nextras/orm/intellij/reference/ModifierClassName.kt b/src/main/kotlin/org/nextras/orm/intellij/reference/ModifierClassName.kt
@@ -6,7 +6,6 @@ import com.intellij.psi.PsiPolyVariantReferenceBase
 import com.intellij.psi.ResolveResult
 import com.jetbrains.php.PhpIndex
 import com.jetbrains.php.lang.psi.resolve.types.PhpType
-import org.nextras.orm.intellij.utils.OrmUtils
 import org.nextras.orm.intellij.utils.PhpIndexUtils
 
 class ModifierClassName(
@@ -17,7 +16,6 @@ class ModifierClassName(
 		val phpIndex = PhpIndex.getInstance(this.element.project)
 		val result = PhpIndexUtils.getByType(PhpType().add(fqnClass), phpIndex)
 		return result
-			.filter { OrmUtils.OrmClass.ENTITY.`is`(it, phpIndex) }
 			.map {
 				object : ResolveResult {
 					override fun getElement(): PsiElement {

diff --git a/src/main/kotlin/org/nextras/orm/intellij/reference/ModifierClassNameProvider.kt b/src/main/kotlin/org/nextras/orm/intellij/reference/ModifierClassNameProvider.kt
@@ -41,6 +41,7 @@ class ModifierClassNameProvider : PsiReferenceProvider() {
 							PlatformPatterns.psiElement(PhpDocTokenTypes.DOC_IDENTIFIER).withText("1")
 						)
 					),
+					PlatformPatterns.psiElement(PhpDocTokenTypes.DOC_IDENTIFIER).withText("wrapper"),
 				)
 			)
 			.withLanguage(PhpLanguage.INSTANCE)