platform.ts

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

import * as os from "os";
import * as path from "path";
import * as process from "process";
import vscode = require("vscode");
import { integer } from "vscode-languageserver-protocol";
import { ILogger } from "./logging";
import { changeSetting, getSettings, PowerShellAdditionalExePathSettings } from "./settings";

// This uses require so we can rewire it in unit tests!
// eslint-disable-next-line @typescript-eslint/no-explicit-any, @typescript-eslint/no-var-requires
const utils = require("./utils");

const WindowsPowerShell64BitLabel = "Windows PowerShell (x64)";
const WindowsPowerShell32BitLabel = "Windows PowerShell (x86)";

const LinuxExePath = "/usr/bin/pwsh";
const LinuxPreviewExePath = "/usr/bin/pwsh-preview";

const SnapExePath = "/snap/bin/pwsh";
const SnapPreviewExePath = "/snap/bin/pwsh-preview";

const MacOSExePath = "/usr/local/bin/pwsh";
const MacOSPreviewExePath = "/usr/local/bin/pwsh-preview";

export enum OperatingSystem {
    Unknown,
    Windows,
    MacOS,
    Linux,
}

export interface IPlatformDetails {
    operatingSystem: OperatingSystem;
    isOS64Bit: boolean;
    isProcess64Bit: boolean;
}

export interface IPowerShellExeDetails {
    readonly displayName: string;
    readonly exePath: string;
    readonly supportsProperArguments: boolean;
}

export function getPlatformDetails(): IPlatformDetails {
    let operatingSystem = OperatingSystem.Unknown;

    if (process.platform === "win32") {
        operatingSystem = OperatingSystem.Windows;
    } else if (process.platform === "darwin") {
        operatingSystem = OperatingSystem.MacOS;
    } else if (process.platform === "linux") {
        operatingSystem = OperatingSystem.Linux;
    }

    const isProcess64Bit = (process.arch === "x64" || process.arch === "arm64");

    return {
        operatingSystem,
        isOS64Bit: isProcess64Bit || (process.env.PROCESSOR_ARCHITEW6432 !== undefined),
        isProcess64Bit,
    };
}

/**
 * Class to lazily find installed PowerShell executables on a machine.
 * When given a list of additional PowerShell executables,
 * this will also surface those at the end of the list.
 */
export class PowerShellExeFinder {
    // This is required, since parseInt("7-preview") will return 7.
    private static IntRegex = /^\d+$/;
    private static PwshMsixRegex = /^Microsoft.PowerShell_.*/;
    private static PwshPreviewMsixRegex = /^Microsoft.PowerShellPreview_.*/;
    private winPS: IPossiblePowerShellExe | undefined;
    private alternateBitnessWinPS: IPossiblePowerShellExe | undefined;

    /**
     * Create a new PowerShellFinder object to discover PowerShell installations.
     * @param platformDetails Information about the machine we are running on.
     * @param additionalPowerShellExes Additional PowerShell installations as configured in the settings.
     */
    constructor(
        // The platform details descriptor for the platform we're on
        private platformDetails: IPlatformDetails,
        // Additional configured PowerShells
        private additionalPowerShellExes: PowerShellAdditionalExePathSettings,
        private logger: ILogger) { }

    /**
     * Returns the first available PowerShell executable found in the search order.
     */
    public async getFirstAvailablePowerShellInstallation(): Promise<IPowerShellExeDetails | undefined> {
        for await (const pwsh of this.enumeratePowerShellInstallations()) {
            return pwsh;
        }
        return undefined;
    }

    /**
     * Get an array of all PowerShell executables found when searching for PowerShell installations.
     */
    public async getAllAvailablePowerShellInstallations(): Promise<IPowerShellExeDetails[]> {
        const array: IPowerShellExeDetails[] = [];
        for await (const pwsh of this.enumeratePowerShellInstallations()) {
            array.push(pwsh);
        }
        return array;
    }

