wip

extensions/typescript-language-features/src/features/semanticColoring.ts

@@ -109,6 +109,7 @@ class SemanticColoringProvider implements vscode.SemanticColoringProvider {
 }
 
 export function register(
+	_selector: vscode.DocumentSelector,
 	client: ITypeScriptServiceClient
 ) {
 	const provider = new SemanticColoringProvider(client);

src/vs/editor/common/model/textModel.ts

@@ -276,6 +276,7 @@ export class TextModel extends Disposable implements model.ITextModel {
 	private _languageIdentifier: LanguageIdentifier;
 	private readonly _languageRegistryListener: IDisposable;
 	private readonly _tokens: TokensStore;
+	private readonly _tokens2: TokensStore;
 	private readonly _tokenization: TextModelTokenization;
 	//#endregion
 
@@ -339,6 +340,7 @@ export class TextModel extends Disposable implements model.ITextModel {
 		this._trimAutoWhitespaceLines = null;
 
 		this._tokens = new TokensStore();
+		this._tokens2 = new TokensStore();
 		this._tokenization = new TextModelTokenization(this);
 	}
 
@@ -414,6 +416,7 @@ export class TextModel extends Disposable implements model.ITextModel {
 
 		// Flush all tokens
 		this._tokens.flush();
+		this._tokens2.flush();
 
 		// Destroy all my decorations
 		this._decorations = Object.create(null);
@@ -1263,6 +1266,7 @@ export class TextModel extends Disposable implements model.ITextModel {
 				const change = contentChanges[i];
 				const [eolCount, firstLineLength] = countEOL(change.text);
 				this._tokens.acceptEdit(change.range, eolCount, firstLineLength);
+				this._tokens2.acceptEdit(change.range, eolCount, firstLineLength);
 				this._onDidChangeDecorations.fire();
 				this._decorationsTree.acceptReplace(change.rangeOffset, change.rangeLength, change.text.length, change.forceMoveMarkers);
 
@@ -1716,6 +1720,10 @@ export class TextModel extends Disposable implements model.ITextModel {
 		});
 	}
 
+	public setSemanticTokens(tokens: MultilineTokens[]): void {
+		
+	}
+
 	public tokenizeViewport(startLineNumber: number, endLineNumber: number): void {
 		startLineNumber = Math.max(1, startLineNumber);
 		endLineNumber = Math.min(this._buffer.getLineCount(), endLineNumber);
@@ -1733,6 +1741,11 @@ export class TextModel extends Disposable implements model.ITextModel {
 		});
 	}
 
+	public clearSemanticTokens(): void {
+		this._tokens2.flush();
+		console.log(`todo - shoud emit event?`);
+	}
+
 	private _emitModelTokensChangedEvent(e: IModelTokensChangedEvent): void {
 		if (!this._isDisposing) {
 			this._onDidChangeTokens.fire(e);

src/vs/editor/common/services/modelServiceImpl.ts

@@ -426,3 +426,28 @@ export class ModelServiceImpl extends Disposable implements IModelService {
 export interface ILineSequence {
 	getLineContent(lineNumber: number): string;
 }
+
+class SemanticColoring extends Disposable {
+	private _watchers: Record<string, ModelSemanticColoring>;
+
+	constructor(modelService: IModelService) {
+		super();
+		this._watchers = Object.create(null);
+		this._register(modelService.onModelAdded((model) => {
+			this._watchers[model.uri.toString()] = new ModelSemanticColoring(model);
+		}));
+		this._register(modelService.onModelRemoved((model) => {
+			this._watchers[model.uri.toString()].dispose();
+			delete this._watchers[model.uri.toString()];
+		}));
+	}
+}
+
+class ModelSemanticColoring extends Disposable {
+
+	constructor(model: ITextModel) {
+		super();
+		this._register
+	}
+
+}