feat: initial powershell startup provider

Author: karthiknadig

src/features/terminal/startup/activateUsingShellStartup.ts

@@ -2,12 +2,36 @@ import { Disposable, GlobalEnvironmentVariableCollection } from 'vscode';
 import { onDidChangeConfiguration } from '../../../common/workspace.apis';
 import { registerCommand } from '../../../common/command.api';
 import { getAutoActivationType } from '../utils';
+import { EnvironmentManagers } from '../../../internal.api';
 
 export interface ActivateUsingShellStartup extends Disposable {}
 
 class ActivateUsingShellStartupImpl implements ActivateUsingShellStartup {
     private readonly disposables: Disposable[] = [];
-    constructor(private readonly envCollection: GlobalEnvironmentVariableCollection) {}
+    constructor(
+        private readonly envCollection: GlobalEnvironmentVariableCollection,
+        private readonly em: EnvironmentManagers,
+    ) {
+        this.disposables.push(
+            onDidChangeConfiguration((e) => {
+                this.handleConfigurationChange(e);
+            }),
+        );
+    }
+
+    private handleConfigurationChange(e) {
+        if (e.affectsConfiguration('python.terminal.autoActivationType')) {
+            const autoActType = getAutoActivationType();
+            if (autoActType === 'shellStartup') {
+                s;
+            } else {
+            }
+        }
+    }
+
+    private async addActivationVariables(): Promise<void> {}
+
+    private async removeActivationVariables(): Promise<void> {}
 
     dispose() {
         this.disposables.forEach((disposable) => disposable.dispose());
@@ -27,6 +51,7 @@ export async function removeAllStartupScripts(): Promise<void> {
 export function registerActivateUsingShellStartup(
     disposables: Disposable[],
     environmentVariableCollection: GlobalEnvironmentVariableCollection,
+    em: EnvironmentManagers,
 ) {
     let activateUsingShellStartup: ActivateUsingShellStartup | undefined;
 

src/features/terminal/startup/powershellStartup.ts

@@ -1,6 +1,9 @@
 import * as fs from 'fs-extra';
 import { getUserHomeDir } from '../../../common/utils/pathUtils';
 import { isWindows } from '../../../common/utils/platformUtils';
+import { ShellStartupProvider } from './startupProvider';
+import { EnvironmentVariableScope, GlobalEnvironmentVariableCollection } from 'vscode';
+import { EnvironmentManagers } from '../../../internal.api';
 
 const pwshActivationEnvVarKey = 'VSCODE_PWSH_ACTIVATE';
 
@@ -74,7 +77,7 @@ async function getPowerShellProfile(): Promise<PowerShellProfile | undefined> {
     return existingProfiles.length > 0 ? existingProfiles.sort((a, b) => a.type - b.type)[0] : undefined;
 }
 
-export async function isPowerShellStartupSetup(): Promise<boolean> {
+async function isPowerShellStartupSetup(): Promise<boolean> {
     const profile = await getPowerShellProfile();
     if (profile) {
         const content = await fs.readFile(profile.path, 'utf8');
@@ -83,7 +86,7 @@ export async function isPowerShellStartupSetup(): Promise<boolean> {
     return false;
 }
 
-export async function setupPowerShellStartup(): Promise<void> {
+async function setupPowerShellStartup(): Promise<void> {
     const lineSep = isWindows() ? '\r\n' : '\n';
     const activationContent = `${lineSep}${lineSep}# VSCODE-PYTHON-ACTIVATION:START${lineSep}if ($env:${pwshActivationEnvVarKey} -ne $null) {${lineSep}    Invoke-Expression $env:${pwshActivationEnvVarKey}${lineSep}}${lineSep}# VSCODE-PYTHON-ACTIVATION:END${lineSep}`;
     const profile = await getPowerShellProfile();
@@ -95,7 +98,7 @@ export async function setupPowerShellStartup(): Promise<void> {
     }
 }
 
-export async function removePowerShellStartup(): Promise<void> {
+async function removePowerShellStartup(): Promise<void> {
     const profile = await getPowerShellProfile();
     if (profile) {
         const content = await fs.readFile(profile.path, 'utf8');
@@ -108,3 +111,30 @@ export async function removePowerShellStartup(): Promise<void> {
         }
     }
 }
+
+export class PowershellStartupProvider implements ShellStartupProvider {
+    constructor(private readonly em: EnvironmentManagers) {}
+    async isSetup(): Promise<boolean> {
+        return await isPowerShellStartupSetup();
+    }
+
+    async setupScripts(): Promise<void> {
+        await setupPowerShellStartup();
+    }
+
+    async teardownScripts(): Promise<void> {
+        await removePowerShellStartup();
+    }
+
+    async updateEnvVariables(
+        global: GlobalEnvironmentVariableCollection,
+        scope: EnvironmentVariableScope,
+    ): Promise<void> {
+        const envVars = global.getScoped(scope);
+    }
+
+    async removeEnvVariables(envCollection: GlobalEnvironmentVariableCollection): Promise<void> {
+        // Implementation for removing environment variables if needed
+        // Currently, this method is not implemented
+    }
+}

src/features/terminal/startup/startupProvider.ts

@@ -0,0 +1,9 @@
+import { GlobalEnvironmentVariableCollection, Uri } from 'vscode';
+
+export interface ShellStartupProvider {
+    isSetup(): Promise<boolean>;
+    setupScripts(): Promise<void>;
+    removeScripts(): Promise<void>;
+    updateEnvVariables(global: GlobalEnvironmentVariableCollection, scope?: Uri): Promise<void>;
+    removeEnvVariables(global: GlobalEnvironmentVariableCollection, scope?: Uri): Promise<void>;
+}

src/features/terminal/utils.ts

@@ -97,3 +97,26 @@ export function getAutoActivationType(): 'off' | 'command' | 'shellStartup' {
     const config = getConfiguration('python-envs');
     return config.get<'off' | 'command' | 'shellStartup'>('terminal.autoActivationType', 'command');
 }
+
+export async function getAllDistinctProjectEnvironments(
+    api: PythonProjectGetterApi & PythonProjectEnvironmentApi,
+): Promise<PythonEnvironment[] | undefined> {
+    const envs: PythonEnvironment[] | undefined = [];
+
+    const projects = api.getPythonProjects();
+    if (projects.length === 0) {
+        const env = await api.getEnvironment(undefined);
+        if (env) {
+            envs.push(env);
+        }
+    } else if (projects.length === 1) {
+        const env = await api.getEnvironment(projects[0].uri);
+        if (env) {
+            envs.push(env);
+        }
+    } else {
+        envs.push(...(await getDistinctProjectEnvs(api, projects)));
+    }
+
+    return envs.length > 0 ? envs : undefined;
+}