extHostLanguageFeatures.ts 28.6 KB
Newer Older
J
Johannes Rieken 已提交
1 2 3 4 5 6 7 8
/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the MIT License. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/
'use strict';

import URI from 'vs/base/common/uri';
import {TPromise} from 'vs/base/common/winjs.base';
J
Joao Moreno 已提交
9
import {IDisposable, dispose} from 'vs/base/common/lifecycle';
10
import {IThreadService} from 'vs/workbench/services/thread/common/threadService';
J
Johannes Rieken 已提交
11
import * as vscode from 'vscode';
J
Johannes Rieken 已提交
12
import * as TypeConverters from 'vs/workbench/api/node/extHostTypeConverters';
13
import {Range, Disposable, SignatureHelp, CompletionList} from 'vs/workbench/api/node/extHostTypes';
14
import {IPosition, IRange, ISingleEditOperation} from 'vs/editor/common/editorCommon';
J
Johannes Rieken 已提交
15
import * as modes from 'vs/editor/common/modes';
16
import {ExtHostDocuments} from 'vs/workbench/api/node/extHostDocuments';
J
Johannes Rieken 已提交
17
import {ExtHostCommands} from 'vs/workbench/api/node/extHostCommands';
18
import {ExtHostDiagnostics} from 'vs/workbench/api/node/extHostDiagnostics';
19 20
import {INavigateTypesSupport, ITypeBearing} from 'vs/workbench/parts/search/common/search';
import {asWinJsPromise, ShallowCancelThenPromise} from 'vs/base/common/async';
21
import {MainContext, MainThreadLanguageFeaturesShape} from './extHost.protocol';
22
import {regExpLeadsToEndlessLoop} from 'vs/base/common/strings';
J
Johannes Rieken 已提交
23 24 25

// --- adapter

26
class OutlineAdapter {
J
Johannes Rieken 已提交
27

28
	private _documents: ExtHostDocuments;
J
Johannes Rieken 已提交
29 30
	private _provider: vscode.DocumentSymbolProvider;

31
	constructor(documents: ExtHostDocuments, provider: vscode.DocumentSymbolProvider) {
J
Johannes Rieken 已提交
32 33 34 35
		this._documents = documents;
		this._provider = provider;
	}

36
	provideDocumentSymbols(resource: URI): TPromise<modes.SymbolInformation[]> {
37
		let doc = this._documents.getDocumentData(resource).document;
J
Johannes Rieken 已提交
38 39
		return asWinJsPromise(token => this._provider.provideDocumentSymbols(doc, token)).then(value => {
			if (Array.isArray(value)) {
40
				return value.map(TypeConverters.SymbolInformation.toOutlineEntry);
J
Johannes Rieken 已提交
41 42 43 44 45
			}
		});
	}
}

46 47 48 49
interface CachedCodeLens {
	symbols: modes.ICodeLensSymbol[];
	lenses: vscode.CodeLens[];
	disposables: IDisposable[];
J
Johannes Rieken 已提交
50
}
51

52
class CodeLensAdapter {
53

54
	private _documents: ExtHostDocuments;
55
	private _commands: ExtHostCommands;
56 57
	private _provider: vscode.CodeLensProvider;

58
	private _cache: { [uri: string]: { version: number; data: TPromise<CachedCodeLens>; } } = Object.create(null);
59

60
	constructor(documents: ExtHostDocuments, commands: ExtHostCommands, provider: vscode.CodeLensProvider) {
61
		this._documents = documents;
62
		this._commands = commands;
63 64 65
		this._provider = provider;
	}

66
	provideCodeLenses(resource: URI): TPromise<modes.ICodeLensSymbol[]> {
67
		const doc = this._documents.getDocumentData(resource).document;
68 69
		const version = doc.version;
		const key = resource.toString();
70

71
		// from cache
72
		let entry = this._cache[key];
73
		if (entry && entry.version === version) {
74
			return new ShallowCancelThenPromise(entry.data.then(cached => cached.symbols));
75
		}
76

77 78
		const newCodeLensData = asWinJsPromise(token => this._provider.provideCodeLenses(doc, token)).then(lenses => {
			if (!Array.isArray(lenses)) {
79 80 81
				return;
			}

82 83 84 85
			const data: CachedCodeLens = {
				lenses,
				symbols: [],
				disposables: [],
J
Johannes Rieken 已提交
86
			};
87

88 89
			lenses.forEach((lens, i) => {
				data.symbols.push(<modes.ICodeLensSymbol>{
90
					id: String(i),
J
Johannes Rieken 已提交
91
					range: TypeConverters.fromRange(lens.range),
92
					command: TypeConverters.Command.from(lens.command, data.disposables)
93
				});
94
			});
95 96 97 98 99 100 101 102 103

			return data;
		});

		this._cache[key] = {
			version,
			data: newCodeLensData
		};

104
		return new ShallowCancelThenPromise(newCodeLensData.then(newCached => {
105 106
			if (entry) {
				// only now dispose old commands et al
J
Joao Moreno 已提交
107
				entry.data.then(oldCached => dispose(oldCached.disposables));
108 109
			}
			return newCached && newCached.symbols;
110
		}));
111

112 113
	}

114
	resolveCodeLens(resource: URI, symbol: modes.ICodeLensSymbol): TPromise<modes.ICodeLensSymbol> {
115

116 117
		const entry = this._cache[resource.toString()];
		if (!entry) {
118 119 120
			return;
		}

121
		return entry.data.then(cachedData => {
122

123 124 125
			if (!cachedData) {
				return;
			}
126

127 128 129 130 131 132 133 134 135 136
			let lens = cachedData.lenses[Number(symbol.id)];
			if (!lens) {
				return;
			}

			let resolve: TPromise<vscode.CodeLens>;
			if (typeof this._provider.resolveCodeLens !== 'function' || lens.isResolved) {
				resolve = TPromise.as(lens);
			} else {
				resolve = asWinJsPromise(token => this._provider.resolveCodeLens(lens, token));
137
			}
J
Johannes Rieken 已提交
138

139 140 141 142 143 144 145 146 147 148
			return resolve.then(newLens => {
				lens = newLens || lens;
				let command = lens.command;
				if (!command) {
					command = {
						title: '<<MISSING COMMAND>>',
						command: 'missing',
					};
				}

149
				symbol.command = TypeConverters.Command.from(command, cachedData.disposables);
150 151
				return symbol;
			});
152 153 154 155
		});
	}
}

