install.ps1

# Vite+ CLI Installer for Windows
# https://vite.plus/ps1
#
# Usage:
#   irm https://vite.plus/ps1 | iex
#
# Environment variables:
#   VP_VERSION - Version to install (default: latest)
#   VP_HOME - Installation directory (default: $env:USERPROFILE\.vite-plus)
#   NPM_CONFIG_REGISTRY - Custom npm registry URL (default: https://registry.npmjs.org)
#   VP_LOCAL_TGZ - Path to local vite-plus.tgz (for development/testing)
#   VP_PR_VERSION - PR number or commit SHA to install from pkg.pr.new
#                   (for temporary testing of unreleased builds, e.g. VP_PR_VERSION=1569).
#                   When set, overrides VP_VERSION and bypasses the npm registry.

$ErrorActionPreference = "Stop"

$ViteVersion = if ($env:VP_VERSION) { $env:VP_VERSION } else { "latest" }
$InstallDir = if ($env:VP_HOME) { $env:VP_HOME } else { "$env:USERPROFILE\.vite-plus" }
# npm registry URL (strip trailing slash if present)
$NpmRegistry = if ($env:NPM_CONFIG_REGISTRY) { $env:NPM_CONFIG_REGISTRY.TrimEnd('/') } else { "https://registry.npmjs.org" }
# Local tarball for development/testing
$LocalTgz = $env:VP_LOCAL_TGZ
# Local binary path (set by install-global-cli.ts for local dev)
$LocalBinary = $env:VP_LOCAL_BINARY
# pkg.pr.new PR number or commit SHA (for temporary testing of unreleased builds)
$PrVersion = $env:VP_PR_VERSION
# pkg.pr.new base URL for fetching tarballs and constructing dependency URLs
$PkgPrNewBase = "https://pkg.pr.new/voidzero-dev/vite-plus"

function Write-Info {
    param([string]$Message)
    Write-Host "info: " -ForegroundColor Blue -NoNewline
    Write-Host $Message
}

function Write-Success {
    param([string]$Message)
    Write-Host "success: " -ForegroundColor Green -NoNewline
    Write-Host $Message
}

function Write-Warn {
    param([string]$Message)
    Write-Host "warn: " -ForegroundColor Yellow -NoNewline
    Write-Host $Message
}

function Write-Error-Exit {
    param([string]$Message)
    Write-Host "error: " -ForegroundColor Red -NoNewline
    Write-Host $Message
    exit 1
}

