mirror of
https://gitflic.ru/project/openide/openide.git
synced 2026-09-27 10:03:11 +07:00
[kotlin] KTIJ-32668 Merge KtWhenExpressionPostfixTemplate and KotlinWhenPostfixTemplate
GitOrigin-RevId: 53e718c72f1b5f702871dbeb6cf59338c621bfc0
This commit is contained in:
Anna Antonova
committed by
intellij-monorepo-bot
intellij-monorepo-bot
parent
9b0621595b
commit
d3efa92dc1
+8
-156
@@ -1,161 +1,13 @@
|
||||
// Copyright 2000-2024 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
|
||||
package org.jetbrains.kotlin.idea.codeInsight.postfix
|
||||
|
||||
import com.intellij.codeInsight.template.postfix.templates.StringBasedPostfixTemplate
|
||||
import com.intellij.psi.PsiElement
|
||||
import org.jetbrains.kotlin.analysis.api.KaContextParameterApi
|
||||
import org.jetbrains.kotlin.analysis.api.KaSession
|
||||
import org.jetbrains.kotlin.analysis.api.analyze
|
||||
import org.jetbrains.kotlin.analysis.api.components.sealedClassInheritors
|
||||
import org.jetbrains.kotlin.analysis.api.components.staticDeclaredMemberScope
|
||||
import org.jetbrains.kotlin.analysis.api.permissions.KaAllowAnalysisFromWriteAction
|
||||
import org.jetbrains.kotlin.analysis.api.permissions.KaAllowAnalysisOnEdt
|
||||
import org.jetbrains.kotlin.analysis.api.permissions.allowAnalysisFromWriteAction
|
||||
import org.jetbrains.kotlin.analysis.api.permissions.allowAnalysisOnEdt
|
||||
import org.jetbrains.kotlin.analysis.api.symbols.KaClassKind
|
||||
import org.jetbrains.kotlin.analysis.api.symbols.KaEnumEntrySymbol
|
||||
import org.jetbrains.kotlin.analysis.api.symbols.KaNamedClassSymbol
|
||||
import org.jetbrains.kotlin.analysis.api.symbols.KaSymbolModality
|
||||
import org.jetbrains.kotlin.analysis.api.types.KaClassType
|
||||
import org.jetbrains.kotlin.analysis.api.types.KaType
|
||||
import org.jetbrains.kotlin.name.CallableId
|
||||
import org.jetbrains.kotlin.name.ClassId
|
||||
import org.jetbrains.kotlin.psi.KtExpression
|
||||
import com.intellij.codeInsight.template.postfix.templates.PostfixTemplateProvider
|
||||
import com.intellij.codeInsight.template.postfix.templates.SurroundPostfixTemplateBase
|
||||
import org.jetbrains.kotlin.idea.codeInsight.surroundWith.expression.KotlinWhenSurrounder
|
||||
|
||||
internal class KotlinWhenPostfixTemplate : StringBasedPostfixTemplate {
|
||||
@Suppress("ConvertSecondaryConstructorToPrimary")
|
||||
constructor(provider: KotlinPostfixTemplateProvider) : super(
|
||||
/* name = */ "when",
|
||||
/* example = */ "when (expr) {}",
|
||||
/* selector = */ allExpressions(ValuedFilter, ExpressionTypeFilter { isSealedType(it) }),
|
||||
/* provider = */ provider
|
||||
)
|
||||
|
||||
override fun getTemplateString(element: PsiElement): String {
|
||||
return buildString {
|
||||
val branches = collectPossibleBranches(element)
|
||||
|
||||
append("when (\$expr\$) {")
|
||||
|
||||
if (branches.isNotEmpty()) {
|
||||
for ((index, branch) in branches.withIndex()) {
|
||||
when (branch) {
|
||||
is CaseBranch.Callable -> {
|
||||
append("\n")
|
||||
append(branch.callableId.asSingleFqName().asString())
|
||||
}
|
||||
is CaseBranch.Object -> {
|
||||
append("\n")
|
||||
append(branch.classId.asFqNameString())
|
||||
}
|
||||
is CaseBranch.Instance -> {
|
||||
append("\nis ")
|
||||
append(branch.classId.asFqNameString())
|
||||
}
|
||||
}
|
||||
append(" -> ")
|
||||
if (index == 0) {
|
||||
append("\$END\$")
|
||||
}
|
||||
append("TODO()")
|
||||
}
|
||||
} else {
|
||||
append("\nelse -> \$END\$TODO()")
|
||||
}
|
||||
|
||||
append("\n}")
|
||||
}
|
||||
}
|
||||
|
||||
@OptIn(KaAllowAnalysisOnEdt::class)
|
||||
private fun collectPossibleBranches(element: PsiElement): List<CaseBranch> {
|
||||
if (element !is KtExpression) {
|
||||
return emptyList()
|
||||
}
|
||||
|
||||
allowAnalysisOnEdt {
|
||||
@OptIn(KaAllowAnalysisFromWriteAction::class)
|
||||
allowAnalysisFromWriteAction {
|
||||
analyze(element) {
|
||||
val type = element.expressionType
|
||||
if (type is KaClassType) {
|
||||
val klass = type.symbol
|
||||
if (klass is KaNamedClassSymbol) {
|
||||
return when (klass.classKind) {
|
||||
KaClassKind.ENUM_CLASS -> collectEnumBranches(klass)
|
||||
else -> collectSealedClassInheritors(klass)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return emptyList()
|
||||
}
|
||||
|
||||
@OptIn(KaContextParameterApi::class)
|
||||
context(_: KaSession)
|
||||
private fun collectEnumBranches(klass: KaNamedClassSymbol): List<CaseBranch> {
|
||||
val enumEntries = klass.staticDeclaredMemberScope
|
||||
.callables
|
||||
.filterIsInstance<KaEnumEntrySymbol>()
|
||||
|
||||
return buildList {
|
||||
for (enumEntry in enumEntries) {
|
||||
val callableId = enumEntry.callableId ?: return emptyList()
|
||||
add(CaseBranch.Callable(callableId))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
context(_: KaSession)
|
||||
private fun collectSealedClassInheritors(klass: KaNamedClassSymbol): List<CaseBranch> {
|
||||
return mutableListOf<CaseBranch>().also { processSealedClassInheritor(klass, it) }
|
||||
}
|
||||
|
||||
@OptIn(KaContextParameterApi::class)
|
||||
context(_: KaSession)
|
||||
private fun processSealedClassInheritor(klass: KaNamedClassSymbol, consumer: MutableList<CaseBranch>): Boolean {
|
||||
val classId = klass.classId ?: return false
|
||||
|
||||
if (klass.classKind == KaClassKind.OBJECT) {
|
||||
consumer.add(CaseBranch.Object(classId))
|
||||
return true
|
||||
}
|
||||
|
||||
if (klass.modality == KaSymbolModality.SEALED) {
|
||||
val inheritors = klass.sealedClassInheritors
|
||||
if (inheritors.isNotEmpty()) {
|
||||
for (inheritor in inheritors) {
|
||||
if (!processSealedClassInheritor(inheritor, consumer)) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
consumer.add(CaseBranch.Instance(classId))
|
||||
return true
|
||||
}
|
||||
|
||||
override fun getElementToRemove(expr: PsiElement): PsiElement = expr
|
||||
}
|
||||
|
||||
private sealed class CaseBranch {
|
||||
class Callable(val callableId: CallableId) : CaseBranch()
|
||||
class Object(val classId: ClassId) : CaseBranch()
|
||||
class Instance(val classId: ClassId) : CaseBranch()
|
||||
}
|
||||
|
||||
private fun isSealedType(type: KaType): Boolean {
|
||||
if (type is KaClassType) {
|
||||
val symbol = type.symbol
|
||||
if (symbol is KaNamedClassSymbol) {
|
||||
return symbol.classKind == KaClassKind.ENUM_CLASS || symbol.modality == KaSymbolModality.SEALED
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
internal class KotlinWhenPostfixTemplate(provider: PostfixTemplateProvider) : SurroundPostfixTemplateBase(
|
||||
"when", "when (expr)",
|
||||
KtPostfixTemplatePsiInfo, createPostfixExpressionSelector(), provider
|
||||
) {
|
||||
override fun getSurrounder(): KotlinWhenSurrounder = KotlinWhenSurrounder()
|
||||
}
|
||||
@@ -1,4 +1,3 @@
|
||||
// IGNORE_K2
|
||||
fun foo(x: String) {
|
||||
x.when<caret>
|
||||
}
|
||||
|
||||
-1
@@ -1,4 +1,3 @@
|
||||
// IGNORE_K2
|
||||
fun foo(x: String) {
|
||||
when (x) {
|
||||
<caret> -> {}
|
||||
|
||||
-1
@@ -1,4 +1,3 @@
|
||||
// IGNORE_K2
|
||||
fun foo(x: String) {
|
||||
val y = x.when<caret>
|
||||
}
|
||||
|
||||
-1
@@ -1,4 +1,3 @@
|
||||
// IGNORE_K2
|
||||
fun foo(x: String) {
|
||||
val y = when (x) {
|
||||
-> {}
|
||||
|
||||
-1
@@ -1,4 +1,3 @@
|
||||
// IGNORE_K2
|
||||
class C {
|
||||
fun foo() {
|
||||
this.when<caret>
|
||||
|
||||
-1
@@ -1,4 +1,3 @@
|
||||
// IGNORE_K2
|
||||
class C {
|
||||
fun foo() {
|
||||
when (this) {
|
||||
|
||||
-1
@@ -1,4 +1,3 @@
|
||||
// IGNORE_K2
|
||||
fun String.foo() {
|
||||
this.when<caret>
|
||||
}
|
||||
-1
@@ -1,4 +1,3 @@
|
||||
// IGNORE_K2
|
||||
fun String.foo() {
|
||||
when (this) {
|
||||
<caret> -> {}
|
||||
|
||||
+4
-1
@@ -1,5 +1,8 @@
|
||||
fun test(f: Foo) {
|
||||
f.when
|
||||
when (f) {
|
||||
-> {}
|
||||
else -> {}
|
||||
}
|
||||
}
|
||||
|
||||
class Foo {
|
||||
|
||||
Reference in New Issue
Block a user