Skip to content

K
kotlin

硅谷海盗 / kotlin
与 Fork 源项目一致

从无法访问的项目Fork

通知 2
Star 0
Fork 0
K
kotlin

体验新版 GitCode，发现更多精彩内容 >>

提交 45e28bef 编写于 作者: M Mikhail Glukhikh
浏览文件
操作

Enhancement for "join declaration and assignment": now can handle also local... 

Enhancement for "join declaration and assignment": now can handle also local variables, relevant inspection added #KT-12095 Fixed
上级 c44ecded
隐藏空白更改
内联 并排
Showing with 245 addition and 62 deletion
idea/resources/inspectionDescriptions/JoinDeclarationAndAssignment.html 0 → 100644
浏览文件 @ 45e28bef
<html>
<body>
This inspection reports any property declaration that can be joined with the following assignment
</body>
</html>
idea/src/META-INF/plugin.xml
浏览文件 @ 45e28bef
......@@ -1928,6 +1928,14 @@
language="kotlin"
/>
<localInspection implementationClass="org.jetbrains.kotlin.idea.intentions.JoinDeclarationAndAssignmentInspection"
displayName="Join declaration and assignment"
groupName="Kotlin"
enabledByDefault="true"
level="WEAK WARNING"
language="kotlin"
/>
<referenceImporter implementation="org.jetbrains.kotlin.idea.quickfix.KotlinReferenceImporter"/>
<fileType.fileViewProviderFactory filetype="KJSM" implementationClass="com.intellij.psi.ClassFileViewProviderFactory"/>
......
idea/src/org/jetbrains/kotlin/idea/intentions/JoinDeclarationAndAssignmentIntention.kt
浏览文件 @ 45e28bef
......@@ -24,38 +24,44 @@ import com.intellij.psi.util.PsiTreeUtil
import org.jetbrains.kotlin.idea.core.canOmitDeclaredType
import org.jetbrains.kotlin.idea.core.moveCaret
import org.jetbrains.kotlin.idea.core.unblockDocument
import org.jetbrains.kotlin.idea.inspections.IntentionBasedInspection
import org.jetbrains.kotlin.lexer.KtTokens
import org.jetbrains.kotlin.psi.*
import org.jetbrains.kotlin.psi.psiUtil.*
import org.jetbrains.kotlin.utils.addToStdlib.firstIsInstanceOrNull
class JoinDeclarationAndAssignmentIntention :
SelfTargetingIntention<KtBinaryExpression>(KtBinaryExpression::class.java, "Join declaration and assignment") {
override fun isApplicableTo(element: KtBinaryExpression, caretOffset: Int): Boolean {
if (element.operationToken != KtTokens.EQ) {
class JoinDeclarationAndAssignmentInspection : IntentionBasedInspection<KtProperty>(
JoinDeclarationAndAssignmentIntention::class,
"Can be joined with assignment"
)
class JoinDeclarationAndAssignmentIntention : SelfTargetingOffsetIndependentIntention<KtProperty>(
KtProperty::class.java,
"Join declaration and assignment"
) {
override fun isApplicableTo(element: KtProperty): Boolean {
if (element.hasDelegate()
|| element.hasInitializer()
|| element.setter != null
|| element.getter != null
|| element.receiverTypeReference != null
|| element.name == null) {
return false
}
val rightExpression = element.right ?: return false
val initializer = PsiTreeUtil.getParentOfType(element,
KtAnonymousInitializer::class.java,
KtSecondaryConstructor::class.java) ?: return false
val target = findTargetProperty(element)
return target != null && target.initializer == null && target.receiverTypeReference == null &&
target.getNonStrictParentOfType<KtClassOrObject>() == element.getNonStrictParentOfType<KtClassOrObject>() &&
hasNoLocalDependencies(rightExpression, initializer)
val assignment = findAssignment(element) ?: return false
return assignment.right?.let { hasNoLocalDependencies(it, element.parent) } ?: false
}
override fun applyTo(element: KtBinaryExpression, editor: Editor?) {
val property = findTargetProperty(element) ?: return
val initializer = element.right ?: return
val newInitializer = property.setInitializer(initializer)!!
override fun applyTo(element: KtProperty, editor: Editor?) {
val typeReference = element.typeReference ?: return
val assignment = findAssignment(element) ?: return
val initializer = assignment.right ?: return
val newInitializer = element.setInitializer(initializer)!!
val initializerBlock = element.getStrictParentOfType<KtAnonymousInitializer>()
element.delete()
val initializerBlock = assignment.parent?.parent as? KtAnonymousInitializer
assignment.delete()
if (initializerBlock != null && (initializerBlock.body as? KtBlockExpression)?.isEmpty() == true) {
initializerBlock.delete()
}
......@@ -63,11 +69,10 @@ class JoinDeclarationAndAssignmentIntention :
editor?.apply {
unblockDocument()
val typeRef = property.typeReference
if (typeRef != null && property.canOmitDeclaredType(newInitializer, canChangeTypeToSubtype = !property.isVar)) {
val colon = property.colon!!
selectionModel.setSelection(colon.startOffset, typeRef.endOffset)
moveCaret(typeRef.endOffset, ScrollType.CENTER)
if (element.canOmitDeclaredType(newInitializer, canChangeTypeToSubtype = !element.isVar)) {
val colon = element.colon!!
selectionModel.setSelection(colon.startOffset, typeReference.endOffset)
moveCaret(typeReference.endOffset, ScrollType.CENTER)
}
else {
moveCaret(newInitializer.startOffset, ScrollType.CENTER)
......@@ -75,16 +80,41 @@ class JoinDeclarationAndAssignmentIntention :
}
}
private fun findTargetProperty(expr: KtBinaryExpression): KtProperty? {
val leftExpression = expr.left as? KtNameReferenceExpression ?: return null
return leftExpression.resolveAllReferences().firstIsInstanceOrNull<KtProperty>()
}
private fun findAssignment(property: KtProperty): KtBinaryExpression? {
val propertyContainer = property.parent as? KtElement ?: return null
property.typeReference ?: return null
fun KtBlockExpression.isEmpty(): Boolean {
// a block that only contains comments is not empty
return contentRange().isEmpty
val assignments = mutableListOf<KtBinaryExpression>()
fun process(binaryExpr: KtBinaryExpression) {
if (binaryExpr.operationToken != KtTokens.EQ) return
val leftReference = binaryExpr.left as? KtNameReferenceExpression ?: return
if (leftReference.getReferencedName() != property.name) return
assignments += binaryExpr
}
propertyContainer.forEachDescendantOfType(::process)
fun PsiElement?.invalidParent(): Boolean {
when {
this == null -> return true
this === propertyContainer -> return false
else -> {
val grandParent = parent
if (grandParent.parent !== propertyContainer) return true
return grandParent !is KtAnonymousInitializer && grandParent !is KtSecondaryConstructor
}
}
}
if (assignments.any { it.parent.invalidParent() }) return null
val first = assignments.firstOrNull() ?: return null
if (assignments.any { it !== first && it.parent?.parent is KtSecondaryConstructor}) return null
return first
}
// a block that only contains comments is not empty
private fun KtBlockExpression.isEmpty() = contentRange().isEmpty
private fun hasNoLocalDependencies(element: KtElement, localContext: PsiElement): Boolean {
return !element.anyDescendantOfType<PsiElement> { child ->
child.resolveAllReferences().any { it != null && PsiTreeUtil.isAncestor(localContext, it, false) }
......@@ -97,5 +127,3 @@ private fun PsiElement.resolveAllReferences(): Sequence<PsiElement?> {
.asSequence()
.map { it.resolve() }
}
idea/testData/intentions/joinDeclarationAndAssignment/assignmentInIf.kt 0 → 100644
浏览文件 @ 45e28bef
// IS_APPLICABLE: false
fun foo(flag: Boolean) {
var x: Double<caret>
if (flag) {
x = 3.14
}
x = 2.71
}
\ No newline at end of file
idea/testData/intentions/joinDeclarationAndAssignment/cannotRemoveType.kt
浏览文件 @ 45e28bef
class A {
var a: String?
var a<caret>: String?
init {
<caret>a = null
a = null
}
}
idea/testData/intentions/joinDeclarationAndAssignment/cannotRemoveType2.kt
浏览文件 @ 45e28bef
// WITH_RUNTIME
class A {
var a: List<String>
var a<caret>: List<String>
init {
<caret>a = emptyList()
a = emptyList()
}
}
idea/testData/intentions/joinDeclarationAndAssignment/capturedInitialization.kt 0 → 100644
浏览文件 @ 45e28bef
// IS_APPLICABLE: false
fun bar() {
var x: <caret>String
fun foo() {
x = "456"
x.hashCode()
}
foo()
x = "123"
x.hashCode()
}
idea/testData/intentions/joinDeclarationAndAssignment/comment.kt
浏览文件 @ 45e28bef
class A {
var a: Int
var a<caret>: Int
init {
// Initialize a
<caret>a = 1
a = 1
}
}
idea/testData/intentions/joinDeclarationAndAssignment/correctConditionalAssignment.kt 0 → 100644
浏览文件 @ 45e28bef
fun foo(flag: Boolean) {
val x: Double<caret>
x = if (flag) 3.14 else 2.71
}
\ No newline at end of file
idea/testData/intentions/joinDeclarationAndAssignment/correctConditionalAssignment.kt.after 0 → 100644
浏览文件 @ 45e28bef
fun foo(flag: Boolean) {
val x<selection>: Double</selection><caret> = if (flag) 3.14 else 2.71
}
\ No newline at end of file
idea/testData/intentions/joinDeclarationAndAssignment/deleteInitBlock.kt
浏览文件 @ 45e28bef
class A {
var a: Int
var a<caret>: Int
init {
<caret>a = 1
a = 1
}
}
idea/testData/intentions/joinDeclarationAndAssignment/incorrectConditionalAssignment.kt 0 → 100644
浏览文件 @ 45e28bef
// IS_APPLICABLE: false
fun foo() {
val x: Double<caret>
val flag = false
x = if (flag) 3.14 else 2.71
}
\ No newline at end of file
idea/testData/intentions/joinDeclarationAndAssignment/multipleConstructors.kt 0 → 100644
浏览文件 @ 45e28bef
// IS_APPLICABLE: false
class A {
constructor() {
a = 1
}
constructor(aa: Int) {
a = aa
}
val a<caret>: Int
}
idea/testData/intentions/joinDeclarationAndAssignment/propertyReassignment.kt 0 → 100644
浏览文件 @ 45e28bef
class C {
var s<caret>: Int
init {
s = 1
s.hashCode()
s = 2
}
}
\ No newline at end of file
idea/testData/intentions/joinDeclarationAndAssignment/propertyReassignment.kt.after 0 → 100644
浏览文件 @ 45e28bef
class C {
var s<selection>: Int</selection><caret> = 1
init {
s.hashCode()
s = 2
}
}
\ No newline at end of file
idea/testData/intentions/joinDeclarationAndAssignment/simple.kt
浏览文件 @ 45e28bef
class A {
var a: Int
var a<caret>: Int
var b: Int
init {
<caret>a = 1
a = 1
b = 2
}
}
idea/testData/intentions/joinDeclarationAndAssignment/simpleLocal.kt 0 → 100644
浏览文件 @ 45e28bef
fun foo() {
val s: String<caret>
s = "Hello"
}
\ No newline at end of file
idea/testData/intentions/joinDeclarationAndAssignment/simpleLocal.kt.after 0 → 100644
浏览文件 @ 45e28bef
fun foo() {
val s<selection>: String</selection><caret> = "Hello"
}
\ No newline at end of file
idea/testData/intentions/joinDeclarationAndAssignment/singleConstructor.kt 0 → 100644
浏览文件 @ 45e28bef
class A {
constructor() {
a = 1
}
val a<caret>: Int
}
idea/testData/intentions/joinDeclarationAndAssignment/singleConstructor.kt.after 0 → 100644
浏览文件 @ 45e28bef
class A {
constructor() {
}
val a<selection>: Int</selection><caret> = 1
}
idea/testData/intentions/joinDeclarationAndAssignment/usedLocal.kt
浏览文件 @ 45e28bef
// IS_APPLICABLE: false
class A {
var b: Int
var b<caret>: Int
init {
val i = 0
b = <caret>i
b = i
}
}
idea/testData/intentions/joinDeclarationAndAssignment/usedLocal2.kt 0 → 100644
浏览文件 @ 45e28bef
// IS_APPLICABLE: false
fun foo() {
// It's possible here to join with assignment but move the result after declaration of i
var b<caret>: Int
val i = 0
b = i
}
idea/testData/intentions/joinDeclarationAndAssignment/varReassignment.kt 0 → 100644
浏览文件 @ 45e28bef
fun foo() {
var s<caret>: Int
s = 1
s.hashCode()
s = 2
}
\ No newline at end of file
idea/testData/intentions/joinDeclarationAndAssignment/varReassignment.kt.after 0 → 100644
浏览文件 @ 45e28bef
fun foo() {
var s<selection>: Int</selection><caret> = 1
s.hashCode()
s = 2
}
\ No newline at end of file
idea/tests/org/jetbrains/kotlin/idea/intentions/IntentionTestGenerated.java
浏览文件 @ 45e28bef
......@@ -8046,6 +8046,12 @@ public class IntentionTestGenerated extends AbstractIntentionTest {
KotlinTestUtils.assertAllTestsPresentByMetadata(this.getClass(), new File("idea/testData/intentions/joinDeclarationAndAssignment"), Pattern.compile("^([\\w\\-_]+)\\.kt$"), TargetBackend.ANY, true);
}
@TestMetadata("assignmentInIf.kt")
public void testAssignmentInIf() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/assignmentInIf.kt");
doTest(fileName);
}
@TestMetadata("cannotRemoveType.kt")
public void testCannotRemoveType() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/cannotRemoveType.kt");
......@@ -8058,9 +8064,9 @@ public class IntentionTestGenerated extends AbstractIntentionTest {
doTest(fileName);
}
@TestMetadata("cannotRemoveType3.kt")
public void testCannotRemoveType3() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/cannotRemoveType3.kt");
@TestMetadata("capturedInitialization.kt")
public void testCapturedInitialization() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/capturedInitialization.kt");
doTest(fileName);
}
......@@ -8070,23 +8076,71 @@ public class IntentionTestGenerated extends AbstractIntentionTest {
doTest(fileName);
}
@TestMetadata("correctConditionalAssignment.kt")
public void testCorrectConditionalAssignment() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/correctConditionalAssignment.kt");
doTest(fileName);
}
@TestMetadata("deleteInitBlock.kt")
public void testDeleteInitBlock() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/deleteInitBlock.kt");
doTest(fileName);
}
@TestMetadata("incorrectConditionalAssignment.kt")
public void testIncorrectConditionalAssignment() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/incorrectConditionalAssignment.kt");
doTest(fileName);
}
@TestMetadata("multipleConstructors.kt")
public void testMultipleConstructors() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/multipleConstructors.kt");
doTest(fileName);
}
@TestMetadata("propertyReassignment.kt")
public void testPropertyReassignment() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/propertyReassignment.kt");
doTest(fileName);
}
@TestMetadata("simple.kt")
public void testSimple() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/simple.kt");
doTest(fileName);
}
@TestMetadata("simpleLocal.kt")
public void testSimpleLocal() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/simpleLocal.kt");
doTest(fileName);
}
@TestMetadata("singleConstructor.kt")
public void testSingleConstructor() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/singleConstructor.kt");
doTest(fileName);
}
@TestMetadata("usedLocal.kt")
public void testUsedLocal() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/usedLocal.kt");
doTest(fileName);
}
@TestMetadata("usedLocal2.kt")
public void testUsedLocal2() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/usedLocal2.kt");
doTest(fileName);
}
@TestMetadata("varReassignment.kt")
public void testVarReassignment() throws Exception {
String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/joinDeclarationAndAssignment/varReassignment.kt");
doTest(fileName);
}
}
@TestMetadata("idea/testData/intentions/loopToCallChain")
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或