Just use the owner name

burst-kotlin-plugin/src/main/kotlin/app/cash/burst/kotlin/Argument.kt

@@ -29,7 +29,6 @@ import org.jetbrains.kotlin.ir.expressions.IrGetEnumValue
 import org.jetbrains.kotlin.ir.expressions.IrVararg
 import org.jetbrains.kotlin.ir.expressions.impl.IrConstImpl
 import org.jetbrains.kotlin.ir.expressions.impl.IrGetEnumValueImpl
-import org.jetbrains.kotlin.ir.symbols.IrEnumEntrySymbol
 import org.jetbrains.kotlin.ir.symbols.UnsafeDuringIrConstructionAPI
 import org.jetbrains.kotlin.ir.types.IrType
 import org.jetbrains.kotlin.ir.types.classFqName
@@ -38,9 +37,7 @@ import org.jetbrains.kotlin.ir.types.isNullable
 import org.jetbrains.kotlin.ir.util.classId
 import org.jetbrains.kotlin.ir.util.deepCopyWithSymbols
 import org.jetbrains.kotlin.ir.util.defaultType
-import org.jetbrains.kotlin.ir.util.findDeclaration
 import org.jetbrains.kotlin.ir.util.isEnumClass
-import org.jetbrains.kotlin.ir.util.parentAsClass
 import org.jetbrains.kotlin.ir.visitors.IrElementVisitor
 import org.jetbrains.kotlin.name.NameUtils
 
@@ -207,16 +204,16 @@ private fun IrExpression.suggestedName(): String? {
 }
 
 @UnsafeDuringIrConstructionAPI
-private fun IrPluginContext.enumValueArguments(
+private fun enumValueArguments(
   referenceClass: IrClass,
   parameter: IrValueParameter,
 ): List<Argument> {
   val enumEntries = referenceClass.declarations.filterIsInstance<IrEnumEntry>()
   val hasDefaultValue = parameter.defaultValue != null
-  val defaultEnumSymbol = parameter.defaultValue?.let { defaultValue ->
+  val defaultEnumSymbolName = parameter.defaultValue?.let { defaultValue ->
     val expression = defaultValue.expression
     when {
-      expression is IrGetEnumValue -> canonicalize(expression.symbol)
+      expression is IrGetEnumValue -> expression.symbol.owner.name
       expression is IrConst<*> && expression.value == null -> null
       else -> unexpectedDefaultValue(parameter)
     }
@@ -228,7 +225,7 @@ private fun IrPluginContext.enumValueArguments(
         EnumValueArgument(
           original = parameter,
           type = parameter.type,
-          isDefault = hasDefaultValue && canonicalize(enumEntry.symbol) == defaultEnumSymbol,
+          isDefault = hasDefaultValue && enumEntry.symbol.owner.name == defaultEnumSymbolName,
           value = enumEntry,
         ),
       )
@@ -238,29 +235,13 @@ private fun IrPluginContext.enumValueArguments(
         NullArgument(
           original = parameter,
           type = parameter.type,
-          isDefault = hasDefaultValue && defaultEnumSymbol == null,
+          isDefault = hasDefaultValue && defaultEnumSymbolName == null,
         ),
       )
     }
   }
 }
 
-/**
- * It's possible we'll see two different [IrEnumEntrySymbol] instances for the same enum constant.
- * This returns a canonical instance by looking up the enum by its name in this [IrPluginContext].
- *
- * https://kotlinlang.slack.com/archives/C7L3JB43G/p1732121246270569
- */
-@UnsafeDuringIrConstructionAPI
-private fun IrPluginContext.canonicalize(
-  enumSymbol: IrEnumEntrySymbol,
-) : IrEnumEntrySymbol {
-  val classId = enumSymbol.owner.parentAsClass.classId ?: return enumSymbol
-  val classSymbol = referenceClass(classId) ?: return enumSymbol
-  val entry = classSymbol.owner.findDeclaration<IrEnumEntry> { it.name == enumSymbol.owner.name }
-  return entry?.symbol ?: enumSymbol
-}
-
 private fun IrPluginContext.booleanArguments(
   parameter: IrValueParameter,
 ): List<Argument> {