Update results on file create

extensions/markdown-language-features/src/features/workspaceSymbolProvider.ts

@@ -12,12 +12,14 @@ export interface WorkspaceMarkdownDocumentProvider {
 	getAllMarkdownDocuments(): Thenable<vscode.TextDocument[]>;
 
 	readonly onDidChangeMarkdownDocument: vscode.Event<vscode.TextDocument>;
+	readonly onDidCreateMarkdownDocument: vscode.Event<vscode.TextDocument>;
 	readonly onDidDeleteMarkdownDocument: vscode.Event<vscode.Uri>;
 }
 
 class VSCodeWorkspaceMarkdownDocumentProvider implements WorkspaceMarkdownDocumentProvider {
 
 	private readonly _onDidChangeMarkdownDocumentEmitter = new vscode.EventEmitter<vscode.TextDocument>();
+	private readonly _onDidCreateMarkdownDocumentEmitter = new vscode.EventEmitter<vscode.TextDocument>();
 	private readonly _onDidDeleteMarkdownDocumentEmitter = new vscode.EventEmitter<vscode.Uri>();
 
 	private _watcher: vscode.FileSystemWatcher | undefined;
@@ -46,6 +48,11 @@ class VSCodeWorkspaceMarkdownDocumentProvider implements WorkspaceMarkdownDocume
 		return this._onDidChangeMarkdownDocumentEmitter.event;
 	}
 
+	public get onDidCreateMarkdownDocument() {
+		this.ensureWatcher();
+		return this._onDidCreateMarkdownDocumentEmitter.event;
+	}
+
 	public get onDidDeleteMarkdownDocument() {
 		this.ensureWatcher();
 		return this._onDidDeleteMarkdownDocumentEmitter.event;
@@ -65,6 +72,13 @@ class VSCodeWorkspaceMarkdownDocumentProvider implements WorkspaceMarkdownDocume
 			}
 		}, this, this._disposables);
 
+		this._watcher.onDidCreate(async resource => {
+			const document = await vscode.workspace.openTextDocument(resource);
+			if (isMarkdownFile(document)) {
+				this._onDidCreateMarkdownDocumentEmitter.fire(document);
+			}
+		}, this, this._disposables);
+
 		this._watcher.onDidDelete(async resource => {
 			this._onDidDeleteMarkdownDocumentEmitter.fire(resource);
 		}, this, this._disposables);
@@ -88,6 +102,7 @@ export default class MarkdownWorkspaceSymbolProvider implements vscode.Workspace
 			this._symbolCachePopulated = true;
 
 			this._workspaceMarkdownDocumentProvider.onDidChangeMarkdownDocument(this.onDidChangeDocument, this, this._disposables);
+			this._workspaceMarkdownDocumentProvider.onDidCreateMarkdownDocument(this.onDidChangeDocument, this, this._disposables);
 			this._workspaceMarkdownDocumentProvider.onDidDeleteMarkdownDocument(this.onDidDeleteDocument, this, this._disposables);
 		}
 

extensions/markdown-language-features/src/test/workspaceSymbolProvider.test.ts

@@ -75,7 +75,6 @@ suite('markdown.WorkspaceSymbolProvider', () => {
 		]);
 
 		const provider = new MarkdownWorkspaceSymbolProvider(symbolProvider, workspaceFileProvider);
-
 		assert.strictEqual((await provider.provideWorkspaceSymbols('')).length, 1);
 
 		// delete file
@@ -83,6 +82,22 @@ suite('markdown.WorkspaceSymbolProvider', () => {
 		const newSymbols = await provider.provideWorkspaceSymbols('');
 		assert.strictEqual(newSymbols.length, 0);
 	});
+
+	test('Should update results when markdown file is created', async () => {
+		const testFileName = vscode.Uri.parse('test.md');
+
+		const workspaceFileProvider = new InMemoryWorkspaceMarkdownDocumentProvider([
+			new InMemoryDocument(testFileName, `# header1`)
+		]);
+
+		const provider = new MarkdownWorkspaceSymbolProvider(symbolProvider, workspaceFileProvider);
+		assert.strictEqual((await provider.provideWorkspaceSymbols('')).length, 1);
+
+		// Creat file
+		workspaceFileProvider.createDocument(new InMemoryDocument(vscode.Uri.parse('test2.md'), `# new header\nabc\n## header2`));
+		const newSymbols = await provider.provideWorkspaceSymbols('');
+		assert.strictEqual(newSymbols.length, 3);
+	});
 });
 
 
@@ -102,6 +117,9 @@ class InMemoryWorkspaceMarkdownDocumentProvider implements WorkspaceMarkdownDocu
 	private readonly _onDidChangeMarkdownDocumentEmitter = new vscode.EventEmitter<vscode.TextDocument>();
 	public onDidChangeMarkdownDocument = this._onDidChangeMarkdownDocumentEmitter.event;
 
+	private readonly _onDidCreateMarkdownDocumentEmitter = new vscode.EventEmitter<vscode.TextDocument>();
+	public onDidCreateMarkdownDocument = this._onDidCreateMarkdownDocumentEmitter.event;
+
 	private readonly _onDidDeleteMarkdownDocumentEmitter = new vscode.EventEmitter<vscode.Uri>();
 	public onDidDeleteMarkdownDocument = this._onDidDeleteMarkdownDocumentEmitter.event;
 
@@ -110,6 +128,13 @@ class InMemoryWorkspaceMarkdownDocumentProvider implements WorkspaceMarkdownDocu
 		this._onDidChangeMarkdownDocumentEmitter.fire(document);
 	}
 
+	public createDocument(document: vscode.TextDocument) {
+		assert.ok(!this._documents.has(document.uri.fsPath));
+
+		this._documents.set(document.uri.fsPath, document);
+		this._onDidCreateMarkdownDocumentEmitter.fire(document);
+	}
+
 	public deleteDocument(resource: vscode.Uri) {
 		this._documents.delete(resource.fsPath);
 		this._onDidDeleteMarkdownDocumentEmitter.fire(resource);