Invoke-TaskFromVSCode

1.1.0

Invoke-TaskFromVSCode.ps1

<#PSScriptInfo
.VERSION 1.1.0
.AUTHOR Roman Kuzmin
.COPYRIGHT (c) Roman Kuzmin
.TAGS Invoke-Build, Task, VSCode
.GUID 1dcf7c94-b68d-4fb7-9e2b-886889b6c42e
.LICENSEURI http://www.apache.org/licenses/LICENSE-2.0
.PROJECTURI https://github.com/nightroman/Invoke-Build
#>

<#
.Synopsis
    Invokes the current Invoke-Build task from VSCode.
 
.Description
    It invokes the current task from the current build script opened in VSCode.
 
    Requires:
    - Invoke-Build
    - VSCode PowerShell
 
    How to use:
    https://github.com/nightroman/Invoke-Build/blob/main/Docs/Invoke-Task-from-VSCode.md
 
.Parameter Console
        Tells to invoke the task in an external console.
 
.Link
    https://github.com/nightroman/Invoke-Build/blob/main/Docs/Invoke-Task-from-VSCode.md
#>

[CmdletBinding()]
param(
    [switch]$Console
)

$ErrorActionPreference = 1; trap {throw $_}

$private:file = $null
try {
    $private:context = $psEditor.GetEditorContext()
    $file = $context.CurrentFile
}
catch {}

if (!$file) {
    return Write-Warning "No current file."
}

# save if modified, #118
$file.Save()

$private:path = $file.Path
if ($path -notlike '*.ps1') {
    return Write-Warning "No current .ps1 file."
}

$private:_Console = $Console
Remove-Variable Console

$goodTasksDic = Invoke-Build ?? $path -Result:Result
$goodTasks = $goodTasksDic.get_Values()
$dupeTasks = $Result.Redefined
$line = $context.CursorPosition.Line

function __find_caret_task($Tasks, $Path, $Line) {
    $bestTaskName = ''
    $bestLineNumber = -1
    foreach($task in $Tasks) {
        $ii = $task.InvocationInfo

        # skip different file
        if ($ii.ScriptName -ne $Path) {
            continue
        }

        # stop on any task below the caret
        if ($ii.ScriptLineNumber -gt $Line) {
            break
        }

        # keep the best
        $bestTaskName = $task.Name
        $bestLineNumber = $ii.ScriptLineNumber
    }
    [pscustomobject]@{Name = $bestTaskName; LineNumber = $bestLineNumber}
}

$goodTask = __find_caret_task $goodTasks $path $line
$dupeTask = __find_caret_task $dupeTasks $path $line

# no dupe or good task?
if (!$dupeTask.Name -and !$goodTask.Name) {
    return Write-Warning "No current task."
}

# dupe task and no good task better than dupe?
if ($dupeTask.Name -and $goodTask.LineNumber -lt $dupeTask.LineNumber) {
    $ii = $goodTasksDic[$dupeTask.Name].InvocationInfo
    Write-Warning "Invoking redefined task at $($ii.ScriptName):$($ii.scriptLineNumber)"
    $goodTask = $dupeTask
}

if ($_Console) {
    $command = "Invoke-Build '$($goodTask.Name.Replace("'", "''"))' '$($path.Replace("'", "''"))'"
    $encoded = [Convert]::ToBase64String(([System.Text.Encoding]::Unicode.GetBytes($command)))
    Start-Process powershell.exe "-NoExit -NoProfile -ExecutionPolicy Bypass -EncodedCommand $encoded"
}
else {
    Invoke-Build $goodTask.Name $path
}