posh-vs

0.2.3

posh-vs.psm1

[Diagnostics.CodeAnalysis.SuppressMessageAttribute("PSUseSingularNouns", "", Scope="Function", Target="*-PoshVS", Justification="PoshVs is a singular noun")]
param()

function Get-VisualStudio2015BatchFile {
    if ($env:VS140ComnTools) {
        Join-Path $env:VS140ComnTools "VsDevCmd.bat"
    }
}

function Get-VisualStudio2017ApplicationDescription {
    & {
        Get-ItemProperty HKLM:\SOFTWARE\WOW6432Node\Microsoft\VisualStudio_*\Capabilities
        Get-ItemProperty HKLM:\SOFTWARE\Microsoft\VisualStudio_*\Capabilities
    } | ForEach-Object { $_.ApplicationDescription }
}

function Get-VisualStudio2017BatchFile {
    Get-VisualStudio2017ApplicationDescription |
    ForEach-Object {
        # @C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\Common7\IDE\devenvdesc.dll,-1004
        if ($_ -match '@(?<Common7>.*)\\IDE\\devenvdesc.dll,-(?:\d*)') {
            return Join-Path $matches.Common7 'Tools\VsDevCmd.bat'
        }

        throw "Cannot parse Visual Studio ApplicationDescription: $_"
    }
}

function Get-VisualStudioBatchFile {
    Get-VisualStudio2017BatchFile
    Get-VisualStudio2015BatchFile
}

<# .SYNOPSIS
Executes a batch file and copies environment variables it sets to the current
PowerShell session #>
function Import-BatchEnvironment {
    [CmdletBinding()] param (
        [Parameter(Mandatory = $true)] [string] $batchFile
    )

    if (-not (Test-Path $batchFile)) {
        Throw "Batch file '$batchFile' does not exist."
    }

    Write-Verbose "Executing '$batchFile' to capture environment variables it sets."
    cmd /c "`"$batchFile`" > nul & set" | ForEach-Object {
        if ($_ -match "^(.+?)=(.*)$") {
            [string] $variable = $matches[1]
            [string] $value = $matches[2]
            Write-Verbose "`$env:$variable=$value"
            Set-Item -Force -Path "env:\$variable" -Value $value
        }
    }
}

<# .SYNOPSIS
Executes Visual Studio 2015's VsDevCmd.bat and copies environment variables it sets
to the current PowerShell session. #>
function Import-VisualStudioEnvironment {
    [string] $batchFile = Get-VisualStudioBatchFile | Select-Object -First 1
    Import-BatchEnvironment $batchFile
}

[string] $importModulePattern = "Import-Module.*posh-vs"
[string] $importEnvironmentPattern = "Import-VisualStudioEnvironment"

<# .SYNOPSIS
Adds Import-Module and Import-VisualStudioEnvironment commands to the current PowerShell profile script. #>
function Install-PoshVs {
    [string[]] $profileScript = @()
    [bool] $importEnvironment = $true
    if (Test-Path $profile) {
        $profileScript = Get-Content $profile
        foreach ($line in $profileScript) {
            if ($line -match $importEnvironmentPattern) {
                $importEnvironment = $false
            }
        }
    }

    if ($importEnvironment) {
        $profileScript += "Import-VisualStudioEnvironment"
    }

    $profileScript | Out-File $profile

    Write-Output "Successfully added posh-vs to profile '$profile'."
    Write-Output "Reload your profile for the changes to take effect:"
    Write-Output " . `$profile"
}

<# .SYNOPSIS
Removes Import-Module and Import-VisualStudioEnvironment from the current PowerShell profile script. #>
function Uninstall-PoshVs {
    if (Test-Path $profile) {
        [string[]] $script = Get-Content $profile
        $script | Where-Object {
            -not ($_ -match $importModulePattern) -and
            -not ($_ -match $importEnvironmentPattern)
        } | Out-File $profile
    }

    Write-Output "Successfully removed posh-vs from profile '$profile'."
    Write-Output "Restart PowerShell for the changes to take effect."
}

Export-ModuleMember -Function Import-BatchEnvironment
Export-ModuleMember -Function Import-VisualStudioEnvironment
Export-ModuleMember -Function Install-PoshVs
Export-ModuleMember -Function Uninstall-PoshVs