156
class DefinitionAdapter {
J
Johannes Rieken 已提交
157

158
	private _documents: ExtHostDocuments;
J
Johannes Rieken 已提交
159 160
	private _provider: vscode.DefinitionProvider;

161
	constructor(documents: ExtHostDocuments, provider: vscode.DefinitionProvider) {
J
Johannes Rieken 已提交
162 163 164 165
		this._documents = documents;
		this._provider = provider;
	}

166
	provideDefinition(resource: URI, position: IPosition): TPromise<modes.Definition> {
167
		let doc = this._documents.getDocumentData(resource).document;
J
Johannes Rieken 已提交
168 169 170
		let pos = TypeConverters.toPosition(position);
		return asWinJsPromise(token => this._provider.provideDefinition(doc, pos, token)).then(value => {
			if (Array.isArray(value)) {
171
				return value.map(DefinitionAdapter._convertLocation);
J
Johannes Rieken 已提交
172
			} else if (value) {
173
				return DefinitionAdapter._convertLocation(value);
J
Johannes Rieken 已提交
174 175 176 177
			}
		});
	}

178
	private static _convertLocation(location: vscode.Location): modes.Location {
J
Johannes Rieken 已提交
179 180 181
		if (!location) {
			return;
		}
182 183
		return <modes.Location>{
			uri: location.uri,
J
Johannes Rieken 已提交
184 185 186 187 188
			range: TypeConverters.fromRange(location.range)
		};
	}
}

189
class HoverAdapter {
J
Johannes Rieken 已提交
190

191
	private _documents: ExtHostDocuments;
J
Johannes Rieken 已提交
192 193
	private _provider: vscode.HoverProvider;

194
	constructor(documents: ExtHostDocuments, provider: vscode.HoverProvider) {
J
Johannes Rieken 已提交
195 196 197 198
		this._documents = documents;
		this._provider = provider;
	}

199
	public provideHover(resource: URI, position: IPosition): TPromise<modes.Hover> {
J
Johannes Rieken 已提交
200

201
		let doc = this._documents.getDocumentData(resource).document;
J
Johannes Rieken 已提交
202 203 204 205 206 207
		let pos = TypeConverters.toPosition(position);

		return asWinJsPromise(token => this._provider.provideHover(doc, pos, token)).then(value => {
			if (!value) {
				return;
			}
208 209
			if (!value.range) {
				value.range = doc.getWordRangeAtPosition(pos);
J
Johannes Rieken 已提交
210
			}
211 212
			if (!value.range) {
				value.range = new Range(pos, pos);
J
Johannes Rieken 已提交
213 214
			}

215
			return TypeConverters.fromHover(value);
J
Johannes Rieken 已提交
216 217 218 219
		});
	}
}

