Fix issue where nothing happens on empty body for implement abstract

members quick fix

Author: themkat

server/src/main/kotlin/org/javacs/kt/codeaction/quickfix/ImplementAbstractMembersQuickFix.kt

@@ -53,14 +53,17 @@ class ImplementAbstractMembersQuickFix : QuickFix {
                 val uri = file.parse.toPath().toUri().toString()
                 // Get the padding to be introduced before the function declarations
                 val padding = getDeclarationPadding(file, kotlinClass)
+
                 // Get the location where the new code will be placed
                 val newFunctionStartPosition = getNewFunctionStartPosition(file, kotlinClass)
+                val bodyAppendBeginning = listOf(TextEdit(Range(newFunctionStartPosition, newFunctionStartPosition), "{")).takeIf { kotlinClass.hasNoBody() } ?: emptyList()
+                val bodyAppendEnd = listOf(TextEdit(Range(newFunctionStartPosition, newFunctionStartPosition), System.lineSeparator() + "}")).takeIf { kotlinClass.hasNoBody() } ?: emptyList()
 
-                val textEdits = functionsToImplement.map {
+                val textEdits = bodyAppendBeginning + functionsToImplement.map {
                     // We leave two new lines before the function is inserted
                     val newText = System.lineSeparator() + System.lineSeparator() + padding + it
                     TextEdit(Range(newFunctionStartPosition, newFunctionStartPosition), newText)
-                }
+                } + bodyAppendEnd
 
                 val codeAction = CodeAction()
                 codeAction.edit = WorkspaceEdit(mapOf(uri to textEdits))
@@ -221,6 +224,11 @@ private fun getNewFunctionStartPosition(file: CompiledFile, kotlinClass: KtClass
         if (body != null) {
             position(file.content, body.startOffset + 1)
         } else {
-            null
+            // function has no body. We have to create one. New position is right after entire kotlin class text (with space)
+            val newPosCorrectLine = position(file.content, kotlinClass.startOffset + 1)
+            newPosCorrectLine.character = (kotlinClass.text.length + 2)
+            newPosCorrectLine
         }
     }
+
+private fun KtClass.hasNoBody() = null == this.body

server/src/test/kotlin/org/javacs/kt/QuickFixesTest.kt

@@ -241,6 +241,36 @@ class ImplementAbstractMembersQuickFixSameFileTest : SingleFileTestFixture("quic
         assertThat(memberToImplementEdit?.range, equalTo(range(38, 31, 38, 31)))
         assertThat(memberToImplementEdit?.newText, equalTo(System.lineSeparator() + System.lineSeparator() + "    override fun myFun() { }"))
     }
+
+    @Test
+    fun `should find abstract members when class has no body (square brackets)`() {
+        val only = listOf(CodeActionKind.QuickFix)
+        val codeActionParams = codeActionParams(file, 47, 1, 47, 12, diagnostics, only)
+
+        val codeActionResult = languageServer.textDocumentService.codeAction(codeActionParams).get()
+
+        assertThat(codeActionResult, hasSize(1))
+        val codeAction = codeActionResult[0].right
+        assertThat(codeAction.kind, equalTo(CodeActionKind.QuickFix))
+        assertThat(codeAction.title, equalTo("Implement abstract members"))
+
+        val textEdit = codeAction.edit.changes
+        val key = workspaceRoot.resolve(file).toUri().toString()
+        assertThat(textEdit.containsKey(key), equalTo(true))
+        assertThat(textEdit[key], hasSize(3))
+
+        val firstMemberToImplementEdit = textEdit[key]?.get(0)
+        assertThat(firstMemberToImplementEdit?.range, equalTo(range(47, 23, 47, 23)))
+        assertThat(firstMemberToImplementEdit?.newText, equalTo("{"))
+
+        val secondMemberToImplementEdit = textEdit[key]?.get(1)
+        assertThat(secondMemberToImplementEdit?.range, equalTo(range(47, 23, 47, 23)))
+        assertThat(secondMemberToImplementEdit?.newText, equalTo(System.lineSeparator() + System.lineSeparator() + "    override fun behaviour() { }"))
+
+        val thirdMemberToImplementEdit = textEdit[key]?.get(2)
+        assertThat(thirdMemberToImplementEdit?.range, equalTo(range(47, 23, 47, 23)))
+        assertThat(thirdMemberToImplementEdit?.newText, equalTo(System.lineSeparator() + "}"))
+    }
 }
 
 class ImplementAbstractMembersQuickFixExternalLibraryTest : SingleFileTestFixture("quickfixes", "standardlib.kt") {

server/src/test/resources/quickfixes/samefile.kt

@@ -37,3 +37,11 @@ class MyImplClass : MyAbstract() {}
 class My2ndClass : MyAbstract() {
     override val name = "Nils"
 }
+
+
+// defect GH-366, part of the solution
+interface IThing {
+    fun behaviour
+}
+
+class Thing : IThing