function Test-ReleaseAgeError {
    param([string]$LogPath)
    if (-not (Test-Path $LogPath)) {
        return $false
    }

    $content = Get-Content -Path $LogPath -Raw
    # This wrapper install path is pinned to pnpm via packageManager, so this
    # detection follows pnpm's resolver/reporter output rather than npm/yarn.
    #
    # pnpm's PnpmError prefixes internal codes with ERR_PNPM_, so
    # NO_MATURE_MATCHING_VERSION is normally printed as
    # ERR_PNPM_NO_MATURE_MATCHING_VERSION. npm-resolver emits that code with the
    # "does not meet the minimumReleaseAge constraint" message when
    # publishedBy/minimumReleaseAge rejects a matching version.
    # https://github.com/pnpm/pnpm/blob/16cfde66ec71125d692ea828eba2a5f9b3cc54fc/core/error/src/index.ts#L18-L20
    # https://github.com/pnpm/pnpm/blob/16cfde66ec71125d692ea828eba2a5f9b3cc54fc/resolving/npm-resolver/src/index.ts#L76-L84
    #
    # default-reporter may append guidance mentioning minimumReleaseAgeExclude
    # when the error has an immatureVersion, so that token is also a useful
    # release-age signal. minimum-release-age is pnpm's .npmrc key; npm's
    # min-release-age is intentionally not treated as a pnpm signal here.
    # https://github.com/pnpm/pnpm/blob/16cfde66ec71125d692ea828eba2a5f9b3cc54fc/cli/default-reporter/src/reportError.ts#L163-L164
    # https://github.com/pnpm/pnpm/blob/16cfde66ec71125d692ea828eba2a5f9b3cc54fc/config/reader/src/types.ts#L73-L74
    $hasReleaseAgeText = $content -match "does not meet the minimumReleaseAge constraint" `
        -or $content -match "minimumReleaseAge" `
        -or $content -match "minimumReleaseAgeExclude" `
        -or $content -match "minimum release age" `
        -or $content -match "minimum-release-age"

    # pnpm can also surface ERR_PNPM_NO_MATCHING_VERSION when minimumReleaseAge
    # filters out all candidates. That code is also used for real missing
    # versions, so require age-gate context before prompting for a bypass.
    # https://github.com/pnpm/pnpm/blob/16cfde66ec71125d692ea828eba2a5f9b3cc54fc/deps/inspection/outdated/src/createManifestGetter.ts#L66-L76
    return $content -match "ERR_PNPM_NO_MATURE_MATCHING_VERSION" `
        -or $content -match "NO_MATURE_MATCHING_VERSION" `
        -or (($content -match "ERR_PNPM_NO_MATCHING_VERSION") -and $hasReleaseAgeText) `
        -or $hasReleaseAgeText
}

function Confirm-ReleaseAgeOverride {
    if ($env:CI -eq "true") {
        return $false
    }
    if (-not [Environment]::UserInteractive) {
        return $false
    }

    Write-Host ""
    Write-Warn "Your minimumReleaseAge setting prevented installing vite-plus@$ViteVersion."
    Write-Host "This setting helps protect against newly published compromised packages."
    Write-Host "Proceeding will disable this protection for this Vite+ install only."
    $response = Read-Host "Do you want to proceed? (y/N)"
    return $response -match "^(?i:y|yes)$"
}

function Write-ReleaseAgeOverride {
    Set-Content -Path (Join-Path $VersionDir ".npmrc") -Value "minimum-release-age=0"
}

function Write-InstallFailure {
    param([string]$LogPath)
    if ($env:CI -eq "true") {
        Write-Host "error: " -ForegroundColor Red -NoNewline
        Write-Host "Failed to install dependencies. Log output:"
        Get-Content -Path $LogPath | ForEach-Object { Write-Host $_ }
    } else {
        Write-Error-Exit "Failed to install dependencies. See log for details: $LogPath"
    }
}

function Write-ReleaseAgeFailure {
    param([string]$LogPath)
    if ($env:CI -eq "true") {
        Write-Host "error: " -ForegroundColor Red -NoNewline
        Write-Host "Install blocked by your minimumReleaseAge setting. Log output:"
        Get-Content -Path $LogPath | ForEach-Object { Write-Host $_ }
    } else {
        Write-Error-Exit "Install blocked by your minimumReleaseAge setting. Wait until the package is old enough or adjust your package manager configuration explicitly. See log for details: $LogPath"
    }
}

function Get-Architecture {
    if ([Environment]::Is64BitOperatingSystem) {
        if ($env:PROCESSOR_ARCHITECTURE -eq "ARM64") {
            return "arm64"
        } else {
            return "x64"
        }
    } else {
        Write-Error-Exit "32-bit Windows is not supported"
    }
}

# Cached package metadata
$script:PackageMetadata = $null

function Get-PackageMetadata {
    if ($null -eq $script:PackageMetadata) {
        $versionPath = if ($ViteVersion -eq "latest") { "latest" } else { $ViteVersion }
        $metadataUrl = "$NpmRegistry/vite-plus/$versionPath"
        try {
            $script:PackageMetadata = Invoke-RestMethod $metadataUrl
        } catch {
            # Try to extract npm error message from response
            $errorMsg = $_.ErrorDetails.Message
            if ($errorMsg) {
                try {
                    $errorJson = $errorMsg | ConvertFrom-Json
                    if ($errorJson.error) {
                        Write-Error-Exit "Failed to fetch version '${versionPath}': $($errorJson.error)`n  URL: $metadataUrl"
                    }
                } catch {
                    # JSON parsing failed, fall through to generic error
                }
            }
            Write-Error-Exit "Failed to fetch package metadata from: $metadataUrl`nError: $_"
        }
        # Check for error in successful response
        # npm can return {"error":"..."} object or a plain string like "version not found: test"
        if ($script:PackageMetadata -is [string]) {
            # Some registries (e.g. JFrog) may return JSON with a non-JSON content type,
            # causing Invoke-RestMethod to return a raw string. Try parsing it as JSON first.
            try {
                $script:PackageMetadata = $script:PackageMetadata | ConvertFrom-Json
            } catch {
                # Not valid JSON - treat as plain string error
                Write-Error-Exit "Failed to fetch version '${versionPath}': $script:PackageMetadata`n  URL: $metadataUrl"
            }
        }
        if ($script:PackageMetadata.error) {
            Write-Error-Exit "Failed to fetch version '${versionPath}': $($script:PackageMetadata.error)`n  URL: $metadataUrl"
        }
    }
    return $script:PackageMetadata
}

