Revert "Minor: Remove hard-coded 'Companion' name"

This reverts commit 871ad2b9
This commit is contained in:
Yan Zhulanow
2020-10-24 00:18:44 +09:00
parent 19f518c037
commit 6cc4d70b07
@@ -12,16 +12,13 @@ import com.intellij.openapi.project.Project
import org.jetbrains.kotlin.diagnostics.Diagnostic
import org.jetbrains.kotlin.idea.KotlinBundle
import org.jetbrains.kotlin.idea.references.mainReference
import org.jetbrains.kotlin.name.SpecialNames
import org.jetbrains.kotlin.psi.*
import org.jetbrains.kotlin.utils.addToStdlib.safeAs
object JavaClassOnCompanionFixes : KotlinIntentionActionsFactory() {
override fun doCreateActions(diagnostic: Diagnostic): List<IntentionAction> {
val expression = diagnostic.psiElement.parent as? KtDotQualifiedExpression ?: return emptyList()
val companionName = expression.receiverExpression.mainReference?.resolve()?.safeAs<KtObjectDeclaration>()?.name
?: SpecialNames.DEFAULT_NAME_FOR_COMPANION_OBJECT.identifier
val companionName = expression.receiverExpression.mainReference?.resolve()?.safeAs<KtObjectDeclaration>()?.name ?: "Companion"
return listOf(
ReplaceWithCompanionClassJavaFix(expression, companionName),
ReplaceWithClassJavaFix(expression)