Function getActivatedEnvVariables

src/test/interpreters/index.node.ts:77–110

(pythonPath: Uri)

const envVariables = new Map<string, Promise<NodeJS.ProcessEnv | undefined>>();
export async function getActivatedEnvVariables(pythonPath: Uri): Promise<NodeJS.ProcessEnv | undefined> {
    const key = getComparisonKey(pythonPath);
    if (envVariables.has(key)) {
        return envVariables.get(key);
    }
    const promise = (async () => {
        const cli = await getPythonCli(pythonPath);
        const processService = new ProcessService();
        const separator = 'e976ee50-99ed-4aba-9b6b-9dcd5634d07d';
        const argv = [...cli, path.join(SCRIPTS_DIR, 'printEnvVariables.py')];
        const cmd = argv.reduce((p, c) => (p ? `${p} "${c}"` : `"${c.replace('\\', '/')}"`), ''); // CodeQL [SM02383] Replace just the first occurrence of \\ as this could be a UNC path.
        const result = await processService.shellExec(cmd, {
            timeout: executionTimeout,
            maxBuffer: 1000 * 1000,
            throwOnStdErr: false,
            env: process.env,
            shell: defaultShell
        });
        if (result.stderr && result.stderr.length) {
            logger.error(`Failed to get env vars for shell ${defaultShell} with ${argv} stderr: ${result.stderr}`);
            return;
        }
        try {
            // Sometimes when environments get activated, we get a lot of noise in the output.
            // Having a separator allows us to filter out the noise.
            const output = result.stdout;
            return JSON.parse(output.substring(output.indexOf(separator) + separator.length).trim());
        } catch (ex) {
            logger.error(`Failed to get env vars for shell ${defaultShell} with ${argv}`, ex);
        }
    })();
    envVariables.set(key, promise);
    return promise;
}

async function getPythonCli(pythonPath: Uri | undefined) {
    const isConda = await isCondaEnvironment(pythonPath);