Added rename

Author: daplf

server/src/main/kotlin/org/javacs/kt/KotlinLanguageServer.kt

@@ -76,6 +76,7 @@ class KotlinLanguageServer : LanguageServer, LanguageClientAware, Closeable {
         serverCapabilities.workspace.workspaceFolders.supported = true
         serverCapabilities.workspace.workspaceFolders.changeNotifications = Either.forRight(true)
         serverCapabilities.hoverProvider = Either.forLeft(true)
+        serverCapabilities.renameProvider = Either.forLeft(true)
         serverCapabilities.completionProvider = CompletionOptions(false, listOf("."))
         serverCapabilities.signatureHelpProvider = SignatureHelpOptions(listOf("(", ","))
         serverCapabilities.definitionProvider = Either.forLeft(true)

server/src/main/kotlin/org/javacs/kt/KotlinTextDocumentService.kt

@@ -26,6 +26,7 @@ import org.javacs.kt.util.parseURI
 import org.javacs.kt.util.describeURI
 import org.javacs.kt.util.describeURIs
 import org.javacs.kt.commands.JAVA_TO_KOTLIN_COMMAND
+import org.javacs.kt.rename.renameSymbol
 import org.jetbrains.kotlin.resolve.diagnostics.Diagnostics
 import java.net.URI
 import java.io.Closeable
@@ -144,8 +145,9 @@ class KotlinTextDocumentService(
         TODO("not implemented")
     }
 
-    override fun rename(params: RenameParams): CompletableFuture<WorkspaceEdit> {
-        TODO("not implemented")
+    override fun rename(params: RenameParams) = async.compute {
+        val (file, cursor) = recover(params, Recompile.NEVER)
+        renameSymbol(file, cursor, sp, params.newName)
     }
 
     override fun completion(position: CompletionParams) = async.compute {

server/src/main/kotlin/org/javacs/kt/references/FindReferences.kt

@@ -4,6 +4,7 @@ import org.eclipse.lsp4j.Location
 import org.javacs.kt.LOG
 import org.javacs.kt.SourcePath
 import org.javacs.kt.position.location
+import org.javacs.kt.position.toURIString
 import org.javacs.kt.util.emptyResult
 import org.javacs.kt.util.findParent
 import org.javacs.kt.util.preOrderTraversal
@@ -31,9 +32,22 @@ fun findReferences(file: Path, cursor: Int, sp: SourcePath): List<Location> {
             .sortedWith(compareBy({ it.getUri() }, { it.getRange().getStart().getLine() }))
 }
 
+fun findReferences(declaration: KtNamedDeclaration, sp: SourcePath): List<Location> {
+    return doFindReferences(declaration, sp)
+        .map { location(it) }
+        .filterNotNull()
+        .toList()
+        .sortedWith(compareBy({ it.getUri() }, { it.getRange().getStart().getLine() }))
+}
+
 private fun doFindReferences(file: Path, cursor: Int, sp: SourcePath): Collection<KtElement> {
     val recover = sp.currentVersion(file.toUri())
     val element = recover.elementAtPoint(cursor)?.findParent<KtNamedDeclaration>() ?: return emptyResult("No declaration at ${recover.describePosition(cursor)}")
+    return doFindReferences(element, sp)
+}
+
+private fun doFindReferences(element: KtNamedDeclaration, sp: SourcePath): Collection<KtElement> {
+    val recover = sp.currentVersion(element.containingFile.toPath().toUri())
     val declaration = recover.compile[BindingContext.DECLARATION_TO_DESCRIPTOR, element] ?: return emptyResult("Declaration ${element.fqName} has no descriptor")
     val maybes = possibleReferences(declaration, sp).map { it.toPath() }
     LOG.debug("Scanning {} files for references to {}", maybes.size, element.fqName)

server/src/main/kotlin/org/javacs/kt/rename/Rename.kt

@@ -0,0 +1,44 @@
+package org.javacs.kt.rename
+
+import org.eclipse.lsp4j.*
+import org.eclipse.lsp4j.jsonrpc.messages.Either
+import org.javacs.kt.CompiledFile
+import org.javacs.kt.SourcePath
+import org.javacs.kt.position.location
+import org.javacs.kt.references.findReferences
+import org.jetbrains.kotlin.js.resolve.diagnostics.findPsi
+import org.jetbrains.kotlin.psi.KtNamedDeclaration
+
+fun renameSymbol(file: CompiledFile, cursor: Int, sp: SourcePath, newName: String): WorkspaceEdit? {
+    val (declaration, location) = findDeclaration(file, cursor) ?: return null
+    return declaration.let {
+        val declarationEdit = Either.forLeft<TextDocumentEdit, ResourceOperation>(TextDocumentEdit(
+            VersionedTextDocumentIdentifier().apply { uri = location.uri },
+            listOf(TextEdit(location.range, newName))
+        ))
+
+        val referenceEdits = findReferences(declaration, sp).map {
+            Either.forLeft<TextDocumentEdit, ResourceOperation>(TextDocumentEdit(
+                VersionedTextDocumentIdentifier().apply { uri = it.uri },
+                listOf(TextEdit(it.range, newName))
+            ))
+        }
+
+        WorkspaceEdit(listOf(declarationEdit) + referenceEdits)
+    }
+}
+
+private fun findDeclaration(file: CompiledFile, cursor: Int): Pair<KtNamedDeclaration, Location>? {
+    val (_, target) = file.referenceAtPoint(cursor) ?: return null
+    val psi = target.findPsi()
+
+    return if (psi is KtNamedDeclaration) {
+        psi.nameIdentifier?.let {
+            location(it)?.let { location ->
+                Pair(psi, location)
+            }
+        }
+    } else {
+        null
+    }
+}