-
Notifications
You must be signed in to change notification settings - Fork 317
/
inlineCompletion.ts
80 lines (67 loc) · 3.89 KB
/
inlineCompletion.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
71
72
73
74
75
76
77
78
79
80
/* --------------------------------------------------------------------------------------------
* Copyright (c) Microsoft Corporation. All rights reserved.
* Licensed under the MIT License. See License.txt in the project root for license information.
* ------------------------------------------------------------------------------------------ */
import {
languages as Languages, TextDocument, Disposable, Position as VPosition, InlineCompletionContext as VInlineCompletionContext,
CancellationToken, ProviderResult, InlineCompletionItem as VInlineCompletionItem, InlineCompletionList as VInlineCompletionList, InlineCompletionItemProvider
} from 'vscode';
import {
ClientCapabilities, InlineCompletionOptions, InlineCompletionRegistrationOptions, InlineCompletionRequest,
DocumentSelector, ServerCapabilities
} from 'vscode-languageserver-protocol';
import {
FeatureClient, ensure, TextDocumentLanguageFeature
} from './features';
import * as UUID from './utils/uuid';
export interface ProvideInlineCompletionItemsSignature {
(this: void, document: TextDocument, position: VPosition, context: VInlineCompletionContext, token: CancellationToken): ProviderResult<VInlineCompletionItem[] | VInlineCompletionList>;
}
export interface InlineCompletionMiddleware {
provideInlineCompletionItems?: (this: void, document: TextDocument, position: VPosition, context: VInlineCompletionContext, token: CancellationToken, next: ProvideInlineCompletionItemsSignature) => ProviderResult<VInlineCompletionItem[] | VInlineCompletionList>;
}
export type InlineCompletionProviderShape = {
provider: InlineCompletionItemProvider;
};
export class InlineCompletionItemFeature extends TextDocumentLanguageFeature<boolean | InlineCompletionOptions, InlineCompletionRegistrationOptions, InlineCompletionItemProvider, InlineCompletionMiddleware> {
constructor(client: FeatureClient<InlineCompletionMiddleware>) {
super(client, InlineCompletionRequest.type);
}
public fillClientCapabilities(capabilities: ClientCapabilities): void {
const inlineCompletion = ensure(ensure(capabilities, 'textDocument')!, 'inlineCompletion')!;
inlineCompletion.dynamicRegistration = true;
}
public initialize(capabilities: ServerCapabilities, documentSelector: DocumentSelector): void {
const options = this.getRegistrationOptions(documentSelector, capabilities.inlineCompletionProvider);
if (!options) {
return;
}
this.register({
id: UUID.generateUuid(),
registerOptions: options
});
}
protected registerLanguageProvider(options: InlineCompletionRegistrationOptions): [Disposable, InlineCompletionItemProvider] {
const selector = options.documentSelector!;
const provider: InlineCompletionItemProvider = {
provideInlineCompletionItems: (document: TextDocument, position: VPosition, context: VInlineCompletionContext, token: CancellationToken): ProviderResult<VInlineCompletionList | VInlineCompletionItem[]> => {
const client = this._client;
const middleware = this._client.middleware;
const provideInlineCompletionItems: ProvideInlineCompletionItemsSignature = (document, position, context, token) => {
return client.sendRequest(InlineCompletionRequest.type, client.code2ProtocolConverter.asInlineCompletionParams(document, position, context), token).then((result) => {
if (token.isCancellationRequested) {
return null;
}
return client.protocol2CodeConverter.asInlineCompletionResult(result, token);
}, (error) => {
return client.handleFailedRequest(InlineCompletionRequest.type, token, error, null);
});
};
return middleware.provideInlineCompletionItems
? middleware.provideInlineCompletionItems(document, position, context, token, provideInlineCompletionItems)
: provideInlineCompletionItems(document, position, context, token);
}
};
return [Languages.registerInlineCompletionItemProvider(this._client.protocol2CodeConverter.asDocumentSelector(selector), provider), provider];
}
}