forked from microsoft/vscode-java-test
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathTestCodeLensController.ts
More file actions
60 lines (46 loc) · 2.15 KB
/
TestCodeLensController.ts
File metadata and controls
60 lines (46 loc) · 2.15 KB
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
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT license.
import { ConfigurationChangeEvent, Disposable, DocumentSelector, languages, RelativePattern, workspace } from 'vscode';
import { testSourceProvider } from '../../extension.bundle';
import { ENABLE_EDITOR_SHORTCUTS_KEY } from '../constants/configs';
import { parseDocumentSelector } from '../utils/uiUtils';
import { TestCodeLensProvider } from './TestCodeLensProvider';
class TestCodeLensController implements Disposable {
private internalProvider: TestCodeLensProvider;
private registeredProvider: Disposable | undefined;
private configurationChangeListener: Disposable;
constructor() {
this.internalProvider = new TestCodeLensProvider();
this.configurationChangeListener = workspace.onDidChangeConfiguration((event: ConfigurationChangeEvent) => {
if (event.affectsConfiguration(ENABLE_EDITOR_SHORTCUTS_KEY)) {
this.setCodeLensVisibility();
}
}, this);
this.setCodeLensVisibility();
}
public async registerCodeLensProvider(): Promise<void> {
if (this.registeredProvider) {
this.registeredProvider.dispose();
}
const patterns: RelativePattern[] = await testSourceProvider.getTestSourcePattern();
const documentSelector: DocumentSelector = parseDocumentSelector(patterns);
this.registeredProvider = languages.registerCodeLensProvider(documentSelector, this.internalProvider);
}
public refresh(): void {
this.internalProvider.refresh();
}
public dispose(): void {
this.internalProvider.dispose();
if (this.registeredProvider) {
this.registeredProvider.dispose();
}
this.configurationChangeListener.dispose();
}
private setCodeLensVisibility(): void {
this.internalProvider.setIsActivated(this.isCodeLensEnabled());
}
private isCodeLensEnabled(): boolean {
return workspace.getConfiguration().get<boolean>(ENABLE_EDITOR_SHORTCUTS_KEY, true);
}
}
export const testCodeLensController: TestCodeLensController = new TestCodeLensController();