220
class DocumentHighlightAdapter {
221

222
	private _documents: ExtHostDocuments;
223 224
	private _provider: vscode.DocumentHighlightProvider;

225
	constructor(documents: ExtHostDocuments, provider: vscode.DocumentHighlightProvider) {
226 227 228 229
		this._documents = documents;
		this._provider = provider;
	}

230
	provideDocumentHighlights(resource: URI, position: IPosition): TPromise<modes.DocumentHighlight[]> {
231

232
		let doc = this._documents.getDocumentData(resource).document;
233 234 235 236
		let pos = TypeConverters.toPosition(position);

		return asWinJsPromise(token => this._provider.provideDocumentHighlights(doc, pos, token)).then(value => {
			if (Array.isArray(value)) {
237
				return value.map(DocumentHighlightAdapter._convertDocumentHighlight);
238 239 240 241
			}
		});
	}

242
	private static _convertDocumentHighlight(documentHighlight: vscode.DocumentHighlight): modes.DocumentHighlight {
243 244
		return {
			range: TypeConverters.fromRange(documentHighlight.range),
245
			kind: documentHighlight.kind
B
Benjamin Pasero 已提交
246
		};
247 248 249
	}
}

250
class ReferenceAdapter {
J
Johannes Rieken 已提交
251

252
	private _documents: ExtHostDocuments;
J
Johannes Rieken 已提交
253 254
	private _provider: vscode.ReferenceProvider;

255
	constructor(documents: ExtHostDocuments, provider: vscode.ReferenceProvider) {
J
Johannes Rieken 已提交
256 257 258 259
		this._documents = documents;
		this._provider = provider;
	}

260
	provideReferences(resource: URI, position: IPosition, context: modes.ReferenceContext): TPromise<modes.Location[]> {
261
		let doc = this._documents.getDocumentData(resource).document;
J
Johannes Rieken 已提交
262 263
		let pos = TypeConverters.toPosition(position);

264
		return asWinJsPromise(token => this._provider.provideReferences(doc, pos, context, token)).then(value => {
J
Johannes Rieken 已提交
265 266 267 268 269 270
			if (Array.isArray(value)) {
				return value.map(ReferenceAdapter._convertLocation);
			}
		});
	}

271 272 273
	private static _convertLocation(location: vscode.Location): modes.Location {
		return <modes.Location>{
			uri: location.uri,
J
Johannes Rieken 已提交
274 275 276 277 278
			range: TypeConverters.fromRange(location.range)
		};
	}
}

279
class QuickFixAdapter {
J
Johannes Rieken 已提交
280

281
	private _documents: ExtHostDocuments;
282
	private _commands: ExtHostCommands;
283
	private _diagnostics: ExtHostDiagnostics;
J
Johannes Rieken 已提交
284 285
	private _provider: vscode.CodeActionProvider;

286 287
	private _cachedCommands: IDisposable[] = [];

288
	constructor(documents: ExtHostDocuments, commands: ExtHostCommands, diagnostics: ExtHostDiagnostics, provider: vscode.CodeActionProvider) {
J
Johannes Rieken 已提交
289 290
		this._documents = documents;
		this._commands = commands;
291
		this._diagnostics = diagnostics;
J
Johannes Rieken 已提交
292 293 294
		this._provider = provider;
	}

A
Alex Dima 已提交
295
	provideCodeActions(resource: URI, range: IRange): TPromise<modes.CodeAction[]> {
J
Johannes Rieken 已提交
296

297
		const doc = this._documents.getDocumentData(resource).document;
J
Johannes Rieken 已提交
298
		const ran = TypeConverters.toRange(range);
299 300 301 302 303 304 305 306 307 308
		const allDiagnostics: vscode.Diagnostic[] = [];

		this._diagnostics.forEach(collection => {
			if (collection.has(resource)) {
				for (let diagnostic of collection.get(resource)) {
					if (diagnostic.range.intersection(ran)) {
						allDiagnostics.push(diagnostic);
					}
				}
			}
J
Johannes Rieken 已提交
309 310
		});

J
Joao Moreno 已提交
311
		this._cachedCommands = dispose(this._cachedCommands);
312

313
		return asWinJsPromise(token => this._provider.provideCodeActions(doc, ran, { diagnostics: allDiagnostics }, token)).then(commands => {
J
Johannes Rieken 已提交
314 315 316 317
			if (!Array.isArray(commands)) {
				return;
			}
			return commands.map((command, i) => {
A
Alex Dima 已提交
318
				return <modes.CodeAction> {
319
					command: TypeConverters.Command.from(command, this._cachedCommands),
320
					score: i
J
Johannes Rieken 已提交
321 322 323 324 325 326
				};
			});
		});
	}
}

327
class DocumentFormattingAdapter {
328

329
	private _documents: ExtHostDocuments;
330 331
	private _provider: vscode.DocumentFormattingEditProvider;

332
	constructor(documents: ExtHostDocuments, provider: vscode.DocumentFormattingEditProvider) {
333 334 335 336
		this._documents = documents;
		this._provider = provider;
	}

A
Alex Dima 已提交
337
	provideDocumentFormattingEdits(resource: URI, options: modes.FormattingOptions): TPromise<ISingleEditOperation[]> {
338

339
		let doc = this._documents.getDocumentData(resource).document;
340 341 342

		return asWinJsPromise(token => this._provider.provideDocumentFormattingEdits(doc, <any>options, token)).then(value => {
			if (Array.isArray(value)) {
343
				return value.map(TypeConverters.TextEdit.from);
344 345 346 347 348
			}
		});
	}
}

