Skip to content

Commit 1e62e32

Browse files
committed
chore: return file type on psi file
1 parent ce68ffe commit 1e62e32

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

core/src/main/kotlin/spp/jetbrains/artifact/service/ArtifactTypeService.kt

+5
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ package spp.jetbrains.artifact.service
1818

1919
import com.intellij.psi.PsiComment
2020
import com.intellij.psi.PsiElement
21+
import com.intellij.psi.PsiFile
2122
import spp.jetbrains.artifact.model.FunctionArtifact
2223
import spp.jetbrains.artifact.service.define.AbstractSourceMarkerService
2324
import spp.jetbrains.artifact.service.define.IArtifactTypeService
@@ -52,6 +53,10 @@ object ArtifactTypeService : AbstractSourceMarkerService<IArtifactTypeService>()
5253

5354
override fun getType(element: PsiElement): ArtifactType? {
5455
if (!isSupported(element)) return null
56+
//language-agnostic check
57+
if (element is PsiFile) return ArtifactType.FILE
58+
59+
//language-specific check
5560
return getService(element.language).getType(element)
5661
}
5762

0 commit comments

Comments
 (0)