Fix reading JUPYTER_RUNTIME_DIR and XDG_RUNTIME_DIR (#16451)

Signed-off-by: Mathew Wicks <[email protected]>
microsoft · Feb 18, 2025 · 245979f · 245979f
1 parent 9f67a49
commit 245979f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/kernels/raw/finder/jupyterPaths.node.ts b/src/kernels/raw/finder/jupyterPaths.node.ts
@@ -133,16 +133,18 @@ export class JupyterPaths {
     private async getRuntimeDirImpl(): Promise<Uri | undefined> {
         let runtimeDir: Uri | undefined;
         const userHomeDir = this.platformService.homeDir;
-        if (userHomeDir) {
+        if (process.env['JUPYTER_RUNTIME_DIR']) {
+            runtimeDir = Uri.file(path.normalize(process.env['JUPYTER_RUNTIME_DIR']));
+        } else if (userHomeDir) {
             if (this.platformService.isWindows) {
                 // On windows the path is not correct if we combine those variables.
                 // It won't point to a path that you can actually read from.
                 runtimeDir = await tryGetRealPath(uriPath.joinPath(userHomeDir, winJupyterRuntimePath));
             } else if (this.platformService.isMac) {
                 runtimeDir = uriPath.joinPath(userHomeDir, macJupyterRuntimePath);
             } else {
-                runtimeDir = process.env['$XDG_RUNTIME_DIR']
-                    ? Uri.file(path.join(process.env['$XDG_RUNTIME_DIR'], 'jupyter', 'runtime'))
+                runtimeDir = process.env['XDG_RUNTIME_DIR']
+                    ? Uri.file(path.join(process.env['XDG_RUNTIME_DIR'], 'jupyter', 'runtime'))
                     : uriPath.joinPath(userHomeDir, '.local', 'share', 'jupyter', 'runtime');
             }
         }