@mcptoolshop/venvkit 0.2.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (63) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +249 -0
  3. package/dist/doctorLite.d.ts +75 -0
  4. package/dist/doctorLite.d.ts.map +1 -0
  5. package/dist/doctorLite.js +705 -0
  6. package/dist/doctorLite.js.map +1 -0
  7. package/dist/doctorLite.test.d.ts +2 -0
  8. package/dist/doctorLite.test.d.ts.map +1 -0
  9. package/dist/doctorLite.test.js +268 -0
  10. package/dist/doctorLite.test.js.map +1 -0
  11. package/dist/index.d.ts +6 -0
  12. package/dist/index.d.ts.map +1 -0
  13. package/dist/index.js +6 -0
  14. package/dist/index.js.map +1 -0
  15. package/dist/integration.test.d.ts +2 -0
  16. package/dist/integration.test.d.ts.map +1 -0
  17. package/dist/integration.test.js +245 -0
  18. package/dist/integration.test.js.map +1 -0
  19. package/dist/mapRender.d.ts +105 -0
  20. package/dist/mapRender.d.ts.map +1 -0
  21. package/dist/mapRender.js +718 -0
  22. package/dist/mapRender.js.map +1 -0
  23. package/dist/mapRender.test.d.ts +2 -0
  24. package/dist/mapRender.test.d.ts.map +1 -0
  25. package/dist/mapRender.test.js +571 -0
  26. package/dist/mapRender.test.js.map +1 -0
  27. package/dist/map_cli.d.ts +3 -0
  28. package/dist/map_cli.d.ts.map +1 -0
  29. package/dist/map_cli.js +278 -0
  30. package/dist/map_cli.js.map +1 -0
  31. package/dist/map_cli.test.d.ts +2 -0
  32. package/dist/map_cli.test.d.ts.map +1 -0
  33. package/dist/map_cli.test.js +276 -0
  34. package/dist/map_cli.test.js.map +1 -0
  35. package/dist/runLog.d.ts +71 -0
  36. package/dist/runLog.d.ts.map +1 -0
  37. package/dist/runLog.js +98 -0
  38. package/dist/runLog.js.map +1 -0
  39. package/dist/runLog.test.d.ts +2 -0
  40. package/dist/runLog.test.d.ts.map +1 -0
  41. package/dist/runLog.test.js +327 -0
  42. package/dist/runLog.test.js.map +1 -0
  43. package/dist/scanEnvPaths.d.ts +18 -0
  44. package/dist/scanEnvPaths.d.ts.map +1 -0
  45. package/dist/scanEnvPaths.js +174 -0
  46. package/dist/scanEnvPaths.js.map +1 -0
  47. package/dist/scanEnvPaths.test.d.ts +2 -0
  48. package/dist/scanEnvPaths.test.d.ts.map +1 -0
  49. package/dist/scanEnvPaths.test.js +250 -0
  50. package/dist/scanEnvPaths.test.js.map +1 -0
  51. package/dist/taskCluster.d.ts +62 -0
  52. package/dist/taskCluster.d.ts.map +1 -0
  53. package/dist/taskCluster.js +180 -0
  54. package/dist/taskCluster.js.map +1 -0
  55. package/dist/taskCluster.test.d.ts +2 -0
  56. package/dist/taskCluster.test.d.ts.map +1 -0
  57. package/dist/taskCluster.test.js +375 -0
  58. package/dist/taskCluster.test.js.map +1 -0
  59. package/dist/vitest.config.d.ts +3 -0
  60. package/dist/vitest.config.d.ts.map +1 -0
  61. package/dist/vitest.config.js +8 -0
  62. package/dist/vitest.config.js.map +1 -0
  63. package/package.json +58 -0
