Add missing opt-in annotations (#407)

drewhamilton · Sep 22, 2024 · 77160db · 77160db
1 parent 2e37d43
commit 77160db
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 0 deletions.
diff --git a/poko-compiler-plugin/src/main/kotlin/dev/drewhamilton/poko/ir/PokoMembersTransformer.kt b/poko-compiler-plugin/src/main/kotlin/dev/drewhamilton/poko/ir/PokoMembersTransformer.kt
@@ -17,6 +17,7 @@ import org.jetbrains.kotlin.ir.declarations.IrFunction
 import org.jetbrains.kotlin.ir.declarations.IrProperty
 import org.jetbrains.kotlin.ir.declarations.IrSimpleFunction
 import org.jetbrains.kotlin.ir.declarations.impl.IrFactoryImpl
+import org.jetbrains.kotlin.ir.symbols.UnsafeDuringIrConstructionAPI
 import org.jetbrains.kotlin.ir.symbols.impl.IrValueParameterSymbolImpl
 import org.jetbrains.kotlin.ir.types.createType
 import org.jetbrains.kotlin.ir.util.IdSignature
@@ -73,6 +74,7 @@ internal class PokoMembersTransformer(
         return super.visitFunctionNew(declaration)
     }
 
+    @OptIn(UnsafeDuringIrConstructionAPI::class)
     private fun IrClass.isPokoClass(): Boolean = when {
         !hasAnnotation(pokoAnnotationName.asSingleFqName()) -> {
             messageCollector.log("Not Poko class")
@@ -106,6 +108,7 @@ internal class PokoMembersTransformer(
         }
     }
 
+    @OptIn(UnsafeDuringIrConstructionAPI::class)
     private inline fun IrFunction.convertToGenerated(
         generateFunctionBody: IrBlockBodyBuilder.(List<IrProperty>) -> Unit
     ) {

diff --git a/poko-compiler-plugin/src/main/kotlin/dev/drewhamilton/poko/ir/equalsGeneration.kt b/poko-compiler-plugin/src/main/kotlin/dev/drewhamilton/poko/ir/equalsGeneration.kt
@@ -33,6 +33,7 @@ import org.jetbrains.kotlin.ir.expressions.IrExpression
 import org.jetbrains.kotlin.ir.symbols.IrClassSymbol
 import org.jetbrains.kotlin.ir.symbols.IrClassifierSymbol
 import org.jetbrains.kotlin.ir.symbols.IrSimpleFunctionSymbol
+import org.jetbrains.kotlin.ir.symbols.UnsafeDuringIrConstructionAPI
 import org.jetbrains.kotlin.ir.types.classifierOrFail
 import org.jetbrains.kotlin.ir.types.classifierOrNull
 import org.jetbrains.kotlin.ir.types.isNullable
@@ -215,6 +216,7 @@ private fun IrBuilderWithScope.irCallContentDeepEquals(
  * function if it represents a primitive array.
  */
 context(IrPluginContext)
+@OptIn(UnsafeDuringIrConstructionAPI::class)
 private fun IrBuilderWithScope.findContentDeepEqualsFunctionSymbol(
     classifier: IrClassifierSymbol,
 ): IrSimpleFunctionSymbol {

diff --git a/poko-compiler-plugin/src/main/kotlin/dev/drewhamilton/poko/ir/functionGeneration.kt b/poko-compiler-plugin/src/main/kotlin/dev/drewhamilton/poko/ir/functionGeneration.kt
@@ -15,6 +15,7 @@ import org.jetbrains.kotlin.ir.symbols.IrClassSymbol
 import org.jetbrains.kotlin.ir.symbols.IrClassifierSymbol
 import org.jetbrains.kotlin.ir.symbols.IrTypeParameterSymbol
 import org.jetbrains.kotlin.ir.symbols.IrValueSymbol
+import org.jetbrains.kotlin.ir.symbols.UnsafeDuringIrConstructionAPI
 import org.jetbrains.kotlin.ir.types.IrSimpleType
 import org.jetbrains.kotlin.ir.types.IrType
 import org.jetbrains.kotlin.ir.types.classOrNull
@@ -124,6 +125,7 @@ private val IrTypeParameterSymbol.hasArrayOrPrimitiveArrayUpperBound: Boolean
         return false
     }
 
+@OptIn(UnsafeDuringIrConstructionAPI::class)
 internal val IrTypeParameter.erasedUpperBound: IrClass
     get() {
         // Pick the (necessarily unique) non-interface upper bound if it exists

diff --git a/poko-compiler-plugin/src/main/kotlin/dev/drewhamilton/poko/ir/hashCodeGeneration.kt b/poko-compiler-plugin/src/main/kotlin/dev/drewhamilton/poko/ir/hashCodeGeneration.kt
@@ -31,6 +31,7 @@ import org.jetbrains.kotlin.ir.symbols.IrClassifierSymbol
 import org.jetbrains.kotlin.ir.symbols.IrSimpleFunctionSymbol
 import org.jetbrains.kotlin.ir.symbols.IrTypeParameterSymbol
 import org.jetbrains.kotlin.ir.symbols.IrVariableSymbol
+import org.jetbrains.kotlin.ir.symbols.UnsafeDuringIrConstructionAPI
 import org.jetbrains.kotlin.ir.symbols.impl.IrVariableSymbolImpl
 import org.jetbrains.kotlin.ir.types.IrType
 import org.jetbrains.kotlin.ir.types.classifierOrFail
@@ -141,6 +142,7 @@ private fun IrBlockBodyBuilder.getHashCodeOfProperty(
  * [org.jetbrains.kotlin.fir.backend.generators.DataClassMembersGenerator].
  */
 context(IrPluginContext)
+@OptIn(UnsafeDuringIrConstructionAPI::class)
 private fun IrBlockBodyBuilder.getHashCodeOf(
     property: IrProperty,
     value: IrExpression,
@@ -265,6 +267,7 @@ private fun IrBlockBodyBuilder.maybeFindArrayContentHashCodeFunction(
 }
 
 context(IrPluginContext)
+@OptIn(UnsafeDuringIrConstructionAPI::class)
 private fun IrBlockBodyBuilder.findArrayContentDeepHashCodeFunction(
     propertyClassifier: IrClassifierSymbol,
 ): IrSimpleFunctionSymbol {
@@ -286,6 +289,7 @@ private fun IrBlockBodyBuilder.findArrayContentDeepHashCodeFunction(
     }
 }
 
+@OptIn(UnsafeDuringIrConstructionAPI::class)
 private fun IrBlockBodyBuilder.getStandardHashCodeFunctionSymbol(
     classifier: IrClassifierSymbol?,
 ): IrSimpleFunctionSymbol = when {
@@ -299,6 +303,7 @@ private fun IrBlockBodyBuilder.getStandardHashCodeFunctionSymbol(
         error("Unknown classifier kind $classifier")
 }
 
+@OptIn(UnsafeDuringIrConstructionAPI::class)
 private fun IrBlockBodyBuilder.getHashCodeFunctionForClass(
     irClass: IrClass
 ): IrSimpleFunctionSymbol {
@@ -311,6 +316,7 @@ private fun IrBlockBodyBuilder.getHashCodeFunctionForClass(
         ?: context.irBuiltIns.anyClass.functions.single { it.owner.name.asString() == "hashCode" }
 }
 
+@OptIn(UnsafeDuringIrConstructionAPI::class)
 private fun IrBlockBodyBuilder.irCallHashCodeFunction(
     hashCodeFunctionSymbol: IrSimpleFunctionSymbol,
     value: IrExpression,

diff --git a/poko-compiler-plugin/src/main/kotlin/dev/drewhamilton/poko/ir/toStringGeneration.kt b/poko-compiler-plugin/src/main/kotlin/dev/drewhamilton/poko/ir/toStringGeneration.kt
@@ -25,6 +25,7 @@ import org.jetbrains.kotlin.ir.expressions.addArgument
 import org.jetbrains.kotlin.ir.symbols.IrClassSymbol
 import org.jetbrains.kotlin.ir.symbols.IrClassifierSymbol
 import org.jetbrains.kotlin.ir.symbols.IrSimpleFunctionSymbol
+import org.jetbrains.kotlin.ir.symbols.UnsafeDuringIrConstructionAPI
 import org.jetbrains.kotlin.ir.types.classifierOrFail
 import org.jetbrains.kotlin.ir.types.classifierOrNull
 import org.jetbrains.kotlin.ir.types.isNullable
@@ -187,6 +188,7 @@ private fun IrBlockBodyBuilder.irArrayTypeCheckAndContentDeepToStringBranch(
  * `contentToString` function if it is a primitive array.
  */
 context(IrPluginContext)
+@OptIn(UnsafeDuringIrConstructionAPI::class)
 private fun IrBuilderWithScope.findContentDeepToStringFunctionSymbol(
     propertyClassifier: IrClassifierSymbol,
 ): IrSimpleFunctionSymbol {
@@ -209,6 +211,7 @@ private fun IrBuilderWithScope.findContentDeepToStringFunctionSymbol(
     }
 }
 
+@OptIn(UnsafeDuringIrConstructionAPI::class)
 private fun IrBlockBodyBuilder.irCallToStringFunction(
     toStringFunctionSymbol: IrSimpleFunctionSymbol,
     value: IrExpression,