function Get-VersionFromMetadata {
    $metadata = Get-PackageMetadata
    if (-not $metadata.version) {
        Write-Error-Exit "Failed to extract version from package metadata"
    }
    return $metadata.version
}

function Get-PlatformSuffix {
    param([string]$Platform)
    # Windows needs -msvc suffix, other platforms map directly
    if ($Platform.StartsWith("win32-")) { return "${Platform}-msvc" }
    return $Platform
}

function Download-AndExtract {
    param(
        [string]$Url,
        [string]$DestDir,
        [string]$Filter
    )

    $tempFile = New-TemporaryFile
    try {
        # Suppress progress bar for cleaner output
        $ProgressPreference = 'SilentlyContinue'
        Invoke-WebRequest -Uri $Url -OutFile $tempFile

        # Create temp extraction directory
        $tempExtract = Join-Path $env:TEMP "vite-install-$(Get-Random)"
        New-Item -ItemType Directory -Force -Path $tempExtract | Out-Null

        # Extract using tar (available in Windows 10+)
        & "$env:SystemRoot\System32\tar.exe" -xzf $tempFile -C $tempExtract

        # Copy the specified file/directory
        $sourcePath = Join-Path (Join-Path $tempExtract "package") $Filter
        if (Test-Path $sourcePath) {
            Copy-Item -Path $sourcePath -Destination $DestDir -Recurse -Force
        }

        Remove-Item -Recurse -Force $tempExtract
    } finally {
        Remove-Item $tempFile -ErrorAction SilentlyContinue
    }
}

function Cleanup-OldVersions {
    param([string]$InstallDir)

    $maxVersions = 5
    # Only cleanup semver format directories (0.1.0, 1.2.3-beta.1, etc.)
    # This excludes 'current' symlink and non-semver directories like 'local-dev'
    $semverPattern = '^\d+\.\d+\.\d+(-[a-zA-Z0-9.-]+)?$'
    $versions = Get-ChildItem -Path $InstallDir -Directory -ErrorAction SilentlyContinue |
        Where-Object { $_.Name -match $semverPattern }

    if ($null -eq $versions -or $versions.Count -le $maxVersions) {
        return
    }

    # Sort by creation time (oldest first) and select excess
    $toDelete = $versions |
        Sort-Object CreationTime |
        Select-Object -First ($versions.Count - $maxVersions)

    foreach ($old in $toDelete) {
        # Remove silently
        Remove-Item -Path $old.FullName -Recurse -Force
    }
}