349
class RangeFormattingAdapter {
350

351
	private _documents: ExtHostDocuments;
352 353
	private _provider: vscode.DocumentRangeFormattingEditProvider;

354
	constructor(documents: ExtHostDocuments, provider: vscode.DocumentRangeFormattingEditProvider) {
355 356 357 358
		this._documents = documents;
		this._provider = provider;
	}

A
Alex Dima 已提交
359
	provideDocumentRangeFormattingEdits(resource: URI, range: IRange, options: modes.FormattingOptions): TPromise<ISingleEditOperation[]> {
360

361
		let doc = this._documents.getDocumentData(resource).document;
362 363 364 365
		let ran = TypeConverters.toRange(range);

		return asWinJsPromise(token => this._provider.provideDocumentRangeFormattingEdits(doc, ran, <any>options, token)).then(value => {
			if (Array.isArray(value)) {
366
				return value.map(TypeConverters.TextEdit.from);
367 368 369 370 371
			}
		});
	}
}

372
class OnTypeFormattingAdapter {
373

374
	private _documents: ExtHostDocuments;
375 376
	private _provider: vscode.OnTypeFormattingEditProvider;

377
	constructor(documents: ExtHostDocuments, provider: vscode.OnTypeFormattingEditProvider) {
378 379 380 381
		this._documents = documents;
		this._provider = provider;
	}

382
	autoFormatTriggerCharacters: string[] = []; // not here
383

A
Alex Dima 已提交
384
	provideOnTypeFormattingEdits(resource: URI, position: IPosition, ch: string, options: modes.FormattingOptions): TPromise<ISingleEditOperation[]> {
385

386
		let doc = this._documents.getDocumentData(resource).document;
387 388 389 390
		let pos = TypeConverters.toPosition(position);

		return asWinJsPromise(token => this._provider.provideOnTypeFormattingEdits(doc, pos, ch, <any> options, token)).then(value => {
			if (Array.isArray(value)) {
391
				return value.map(TypeConverters.TextEdit.from);
392 393 394 395 396
			}
		});
	}
}

397 398 399 400 401 402 403 404 405 406 407
class NavigateTypeAdapter implements INavigateTypesSupport {

	private _provider: vscode.WorkspaceSymbolProvider;

	constructor(provider: vscode.WorkspaceSymbolProvider) {
		this._provider = provider;
	}

	getNavigateToItems(search: string): TPromise<ITypeBearing[]> {
		return asWinJsPromise(token => this._provider.provideWorkspaceSymbols(search, token)).then(value => {
			if (Array.isArray(value)) {
408
				return value.map(TypeConverters.fromSymbolInformation);
409 410 411 412 413
			}
		});
	}
}

414
class RenameAdapter {
J
Johannes Rieken 已提交
415

416
	private _documents: ExtHostDocuments;
J
Johannes Rieken 已提交
417 418
	private _provider: vscode.RenameProvider;

419
	constructor(documents: ExtHostDocuments, provider: vscode.RenameProvider) {
J
Johannes Rieken 已提交
420 421 422 423
		this._documents = documents;
		this._provider = provider;
	}

424
	provideRenameEdits(resource: URI, position: IPosition, newName: string): TPromise<modes.WorkspaceEdit> {
J
Johannes Rieken 已提交
425

426
		let doc = this._documents.getDocumentData(resource).document;
J
Johannes Rieken 已提交
427 428 429 430 431 432 433 434
		let pos = TypeConverters.toPosition(position);

		return asWinJsPromise(token => this._provider.provideRenameEdits(doc, pos, newName, token)).then(value => {

			if (!value) {
				return;
			}

435
			let result = <modes.WorkspaceEdit>{
J
Johannes Rieken 已提交
436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451
				edits: []
			};

			for (let entry of value.entries()) {
				let [uri, textEdits] = entry;
				for (let textEdit of textEdits) {
					result.edits.push({
						resource: <URI>uri,
						newText: textEdit.newText,
						range: TypeConverters.fromRange(textEdit.range)
					});
				}
			}
			return result;
		}, err => {
			if (typeof err === 'string') {
452
				return <modes.WorkspaceEdit>{
J
Johannes Rieken 已提交
453 454 455 456 457 458 459 460 461
					edits: undefined,
					rejectReason: err
				};
			}
			return TPromise.wrapError(err);
		});
	}
}

462 463 464 465
interface ISuggestion2 extends modes.ISuggestion {
	id: string;
}

