refactor: compare FunctionArtifact

BFergerson · BFergerson · commit 2b86431bb577 · 2023-01-31T23:27:11.000+04:00
diff --git a/common/src/main/kotlin/spp/jetbrains/artifact/service/ArtifactTypeService.kt b/common/src/main/kotlin/spp/jetbrains/artifact/service/ArtifactTypeService.kt
@@ -18,6 +18,7 @@ package spp.jetbrains.artifact.service
 
 import com.intellij.psi.PsiComment
 import com.intellij.psi.PsiElement
+import spp.jetbrains.artifact.model.FunctionArtifact
 import spp.jetbrains.artifact.service.define.AbstractSourceMarkerService
 import spp.jetbrains.artifact.service.define.IArtifactTypeService
 import spp.protocol.artifact.ArtifactType
@@ -50,7 +51,7 @@ object ArtifactTypeService : AbstractSourceMarkerService<IArtifactTypeService>()
     }
 
     fun isFunction(element: PsiElement): Boolean {
-        return getType(element) == ArtifactType.FUNCTION
+        return element is FunctionArtifact || getType(element) == ArtifactType.FUNCTION
     }
 
     fun isPython(element: PsiElement): Boolean {

Original file line number	Diff line number	Diff line change
`@@ -18,6 +18,7 @@ package spp.jetbrains.artifact.service`
`18`	`18`
`19`	`19`	`import com.intellij.psi.PsiComment`
`20`	`20`	`import com.intellij.psi.PsiElement`
	`21`	`+import spp.jetbrains.artifact.model.FunctionArtifact`
`21`	`22`	`import spp.jetbrains.artifact.service.define.AbstractSourceMarkerService`
`22`	`23`	`import spp.jetbrains.artifact.service.define.IArtifactTypeService`
`23`	`24`	`import spp.protocol.artifact.ArtifactType`
`@@ -50,7 +51,7 @@ object ArtifactTypeService : AbstractSourceMarkerService<IArtifactTypeService>()`
`50`	`51`	`}`
`51`	`52`
`52`	`53`	`fun isFunction(element: PsiElement): Boolean {`
`53`		`- return getType(element) == ArtifactType.FUNCTION`
	`54`	`+ return element is FunctionArtifact \|\| getType(element) == ArtifactType.FUNCTION`
`54`	`55`	`}`
`55`	`56`
`56`	`57`	`fun isPython(element: PsiElement): Boolean {`