    /**
     * Fixes PowerShell paths when Windows PowerShell is set to the non-native bitness.
     * @param configuredPowerShellPath the PowerShell path configured by the user.
     */
    public fixWindowsPowerShellPath(configuredPowerShellPath: string): string | undefined {
        const altWinPS = this.findWinPS({ useAlternateBitness: true });

        if (!altWinPS) {
            return configuredPowerShellPath;
        }

        const lowerAltWinPSPath = altWinPS.exePath.toLocaleLowerCase();
        const lowerConfiguredPath = configuredPowerShellPath.toLocaleLowerCase();

        if (lowerConfiguredPath === lowerAltWinPSPath) {
            return this.findWinPS()?.exePath;
        }

        return configuredPowerShellPath;
    }

    /**
     * Iterates through PowerShell installations on the machine according
     * to configuration passed in through the constructor.
     * PowerShell items returned by this object are verified
     * to exist on the filesystem.
     */
    public async *enumeratePowerShellInstallations(): AsyncIterable<IPowerShellExeDetails> {
        // Get the default PowerShell installations first
        for await (const defaultPwsh of this.enumerateDefaultPowerShellInstallations()) {
            if (defaultPwsh && await defaultPwsh.exists()) {
                yield defaultPwsh;
            }
        }

        // Also show any additionally configured PowerShells
        // These may be duplicates of the default installations, but given a different name.
        for (const additionalPwsh of this.enumerateAdditionalPowerShellInstallations()) {
            if (await additionalPwsh.exists()) {
                yield additionalPwsh;
            } else if (!additionalPwsh.suppressWarning) {
                const message = `Additional PowerShell '${additionalPwsh.displayName}' not found at '${additionalPwsh.exePath}'!`;
                this.logger.writeWarning(message);

                if (!getSettings().suppressAdditionalExeNotFoundWarning) {
                    const selection = await vscode.window.showWarningMessage(message, "Don't Show Again");
                    if (selection !== undefined) {
                        await changeSetting("suppressAdditionalExeNotFoundWarning", true, true, this.logger);
                    }
                }
            }
        }
    }

    /**
     * Iterates through all the possible well-known PowerShell installations on a machine.
     * Returned values may not exist, but come with an .exists property
     * which will check whether the executable exists.
     */
    private async *enumerateDefaultPowerShellInstallations(): AsyncIterable<IPossiblePowerShellExe | undefined> {
        // Find PSCore stable first
        yield this.findPSCoreStable();

        switch (this.platformDetails.operatingSystem) {
        case OperatingSystem.Linux:
            // On Linux, find the snap
            yield this.findPSCoreStableSnap();
            break;

        case OperatingSystem.Windows:
            // Windows may have a 32-bit pwsh.exe
            yield this.findPSCoreWindowsInstallation({ useAlternateBitness: true });

            // Also look for the MSIX/UWP installation
            yield await this.findPSCoreMsix();

            break;
        }

        // Look for the .NET global tool
        // Some older versions of PowerShell have a bug in this where startup will fail,
        // but this is fixed in newer versions
        yield this.findPSCoreDotnetGlobalTool();

        // Look for PSCore preview
        yield this.findPSCorePreview();

        switch (this.platformDetails.operatingSystem) {
        // On Linux, there might be a preview snap
        case OperatingSystem.Linux:
            yield this.findPSCorePreviewSnap();
            break;

        case OperatingSystem.Windows:
            // Find a preview MSIX
            yield this.findPSCoreMsix({ findPreview: true });

            // Look for pwsh-preview with the opposite bitness
            yield this.findPSCoreWindowsInstallation({ useAlternateBitness: true, findPreview: true });

            // Finally, get Windows PowerShell

            // Get the natural Windows PowerShell for the process bitness
            yield this.findWinPS();

            // Get the alternate bitness Windows PowerShell
            yield this.findWinPS({ useAlternateBitness: true });

            break;
        }

        // Look for PSCore daily
        yield this.findPSCoreDaily();
    }