466
class SuggestAdapter {
467

468
	private _documents: ExtHostDocuments;
469
	private _provider: vscode.CompletionItemProvider;
470
	private _cache: { [key: string]: CompletionList } = Object.create(null);
471

472
	constructor(documents: ExtHostDocuments, provider: vscode.CompletionItemProvider) {
473 474 475 476
		this._documents = documents;
		this._provider = provider;
	}

477
	provideCompletionItems(resource: URI, position: IPosition): TPromise<modes.ISuggestResult[]> {
478

479
		const doc = this._documents.getDocumentData(resource).document;
480 481 482 483 484 485
		const pos = TypeConverters.toPosition(position);
		const ran = doc.getWordRangeAtPosition(pos);

		const key = resource.toString();
		delete this._cache[key];

486
		return asWinJsPromise<vscode.CompletionItem[]|vscode.CompletionList>(token => this._provider.provideCompletionItems(doc, pos, token)).then(value => {
487

488
			let defaultSuggestions: modes.ISuggestResult = {
489
				suggestions: [],
490
				currentWord: ran ? doc.getText(new Range(ran.start.line, ran.start.character, pos.line, pos.character)) : '',
491
			};
492
			let allSuggestions: modes.ISuggestResult[] = [defaultSuggestions];
493

494 495 496 497 498 499
			let list: CompletionList;
			if (Array.isArray(value)) {
				list = new CompletionList(value);
			} else if (value instanceof CompletionList) {
				list = value;
				defaultSuggestions.incomplete = list.isIncomplete;
500 501 502
			} else if (!value) {
				// undefined and null are valid results
				return;
503
			} else {
504 505
				// warn about everything else
				console.warn('INVALID result from completion provider. expected CompletionItem-array or CompletionList but got:', value);
506 507
				return;
			}
508

509 510
			for (let i = 0; i < list.items.length; i++) {
				const item = list.items[i];
511
				const suggestion = <ISuggestion2> TypeConverters.Suggest.from(item);
512 513 514 515 516 517 518 519 520 521 522 523 524 525

				if (item.textEdit) {

					let editRange = item.textEdit.range;

					// invalid text edit
					if (!editRange.isSingleLine || editRange.start.line !== pos.line) {
						console.warn('INVALID text edit, must be single line and on the same line');
						continue;
					}

					// insert the text of the edit and create a dedicated
					// suggestion-container with overwrite[Before|After]
					suggestion.codeSnippet = item.textEdit.newText;
B
Benjamin Pasero 已提交
526 527
					suggestion.overwriteBefore = pos.character - editRange.start.character;
					suggestion.overwriteAfter = editRange.end.character - pos.character;
528 529

					allSuggestions.push({
530
						currentWord: doc.getText(editRange),
531 532
						suggestions: [suggestion],
						incomplete: list.isIncomplete
533 534 535 536 537 538 539 540 541 542 543
					});

				} else {
					defaultSuggestions.suggestions.push(suggestion);
				}

				// assign identifier to suggestion
				suggestion.id = String(i);
			}

			// cache for details call
544
			this._cache[key] = list;
545 546 547 548 549

			return allSuggestions;
		});
	}

550
	resolveCompletionItem(resource: URI, position: IPosition, suggestion: modes.ISuggestion): TPromise<modes.ISuggestion> {
551 552 553
		if (typeof this._provider.resolveCompletionItem !== 'function') {
			return TPromise.as(suggestion);
		}
554 555
		let list = this._cache[resource.toString()];
		if (!list) {
556 557
			return TPromise.as(suggestion);
		}
558
		let item = list.items[Number((<ISuggestion2> suggestion).id)];
559 560 561 562
		if (!item) {
			return TPromise.as(suggestion);
		}
		return asWinJsPromise(token => this._provider.resolveCompletionItem(item, token)).then(resolvedItem => {
563
			return TypeConverters.Suggest.from(resolvedItem || item);
564 565 566 567
		});
	}
}

A
Alex Dima 已提交
568
class SignatureHelpAdapter {
569

570
	private _documents: ExtHostDocuments;
571 572
	private _provider: vscode.SignatureHelpProvider;

573
	constructor(documents: ExtHostDocuments, provider: vscode.SignatureHelpProvider) {
574 575 576 577
		this._documents = documents;
		this._provider = provider;
	}

578
	provideSignatureHelp(resource: URI, position: IPosition): TPromise<modes.SignatureHelp> {
579

580
		const doc = this._documents.getDocumentData(resource).document;
581 582 583 584
		const pos = TypeConverters.toPosition(position);

		return asWinJsPromise(token => this._provider.provideSignatureHelp(doc, pos, token)).then(value => {
			if (value instanceof SignatureHelp) {
585
				return TypeConverters.SignatureHelp.from(value);
586 587 588 589 590
			}
		});
	}
}

591
type Adapter = OutlineAdapter | CodeLensAdapter | DefinitionAdapter | HoverAdapter
592
	| DocumentHighlightAdapter | ReferenceAdapter | QuickFixAdapter | DocumentFormattingAdapter
