Create From Usage: Do not generate unnecessary/skip necessary empty lines
This commit is contained in:
Alexey Sedunov
@@ -109,10 +109,15 @@ public class JetPsiFactory(private val project: Project) {
|
||||
return createProperty("val${text}x").findElementAt(3)!!
|
||||
}
|
||||
|
||||
// Remove when all Java usages are rewritten to Kotlin
|
||||
public fun createNewLine(): PsiElement {
|
||||
return createWhiteSpace("\n")
|
||||
}
|
||||
|
||||
public fun createNewLine(lineBreaks: Int): PsiElement {
|
||||
return createWhiteSpace("\n".repeat(lineBreaks))
|
||||
}
|
||||
|
||||
public fun createClass(text: String): JetClass {
|
||||
return createDeclaration(text)
|
||||
}
|
||||
|
||||
+41
-36
@@ -66,6 +66,8 @@ import org.jetbrains.jet.plugin.util.IdeDescriptorRenderers
|
||||
import org.jetbrains.jet.plugin.quickfix.createFromUsage.createClass.ClassKind
|
||||
import org.jetbrains.jet.plugin.util.isAny
|
||||
import org.jetbrains.jet.utils.addToStdlib.singletonOrEmptyList
|
||||
import org.jetbrains.jet.lang.psi.psiUtil.siblings
|
||||
import org.jetbrains.jet.plugin.refactoring.getLineCount
|
||||
|
||||
private val TYPE_PARAMETER_LIST_VARIABLE_NAME = "typeParameterList"
|
||||
private val TEMPLATE_FROM_USAGE_FUNCTION_BODY = "New Kotlin Function Body.kt"
|
||||
@@ -427,40 +429,31 @@ class CallableBuilder(val config: CallableBuilderConfiguration) {
|
||||
|
||||
val newLine = psiFactory.createNewLine()
|
||||
|
||||
fun prepend(element: PsiElement, elementBeforeStart: PsiElement, skipInitial: Boolean): PsiElement {
|
||||
val parent = elementBeforeStart.getParent()!!
|
||||
val anchor =
|
||||
if (!skipInitial && elementBeforeStart !is PsiWhiteSpace) {
|
||||
elementBeforeStart
|
||||
fun calcNecessaryEmptyLines(decl: JetDeclaration, after: Boolean): Int {
|
||||
var lineBreaksPresent: Int = 0
|
||||
var neighbor: PsiElement? = null
|
||||
for (sibling in decl.siblings(forward = after, withItself = false)) {
|
||||
when (sibling) {
|
||||
is PsiWhiteSpace -> lineBreaksPresent += (sibling.getText() ?: "").count { it == '\n' }
|
||||
else -> {
|
||||
neighbor = sibling
|
||||
break
|
||||
}
|
||||
else {
|
||||
PsiTreeUtil.skipSiblingsForward(elementBeforeStart, javaClass<PsiWhiteSpace>())
|
||||
}
|
||||
val addedElement = parent.addBefore(element, anchor)!!
|
||||
parent.addAfter(newLine, addedElement)
|
||||
parent.addAfter(newLine, addedElement)
|
||||
return addedElement
|
||||
}
|
||||
|
||||
fun append(element: PsiElement, elementAfterEnd: PsiElement, skipInitial: Boolean): PsiElement {
|
||||
val parent = elementAfterEnd.getParent()!!
|
||||
val anchor =
|
||||
if (!skipInitial && elementAfterEnd !is PsiWhiteSpace) {
|
||||
elementAfterEnd
|
||||
}
|
||||
else {
|
||||
PsiTreeUtil.skipSiblingsBackward(elementAfterEnd, javaClass<PsiWhiteSpace>())
|
||||
}
|
||||
val addedElement = parent.addAfter(element, anchor)!!
|
||||
if (anchor?.getNode()?.getElementType() != JetTokens.LBRACE) {
|
||||
parent.addAfter(newLine, anchor)
|
||||
parent.addAfter(newLine, anchor)
|
||||
}
|
||||
}
|
||||
return addedElement
|
||||
|
||||
val neighborType = neighbor?.getNode()?.getElementType()
|
||||
val lineBreaksNeeded = when {
|
||||
neighborType == JetTokens.LBRACE, neighborType == JetTokens.RBRACE -> 1
|
||||
neighbor is JetDeclaration && (neighbor !is JetProperty || decl !is JetProperty) -> 2
|
||||
else -> 1
|
||||
}
|
||||
|
||||
return Math.max(lineBreaksNeeded - lineBreaksPresent, 0)
|
||||
}
|
||||
|
||||
when (containingElement) {
|
||||
is JetFile -> return append(declaration, containingElement.getLastChild()!!, false) as JetNamedDeclaration
|
||||
val declarationInPlace = when (containingElement) {
|
||||
is JetFile -> containingElement.add(declaration) as JetNamedDeclaration
|
||||
|
||||
is JetClassOrObject -> {
|
||||
var classBody = containingElement.getBody()
|
||||
@@ -470,11 +463,13 @@ class CallableBuilder(val config: CallableBuilderConfiguration) {
|
||||
}
|
||||
|
||||
if (declaration is JetNamedFunction) {
|
||||
val rBrace = classBody!!.getRBrace()
|
||||
return (rBrace?.let { append(declaration, it, true) }
|
||||
?: append(declaration, classBody!!.getLastChild()!!, false)) as JetNamedDeclaration
|
||||
val anchor = PsiTreeUtil.skipSiblingsBackward(
|
||||
classBody!!.getRBrace() ?: classBody!!.getLastChild()!!,
|
||||
javaClass<PsiWhiteSpace>()
|
||||
)
|
||||
classBody.addAfter(declaration, anchor) as JetNamedDeclaration
|
||||
}
|
||||
return prepend(declaration, classBody!!.getLBrace()!!, true) as JetNamedDeclaration
|
||||
else classBody.addAfter(declaration, classBody!!.getLBrace()!!) as JetNamedDeclaration
|
||||
}
|
||||
|
||||
is JetBlockExpression -> {
|
||||
@@ -485,13 +480,23 @@ class CallableBuilder(val config: CallableBuilderConfiguration) {
|
||||
parent.addAfter(newLine, containingElement)
|
||||
}
|
||||
|
||||
return prepend(declaration, containingElement.getFirstChild()!!, false) as JetNamedDeclaration
|
||||
containingElement.addBefore(declaration, containingElement.getFirstChild()!!) as JetNamedDeclaration
|
||||
}
|
||||
return prepend(declaration, containingElement.getLBrace()!!, true) as JetNamedDeclaration
|
||||
else containingElement.addAfter(declaration, containingElement.getLBrace()!!) as JetNamedDeclaration
|
||||
}
|
||||
|
||||
else -> throw AssertionError("Invalid containing element: ${containingElement.getText()}")
|
||||
}
|
||||
|
||||
val parent = declarationInPlace.getParent()
|
||||
calcNecessaryEmptyLines(declarationInPlace, false).let {
|
||||
if (it > 0) parent.addBefore(psiFactory.createNewLine(it), declarationInPlace)
|
||||
}
|
||||
calcNecessaryEmptyLines(declarationInPlace, true).let {
|
||||
if (it > 0) parent.addAfter(psiFactory.createNewLine(it), declarationInPlace)
|
||||
}
|
||||
|
||||
return declarationInPlace
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
-1
@@ -1,7 +1,6 @@
|
||||
// "Create class 'Foo'" "true"
|
||||
|
||||
class A<T>(val n: T) {
|
||||
|
||||
class Foo(i: Int) {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -3,7 +3,6 @@
|
||||
// ERROR: <html>Type mismatch.<table><tr><td>Required:</td><td>U</td></tr><tr><td>Found:</td><td>U</td></tr></table></html>
|
||||
|
||||
class A<T>(val n: T) {
|
||||
|
||||
inner class Foo<U>(u: U) {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -1,7 +1,6 @@
|
||||
// "Create class 'Foo'" "true"
|
||||
|
||||
object A {
|
||||
|
||||
class Foo(i: Int) {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -1,7 +1,6 @@
|
||||
// "Create class 'Foo'" "true"
|
||||
|
||||
class A<T>(val n: T) {
|
||||
|
||||
inner class Foo(i: Int) {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -1,7 +1,6 @@
|
||||
// "Create class 'Foo'" "true"
|
||||
|
||||
class A<T>(val n: T) {
|
||||
|
||||
inner class Foo(i: Int, s: String) {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -1,7 +1,6 @@
|
||||
// "Create class 'Foo'" "true"
|
||||
|
||||
class B<T>(val t: T) {
|
||||
|
||||
class Foo<U, V>(u: U, v: V) {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -3,7 +3,6 @@
|
||||
// ERROR: An integer literal does not conform to the expected type U
|
||||
|
||||
class B<T>(val t: T) {
|
||||
|
||||
inner class Foo<U, V>(u: U, v: V) {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -2,7 +2,6 @@
|
||||
// ERROR: <html>Type mismatch.<table><tr><td>Required:</td><td>U</td></tr><tr><td>Found:</td><td>kotlin.String</td></tr></table></html>
|
||||
|
||||
class B<T>(val t: T) {
|
||||
|
||||
inner class Foo<U>(i: Int, u: U) {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -3,7 +3,6 @@
|
||||
// ERROR: An integer literal does not conform to the expected type V
|
||||
|
||||
class B<T>(val t: T) {
|
||||
|
||||
inner class Foo<U, V, W>(v: V, w: W) {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -1,7 +1,6 @@
|
||||
// "Create class 'Foo'" "true"
|
||||
|
||||
class B<T>(val t: T) {
|
||||
|
||||
class Foo<U>(i: Int, u: U) {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -2,7 +2,6 @@
|
||||
package p
|
||||
|
||||
class X {
|
||||
|
||||
open class A(i: Int, s: String) {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -4,7 +4,6 @@ package p
|
||||
import p.X.A
|
||||
|
||||
class X {
|
||||
|
||||
annotation class A
|
||||
|
||||
}
|
||||
-1
@@ -4,7 +4,6 @@ package p
|
||||
import p.X.A
|
||||
|
||||
class X {
|
||||
|
||||
class A {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -4,7 +4,6 @@ package p
|
||||
import p.E.A
|
||||
|
||||
enum class E {
|
||||
|
||||
A
|
||||
|
||||
}
|
||||
-1
@@ -4,7 +4,6 @@ package p
|
||||
import p.X.A
|
||||
|
||||
class X {
|
||||
|
||||
enum class A {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -4,7 +4,6 @@ package p
|
||||
import p.X.A
|
||||
|
||||
class X {
|
||||
|
||||
object A {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -4,7 +4,6 @@ package p
|
||||
import p.X.A
|
||||
|
||||
class X {
|
||||
|
||||
trait A {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -5,7 +5,6 @@ package p
|
||||
fun foo() = A.B.C
|
||||
|
||||
class A {
|
||||
|
||||
class B {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -4,7 +4,6 @@ package p
|
||||
fun foo() = X.A
|
||||
|
||||
enum class X {
|
||||
|
||||
A
|
||||
|
||||
}
|
||||
-1
@@ -5,7 +5,6 @@ package p
|
||||
fun foo() = X.A
|
||||
|
||||
enum class X(n: Int) {
|
||||
|
||||
A : X()
|
||||
|
||||
}
|
||||
-1
@@ -4,7 +4,6 @@ package p
|
||||
fun foo(): E = E.A
|
||||
|
||||
enum class E {
|
||||
|
||||
A
|
||||
|
||||
}
|
||||
-1
@@ -4,7 +4,6 @@ package p
|
||||
fun foo() = X.A
|
||||
|
||||
class X {
|
||||
|
||||
object A {
|
||||
|
||||
}
|
||||
|
||||
-1
@@ -2,7 +2,6 @@
|
||||
package p
|
||||
|
||||
class T {
|
||||
|
||||
class A {
|
||||
|
||||
}
|
||||
|
||||
+10
@@ -0,0 +1,10 @@
|
||||
// "Create class 'A'" "true"
|
||||
package p
|
||||
|
||||
class T {
|
||||
class A {
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
fun foo(): T.A = throw Throwable("")
|
||||
+6
@@ -0,0 +1,6 @@
|
||||
// "Create class 'A'" "true"
|
||||
package p
|
||||
|
||||
class T
|
||||
|
||||
fun foo(): T.<caret>A = throw Throwable("")
|
||||
@@ -5,7 +5,6 @@
|
||||
class A {
|
||||
val t: Int get() {
|
||||
val foo: Int
|
||||
|
||||
return foo
|
||||
}
|
||||
}
|
||||
-1
@@ -5,7 +5,6 @@
|
||||
class A {
|
||||
val t: Int get() {
|
||||
val foo: Int
|
||||
|
||||
return foo
|
||||
}
|
||||
}
|
||||
@@ -4,6 +4,5 @@
|
||||
|
||||
fun test(): Int {
|
||||
val foo: Int
|
||||
|
||||
return foo
|
||||
}
|
||||
-1
@@ -4,6 +4,5 @@
|
||||
|
||||
fun test(): Int {
|
||||
val foo: Int
|
||||
|
||||
return foo
|
||||
}
|
||||
-1
@@ -5,7 +5,6 @@
|
||||
fun test(n: Int) {
|
||||
val f: () -> Int = {
|
||||
val foo: Int
|
||||
|
||||
foo
|
||||
}
|
||||
}
|
||||
-1
@@ -5,7 +5,6 @@
|
||||
fun test(n: Int) {
|
||||
val f: (Int, Int) -> Int = { (a, b) ->
|
||||
val foo: Int
|
||||
|
||||
foo
|
||||
}
|
||||
}
|
||||
-1
@@ -5,7 +5,6 @@
|
||||
fun test(n: Int) {
|
||||
val f: () -> Int = {
|
||||
val foo: Int
|
||||
|
||||
foo
|
||||
}
|
||||
}
|
||||
-1
@@ -5,7 +5,6 @@
|
||||
fun test(n: Int) {
|
||||
val f: (Int, Int) -> Int = { (a, b) ->
|
||||
val foo: Int
|
||||
|
||||
foo
|
||||
}
|
||||
}
|
||||
@@ -5,7 +5,6 @@ fun test(n: Int): Int {
|
||||
return when (n) {
|
||||
1 -> {
|
||||
val foo: Int
|
||||
|
||||
foo
|
||||
}
|
||||
else -> {
|
||||
|
||||
-1
@@ -6,6 +6,5 @@ package foo
|
||||
|
||||
fun test(): Int {
|
||||
val foo: Int
|
||||
|
||||
return foo
|
||||
}
|
||||
-1
@@ -3,6 +3,5 @@
|
||||
|
||||
fun test(): Int? {
|
||||
val foo: Int?
|
||||
|
||||
return foo
|
||||
}
|
||||
-1
@@ -4,6 +4,5 @@
|
||||
|
||||
fun test() {
|
||||
val foo: Unit
|
||||
|
||||
val u: Unit = foo
|
||||
}
|
||||
-1
@@ -4,7 +4,6 @@
|
||||
|
||||
class A<T>(val n: T) {
|
||||
val foo: Any
|
||||
|
||||
}
|
||||
|
||||
fun test(): Int {
|
||||
|
||||
-1
@@ -6,6 +6,5 @@ import kotlin.properties.ReadOnlyProperty
|
||||
|
||||
class A<T> {
|
||||
val foo: ReadOnlyProperty<A<T>, A<Int>>
|
||||
|
||||
val x: A<Int> by foo
|
||||
}
|
||||
|
||||
-1
@@ -6,6 +6,5 @@ import kotlin.properties.ReadWriteProperty
|
||||
|
||||
class A<T> {
|
||||
val foo: ReadWriteProperty<A<T>, A<Int>>
|
||||
|
||||
var x: A<Int> by foo
|
||||
}
|
||||
|
||||
@@ -3,7 +3,6 @@
|
||||
|
||||
class A<T>(val n: T) {
|
||||
val foo: A<Int>
|
||||
|
||||
}
|
||||
|
||||
fun <U> A<U>.test(): A<Int> {
|
||||
|
||||
-1
@@ -3,7 +3,6 @@
|
||||
|
||||
class A<T>(val n: T) {
|
||||
class object {
|
||||
|
||||
val foo: Int
|
||||
|
||||
}
|
||||
|
||||
@@ -3,7 +3,6 @@
|
||||
|
||||
object A {
|
||||
val foo: Int
|
||||
|
||||
}
|
||||
|
||||
fun test() {
|
||||
|
||||
@@ -3,7 +3,6 @@
|
||||
|
||||
class A<T>(val n: T) {
|
||||
val foo: A<T>
|
||||
|
||||
}
|
||||
|
||||
fun test() {
|
||||
|
||||
-1
@@ -3,7 +3,6 @@
|
||||
|
||||
class A<T>(val n: T) {
|
||||
val foo: A<T>
|
||||
|
||||
}
|
||||
|
||||
fun test<U>(u: U) {
|
||||
|
||||
-1
@@ -3,7 +3,6 @@
|
||||
|
||||
class A<T>(val n: T) {
|
||||
val foo: A<T>?
|
||||
|
||||
}
|
||||
|
||||
fun test() {
|
||||
|
||||
@@ -3,7 +3,6 @@
|
||||
|
||||
class A<T>(val n: T) {
|
||||
var foo: String
|
||||
|
||||
}
|
||||
|
||||
fun test() {
|
||||
|
||||
@@ -1334,6 +1334,12 @@ public class QuickFixTestGenerated extends AbstractQuickFixTest {
|
||||
doTest(fileName);
|
||||
}
|
||||
|
||||
@TestMetadata("beforeClassUserTypeReceiverNoBody.kt")
|
||||
public void testClassUserTypeReceiverNoBody() throws Exception {
|
||||
String fileName = JetTestUtils.navigationMetadata("idea/testData/quickfix/createFromUsage/createClass/typeReference/beforeClassUserTypeReceiverNoBody.kt");
|
||||
doTest(fileName);
|
||||
}
|
||||
|
||||
@TestMetadata("beforeEnumEntryNotQualifierNoTypeArgs.kt")
|
||||
public void testEnumEntryNotQualifierNoTypeArgs() throws Exception {
|
||||
String fileName = JetTestUtils.navigationMetadata("idea/testData/quickfix/createFromUsage/createClass/typeReference/beforeEnumEntryNotQualifierNoTypeArgs.kt");
|
||||
|
||||
Reference in New Issue
Block a user