chore: convenience functions

sourceplusplus · Mar 19, 2024 · 9b65491 · 9b65491
1 parent 1e62e32
commit 9b65491
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/core/src/main/kotlin/spp/jetbrains/artifact/service/ArtifactTypeService.kt b/core/src/main/kotlin/spp/jetbrains/artifact/service/ArtifactTypeService.kt
@@ -72,6 +72,14 @@ object ArtifactTypeService : AbstractSourceMarkerService<IArtifactTypeService>()
         return element is FunctionArtifact || getType(element) == ArtifactType.FUNCTION
     }
 
+    fun isClass(element: PsiElement): Boolean {
+        return getType(element) == ArtifactType.CLASS
+    }
+
+    fun isFile(element: PsiElement): Boolean {
+        return getType(element) == ArtifactType.FILE
+    }
+
     fun isPython(element: PsiElement): Boolean {
         return element.language.id == "Python"
     }
@@ -127,6 +135,14 @@ fun PsiElement.isFunction(): Boolean {
     return ArtifactTypeService.isFunction(this)
 }
 
+fun PsiElement.isClass(): Boolean {
+    return ArtifactTypeService.isClass(this)
+}
+
+fun PsiElement.isFile(): Boolean {
+    return ArtifactTypeService.isFile(this)
+}
+
 fun PsiElement.isPython(): Boolean {
     return ArtifactTypeService.isPython(this)
 }