593
	| RangeFormattingAdapter | OnTypeFormattingAdapter | NavigateTypeAdapter | RenameAdapter
A
Alex Dima 已提交
594
	| SuggestAdapter | SignatureHelpAdapter;
J
Johannes Rieken 已提交
595 596 597

export class ExtHostLanguageFeatures {

598
	private static _handlePool: number = 0;
J
Johannes Rieken 已提交
599

600
	private _proxy: MainThreadLanguageFeaturesShape;
601
	private _documents: ExtHostDocuments;
602
	private _commands: ExtHostCommands;
603
	private _diagnostics: ExtHostDiagnostics;
J
Johannes Rieken 已提交
604 605
	private _adapter: { [handle: number]: Adapter } = Object.create(null);

606 607
	constructor(
		threadService: IThreadService,
608
		documents: ExtHostDocuments,
609 610 611 612 613 614 615
		commands: ExtHostCommands,
		diagnostics: ExtHostDiagnostics
	) {
		this._proxy = threadService.get(MainContext.MainThreadLanguageFeatures);
		this._documents = documents;
		this._commands = commands;
		this._diagnostics = diagnostics;
J
Johannes Rieken 已提交
616 617 618 619 620 621 622 623 624
	}

	private _createDisposable(handle: number): Disposable {
		return new Disposable(() => {
			delete this._adapter[handle];
			this._proxy.$unregister(handle);
		});
	}

625 626 627 628
	private _nextHandle(): number {
		return ExtHostLanguageFeatures._handlePool++;
	}

J
Johannes Rieken 已提交
629 630 631 632 633 634
	private _withAdapter<A, R>(handle: number, ctor: { new (...args: any[]): A }, callback: (adapter: A) => TPromise<R>): TPromise<R> {
		let adapter = this._adapter[handle];
		if (!(adapter instanceof ctor)) {
			return TPromise.wrapError(new Error('no adapter found'));
		}
		return callback(<any> adapter);
635 636
	}

J
Johannes Rieken 已提交
637 638 639
	// --- outline

	registerDocumentSymbolProvider(selector: vscode.DocumentSelector, provider: vscode.DocumentSymbolProvider): vscode.Disposable {
640 641
		const handle = this._nextHandle();
		this._adapter[handle] = new OutlineAdapter(this._documents, provider);
642
		this._proxy.$registerOutlineSupport(handle, selector);
J
Johannes Rieken 已提交
643 644 645
		return this._createDisposable(handle);
	}

646 647
	$provideDocumentSymbols(handle: number, resource: URI): TPromise<modes.SymbolInformation[]> {
		return this._withAdapter(handle, OutlineAdapter, adapter => adapter.provideDocumentSymbols(resource));
J
Johannes Rieken 已提交
648
	}
649 650 651 652 653

	// --- code lens

	registerCodeLensProvider(selector: vscode.DocumentSelector, provider: vscode.CodeLensProvider): vscode.Disposable {
		const handle = this._nextHandle();
654
		this._adapter[handle] = new CodeLensAdapter(this._documents, this._commands, provider);
655 656 657 658
		this._proxy.$registerCodeLensSupport(handle, selector);
		return this._createDisposable(handle);
	}

659 660
	$provideCodeLenses(handle: number, resource: URI): TPromise<modes.ICodeLensSymbol[]> {
		return this._withAdapter(handle, CodeLensAdapter, adapter => adapter.provideCodeLenses(resource));
661 662
	}

663 664
	$resolveCodeLens(handle: number, resource: URI, symbol: modes.ICodeLensSymbol): TPromise<modes.ICodeLensSymbol> {
		return this._withAdapter(handle, CodeLensAdapter, adapter => adapter.resolveCodeLens(resource, symbol));
J
Johannes Rieken 已提交
665 666 667 668 669 670
	}

	// --- declaration

	registerDefinitionProvider(selector: vscode.DocumentSelector, provider: vscode.DefinitionProvider): vscode.Disposable {
		const handle = this._nextHandle();
671
		this._adapter[handle] = new DefinitionAdapter(this._documents, provider);
J
Johannes Rieken 已提交
672 673
		this._proxy.$registerDeclaractionSupport(handle, selector);
		return this._createDisposable(handle);
674 675
	}

676 677
	$provideDefinition(handle: number, resource: URI, position: IPosition): TPromise<modes.Definition> {
		return this._withAdapter(handle, DefinitionAdapter, adapter => adapter.provideDefinition(resource, position));
J
Johannes Rieken 已提交
678
	}
J
Johannes Rieken 已提交
679 680 681 682 683

	// --- extra info

