Add isProperty and isFunction

Author: JavierSegoviaCordoba

CHANGELOG.md

@@ -4,6 +4,8 @@
 
 ### Added
 
+- `FirBasedSymbol<*>.isProperty(): Boolean`
+- `FirBasedSymbol<*>.isFunction(): Boolean`
 - `FirBasedSymbol<*>.hasOwner: Boolean`
 - `<reified T> classId(): ClassId`
 - `ClassId.toCallableId(): CallableId`

kotlin-compiler-extensions/api/kotlin-compiler-extensions.api

@@ -20,6 +20,8 @@ public final class com/javiersc/kotlin/compiler/extensions/common/NameKt {
 
 public final class com/javiersc/kotlin/compiler/extensions/fir/FirBasedSymbolKt {
 	public static final fun getHasOwner (Lorg/jetbrains/kotlin/fir/symbols/FirBasedSymbol;)Z
+	public static final fun isFunction (Lorg/jetbrains/kotlin/fir/symbols/FirBasedSymbol;)Z
+	public static final fun isProperty (Lorg/jetbrains/kotlin/fir/symbols/FirBasedSymbol;)Z
 	public static final fun packageFqName (Lorg/jetbrains/kotlin/fir/symbols/FirBasedSymbol;)Lorg/jetbrains/kotlin/name/FqName;
 }
 

kotlin-compiler-extensions/build.gradle.kts

@@ -9,6 +9,9 @@ hubdle {
         // TODO: https://github.com/facebook/ktfmt/pull/400
         //  https://github.com/facebook/ktfmt/issues/314
         format.isEnabled.set(false)
+        languageSettings { //
+            experimentalContracts()
+        }
         publishing()
     }
 

kotlin-compiler-extensions/main/kotlin/com/javiersc/kotlin/compiler/extensions/fir/firBasedSymbol.kt

@@ -1,9 +1,12 @@
 package com.javiersc.kotlin.compiler.extensions.fir
 
+import kotlin.contracts.contract
 import org.jetbrains.kotlin.fir.getOwnerLookupTag
 import org.jetbrains.kotlin.fir.symbols.FirBasedSymbol
 import org.jetbrains.kotlin.fir.symbols.impl.FirCallableSymbol
 import org.jetbrains.kotlin.fir.symbols.impl.FirClassLikeSymbol
+import org.jetbrains.kotlin.fir.symbols.impl.FirFunctionSymbol
+import org.jetbrains.kotlin.fir.symbols.impl.FirPropertySymbol
 import org.jetbrains.kotlin.name.FqName
 
 public fun FirBasedSymbol<*>.packageFqName(): FqName? {
@@ -16,3 +19,13 @@ public fun FirBasedSymbol<*>.packageFqName(): FqName? {
 
 public val FirBasedSymbol<*>.hasOwner: Boolean
     get() = runCatching { getOwnerLookupTag() }.getOrNull() != null
+
+public fun FirBasedSymbol<*>.isProperty(): Boolean {
+    contract { returns(true) implies (this@FirBasedSymbol is FirPropertySymbol) }
+    return this is FirPropertySymbol
+}
+
+public fun FirBasedSymbol<*>.isFunction(): Boolean {
+    contract { returns(true) implies (this@FirBasedSymbol is FirFunctionSymbol) }
+    return this is FirFunctionSymbol
+}