(feat) basic rename

sveltejs#110
dummdidumm · Jun 6, 2020 · 675d4e1 · 675d4e1
1 parent 58c1c59
commit 675d4e1
Show file tree

Hide file tree

Showing 4 changed files with 84 additions and 1 deletion.
diff --git a/packages/language-server/src/plugins/PluginHost.ts b/packages/language-server/src/plugins/PluginHost.ts
@@ -237,6 +237,23 @@ export class PluginHost implements LSProvider, OnWatchFileChanges {
         );
     }
 
+    async rename(
+        textDocument: TextDocumentIdentifier,
+        position: Position,
+        newName: string,
+    ): Promise<WorkspaceEdit | null> {
+        const document = this.getDocument(textDocument.uri);
+        if (!document) {
+            throw new Error('Cannot call methods on an unopened document');
+        }
+
+        return await this.execute<any>(
+            'rename',
+            [document, position, newName],
+            ExecuteMode.FirstNonNull,
+        );
+    }
+
     onWatchFileChanges(fileName: string, changeType: FileChangeType): void {
         for (const support of this.plugins) {
             support.onWatchFileChanges?.(fileName, changeType);

diff --git a/packages/language-server/src/plugins/interfaces.ts b/packages/language-server/src/plugins/interfaces.ts
@@ -95,6 +95,14 @@ export interface UpdateImportsProvider {
     updateImports(fileRename: FileRename): Resolvable<WorkspaceEdit | null>;
 }
 
+export interface RenameProvider {
+    rename(
+        document: Document,
+        position: Position,
+        newName: string,
+    ): Resolvable<WorkspaceEdit | null>;
+}
+
 export interface OnWatchFileChanges {
     onWatchFileChanges(fileName: string, changeType: FileChangeType): void;
 }
@@ -109,6 +117,7 @@ export type LSProvider = DiagnosticsProvider &
     DocumentSymbolsProvider &
     DefinitionsProvider &
     UpdateImportsProvider &
-    CodeActionsProvider;
+    CodeActionsProvider &
+    RenameProvider;
 
 export type Plugin = Partial<LSProvider & OnWatchFileChanges>;
diff --git a/packages/language-server/src/plugins/typescript/TypeScriptPlugin.ts b/packages/language-server/src/plugins/typescript/TypeScriptPlugin.ts
@@ -17,6 +17,7 @@ import {
     Document,
     DocumentManager,
     mapHoverToParent,
+    mapRangeToOriginal,
     mapSymbolInformationToOriginal,
 } from '../../lib/documents';
 import { LSConfigManager, LSTypescriptConfig } from '../../ls-config';
@@ -32,6 +33,7 @@ import {
     FileRename,
     HoverProvider,
     OnWatchFileChanges,
+    RenameProvider,
     UpdateImportsProvider,
 } from '../interfaces';
 import { DocumentSnapshot, SnapshotFragment } from './DocumentSnapshot';
@@ -58,6 +60,7 @@ export class TypeScriptPlugin
         DefinitionsProvider,
         CodeActionsProvider,
         UpdateImportsProvider,
+        RenameProvider,
         OnWatchFileChanges,
         CompletionsProvider<CompletionEntryWithIdentifer> {
     private readonly configManager: LSConfigManager;
@@ -227,6 +230,55 @@ export class TypeScriptPlugin
         );
     }
 
+    async rename(
+        document: Document,
+        position: Position,
+        newName: string,
+    ): Promise<WorkspaceEdit | null> {
+        const { lang, tsDoc } = this.getLSAndTSDoc(document);
+        const fragment = await tsDoc.getFragment();
+
+        const renameLocations = lang.findRenameLocations(
+            tsDoc.filePath,
+            fragment.offsetAt(fragment.getGeneratedPosition(position)),
+            true,
+            false,
+        );
+        if (!renameLocations) {
+            return null;
+        }
+
+        const docs = new Map<string, SnapshotFragment>([[tsDoc.filePath, fragment]]);
+        const convertedRenameLocations = await Promise.all(
+            renameLocations.map(async (loc) => {
+                let doc = docs.get(loc.fileName);
+                if (!doc) {
+                    doc = await this.getSnapshot(loc.fileName).getFragment();
+                    docs.set(loc.fileName, doc);
+                }
+
+                return {
+                    ...loc,
+                    range: mapRangeToOriginal(doc, convertRange(doc, loc.textSpan)),
+                };
+            }),
+        );
+
+        return convertedRenameLocations
+            .filter((loc) => loc.range.start.line >= 0 && loc.range.end.line >= 0)
+            .reduce(
+                (acc, loc) => {
+                    const uri = pathToUrl(loc.fileName);
+                    if (!acc.changes[uri]) {
+                        acc.changes[uri] = [];
+                    }
+                    acc.changes[uri].push({ newText: newName, range: loc.range });
+                    return acc;
+                },
+                <Required<Pick<WorkspaceEdit, 'changes'>>>{ changes: {} },
+            );
+    }
+
     async getCodeActions(
         document: Document,
         range: Range,

diff --git a/packages/language-server/src/server.ts b/packages/language-server/src/server.ts
@@ -130,10 +130,15 @@ export function startServer(options?: LSOptions) {
                           ],
                       }
                     : true,
+                renameProvider: true,
             },
         };
     });
 
+    connection.onRenameRequest((req) =>
+        pluginHost.rename(req.textDocument, req.position, req.newName),
+    );
+
     connection.onDidChangeConfiguration(({ settings }) => {
         pluginHost.updateConfig(settings.svelte?.plugin);
     });