    /**
     * Iterates through the configured additional PowerShell executable locations,
     * without checking for their existence.
     */
    private *enumerateAdditionalPowerShellInstallations(): Iterable<IPossiblePowerShellExe> {
        for (const versionName in this.additionalPowerShellExes) {
            if (Object.prototype.hasOwnProperty.call(this.additionalPowerShellExes, versionName)) {
                const exePath = utils.stripQuotePair(this.additionalPowerShellExes[versionName]);
                if (!exePath) {
                    continue;
                }

                // Always search for what the user gave us first
                yield new PossiblePowerShellExe(exePath, versionName);

                // Also search for `pwsh[.exe]` and `powershell[.exe]` if missing
                const args: [string, undefined, boolean, boolean]
                    // Must be a tuple type and is suppressing the warning
                    = [versionName, undefined, true, true];

                // Handle Windows where '.exe' and 'powershell' are things
                if (this.platformDetails.operatingSystem === OperatingSystem.Windows) {
                    if (!exePath.endsWith("pwsh.exe") && !exePath.endsWith("powershell.exe")) {
                        if (exePath.endsWith("pwsh") || exePath.endsWith("powershell")) {
                            // Add extension if that was missing
                            yield new PossiblePowerShellExe(exePath + ".exe", ...args);
                        }
                        // Also add full exe names (this isn't an else just in case
                        // the folder was named "pwsh" or "powershell")
                        yield new PossiblePowerShellExe(path.join(exePath, "pwsh.exe"), ...args);
                        yield new PossiblePowerShellExe(path.join(exePath, "powershell.exe"), ...args);
                    }
                } else if (!exePath.endsWith("pwsh")) { // Always just 'pwsh' on non-Windows
                    yield new PossiblePowerShellExe(path.join(exePath, "pwsh"), ...args);
                }
            }
        }
    }

    private async findPSCoreStable(): Promise<IPossiblePowerShellExe | undefined> {
        switch (this.platformDetails.operatingSystem) {
        case OperatingSystem.Linux:
            return new PossiblePowerShellExe(LinuxExePath, "PowerShell");

        case OperatingSystem.MacOS:
            return new PossiblePowerShellExe(MacOSExePath, "PowerShell");

        case OperatingSystem.Windows:
            return await this.findPSCoreWindowsInstallation();

        case OperatingSystem.Unknown:
            return undefined;
        }
    }

    private async findPSCorePreview(): Promise<IPossiblePowerShellExe | undefined> {
        switch (this.platformDetails.operatingSystem) {
        case OperatingSystem.Linux:
            return new PossiblePowerShellExe(LinuxPreviewExePath, "PowerShell Preview");

        case OperatingSystem.MacOS:
            return new PossiblePowerShellExe(MacOSPreviewExePath, "PowerShell Preview");

        case OperatingSystem.Windows:
            return await this.findPSCoreWindowsInstallation({ findPreview: true });

        case OperatingSystem.Unknown:
            return undefined;
        }
    }

    /**
    * If the daily was installed via 'https://aka.ms/install-powershell.ps1', then
    * this is the default installation location:
    *
    * if ($IsWinEnv) {
    *     $Destination = "$env:LOCALAPPDATA\Microsoft\powershell"
    * } else {
    *     $Destination = "~/.powershell"
    * }
    *
    * if ($Daily) {
    *     $Destination = "${Destination}-daily"
    * }
    */
    private findPSCoreDaily(): IPossiblePowerShellExe | undefined {
        switch (this.platformDetails.operatingSystem) {
        case OperatingSystem.Linux:
        case OperatingSystem.MacOS: {
            const exePath = path.join(os.homedir(), ".powershell-daily", "pwsh");
            return new PossiblePowerShellExe(exePath, "PowerShell Daily");
        }

        case OperatingSystem.Windows: {
            // We can't proceed if there's no LOCALAPPDATA path
            if (!process.env.LOCALAPPDATA) {
                return undefined;
            }
            const exePath = path.join(process.env.LOCALAPPDATA, "Microsoft", "powershell-daily", "pwsh.exe");
            return new PossiblePowerShellExe(exePath, "PowerShell Daily");
        }

        case OperatingSystem.Unknown:
            return undefined;
        }
    }

