-
Notifications
You must be signed in to change notification settings - Fork 41
Expand file tree
/
Copy pathmain.ts
More file actions
78 lines (74 loc) · 2.93 KB
/
main.ts
File metadata and controls
78 lines (74 loc) · 2.93 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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
import { Disposable, LogOutputChannel } from 'vscode';
import { PythonEnvironmentApi } from '../../api';
import { createSimpleDebounce } from '../../common/utils/debounce';
import { createFileSystemWatcher, onDidDeleteFiles } from '../../common/workspace.apis';
import { getPythonApi } from '../../features/pythonApi';
import { NativePythonFinder } from '../common/nativePythonFinder';
import { PipPackageManager } from './pipManager';
import { SysPythonManager } from './sysPythonManager';
import { VenvManager } from './venvManager';
export async function registerSystemPythonFeatures(
nativeFinder: NativePythonFinder,
disposables: Disposable[],
log: LogOutputChannel,
envManager: SysPythonManager,
): Promise<void> {
const api: PythonEnvironmentApi = await getPythonApi();
const venvManager = new VenvManager(nativeFinder, api, envManager, log);
const pkgManager = new PipPackageManager(api, log, venvManager);
disposables.push(
api.registerPackageManager(pkgManager),
api.registerEnvironmentManager(envManager),
api.registerEnvironmentManager(venvManager),
);
const venvDebouncedRefresh = createSimpleDebounce(500, () => {
venvManager.watcherRefresh();
});
const activationWatcher = createFileSystemWatcher('{**/activate}', false, true, false);
disposables.push(
activationWatcher,
activationWatcher.onDidCreate(() => {
venvDebouncedRefresh.trigger();
}),
activationWatcher.onDidDelete(() => {
venvDebouncedRefresh.trigger();
}),
onDidDeleteFiles(() => {
venvDebouncedRefresh.trigger();
}),
);
const packageDebouncedRefresh = createSimpleDebounce(500, async () => {
const projects = await api.getPythonProjects();
await Promise.all(
projects.map(async (project) => {
const env = await api.getEnvironment(project.uri);
if (!env) {
return;
}
try {
await api.refreshPackages(env);
} catch (ex) {
log.error(
`Failed to refresh packages for environment ${env.envId}: ${ex instanceof Error ? ex.message : String(ex)}`,
);
}
}),
);
});
const packageWatcher = createFileSystemWatcher(
'**/site-packages/*.dist-info/METADATA',
false, // don't ignore create events (pip install)
true, // ignore change events (content changes in METADATA don't affect package list)
false // don't ignore delete events (pip uninstall)
);
disposables.push(
packageDebouncedRefresh,
packageWatcher,
packageWatcher.onDidCreate(() => {
packageDebouncedRefresh.trigger();
}),
packageWatcher.onDidDelete(() => {
packageDebouncedRefresh.trigger();
}),
);
}