claude-flow
Version:
Enterprise-grade AI agent orchestration with ruv-swarm integration (Alpha Release)
55 lines (54 loc) • 3.05 kB
JavaScript
;
/* --------------------------------------------------------------------------------------------
* Copyright (c) Microsoft Corporation. All rights reserved.
* Licensed under the MIT License. See License.txt in the project root for license information.
* ------------------------------------------------------------------------------------------ */
Object.defineProperty(exports, "__esModule", { value: true });
exports.InlineCompletionItemFeature = void 0;
const vscode_1 = require("vscode");
const vscode_languageserver_protocol_1 = require("vscode-languageserver-protocol");
const features_1 = require("./features");
const UUID = require("./utils/uuid");
class InlineCompletionItemFeature extends features_1.TextDocumentLanguageFeature {
constructor(client) {
super(client, vscode_languageserver_protocol_1.InlineCompletionRequest.type);
}
fillClientCapabilities(capabilities) {
let inlineCompletion = (0, features_1.ensure)((0, features_1.ensure)(capabilities, 'textDocument'), 'inlineCompletion');
inlineCompletion.dynamicRegistration = true;
}
initialize(capabilities, documentSelector) {
const options = this.getRegistrationOptions(documentSelector, capabilities.inlineCompletionProvider);
if (!options) {
return;
}
this.register({
id: UUID.generateUuid(),
registerOptions: options
});
}
registerLanguageProvider(options) {
const selector = options.documentSelector;
const provider = {
provideInlineCompletionItems: (document, position, context, token) => {
const client = this._client;
const middleware = this._client.middleware;
const provideInlineCompletionItems = (document, position, context, token) => {
return client.sendRequest(vscode_languageserver_protocol_1.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(vscode_languageserver_protocol_1.InlineCompletionRequest.type, token, error, null);
});
};
return middleware.provideInlineCompletionItems
? middleware.provideInlineCompletionItems(document, position, context, token, provideInlineCompletionItems)
: provideInlineCompletionItems(document, position, context, token);
}
};
return [vscode_1.languages.registerInlineCompletionItemProvider(this._client.protocol2CodeConverter.asDocumentSelector(selector), provider), provider];
}
}
exports.InlineCompletionItemFeature = InlineCompletionItemFeature;