-
Notifications
You must be signed in to change notification settings - Fork 317
/
linkedEditingRange.ts
70 lines (60 loc) · 3.36 KB
/
linkedEditingRange.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
59
60
61
62
63
64
65
66
67
68
69
70
/* --------------------------------------------------------------------------------------------
* Copyright (c) Microsoft Corporation. All rights reserved.
* Licensed under the MIT License. See License.txt in the project root for license information.
* ------------------------------------------------------------------------------------------ */
import * as code from 'vscode';
import * as proto from 'vscode-languageserver-protocol';
import { TextDocumentLanguageFeature, FeatureClient, ensure } from './features';
export interface ProvideLinkedEditingRangeSignature {
(this: void, document: code.TextDocument, position: code.Position, token: code.CancellationToken): code.ProviderResult<code.LinkedEditingRanges>;
}
/**
* Linked editing middleware
*
* @since 3.16.0
*/
export interface LinkedEditingRangeMiddleware {
provideLinkedEditingRange?: (this: void, document: code.TextDocument, position: code.Position, token: code.CancellationToken, next: ProvideLinkedEditingRangeSignature) => code.ProviderResult<code.LinkedEditingRanges>;
}
export class LinkedEditingFeature extends TextDocumentLanguageFeature<boolean | proto.LinkedEditingRangeOptions, proto.LinkedEditingRangeRegistrationOptions, code.LinkedEditingRangeProvider, LinkedEditingRangeMiddleware> {
constructor(client: FeatureClient<LinkedEditingRangeMiddleware>) {
super(client, proto.LinkedEditingRangeRequest.type);
}
public fillClientCapabilities(capabilities: proto.ClientCapabilities): void {
const linkedEditingSupport = ensure(ensure(capabilities, 'textDocument')!, 'linkedEditingRange')!;
linkedEditingSupport.dynamicRegistration = true;
}
public initialize(capabilities: proto.ServerCapabilities, documentSelector: proto.DocumentSelector): void {
const [id, options] = this.getRegistration(documentSelector, capabilities.linkedEditingRangeProvider);
if (!id || !options) {
return;
}
this.register({ id: id, registerOptions: options });
}
protected registerLanguageProvider(options: proto.LinkedEditingRangeRegistrationOptions): [code.Disposable, code.LinkedEditingRangeProvider] {
const selector = options.documentSelector!;
const provider: code.LinkedEditingRangeProvider = {
provideLinkedEditingRanges: (document, position, token) => {
const client = this._client;
const provideLinkedEditing: ProvideLinkedEditingRangeSignature = (document, position, token) => {
return client.sendRequest(proto.LinkedEditingRangeRequest.type, client.code2ProtocolConverter.asTextDocumentPositionParams(document, position), token).then((result) => {
if (token.isCancellationRequested) {
return null;
}
return client.protocol2CodeConverter.asLinkedEditingRanges(result, token);
}, (error) => {
return client.handleFailedRequest(proto.LinkedEditingRangeRequest.type, token, error, null);
});
};
const middleware = client.middleware;
return middleware.provideLinkedEditingRange
? middleware.provideLinkedEditingRange(document, position, token, provideLinkedEditing)
: provideLinkedEditing(document, position, token);
}
};
return [this.registerProvider(selector, provider), provider];
}
private registerProvider(selector: proto.DocumentSelector, provider: code.LinkedEditingRangeProvider): code.Disposable {
return code.languages.registerLinkedEditingRangeProvider(this._client.protocol2CodeConverter.asDocumentSelector(selector), provider);
}
}