extension.js

const { PythonExtension } = require("@vscode/python-extension");
const vscode = require("vscode");
const fs = require("fs");
const path = require("path");

/**
 * @param {vscode.ExtensionContext} context
 */
async function activate(context) {
    let pythonApi = await PythonExtension.api();
    const activeEditor = vscode.window.activeTextEditor;

    if (activeEditor) {
        await setupPythonEnvironment(activeEditor, pythonApi);
    }

    let disposable = vscode.window.onDidChangeActiveTextEditor(async (editor) => {
        if (editor) {
            await setupPythonEnvironment(editor, pythonApi);
        }
    });

    context.subscriptions.push(disposable);
}

async function setupPythonEnvironment(editor, pythonApi) {
    let currentDir = path.dirname(editor.document.uri.fsPath);
    const root = path.parse(currentDir).root;
    const currentWorkspaceFolder = vscode.workspace.getWorkspaceFolder(vscode.Uri.file(editor.document.uri.path));
    const currentWorkspaceFolderPath = currentWorkspaceFolder ? currentWorkspaceFolder.uri.path : null;

    // Get configuration settings
    const config = vscode.workspace.getConfiguration('pythonEnvy');
    const venvName = config.get('venvName');
    const showNotifications = config.get('showNotifications', true); // Default to true for backward compatibility

    while (currentDir !== root) {
        const venvPath = path.join(currentDir, venvName);

        if (fs.existsSync(venvPath) && fs.lstatSync(venvPath).isDirectory()) {
            const currentPythonPath =
                pythonApi.environments.getActiveEnvironmentPath(currentWorkspaceFolder.uri);
            let pythonPath = path.join(venvPath, "bin", "python");

            if (!fs.existsSync(pythonPath)) {
                pythonPath = path.join(venvPath, "Scripts", "python.exe");
            }

            if (currentPythonPath.path !== pythonPath) {
                try {
                    const relativePath = path.relative(currentWorkspaceFolderPath, venvPath);

                    await pythonApi.environments.updateActiveEnvironmentPath(pythonPath, currentWorkspaceFolder.uri);

                    // Only show notification if enabled in settings
                    if (showNotifications) {
                        vscode.window.showInformationMessage(
                            `Python Envy: interpreter set to ${relativePath} for ${currentWorkspaceFolder.name}`
                        );
                    }
                } catch (error) {
                    vscode.window.showErrorMessage(
                        `Python Envy: error setting Python interpreter: ${error.message}`
                    );
                }
            }
            return;
        }

        if (currentDir === currentWorkspaceFolderPath) {
            break;
        }

        currentDir = path.dirname(currentDir);
        if (currentDir === ".") {
            currentDir = "";
        }
    }
}

function deactivate() { }

module.exports = {
    activate,
    deactivate,
};