function Remove-CurrentLink {
    param([string]$Path)

    try {
        $item = Get-Item -LiteralPath $Path -Force -ErrorAction Stop
    } catch [System.Management.Automation.ItemNotFoundException] {
        return
    }

    $isReparsePoint = ($item.Attributes -band [System.IO.FileAttributes]::ReparsePoint) -ne 0

    try {
        if ($isReparsePoint) {
            if ($item.PSIsContainer) {
                [System.IO.Directory]::Delete($item.FullName)
            } else {
                [System.IO.File]::Delete($item.FullName)
            }
            return
        }

        Remove-Item -LiteralPath $item.FullName -Recurse -Force -ErrorAction Stop
    } catch {
        Write-Error-Exit "Failed to remove existing current link at ${Path}: $_"
    }
}

# Configure user PATH for ~/.vite-plus/bin
# Returns: "true" = added, "already" = already configured
function Configure-UserPath {
    $binPath = "$InstallDir\bin"
    $userPath = [Environment]::GetEnvironmentVariable("Path", "User")

    if ($userPath -like "*$binPath*") {
        return "already"
    }

    $newPath = "$binPath;$userPath"
    try {
        [Environment]::SetEnvironmentVariable("Path", $newPath, "User")
        $env:Path = "$binPath;$env:Path"
        return "true"
    } catch {
        Write-Warn "Could not update user PATH automatically."
        return "failed"
    }
}

function Get-NushellVendorAutoloadDir {
    $nushellCommand = Get-Command nu -ErrorAction SilentlyContinue
    if ($null -eq $nushellCommand) {
        return $null
    }

    try {
        $dirsOutput = & $nushellCommand.Source -c '$nu.vendor-autoload-dirs | reverse | each {|dir| $dir } | str join (char nl)' 2>$null
    } catch {
        return $null
    }

    foreach ($dir in ($dirsOutput -split "\r?\n")) {
        if (-not [string]::IsNullOrWhiteSpace($dir)) {
            return $dir
        }
    }

    return $null
}

function Configure-Nushell {
    $autoloadDir = Get-NushellVendorAutoloadDir
    if ($null -eq $autoloadDir) {
        if ($null -eq (Get-Command nu -ErrorAction SilentlyContinue)) {
            return [pscustomobject]@{
                Status = "skipped"
                Message = "skipped (not installed)"
            }
        }

        return [pscustomobject]@{
            Status = "failed"
            Message = "failed (could not determine vendor autoload dir)"
        }
    }

    $autoloadFile = Join-Path $autoloadDir "vite-plus.nu"
    $nuEnvRef= (Join-Path $InstallDir "env.nu") -replace [regex]::Escape($env:USERPROFILE), '~'
    $content = "# Vite+ bin (https://viteplus.dev)`n" + ("source '"+ $nuEnvRef +"'") + "`n"

    try {
        New-Item -ItemType Directory -Force -Path $autoloadDir | Out-Null
        if (Test-Path $autoloadFile) {
            $existing = Get-Content -Path $autoloadFile -Raw
            if ($existing -eq $content) {
                return [pscustomobject]@{
                    Status = "already"
                    Message = "already configured $autoloadFile"
                }
            }
        }

        [System.IO.File]::WriteAllText($autoloadFile, $content)
        return [pscustomobject]@{
            Status = "true"
            Message = "updated $autoloadFile"
        }
    } catch {
        Write-Warn "Could not configure Nushell automatically."
        return [pscustomobject]@{
            Status = "failed"
            Message = "failed $autoloadFile"
        }
    }
}

# Run vp env setup --refresh, showing output only on failure
function Refresh-Shims {
    param([string]$BinDir)
    $setupOutput = & "$BinDir\vp.exe" env setup --refresh 2>&1
    if ($LASTEXITCODE -ne 0) {
        Write-Warn "Failed to refresh shims:"
        Write-Host "$setupOutput"
    }
}

