editors/code/src/commands/apply_source_change.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

import * as vscode from 'vscode';
import * as lc from 'vscode-languageclient';

import { Server } from '../server';

interface FileSystemEdit {
    type: string;
    uri?: string;
    src?: string;
    dst?: string;
}

export interface SourceChange {
    label: string;
    sourceFileEdits: lc.TextDocumentEdit[];
    fileSystemEdits: FileSystemEdit[];
    cursorPosition?: lc.TextDocumentPositionParams;
}

export async function handle(change: SourceChange) {
    console.log(`applySOurceChange ${JSON.stringify(change)}`);
    const wsEdit = new vscode.WorkspaceEdit();
    for (const sourceEdit of change.sourceFileEdits) {
        const uri = Server.client.protocol2CodeConverter.asUri(sourceEdit.textDocument.uri);
        const edits = Server.client.protocol2CodeConverter.asTextEdits(sourceEdit.edits);
        wsEdit.set(uri, edits);
    }
    let created;
    let moved;
    for (const fsEdit of change.fileSystemEdits) {
        if (fsEdit.type == 'createFile') {
            const uri = vscode.Uri.parse(fsEdit.uri!);
            wsEdit.createFile(uri);
            created = uri;
        } else if (fsEdit.type == 'moveFile') {
            const src = vscode.Uri.parse(fsEdit.src!);
            const dst = vscode.Uri.parse(fsEdit.dst!);
            wsEdit.renameFile(src, dst);
            moved = dst;
        } else {
            console.error(`unknown op: ${JSON.stringify(fsEdit)}`);
        }
    }
    const toOpen = created || moved;
    const toReveal = change.cursorPosition;
    await vscode.workspace.applyEdit(wsEdit);
    if (toOpen) {
        const doc = await vscode.workspace.openTextDocument(toOpen);
        await vscode.window.showTextDocument(doc);
    } else if (toReveal) {
        const uri = Server.client.protocol2CodeConverter.asUri(toReveal.textDocument.uri);
        const position = Server.client.protocol2CodeConverter.asPosition(toReveal.position);
        const editor = vscode.window.activeTextEditor;
        if (!editor || editor.document.uri.toString() != uri.toString()) { return; }
        if (!editor.selection.isEmpty) { return; }
        editor!.selection = new vscode.Selection(position, position);
    }
}