    private findPSCoreDotnetGlobalTool(): IPossiblePowerShellExe {
        const exeName: string = this.platformDetails.operatingSystem === OperatingSystem.Windows
            ? "pwsh.exe"
            : "pwsh";

        const dotnetGlobalToolExePath: string = path.join(os.homedir(), ".dotnet", "tools", exeName);

        // The dotnet installed version of PowerShell does not support proper argument parsing, and so it fails with our multi-line startup banner.
        return new PossiblePowerShellExe(dotnetGlobalToolExePath, ".NET Core PowerShell Global Tool", undefined, false);
    }

    private async findPSCoreMsix({ findPreview }: { findPreview?: boolean } = {}): Promise<IPossiblePowerShellExe | undefined> {
        // We can't proceed if there's no LOCALAPPDATA path
        if (!process.env.LOCALAPPDATA) {
            return undefined;
        }

        // Find the base directory for MSIX application exe shortcuts
        const msixAppDir = path.join(process.env.LOCALAPPDATA, "Microsoft", "WindowsApps");

        if (!await utils.checkIfDirectoryExists(msixAppDir)) {
            return undefined;
        }

        // Define whether we're looking for the preview or the stable
        const { pwshMsixDirRegex, pwshMsixName } = findPreview
            ? { pwshMsixDirRegex: PowerShellExeFinder.PwshPreviewMsixRegex, pwshMsixName: "PowerShell Preview (Store)" }
            : { pwshMsixDirRegex: PowerShellExeFinder.PwshMsixRegex, pwshMsixName: "PowerShell (Store)" };

        // We should find only one such application, so return on the first one
        for (const name of await utils.readDirectory(msixAppDir)) {
            if (pwshMsixDirRegex.test(name)) {
                return new PossiblePowerShellExe(path.join(msixAppDir, name, "pwsh.exe"), pwshMsixName);
            }
        }

        return undefined;
    }

    private findPSCoreStableSnap(): IPossiblePowerShellExe {
        return new PossiblePowerShellExe(SnapExePath, "PowerShell Snap");
    }

    private findPSCorePreviewSnap(): IPossiblePowerShellExe {
        return new PossiblePowerShellExe(SnapPreviewExePath, "PowerShell Preview Snap");
    }

    private async findPSCoreWindowsInstallation(
        { useAlternateBitness = false, findPreview = false }:
            { useAlternateBitness?: boolean; findPreview?: boolean } = {}): Promise<IPossiblePowerShellExe | undefined> {

        const programFilesPath = this.getProgramFilesPath({ useAlternateBitness });

        if (!programFilesPath) {
            return undefined;
        }

        const powerShellInstallBaseDir = path.join(programFilesPath, "PowerShell");

        // Ensure the base directory exists
        if (!await utils.checkIfDirectoryExists(powerShellInstallBaseDir)) {
            return undefined;
        }

        let highestSeenVersion = -1;
        let pwshExePath: string | undefined;
        for (const item of await utils.readDirectory(powerShellInstallBaseDir)) {
            let currentVersion = -1;
            if (findPreview) {
                // We are looking for something like "7-preview"

                // Preview dirs all have dashes in them
                const dashIndex: integer = item.indexOf("-");
                if (dashIndex < 0) {
                    continue;
                }

                // Verify that the part before the dash is an integer
                const intPart: string = item.substring(0, dashIndex);
                if (!PowerShellExeFinder.IntRegex.test(intPart)) {
                    continue;
                }

                // Verify that the part after the dash is "preview"
                if (item.substring(dashIndex + 1) !== "preview") {
                    continue;
                }

                currentVersion = parseInt(intPart, 10);
            } else {
                // Search for a directory like "6" or "7"
                if (!PowerShellExeFinder.IntRegex.test(item)) {
                    continue;
                }

                currentVersion = parseInt(item, 10);
            }

            // Ensure we haven't already seen a higher version
            if (currentVersion <= highestSeenVersion) {
                continue;
            }

            // Now look for the file
            const exePath = path.join(powerShellInstallBaseDir, item, "pwsh.exe");
            if (!await utils.checkIfFileExists(exePath)) {
                continue;
            }

            pwshExePath = exePath;
            highestSeenVersion = currentVersion;
        }

        if (!pwshExePath) {
            return undefined;
        }

        const bitness: string = programFilesPath.includes("x86")
            ? "(x86)"
            : "(x64)";

        const preview: string = findPreview ? " Preview" : "";

        return new PossiblePowerShellExe(pwshExePath, `PowerShell${preview} ${bitness}`);
    }

