Migrate to DeclarationFinder API

Author: lupuuss

mokkery-plugin/src/main/kotlin/dev/mokkery/plugin/ir/IrReferencer.kt

@@ -24,13 +24,26 @@ class IrFunctionById(
     private val id: CallableId,
     private val predicate: (IrSimpleFunctionSymbol) -> Boolean = { true }
 ) : IrFunctionReferencer {
-    override fun reference(context: IrPluginContext): IrSimpleFunction = context.referenceFunctions(id).first(predicate).owner
+    override fun reference(context: IrPluginContext): IrSimpleFunction = context.finderForBuiltins()
+        .findFunctions(id)
+        .find(predicate)
+        ?.owner
+        ?: runtimeDependencyError(id.toString())
 }
 
 class IrClassById(private val id: ClassId) : IrClassReferencer {
-    override fun reference(context: IrPluginContext): IrClass = context.referenceClass(id)!!.owner
+    override fun reference(context: IrPluginContext): IrClass = context.finderForBuiltins()
+        .findClass(id)
+        ?.owner
+        ?: runtimeDependencyError(id.toString())
 }
 
 class IrPropertyById(private val id: CallableId) : IrPropertyReferencer {
-    override fun reference(context: IrPluginContext): IrProperty = context.referenceProperties(id).first().owner
+    override fun reference(context: IrPluginContext): IrProperty = context.finderForBuiltins()
+        .findProperties(id)
+        .firstOrNull()
+        ?.owner
+        ?: runtimeDependencyError(id.toString())
 }
+
+private fun runtimeDependencyError(id: String): Nothing = error("Declaration $id could not be found!")

mokkery-plugin/src/main/kotlin/dev/mokkery/plugin/ir/transformer/mock/stubs/ArrayStubStrategy.kt

@@ -25,7 +25,8 @@ object ArrayStubStrategy : StubStrategy {
         }
         return stub {
             val arrayFunc = pluginContext
-                .referenceFunctions(CallableId(Kotlin.kotlin, Name.identifier(name)))
+                .finderForBuiltins()
+                .findFunctions(CallableId(Kotlin.kotlin, Name.identifier(name)))
                 .first()
             scope.builder.irCall(arrayFunc) {
                 val typeArgs = (type as IrSimpleType).arguments

mokkery-plugin/src/main/kotlin/dev/mokkery/plugin/ir/transformer/mock/stubs/NullableStubStrategy.kt

@@ -6,9 +6,9 @@ import org.jetbrains.kotlin.ir.util.isNullable
 
 object NullableStubStrategy : StubStrategy {
 
-    context(context: StubStrategyScope)
+    context(scope: StubStrategyScope)
     override fun provide(type: IrType): Stub? = when {
-        type.isNullable() -> stub(context.builder.irNull(type))
+        type.isNullable() -> stub(scope.builder.irNull(type))
         else -> null
     }
 }