@@ -0,0 +1,174 @@
1
+ // scanEnvPaths.ts
2
+ //
3
+ // Discover Python environments on disk (venv + base interpreters) in a fast, pragmatic way.
4
+ // Outputs python executable paths you can feed into doctorLite + mapRender.
5
+ //
6
+ // Strategy (safe + useful defaults):
7
+ // - Project-local: .venv, venv, env, .python, .pyenv (common patterns)
8
+ // - Workspace roots you pass in
9
+ // - Global cache: ~/.venvkit/envs (if you use it)
10
+ // - Also looks for any directory containing pyvenv.cfg and a python executable.
11
+ //
12
+ // Notes:
13
+ // - This is discovery only. You still call doctorLite for health + facts.
14
+ // - On Windows we look for Scripts/python.exe; on POSIX bin/python.
15
+ import * as fs from "node:fs/promises";
16
+ import * as path from "node:path";
17
+ import * as os from "node:os";
18
+ function isWin() {
19
+ return os.platform() === "win32";
20
+ }
21
+ function norm(p) {
22
+ return isWin() ? p.toLowerCase() : p;
23
+ }
24
+ async function exists(p) {
25
+ try {
26
+ await fs.stat(p);
27
+ return true;
28
+ }
29
+ catch {
30
+ return false;
31
+ }
32
+ }
33
+ async function isDir(p) {
34
+ try {
35
+ return (await fs.stat(p)).isDirectory();
36
+ }
37
+ catch {
38
+ return false;
39
+ }
40
+ }
41
+ function venvPythonCandidates(venvRoot) {
42
+ return isWin()
43
+ ? [path.join(venvRoot, "Scripts", "python.exe"), path.join(venvRoot, "Scripts", "python")]
44
+ : [path.join(venvRoot, "bin", "python"), path.join(venvRoot, "bin", "python3")];
45
+ }
46
+ async function detectVenvAt(dir) {
47
+ const cfg = path.join(dir, "pyvenv.cfg");
48
+ if (!(await exists(cfg)))
49
+ return null;
50
+ for (const py of venvPythonCandidates(dir)) {
51
+ if (await exists(py))
52
+ return py;
53
+ }
54
+ return null;
55
+ }
56
+ async function listDirSafe(dir) {
57
+ try {
58
+ return await fs.readdir(dir);
59
+ }
60
+ catch {
61
+ return [];
62
+ }
63
+ }
64
+ async function walkForVenvs(root, maxDepth, includeHidden, out) {
65
+ const stack = [{ dir: root, depth: 0 }];
66
+ while (stack.length) {
67
+ const cur = stack.pop();
68
+ if (cur.depth > maxDepth)
69
+ continue;
70
+ // quick venv detection at this directory
71
+ const venvPy = await detectVenvAt(cur.dir);
72
+ if (venvPy) {
73
+ out.push(venvPy);
74
+ // Do not descend into this venv further (keeps scan fast)
75
+ continue;
76
+ }
77
+ const children = await listDirSafe(cur.dir);
78
+ for (const name of children) {
79
+ if (!includeHidden && name.startsWith("."))
80
+ continue;
81
+ const full = path.join(cur.dir, name);
82
+ if (!(await isDir(full)))
83
+ continue;
84
+ // Skip common heavy directories
85
+ if (name === "node_modules" || name === "dist" || name === "build" || name === ".git")
86
+ continue;
87
+ stack.push({ dir: full, depth: cur.depth + 1 });
88
+ }
89
+ }
90
+ }
91
+ async function discoverProjectLocalRoots(root) {
92
+ const candidates = [".venv", "venv", "env", ".python", ".pyenv"];
93
+ const out = [];
94
+ for (const c of candidates) {
95
+ const p = path.join(root, c);
96
+ if (await isDir(p))
97
+ out.push(p);
98
+ }
99
+ return out;
100
+ }
101
+ async function discoverBasePythons() {
102
+ // Best-effort:
103
+ // - On Windows: common install locations
104
+ // - On POSIX: common binaries
105
+ const out = [];
106
+ if (isWin()) {
107
+ const possible = [
108
+ "C:\\Python312\\python.exe",
109
+ "C:\\Python311\\python.exe",
110
+ "C:\\Python310\\python.exe",
111
+ "C:\\Program Files\\Python312\\python.exe",
112
+ "C:\\Program Files\\Python311\\python.exe",
113
+ "C:\\Program Files\\Python310\\python.exe",
114
+ path.join(os.homedir(), "AppData", "Local", "Programs", "Python", "Python312", "python.exe"),
115
+ path.join(os.homedir(), "AppData", "Local", "Programs", "Python", "Python311", "python.exe"),
116
+ path.join(os.homedir(), "AppData", "Local", "Programs", "Python", "Python310", "python.exe"),
117
+ ];
118
+ for (const p of possible)
119
+ if (await exists(p))
120
+ out.push(p);
121
+ }
122
+ else {
123
+ const possible = ["/usr/bin/python3", "/usr/local/bin/python3", "/opt/homebrew/bin/python3", "/usr/bin/python"];
124
+ for (const p of possible)
125
+ if (await exists(p))
126
+ out.push(p);
127
+ }
128
+ return out;
129
+ }
130
+ export async function scanEnvPaths(options) {
131
+ const maxDepth = options.maxDepth ?? 5;
132
+ const includeHidden = options.includeHidden ?? false;
133
+ const includeUserHomeCache = options.includeUserHomeCache ?? true;
134
+ const dedupe = options.dedupe ?? true;
135
+ const roots = [...options.roots];
136
+ if (includeUserHomeCache) {
137
+ roots.push(path.join(os.homedir(), ".venvkit", "envs"));
138
+ roots.push(path.join(os.homedir(), ".virtualenvs"));
139
+ }
140
+ const pythonPaths = [];
141
+ // Fast pass: project-local patterns in each root
142
+ for (const r of options.roots) {
143
+ const locals = await discoverProjectLocalRoots(r);
144
+ for (const venvRoot of locals) {
145
+ const py = await detectVenvAt(venvRoot);
146
+ if (py)
147
+ pythonPaths.push(py);
148
+ }
149
+ }
150
+ // Walk: find pyvenv.cfg
151
+ for (const root of roots) {
152
+ if (!(await isDir(root)))
153
+ continue;
154
+ await walkForVenvs(root, maxDepth, includeHidden, pythonPaths);
155
+ }
156
+ // Optional: base interpreter candidates
157
+ const bases = await discoverBasePythons();
158
+ for (const b of bases)
159
+ pythonPaths.push(b);
160
+ const final = dedupe ? [...new Map(pythonPaths.map((p) => [norm(p), p])).values()] : pythonPaths;
161
+ // crude counts
162
+ const foundBases = bases.length;
163
+ const foundVenvs = Math.max(0, final.length - foundBases);
164
+ return {
165
+ pythonPaths: final,
166
+ meta: {
167
+ scannedRoots: roots,
168
+ maxDepth,
169
+ foundVenvs,
170
+ foundBases,
171
+ },
172
+ };
173
+ }
174
+ //# sourceMappingURL=scanEnvPaths.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"scanEnvPaths.js","sourceRoot":"","sources":["../scanEnvPaths.ts"],"names":[],"mappings":"AAAA,kBAAkB;AAClB,EAAE;AACF,4FAA4F;AAC5F,4EAA4E;AAC5E,EAAE;AACF,qCAAqC;AACrC,uEAAuE;AACvE,gCAAgC;AAChC,kDAAkD;AAClD,gFAAgF;AAChF,EAAE;AACF,SAAS;AACT,0EAA0E;AAC1E,oEAAoE;AAEpE,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAoB9B,SAAS,KAAK;IACZ,OAAO,EAAE,CAAC,QAAQ,EAAE,KAAK,OAAO,CAAC;AACnC,CAAC;AAED,SAAS,IAAI,CAAC,CAAS;IACrB,OAAO,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;AACvC,CAAC;AAED,KAAK,UAAU,MAAM,CAAC,CAAS;IAC7B,IAAI,CAAC;QACH,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,KAAK,UAAU,KAAK,CAAC,CAAS;IAC5B,IAAI,CAAC;QACH,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;IAC1C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAAC,QAAgB;IAC5C,OAAO,KAAK,EAAE;QACZ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,YAAY,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC1F,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,QAAQ,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC;AACpF,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,GAAW;IACrC,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;IACzC,IAAI,CAAC,CAAC,MAAM,MAAM,CAAC,GAAG,CAAC,CAAC;QAAE,OAAO,IAAI,CAAC;IAEtC,KAAK,MAAM,EAAE,IAAI,oBAAoB,CAAC,GAAG,CAAC,EAAE,CAAC;QAC3C,IAAI,MAAM,MAAM,CAAC,EAAE,CAAC;YAAE,OAAO,EAAE,CAAC;IAClC,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,GAAW;IACpC,IAAI,CAAC;QACH,OAAO,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,IAAY,EAAE,QAAgB,EAAE,aAAsB,EAAE,GAAa;IAC/F,MAAM,KAAK,GAA0C,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;IAE/E,OAAO,KAAK,CAAC,MAAM,EAAE,CAAC;QACpB,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,EAAG,CAAC;QACzB,IAAI,GAAG,CAAC,KAAK,GAAG,QAAQ;YAAE,SAAS;QAEnC,yCAAyC;QACzC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC3C,IAAI,MAAM,EAAE,CAAC;YACX,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACjB,0DAA0D;YAC1D,SAAS;QACX,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC5C,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC5B,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,SAAS;YAErD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;YACtC,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC;gBAAE,SAAS;YAEnC,gCAAgC;YAChC,IAAI,IAAI,KAAK,cAAc,IAAI,IAAI,KAAK,MAAM,IAAI,IAAI,KAAK,OAAO,IAAI,IAAI,KAAK,MAAM;gBAAE,SAAS;YAEhG,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;AACH,CAAC;AAED,KAAK,UAAU,yBAAyB,CAAC,IAAY;IACnD,MAAM,UAAU,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;IACjE,MAAM,GAAG,GAAa,EAAE,CAAC;IACzB,KAAK,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAC7B,IAAI,MAAM,KAAK,CAAC,CAAC,CAAC;YAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,KAAK,UAAU,mBAAmB;IAChC,eAAe;IACf,yCAAyC;IACzC,8BAA8B;IAC9B,MAAM,GAAG,GAAa,EAAE,CAAC;IAEzB,IAAI,KAAK,EAAE,EAAE,CAAC;QACZ,MAAM,QAAQ,GAAG;YACf,2BAA2B;YAC3B,2BAA2B;YAC3B,2BAA2B;YAC3B,0CAA0C;YAC1C,0CAA0C;YAC1C,0CAA0C;YAC1C,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,CAAC;YAC5F,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,CAAC;YAC5F,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,CAAC;SAC7F,CAAC;QACF,KAAK,MAAM,CAAC,IAAI,QAAQ;YAAE,IAAI,MAAM,MAAM,CAAC,CAAC,CAAC;gBAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7D,CAAC;SAAM,CAAC;QACN,MAAM,QAAQ,GAAG,CAAC,kBAAkB,EAAE,wBAAwB,EAAE,2BAA2B,EAAE,iBAAiB,CAAC,CAAC;QAChH,KAAK,MAAM,CAAC,IAAI,QAAQ;YAAE,IAAI,MAAM,MAAM,CAAC,CAAC,CAAC;gBAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7D,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,OAAoB;IACrD,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,CAAC,CAAC;IACvC,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,KAAK,CAAC;IACrD,MAAM,oBAAoB,GAAG,OAAO,CAAC,oBAAoB,IAAI,IAAI,CAAC;IAClE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC;IAEtC,MAAM,KAAK,GAAG,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;IAEjC,IAAI,oBAAoB,EAAE,CAAC;QACzB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;QACxD,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,cAAc,CAAC,CAAC,CAAC;IACtD,CAAC;IAED,MAAM,WAAW,GAAa,EAAE,CAAC;IAEjC,iDAAiD;IACjD,KAAK,MAAM,CAAC,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,yBAAyB,CAAC,CAAC,CAAC,CAAC;QAClD,KAAK,MAAM,QAAQ,IAAI,MAAM,EAAE,CAAC;YAC9B,MAAM,EAAE,GAAG,MAAM,YAAY,CAAC,QAAQ,CAAC,CAAC;YACxC,IAAI,EAAE;gBAAE,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED,wBAAwB;IACxB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC;YAAE,SAAS;QACnC,MAAM,YAAY,CAAC,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,WAAW,CAAC,CAAC;IACjE,CAAC;IAED,wCAAwC;IACxC,MAAM,KAAK,GAAG,MAAM,mBAAmB,EAAE,CAAC;IAC1C,KAAK,MAAM,CAAC,IAAI,KAAK;QAAE,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAE3C,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC;IAEjG,eAAe;IACf,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC;IAChC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,GAAG,UAAU,CAAC,CAAC;IAE1D,OAAO;QACL,WAAW,EAAE,KAAK;QAClB,IAAI,EAAE;YACJ,YAAY,EAAE,KAAK;YACnB,QAAQ;YACR,UAAU;YACV,UAAU;SACX;KACF,CAAC;AACJ,CAAC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=scanEnvPaths.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"scanEnvPaths.test.d.ts","sourceRoot":"","sources":["../scanEnvPaths.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,250 @@
1
+ // scanEnvPaths.test.ts
2
+ // Unit tests for environment path discovery
3
+ import { describe, it, expect, beforeEach, afterEach } from "vitest";
4
+ import * as fs from "node:fs/promises";
5
+ import * as path from "node:path";
6
+ import * as os from "node:os";
7
+ import { scanEnvPaths } from "./scanEnvPaths.js";
8
+ describe("scanEnvPaths", () => {
9
+ let tempDir;
10
+ beforeEach(async () => {
11
+ tempDir = await fs.mkdtemp(path.join(os.tmpdir(), "venvkit-scan-test-"));
12
+ });
13
+ afterEach(async () => {
14
+ await fs.rm(tempDir, { recursive: true, force: true });
15
+ });
16
+ // Helper to create a mock venv structure
17
+ async function createMockVenv(venvPath) {
18
+ await fs.mkdir(venvPath, { recursive: true });
19
+ await fs.writeFile(path.join(venvPath, "pyvenv.cfg"), "home = C:\\Python311\nversion = 3.11.5\n");
20
+ const binDir = os.platform() === "win32" ? "Scripts" : "bin";
21
+ const pyExe = os.platform() === "win32" ? "python.exe" : "python";
22
+ await fs.mkdir(path.join(venvPath, binDir), { recursive: true });
23
+ await fs.writeFile(path.join(venvPath, binDir, pyExe), "fake python");
24
+ return path.join(venvPath, binDir, pyExe);
25
+ }
26
+ describe("finding Python venvs", () => {
27
+ it("finds venv with pyvenv.cfg", async () => {
28
+ const venvPath = path.join(tempDir, "project", ".venv");
29
+ await createMockVenv(venvPath);
30
+ const result = await scanEnvPaths({
31
+ roots: [path.join(tempDir, "project")],
32
+ maxDepth: 3,
33
+ includeUserHomeCache: false,
34
+ });
35
+ expect(result.pythonPaths.length).toBeGreaterThanOrEqual(1);
36
+ expect(result.meta.foundVenvs).toBeGreaterThanOrEqual(1);
37
+ });
38
+ it("finds multiple venvs in project tree", async () => {
39
+ await createMockVenv(path.join(tempDir, "project1", ".venv"));
40
+ await createMockVenv(path.join(tempDir, "project2", "venv"));
41
+ await createMockVenv(path.join(tempDir, "project3", "env"));
42
+ const result = await scanEnvPaths({
43
+ roots: [tempDir],
44
+ maxDepth: 3,
45
+ includeUserHomeCache: false,
46
+ });
47
+ expect(result.meta.foundVenvs).toBeGreaterThanOrEqual(3);
48
+ });
49
+ it("finds venv in standard project locations", async () => {
50
+ // Test standard venv folder names: .venv, venv, env
51
+ const projectRoot = path.join(tempDir, "myproject");
52
+ await fs.mkdir(projectRoot, { recursive: true });
53
+ const venvPath = path.join(projectRoot, ".venv");
54
+ await createMockVenv(venvPath);
55
+ const result = await scanEnvPaths({
56
+ roots: [projectRoot],
57
+ maxDepth: 3,
58
+ includeUserHomeCache: false,
59
+ });
60
+ expect(result.pythonPaths.some(p => p.includes(".venv"))).toBe(true);
61
+ });
62
+ });
63
+ describe("scan options", () => {
64
+ it("respects maxDepth limit", async () => {
65
+ // Create venv at depth 2
66
+ await createMockVenv(path.join(tempDir, "level1", ".venv"));
67
+ // Create venv at depth 4 (should not be found with maxDepth=2)
68
+ await createMockVenv(path.join(tempDir, "level1", "level2", "level3", ".venv"));
69
+ const result = await scanEnvPaths({
70
+ roots: [tempDir],
71
+ maxDepth: 2,
72
+ includeUserHomeCache: false,
73
+ });
74
+ // Should find shallow venv but not deeply nested one
75
+ const foundPaths = result.pythonPaths.filter(p => p.includes(tempDir));
76
+ expect(foundPaths.length).toBe(1);
77
+ });
78
+ it("ignores hidden directories by default", async () => {
79
+ await createMockVenv(path.join(tempDir, ".hidden", "venv"));
80
+ await createMockVenv(path.join(tempDir, "visible", "venv"));
81
+ const result = await scanEnvPaths({
82
+ roots: [tempDir],
83
+ maxDepth: 3,
84
+ includeHidden: false,
85
+ includeUserHomeCache: false,
86
+ });
87
+ const foundPaths = result.pythonPaths.filter(p => p.includes(tempDir));
88
+ expect(foundPaths.some(p => p.includes(".hidden"))).toBe(false);
89
+ expect(foundPaths.some(p => p.includes("visible"))).toBe(true);
90
+ });
91
+ it("includes hidden directories when flag is set", async () => {
92
+ await createMockVenv(path.join(tempDir, ".hidden", "venv"));
93
+ const result = await scanEnvPaths({
94
+ roots: [tempDir],
95
+ maxDepth: 3,
96
+ includeHidden: true,
97
+ includeUserHomeCache: false,
98
+ });
99
+ const foundPaths = result.pythonPaths.filter(p => p.includes(tempDir));
100
+ expect(foundPaths.some(p => p.includes(".hidden"))).toBe(true);
101
+ });
102
+ it("deduplicates paths by default", async () => {
103
+ const venvPath = path.join(tempDir, "project", ".venv");
104
+ await createMockVenv(venvPath);
105
+ // Scan same root twice
106
+ const result = await scanEnvPaths({
107
+ roots: [path.join(tempDir, "project"), path.join(tempDir, "project")],
108
+ maxDepth: 3,
109
+ dedupe: true,
110
+ includeUserHomeCache: false,
111
+ });
112
+ const foundPaths = result.pythonPaths.filter(p => p.includes(tempDir));
113
+ // Should only have one entry despite scanning same location twice
114
+ expect(new Set(foundPaths).size).toBe(foundPaths.length);
115
+ });
116
+ });
117
+ describe("handling edge cases", () => {
118
+ it("handles empty directory", async () => {
119
+ const emptyDir = path.join(tempDir, "empty");
120
+ await fs.mkdir(emptyDir, { recursive: true });
121
+ const result = await scanEnvPaths({
122
+ roots: [emptyDir],
123
+ maxDepth: 3,
124
+ includeUserHomeCache: false,
125
+ });
126
+ const foundInTemp = result.pythonPaths.filter(p => p.includes(tempDir));
127
+ expect(foundInTemp.length).toBe(0);
128
+ });
129
+ it("handles permission errors gracefully", async () => {
130
+ // This test might behave differently on Windows vs POSIX
131
+ // The key is that it doesn't throw
132
+ const result = await scanEnvPaths({
133
+ roots: ["/nonexistent/path/that/should/not/exist"],
134
+ maxDepth: 3,
135
+ includeUserHomeCache: false,
136
+ });
137
+ expect(result).toBeDefined();
138
+ expect(result.pythonPaths).toBeDefined();
139
+ });
140
+ it("skips node_modules directory", async () => {
141
+ // Create a venv inside node_modules (should be skipped)
142
+ await createMockVenv(path.join(tempDir, "node_modules", "some-pkg", ".venv"));
143
+ await createMockVenv(path.join(tempDir, "src", ".venv"));
144
+ const result = await scanEnvPaths({
145
+ roots: [tempDir],
146
+ maxDepth: 5,
147
+ includeUserHomeCache: false,
148
+ });
149
+ const foundPaths = result.pythonPaths.filter(p => p.includes(tempDir));
150
+ expect(foundPaths.some(p => p.includes("node_modules"))).toBe(false);
151
+ expect(foundPaths.some(p => p.includes("src"))).toBe(true);
152
+ });
153
+ it("skips .git directory", async () => {
154
+ await createMockVenv(path.join(tempDir, ".git", "hooks", ".venv"));
155
+ await createMockVenv(path.join(tempDir, "app", ".venv"));
156
+ const result = await scanEnvPaths({
157
+ roots: [tempDir],
158
+ maxDepth: 5,
159
+ includeHidden: true, // Even with hidden enabled, .git should be skipped
160
+ includeUserHomeCache: false,
161
+ });
162
+ const foundPaths = result.pythonPaths.filter(p => p.includes(tempDir));
163
+ expect(foundPaths.some(p => p.includes(".git"))).toBe(false);
164
+ });
165
+ it("handles venv without python executable", async () => {
166
+ // Create pyvenv.cfg but no python executable
167
+ const venvPath = path.join(tempDir, "broken-venv");
168
+ await fs.mkdir(venvPath, { recursive: true });
169
+ await fs.writeFile(path.join(venvPath, "pyvenv.cfg"), "home = C:\\Python311\n");
170
+ // Deliberately don't create Scripts/python.exe
171
+ const result = await scanEnvPaths({
172
+ roots: [tempDir],
173
+ maxDepth: 3,
174
+ includeUserHomeCache: false,
175
+ });
176
+ // Should not crash, and should not find the broken venv
177
+ const foundBroken = result.pythonPaths.filter(p => p.includes("broken-venv"));
178
+ expect(foundBroken.length).toBe(0);
179
+ });
180
+ });
181
+ describe("meta information", () => {
182
+ it("reports scanned roots in meta", async () => {
183
+ const root1 = path.join(tempDir, "root1");
184
+ const root2 = path.join(tempDir, "root2");
185
+ await fs.mkdir(root1, { recursive: true });
186
+ await fs.mkdir(root2, { recursive: true });
187
+ const result = await scanEnvPaths({
188
+ roots: [root1, root2],
189
+ maxDepth: 3,
190
+ includeUserHomeCache: false,
191
+ });
192
+ expect(result.meta.scannedRoots).toContain(root1);
193
+ expect(result.meta.scannedRoots).toContain(root2);
194
+ expect(result.meta.maxDepth).toBe(3);
195
+ });
196
+ it("counts venvs and bases separately", async () => {
197
+ await createMockVenv(path.join(tempDir, "venv1"));
198
+ await createMockVenv(path.join(tempDir, "venv2"));
199
+ const result = await scanEnvPaths({
200
+ roots: [tempDir],
201
+ maxDepth: 3,
202
+ includeUserHomeCache: false,
203
+ });
204
+ expect(result.meta.foundVenvs).toBeGreaterThanOrEqual(2);
205
+ expect(typeof result.meta.foundBases).toBe("number");
206
+ });
207
+ });
208
+ describe("nested environment handling", () => {
209
+ it("detects venv and does not descend further", async () => {
210
+ // Create a venv with a nested venv inside (unusual but possible)
211
+ const outerVenv = path.join(tempDir, "outer-venv");
212
+ await createMockVenv(outerVenv);
213
+ // Create another venv inside (should not be found because we stop at outer venv)
214
+ const innerVenv = path.join(outerVenv, "nested", "inner-venv");
215
+ await createMockVenv(innerVenv);
216
+ const result = await scanEnvPaths({
217
+ roots: [tempDir],
218
+ maxDepth: 10,
219
+ includeUserHomeCache: false,
220
+ });
221
+ const foundPaths = result.pythonPaths.filter(p => p.includes(tempDir));
222
+ const foundOuter = foundPaths.filter(p => p.includes("outer-venv") && !p.includes("inner-venv"));
223
+ const foundInner = foundPaths.filter(p => p.includes("inner-venv"));
224
+ // Should find outer venv
225
+ expect(foundOuter.length).toBe(1);
226
+ // Should NOT find inner venv (scan stops at venv boundary)
227
+ expect(foundInner.length).toBe(0);
228
+ });
229
+ });
230
+ describe("large directory trees", () => {
231
+ it("handles directory with many subdirectories", async () => {
232
+ // Create many directories to test performance doesn't degrade badly
233
+ for (let i = 0; i < 20; i++) {
234
+ await fs.mkdir(path.join(tempDir, `dir${i}`), { recursive: true });
235
+ }
236
+ await createMockVenv(path.join(tempDir, "dir10", ".venv"));
237
+ const start = Date.now();
238
+ const result = await scanEnvPaths({
239
+ roots: [tempDir],
240
+ maxDepth: 3,
241
+ includeUserHomeCache: false,
242
+ });
243
+ const elapsed = Date.now() - start;
244
+ expect(result.meta.foundVenvs).toBeGreaterThanOrEqual(1);
245
+ // Should complete in reasonable time (less than 5 seconds)
246
+ expect(elapsed).toBeLessThan(5000);
247
+ });
248
+ });
249
+ });
250
+ //# sourceMappingURL=scanEnvPaths.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"scanEnvPaths.test.js","sourceRoot":"","sources":["../scanEnvPaths.test.ts"],"names":[],"mappings":"AAAA,uBAAuB;AACvB,4CAA4C;AAE5C,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAM,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EAAE,YAAY,EAAqC,MAAM,mBAAmB,CAAC;AAEpF,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,OAAe,CAAC;IAEpB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,oBAAoB,CAAC,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,yCAAyC;IACzC,KAAK,UAAU,cAAc,CAAC,QAAgB;QAC5C,MAAM,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9C,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,YAAY,CAAC,EAAE,0CAA0C,CAAC,CAAC;QAElG,MAAM,MAAM,GAAG,EAAE,CAAC,QAAQ,EAAE,KAAK,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC;QAC7D,MAAM,KAAK,GAAG,EAAE,CAAC,QAAQ,EAAE,KAAK,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC;QAElE,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACjE,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,aAAa,CAAC,CAAC;QAEtE,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;IAC5C,CAAC;IAED,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;YACxD,MAAM,cAAc,CAAC,QAAQ,CAAC,CAAC;YAE/B,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;gBACtC,QAAQ,EAAE,CAAC;gBACX,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;YAC9D,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;YAC7D,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC,CAAC;YAE5D,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,OAAO,CAAC;gBAChB,QAAQ,EAAE,CAAC;gBACX,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,oDAAoD;YACpD,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;YACpD,MAAM,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAEjD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACjD,MAAM,cAAc,CAAC,QAAQ,CAAC,CAAC;YAE/B,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,WAAW,CAAC;gBACpB,QAAQ,EAAE,CAAC;gBACX,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,yBAAyB;YACzB,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;YAC5D,+DAA+D;YAC/D,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;YAEhF,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,OAAO,CAAC;gBAChB,QAAQ,EAAE,CAAC;gBACX,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,qDAAqD;YACrD,MAAM,UAAU,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YACvE,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC;YAC5D,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC;YAE5D,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,OAAO,CAAC;gBAChB,QAAQ,EAAE,CAAC;gBACX,aAAa,EAAE,KAAK;gBACpB,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YACvE,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChE,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC;YAE5D,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,OAAO,CAAC;gBAChB,QAAQ,EAAE,CAAC;gBACX,aAAa,EAAE,IAAI;gBACnB,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YACvE,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;YACxD,MAAM,cAAc,CAAC,QAAQ,CAAC,CAAC;YAE/B,uBAAuB;YACvB,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;gBACrE,QAAQ,EAAE,CAAC;gBACX,MAAM,EAAE,IAAI;gBACZ,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YACvE,kEAAkE;YAClE,MAAM,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC7C,MAAM,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAE9C,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,QAAQ,CAAC;gBACjB,QAAQ,EAAE,CAAC;gBACX,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YACxE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,yDAAyD;YACzD,mCAAmC;YACnC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,yCAAyC,CAAC;gBAClD,QAAQ,EAAE,CAAC;gBACX,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,wDAAwD;YACxD,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;YAC9E,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;YAEzD,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,OAAO,CAAC;gBAChB,QAAQ,EAAE,CAAC;gBACX,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YACvE,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrE,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;YACpC,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YACnE,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;YAEzD,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,OAAO,CAAC;gBAChB,QAAQ,EAAE,CAAC;gBACX,aAAa,EAAE,IAAI,EAAE,mDAAmD;gBACxE,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YACvE,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,6CAA6C;YAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YACnD,MAAM,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC9C,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,YAAY,CAAC,EAAE,wBAAwB,CAAC,CAAC;YAChF,+CAA+C;YAE/C,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,OAAO,CAAC;gBAChB,QAAQ,EAAE,CAAC;gBACX,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,wDAAwD;YACxD,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC;YAC9E,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC1C,MAAM,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3C,MAAM,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAE3C,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC;gBACrB,QAAQ,EAAE,CAAC;gBACX,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAClD,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAElD,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,OAAO,CAAC;gBAChB,QAAQ,EAAE,CAAC;gBACX,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;YACzD,MAAM,CAAC,OAAO,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;QAC3C,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,iEAAiE;YACjE,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;YACnD,MAAM,cAAc,CAAC,SAAS,CAAC,CAAC;YAEhC,iFAAiF;YACjF,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;YAC/D,MAAM,cAAc,CAAC,SAAS,CAAC,CAAC;YAEhC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,OAAO,CAAC;gBAChB,QAAQ,EAAE,EAAE;gBACZ,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YACvE,MAAM,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC;YACjG,MAAM,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC;YAEpE,yBAAyB;YACzB,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAClC,2DAA2D;YAC3D,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,oEAAoE;YACpE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,EAAE,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACrE,CAAC;YACD,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAE3D,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACzB,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK,EAAE,CAAC,OAAO,CAAC;gBAChB,QAAQ,EAAE,CAAC;gBACX,oBAAoB,EAAE,KAAK;aAC5B,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;YAEnC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;YACzD,2DAA2D;YAC3D,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,62 @@
1
+ import type { RunLogEventV1 } from "./runLog.js";
2
+ export type TaskSignature = {
3
+ sigId: string;
4
+ name: string;
5
+ command: string;
6
+ requirementsKey: string;
7
+ };
8
+ export type TaskCluster = {
9
+ sig: TaskSignature;
10
+ runs: number;
11
+ ok: number;
12
+ fail: number;
13
+ successRate: number;
14
+ lastAt: string;
15
+ dominantFailure?: string;
16
+ failureCounts: Record<string, number>;
17
+ envCounts: Record<string, number>;
18
+ envFailCounts: Record<string, number>;
19
+ envOkCounts: Record<string, number>;
20
+ };
21
+ /**
22
+ * Extract a stable signature from a run event.
23
+ * Runs with the same signature are clustered together.
24
+ */
25
+ export declare function signatureForRun(run: RunLogEventV1): TaskSignature;
26
+ /**
27
+ * Cluster runs by task signature.
28
+ * Returns clusters sorted by run count (descending).
29
+ */
30
+ export declare function clusterRuns(runs: RunLogEventV1[]): TaskCluster[];
31
+ /**
32
+ * Determine if a task cluster is flaky.
33
+ * Flaky = both succeeds and fails, with success rate not extreme.
34
+ */
35
+ export declare function isFlaky(cluster: TaskCluster): boolean;
36
+ /**
37
+ * Determine if a task cluster is env-dependent flaky.
38
+ * This means it succeeds on some envs and fails on others.
39
+ */
40
+ export declare function isEnvDependentFlaky(cluster: TaskCluster): boolean;
41
+ /**
42
+ * Get the envs where this task fails most.
43
+ */
44
+ export declare function getFailingEnvs(cluster: TaskCluster, limit?: number): Array<{
45
+ pythonPath: string;
46
+ failCount: number;
47
+ totalCount: number;
48
+ failRate: number;
49
+ }>;
50
+ /**
51
+ * Get summary statistics for a set of clusters.
52
+ */
53
+ export declare function summarizeClusters(clusters: TaskCluster[]): {
54
+ totalTasks: number;
55
+ totalRuns: number;
56
+ totalOk: number;
57
+ totalFail: number;
58
+ overallSuccessRate: number;
59
+ flakyCount: number;
60
+ envDependentFlakyCount: number;
61
+ };
62
+ //# sourceMappingURL=taskCluster.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"taskCluster.d.ts","sourceRoot":"","sources":["../taskCluster.ts"],"names":[],"mappings":"AAcA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAEjD,MAAM,MAAM,aAAa,GAAG;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,eAAe,EAAE,MAAM,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG;IACxB,GAAG,EAAE,aAAa,CAAC;IAEnB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IAEf,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAEtC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAClC,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACtC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACrC,CAAC;AA6BF;;;GAGG;AACH,wBAAgB,eAAe,CAAC,GAAG,EAAE,aAAa,GAAG,aAAa,CAMjE;AAED;;;GAGG;AACH,wBAAgB,WAAW,CAAC,IAAI,EAAE,aAAa,EAAE,GAAG,WAAW,EAAE,CAuDhE;AAED;;;GAGG;AACH,wBAAgB,OAAO,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAKrD;AAED;;;GAGG;AACH,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAejE;AAED;;GAEG;AACH,wBAAgB,cAAc,CAC5B,OAAO,EAAE,WAAW,EACpB,KAAK,SAAI,GACR,KAAK,CAAC;IAAE,UAAU,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,CAAC,CAUxF;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,WAAW,EAAE,GAAG;IAC1D,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,sBAAsB,EAAE,MAAM,CAAC;CAChC,CAwBA"}