# Setup Node.js version manager (node/npm/npx shims)
# Returns: "true" = enabled, "false" = not enabled, "already" = already configured
function Setup-NodeManager {
    param([string]$BinDir)

    $binPath = "$InstallDir\bin"

    # Explicit override via environment variable
    if ($env:VP_NODE_MANAGER -eq "yes") {
        Refresh-Shims -BinDir $BinDir
        return "true"
    } elseif ($env:VP_NODE_MANAGER -eq "no") {
        return "false"
    }

    # Check if Vite+ is already managing Node.js (bin\node.exe exists)
    if (Test-Path "$binPath\node.exe") {
        # Already managing Node.js, just refresh shims
        Refresh-Shims -BinDir $BinDir
        return "already"
    }

    # Auto-enable on CI or devcontainer environments
    # CI: standard CI environment variable (GitHub Actions, Travis, CircleCI, etc.)
    # CODESPACES: set by GitHub Codespaces (https://docs.github.com/en/codespaces)
    # REMOTE_CONTAINERS: set by VS Code Dev Containers extension
    # DEVPOD: set by DevPod (https://devpod.sh)
    if ($env:CI -or $env:CODESPACES -or $env:REMOTE_CONTAINERS -or $env:DEVPOD) {
        Refresh-Shims -BinDir $BinDir
        return "true"
    }

    # Check if node is available on the system
    $nodeAvailable = $null -ne (Get-Command node -ErrorAction SilentlyContinue)

    # Auto-enable if no node available on system
    if (-not $nodeAvailable) {
        Refresh-Shims -BinDir $BinDir
        return "true"
    }

    # Prompt user in interactive mode
    $isInteractive = [Environment]::UserInteractive
    if ($isInteractive) {
        Write-Host ""
        Write-Host "Would you like Vite+ to manage your Node.js versions?"
        Write-Host "It adds ``node``, ``npm``, and ``npx`` shims to ~/.vite-plus/bin/ and automatically uses the right version."
        Write-Host "Opt out anytime with ``vp env off``."
        $response = Read-Host "Press Enter to accept (Y/n)"

        if ($response -eq '' -or $response -eq 'y' -or $response -eq 'Y') {
            Refresh-Shims -BinDir $BinDir
            return "true"
        }
    }

    return "false"
}