	registerHoverProvider(selector: vscode.DocumentSelector, provider: vscode.HoverProvider): vscode.Disposable {
		const handle = this._nextHandle();
684
		this._adapter[handle] = new HoverAdapter(this._documents, provider);
A
Alex Dima 已提交
685
		this._proxy.$registerHoverProvider(handle, selector);
J
Johannes Rieken 已提交
686 687 688
		return this._createDisposable(handle);
	}

689
	$provideHover(handle: number, resource: URI, position: IPosition): TPromise<modes.Hover> {
690
		return this._withAdapter(handle, HoverAdapter, adpater => adpater.provideHover(resource, position));
J
Johannes Rieken 已提交
691
	}
692

J
Johannes Rieken 已提交
693
	// --- occurrences
694 695 696

	registerDocumentHighlightProvider(selector: vscode.DocumentSelector, provider: vscode.DocumentHighlightProvider): vscode.Disposable {
		const handle = this._nextHandle();
697 698
		this._adapter[handle] = new DocumentHighlightAdapter(this._documents, provider);
		this._proxy.$registerDocumentHighlightProvider(handle, selector);
699 700 701
		return this._createDisposable(handle);
	}

702 703
	$provideDocumentHighlights(handle: number, resource: URI, position: IPosition): TPromise<modes.DocumentHighlight[]> {
		return this._withAdapter(handle, DocumentHighlightAdapter, adapter => adapter.provideDocumentHighlights(resource, position));
704
	}
J
Johannes Rieken 已提交
705 706 707 708 709 710 711 712 713 714

	// --- references

	registerReferenceProvider(selector: vscode.DocumentSelector, provider: vscode.ReferenceProvider): vscode.Disposable {
		const handle = this._nextHandle();
		this._adapter[handle] = new ReferenceAdapter(this._documents, provider);
		this._proxy.$registerReferenceSupport(handle, selector);
		return this._createDisposable(handle);
	}

715 716
	$provideReferences(handle: number, resource: URI, position: IPosition, context: modes.ReferenceContext): TPromise<modes.Location[]> {
		return this._withAdapter(handle, ReferenceAdapter, adapter => adapter.provideReferences(resource, position, context));
J
Johannes Rieken 已提交
717 718
	}

J
Johannes Rieken 已提交
719 720 721 722
	// --- quick fix

	registerCodeActionProvider(selector: vscode.DocumentSelector, provider: vscode.CodeActionProvider): vscode.Disposable {
		const handle = this._nextHandle();
723
		this._adapter[handle] = new QuickFixAdapter(this._documents, this._commands, this._diagnostics, provider);
J
Johannes Rieken 已提交
724 725 726 727
		this._proxy.$registerQuickFixSupport(handle, selector);
		return this._createDisposable(handle);
	}

A
Alex Dima 已提交
728
	$provideCodeActions(handle: number, resource: URI, range: IRange): TPromise<modes.CodeAction[]> {
729
		return this._withAdapter(handle, QuickFixAdapter, adapter => adapter.provideCodeActions(resource, range));
J
Johannes Rieken 已提交
730
	}
731 732 733 734 735 736 737 738 739 740

	// --- formatting

	registerDocumentFormattingEditProvider(selector: vscode.DocumentSelector, provider: vscode.DocumentFormattingEditProvider): vscode.Disposable {
		const handle = this._nextHandle();
		this._adapter[handle] = new DocumentFormattingAdapter(this._documents, provider);
		this._proxy.$registerDocumentFormattingSupport(handle, selector);
		return this._createDisposable(handle);
	}

A
Alex Dima 已提交
741
	$provideDocumentFormattingEdits(handle: number, resource: URI, options: modes.FormattingOptions): TPromise<ISingleEditOperation[]> {
742
		return this._withAdapter(handle, DocumentFormattingAdapter, adapter => adapter.provideDocumentFormattingEdits(resource, options));
743 744 745 746 747 748 749 750 751
	}

	registerDocumentRangeFormattingEditProvider(selector: vscode.DocumentSelector, provider: vscode.DocumentRangeFormattingEditProvider): vscode.Disposable {
		const handle = this._nextHandle();
		this._adapter[handle] = new RangeFormattingAdapter(this._documents, provider);
		this._proxy.$registerRangeFormattingSupport(handle, selector);
		return this._createDisposable(handle);
	}

A
Alex Dima 已提交
752
	$provideDocumentRangeFormattingEdits(handle: number, resource: URI, range: IRange, options: modes.FormattingOptions): TPromise<ISingleEditOperation[]> {
753
		return this._withAdapter(handle, RangeFormattingAdapter, adapter => adapter.provideDocumentRangeFormattingEdits(resource, range, options));
754 755 756 757 758 759 760 761 762
	}