    private findWinPS({ useAlternateBitness = false }: { useAlternateBitness?: boolean } = {}): IPossiblePowerShellExe | undefined {

        // 32-bit OSes only have one WinPS on them
        if (!this.platformDetails.isOS64Bit && useAlternateBitness) {
            return undefined;
        }

        let winPS = useAlternateBitness ? this.alternateBitnessWinPS : this.winPS;
        if (winPS === undefined) {
            const systemFolderPath = this.getSystem32Path({ useAlternateBitness });

            if (!systemFolderPath) {
                return undefined;
            }

            const winPSPath = path.join(systemFolderPath, "WindowsPowerShell", "v1.0", "powershell.exe");

            let displayName: string;
            if (this.platformDetails.isProcess64Bit) {
                displayName = useAlternateBitness
                    ? WindowsPowerShell32BitLabel
                    : WindowsPowerShell64BitLabel;
            } else if (this.platformDetails.isOS64Bit) {
                displayName = useAlternateBitness
                    ? WindowsPowerShell64BitLabel
                    : WindowsPowerShell32BitLabel;
            } else {
                displayName = WindowsPowerShell32BitLabel;
            }

            winPS = new PossiblePowerShellExe(winPSPath, displayName, true);

            if (useAlternateBitness) {
                this.alternateBitnessWinPS = winPS;
            } else {
                this.winPS = winPS;
            }
        }

        return winPS;
    }

    private getProgramFilesPath(
        { useAlternateBitness = false }: { useAlternateBitness?: boolean } = {}): string | undefined {

        if (!useAlternateBitness) {
            // Just use the native system bitness
            return process.env.ProgramFiles;
        }

        // We might be a 64-bit process looking for 32-bit program files
        if (this.platformDetails.isProcess64Bit) {
            return process.env["ProgramFiles(x86)"];
        }

        // We might be a 32-bit process looking for 64-bit program files
        if (this.platformDetails.isOS64Bit) {
            return process.env.ProgramW6432;
        }

        // We're a 32-bit process on 32-bit Windows, there is no other Program Files dir
        return undefined;
    }

    private getSystem32Path({ useAlternateBitness = false }: { useAlternateBitness?: boolean } = {}): string | undefined {
        const windir = process.env.windir;

        if (!windir) {
            return undefined;
        }

        if (!useAlternateBitness) {
            // Just use the native system bitness
            return path.join(windir, "System32");
        }

        // We might be a 64-bit process looking for 32-bit system32
        if (this.platformDetails.isProcess64Bit) {
            return path.join(windir, "SysWOW64");
        }

        // We might be a 32-bit process looking for 64-bit system32
        if (this.platformDetails.isOS64Bit) {
            return path.join(windir, "Sysnative");
        }

        // We're on a 32-bit Windows, so no alternate bitness
        return undefined;
    }
}

export function getWindowsSystemPowerShellPath(systemFolderName: string): string | undefined {
    if (process.env.windir === undefined) {
        return undefined;
    } else return path.join(
        process.env.windir,
        systemFolderName,
        "WindowsPowerShell",
        "v1.0",
        "powershell.exe");
}

interface IPossiblePowerShellExe extends IPowerShellExeDetails {
    exists(): Promise<boolean>;
    readonly suppressWarning: boolean;
}

class PossiblePowerShellExe implements IPossiblePowerShellExe {
    constructor(
        public readonly exePath: string,
        public readonly displayName: string,
        private knownToExist?: boolean,
        public readonly supportsProperArguments = true,
        public readonly suppressWarning = false) { }

    public async exists(): Promise<boolean> {
        if (this.knownToExist === undefined) {
            this.knownToExist = await utils.checkIfFileExists(this.exePath);
        }
        return this.knownToExist ?? false;
    }
}