function Main {
    Write-Host ""
    Write-Host "Setting up " -NoNewline
    Write-Host "VITE+" -ForegroundColor Blue -NoNewline
    Write-Host "..."

    if ($PrVersion -and $LocalTgz) {
        Write-Error-Exit "VP_PR_VERSION and VP_LOCAL_TGZ cannot be used together"
    }

    # Suppress progress bars for cleaner output
    $ProgressPreference = 'SilentlyContinue'

    $arch = Get-Architecture
    $platform = "win32-$arch"

    # Local development mode: use local tgz
    if ($LocalTgz) {
        # Validate local tgz
        if (-not (Test-Path $LocalTgz)) {
            Write-Error-Exit "Local tarball not found: $LocalTgz"
        }
        # Use version as-is (default to "local-dev")
        if ($ViteVersion -eq "latest" -or $ViteVersion -eq "test") {
            $ViteVersion = "local-dev"
        }
    } elseif ($PrVersion) {
        # pkg.pr.new mode: skip npm metadata, use a synthetic version label.
        # Non-semver label keeps the directory out of Cleanup-OldVersions and
        # makes it obvious in ~/.vite-plus which install is the PR build.
        $ViteVersion = "pkg-pr-new-$PrVersion"
        Write-Info "Using pkg.pr.new version: $PrVersion"
    } else {
        # Fetch package metadata and resolve version from npm
        $ViteVersion = Get-VersionFromMetadata
    }

    # Set up version-specific directories
    $VersionDir = "$InstallDir\$ViteVersion"
    $BinDir = "$VersionDir\bin"
    $CurrentLink = "$InstallDir\current"

    $binaryName = "vp.exe"

    # Create bin directory
    New-Item -ItemType Directory -Force -Path $BinDir | Out-Null

    if ($LocalTgz) {
        # Local development mode: only need the binary
        Write-Info "Using local tarball: $LocalTgz"

        # Copy binary from LOCAL_BINARY env var (set by install-global-cli.ts)
        if ($LocalBinary -and (Test-Path $LocalBinary)) {
            Copy-Item -Path $LocalBinary -Destination (Join-Path $BinDir $binaryName) -Force
            # Also copy trampoline shim binary if available (sibling to vp.exe)
            $shimSource = Join-Path (Split-Path $LocalBinary) "vp-shim.exe"
            if (Test-Path $shimSource) {
                Copy-Item -Path $shimSource -Destination (Join-Path $BinDir "vp-shim.exe") -Force
            }
        } else {
            Write-Error-Exit "VP_LOCAL_BINARY must be set when using VP_LOCAL_TGZ"
        }
    } else {
        # Download CLI platform tarball — npm registry or pkg.pr.new (when PrVersion is set)
        $platformSuffix = Get-PlatformSuffix -Platform $platform
        if ($PrVersion) {
            # pkg.pr.new redirects this URL to the platform tarball for the matching PR/commit
            $platformUrl = "$PkgPrNewBase/@voidzero-dev/vite-plus-cli-$platformSuffix@$PrVersion"
        } else {
            $packageName = "@voidzero-dev/vite-plus-cli-$platformSuffix"
            $platformUrl = "$NpmRegistry/$packageName/-/vite-plus-cli-$platformSuffix-$ViteVersion.tgz"
        }

        $platformTempFile = New-TemporaryFile
        try {
            Invoke-WebRequest -Uri $platformUrl -OutFile $platformTempFile

            # Create temp extraction directory
            $platformTempExtract = Join-Path $env:TEMP "vite-platform-$(Get-Random)"
            New-Item -ItemType Directory -Force -Path $platformTempExtract | Out-Null

            # Extract the package
            & "$env:SystemRoot\System32\tar.exe" -xzf $platformTempFile -C $platformTempExtract

            # Copy binary to BinDir
            $packageDir = Join-Path $platformTempExtract "package"
            $binarySource = Join-Path $packageDir $binaryName
            if (Test-Path $binarySource) {
                Copy-Item -Path $binarySource -Destination $BinDir -Force
            }
            # Also copy trampoline shim binary if present in the package
            $shimSource = Join-Path $packageDir "vp-shim.exe"
            if (Test-Path $shimSource) {
                Copy-Item -Path $shimSource -Destination $BinDir -Force
            }

            Remove-Item -Recurse -Force $platformTempExtract
        } finally {
            Remove-Item $platformTempFile -ErrorAction SilentlyContinue
        }
    }

    # Remove Zone.Identifier (Mark of the Web) from downloaded binaries so
    # Windows SmartScreen / Defender won't block execution.
    Get-ChildItem -Path $BinDir -Filter "*.exe" | Unblock-File

    # Generate wrapper package.json that declares vite-plus as a dependency.
    # pnpm will install vite-plus and all transitive deps via `vp install`.
    # The packageManager field pins pnpm to a known-good version.
    # pkg.pr.new tarballs pre-rewrite scoped workspace deps to matching URLs by
    # commit SHA, so pointing vite-plus at one URL pulls in a coherent PR build.
    $vitePlusSpec = if ($PrVersion) { "$PkgPrNewBase@$PrVersion" } else { $ViteVersion }
    $wrapperJson = @{
        name = "vp-global"
        version = $ViteVersion
        private = $true
        packageManager = "pnpm@10.33.0"
        dependencies = @{
            "vite-plus" = $vitePlusSpec
        }
    } | ConvertTo-Json -Depth 10
    Set-Content -Path (Join-Path $VersionDir "package.json") -Value $wrapperJson

    # Install production dependencies (skip if VP_SKIP_DEPS_INSTALL is set,
    # e.g. during local dev where install-global-cli.ts handles deps separately)
    if (-not $env:VP_SKIP_DEPS_INSTALL) {
        $installLog = Join-Path $VersionDir "install.log"
        Push-Location $VersionDir
        try {
            # Use cmd /c so CI=true is scoped to the child process only,
            # avoiding leaking it into the user's shell session.
            # Do not pass --silent to the inner install: pnpm suppresses the
            # release-age error body in silent mode, which would leave
            # install.log empty and make the release-age gate impossible to
            # detect. Output is already captured to install.log here.
            $output = cmd /c "set CI=true && `"$BinDir\vp.exe`" install" 2>&1
            $installExitCode = $LASTEXITCODE
            $output | Out-File $installLog
            if ($installExitCode -ne 0) {
                if (Test-ReleaseAgeError $installLog) {
                    if (Confirm-ReleaseAgeOverride) {
                        # Write the override only after explicit consent, then retry once.
                        Write-ReleaseAgeOverride
                        $retryOutput = cmd /c "set CI=true && `"$BinDir\vp.exe`" install" 2>&1
                        $retryExitCode = $LASTEXITCODE
                        $retryOutput | Out-File $installLog
                        if ($retryExitCode -ne 0) {
                            Write-InstallFailure $installLog
                            exit 1
                        }
                    } else {
                        Write-ReleaseAgeFailure $installLog
                        exit 1
                    }
                } else {
                    Write-InstallFailure $installLog
                    exit 1
                }
            }
        } finally {
            Pop-Location
        }
    }

    # Create/update current junction (symlink)
    Remove-CurrentLink $CurrentLink
    # Create new junction pointing to the version directory
    cmd /c mklink /J "$CurrentLink" "$VersionDir" | Out-Null

    # Create bin directory and vp wrapper (always done)
    New-Item -ItemType Directory -Force -Path "$InstallDir\bin" | Out-Null
    $trampolineSrc = "$VersionDir\bin\vp-shim.exe"
    if (Test-Path $trampolineSrc) {
        # New versions: use trampoline exe to avoid "Terminate batch job (Y/N)?" on Ctrl+C
        Copy-Item -Path $trampolineSrc -Destination "$InstallDir\bin\vp.exe" -Force
        # Remove legacy .cmd and shell script wrappers from previous versions
        foreach ($legacy in @("$InstallDir\bin\vp.cmd", "$InstallDir\bin\vp")) {
            if (Test-Path $legacy) {
                Remove-Item -Path $legacy -Force -ErrorAction SilentlyContinue
            }
        }
    } else {
        # Pre-trampoline versions: fall back to legacy .cmd and shell script wrappers.
        # Remove any stale trampoline .exe shims left by a newer install — .exe wins
        # over .cmd on Windows PATH, so leftover trampolines would bypass the wrappers.
        foreach ($stale in @("vp.exe", "node.exe", "npm.exe", "npx.exe", "vpx.exe")) {
            $stalePath = Join-Path "$InstallDir\bin" $stale
            if (Test-Path $stalePath) {
                Remove-Item -Path $stalePath -Force -ErrorAction SilentlyContinue
            }
        }
        # Keep consistent with the original install.ps1 wrapper format
        $wrapperContent = @"
@echo off
set VP_HOME=%~dp0..
"%VP_HOME%\current\bin\vp.exe" %*
exit /b %ERRORLEVEL%
"@
        Set-Content -Path "$InstallDir\bin\vp.cmd" -Value $wrapperContent -NoNewline

        # Also create shell script wrapper for Git Bash/MSYS
        $shContent = @"
#!/bin/sh
VP_HOME="`$(dirname "`$(dirname "`$(readlink -f "`$0" 2>/dev/null || echo "`$0")")")"
export VP_HOME
exec "`$VP_HOME/current/bin/vp.exe" "`$@"
"@
        Set-Content -Path "$InstallDir\bin\vp" -Value $shContent -NoNewline
    }

    # Cleanup old versions
    Cleanup-OldVersions -InstallDir $InstallDir

    # Configure Windows-native shell access via the User PATH
    $pathResult = Configure-UserPath

    # Configure Nushell autoload if Nushell is installed
    $nushellResult = Configure-Nushell

    # Setup Node.js version manager (shims) - separate component
    $nodeManagerResult = Setup-NodeManager -BinDir $BinDir

    # Use ~ shorthand if install dir is under USERPROFILE, otherwise show full path
    $displayDir = $InstallDir -replace [regex]::Escape($env:USERPROFILE), '~'

    # ANSI color codes for consistent output
    $e = [char]27
    $GREEN = "$e[32m"
    $YELLOW = "$e[33m"
    $BRIGHT_BLUE = "$e[94m"
    $BOLD = "$e[1m"
    $DIM = "$e[2m"
    $BOLD_BRIGHT_BLUE = "$e[1;94m"
    $NC = "$e[0m"
    $CHECKMARK = [char]0x2714

    # Print success message
    Write-Host ""
    Write-Host "${GREEN}${CHECKMARK}${NC} ${BOLD_BRIGHT_BLUE}VITE+${NC} successfully installed!"
    Write-Host ""
    Write-Host "  The Unified Toolchain for the Web."
    Write-Host ""
    Write-Host "  ${BOLD}Get started:${NC}"
    Write-Host "    ${BRIGHT_BLUE}vp create${NC}       Create a new project"
    Write-Host "    ${BRIGHT_BLUE}vp env${NC}          Manage Node.js versions"
    Write-Host "    ${BRIGHT_BLUE}vp install${NC}      Install dependencies"
    Write-Host "    ${BRIGHT_BLUE}vp migrate${NC}      Migrate to Vite+"

    # Show Node.js manager status
    if ($nodeManagerResult -eq "true" -or $nodeManagerResult -eq "already") {
        Write-Host ""
        Write-Host "  Vite+ is now managing Node.js via ${BRIGHT_BLUE}vp env${NC}."
        Write-Host "  Run ${BRIGHT_BLUE}vp env doctor${NC} to verify your setup, or ${BRIGHT_BLUE}vp env off${NC} to opt out."
    }

    Write-Host ""
    Write-Host "  Run ${BRIGHT_BLUE}vp help${NC} to see available commands."

    Write-Host ""
    Write-Host "  Shell configuration:"
    switch ($pathResult) {
        "true" { Write-Host "    - Windows PATH: updated" }
        "already" { Write-Host "    - Windows PATH: already configured" }
        "failed" { Write-Host "    - Windows PATH: failed" }
        default { Write-Host "    - Windows PATH: skipped" }
    }
    if ($nushellResult.Status -ne "skipped") {
      Write-Host "    - Nushell: $($nushellResult.Message)"
    }

    # Show note if PATH or Nushell was updated
    if ($pathResult -eq "true" -or $nushellResult.Status -eq "true") {
        Write-Host ""
        Write-Host "  Note: Restart your terminal and IDE for changes to take effect."
    }

    # Show manual PATH/Nushell instructions if anything still needs manual setup
    if ($pathResult -eq "failed" -or $nushellResult.Status -eq "failed") {
        Write-Host ""
        Write-Host "  ${YELLOW}note${NC}: Some shells still need manual setup."
        Write-Host ""
        Write-Host "  vp was installed to: ${BOLD}${displayDir}\bin${NC}"
        Write-Host ""
        if ($pathResult -eq "failed") {
            Write-Host "  To use vp in Powershell/cmd, manually add it to your PATH:"
            Write-Host ""
            Write-Host "    [Environment]::SetEnvironmentVariable('Path', '$InstallDir\bin;' + [Environment]::GetEnvironmentVariable('Path', 'User'), 'User')"
            Write-Host ""
        }
        if ($nushellResult.Status -eq "failed") {
            Write-Host "  To use vp in Nushell, create a vite-plus.nu file in your preferred vendor autoload directory with:"
            Write-Host ""
            Write-Host "    source '$displayDir\env.nu'"
            Write-Host ""
        }
        Write-Host "  Or run vp directly:"
        Write-Host ""
        Write-Host "    & `"$InstallDir\bin\vp.exe`""
    }

    Write-Host ""
}

Main