	registerOnTypeFormattingEditProvider(selector: vscode.DocumentSelector, provider: vscode.OnTypeFormattingEditProvider, triggerCharacters: string[]): vscode.Disposable {
		const handle = this._nextHandle();
		this._adapter[handle] = new OnTypeFormattingAdapter(this._documents, provider);
		this._proxy.$registerOnTypeFormattingSupport(handle, selector, triggerCharacters);
		return this._createDisposable(handle);
	}

A
Alex Dima 已提交
763
	$provideOnTypeFormattingEdits(handle: number, resource: URI, position: IPosition, ch: string, options: modes.FormattingOptions): TPromise<ISingleEditOperation[]> {
764
		return this._withAdapter(handle, OnTypeFormattingAdapter, adapter => adapter.provideOnTypeFormattingEdits(resource, position, ch, options));
765 766
	}

767 768 769 770 771 772 773 774 775 776 777 778
	// --- navigate types

	registerWorkspaceSymbolProvider(provider: vscode.WorkspaceSymbolProvider): vscode.Disposable {
		const handle = this._nextHandle();
		this._adapter[handle] = new NavigateTypeAdapter(provider);
		this._proxy.$registerNavigateTypeSupport(handle);
		return this._createDisposable(handle);
	}

	$getNavigateToItems(handle: number, search: string): TPromise<ITypeBearing[]> {
		return this._withAdapter(handle, NavigateTypeAdapter, adapter => adapter.getNavigateToItems(search));
	}
J
Johannes Rieken 已提交
779 780 781 782 783 784 785 786 787 788

	// --- rename

	registerRenameProvider(selector: vscode.DocumentSelector, provider: vscode.RenameProvider): vscode.Disposable {
		const handle = this._nextHandle();
		this._adapter[handle] = new RenameAdapter(this._documents, provider);
		this._proxy.$registerRenameSupport(handle, selector);
		return this._createDisposable(handle);
	}

789 790
	$provideRenameEdits(handle: number, resource: URI, position: IPosition, newName: string): TPromise<modes.WorkspaceEdit> {
		return this._withAdapter(handle, RenameAdapter, adapter => adapter.provideRenameEdits(resource, position, newName));
J
Johannes Rieken 已提交
791
	}
792 793 794 795 796 797 798 799 800 801

	// --- suggestion

	registerCompletionItemProvider(selector: vscode.DocumentSelector, provider: vscode.CompletionItemProvider, triggerCharacters: string[]): vscode.Disposable {
		const handle = this._nextHandle();
		this._adapter[handle] = new SuggestAdapter(this._documents, provider);
		this._proxy.$registerSuggestSupport(handle, selector, triggerCharacters);
		return this._createDisposable(handle);
	}

802 803
	$provideCompletionItems(handle: number, resource: URI, position: IPosition): TPromise<modes.ISuggestResult[]> {
		return this._withAdapter(handle, SuggestAdapter, adapter => adapter.provideCompletionItems(resource, position));
804 805
	}

806 807
	$resolveCompletionItem(handle: number, resource: URI, position: IPosition, suggestion: modes.ISuggestion): TPromise<modes.ISuggestion> {
		return this._withAdapter(handle, SuggestAdapter, adapter => adapter.resolveCompletionItem(resource, position, suggestion));
808
	}
809 810 811 812 813

	// --- parameter hints

	registerSignatureHelpProvider(selector: vscode.DocumentSelector, provider: vscode.SignatureHelpProvider, triggerCharacters: string[]): vscode.Disposable {
		const handle = this._nextHandle();
A
Alex Dima 已提交
814 815
		this._adapter[handle] = new SignatureHelpAdapter(this._documents, provider);
		this._proxy.$registerSignatureHelpProvider(handle, selector, triggerCharacters);
816 817 818
		return this._createDisposable(handle);
	}

819
	$provideSignatureHelp(handle: number, resource: URI, position: IPosition): TPromise<modes.SignatureHelp> {
A
Alex Dima 已提交
820
		return this._withAdapter(handle, SignatureHelpAdapter, adapter => adapter.provideSignatureHelp(resource, position));
821
	}
J
Johannes Rieken 已提交
822

823
	// --- configuration
J
Johannes Rieken 已提交
824

825 826
	setLanguageConfiguration(languageId:string, configuration: vscode.LanguageConfiguration): vscode.Disposable {
		let {wordPattern} = configuration;
J
Johannes Rieken 已提交
827

828 829 830
		// check for a valid word pattern
		if (wordPattern && regExpLeadsToEndlessLoop(wordPattern)) {
			throw new Error(`Invalid language configuration: wordPattern '${wordPattern}' is not allowed to match the empty string.`);
J
Johannes Rieken 已提交
831
		}
J
Johannes Rieken 已提交
832

833 834 835 836 837 838
		// word definition
		if (wordPattern) {
			this._documents.setWordDefinitionFor(languageId, wordPattern);
		} else {
			this._documents.setWordDefinitionFor(languageId, null);
		}
839

840 841 842
		const handle = this._nextHandle();
		this._proxy.$setLanguageConfiguration(handle, languageId, configuration);
		return this._createDisposable(handle);
843
	}
844
}