copilot.ps1

function Get-Runnable {
    <#
        .SYNOPSIS
        Gets the runnable code from the result

        .DESCRIPTION
        Gets the runnable code from the result

        .EXAMPLE
        Get-Runnable -result $result
    #>
    [CmdletBinding()]
    param(
        $result
    )

    $runnable = for ($idx = 1; $idx -lt $result.Count; $idx++) {
        $line = $result[$idx]
        if ([string]::IsNullOrEmpty($line)) {
            continue
        }

        $line = $line.Trim()
        if ($line.StartsWith('#')) {
            continue
        }

        $line
    }

    return ($runnable -join "`n")
}

function copilot {
    <#
        .SYNOPSIS
        Use GPT to help you remember PowerShell commands and other command line tools

        .DESCRIPTION
        Makes the request to GPT, parses the response and displays it in a box and then prompts the user to run the code or not

        .EXAMPLE
        # via https://twitter.com/ClemMesserli/status/1616312238209376260?s=20&t=KknO2iPk3yrQ7x42ZayS7g

        copilot "using PowerShell regex, just code. split user from domain of email address with match:  demo.user@google.com"

        .EXAMPLE
        copilot 'how to get ImportExcel'

        .EXAMPLE
        copilot 'processes running with more than 700 handles'

        .EXAMPLE
        copilot 'processes running with more than 700 handles select first 5, company and name, as json'

        .EXAMPLE
        copilot 'for each file in the current dir list the name and length'
        
        .EXAMPLE
        copilot 'Find all enabled users that have a samaccountname similar to Mazi; List SAMAccountName and DisplayName'
    #>
    param(
        [Parameter(Mandatory)]
        $inputPrompt,
        [ValidateRange(0, 2)]
        [decimal]$temperature = 0.0,
        # The maximum number of tokens to generate. default 256
        $max_tokens = 256,
        # Don't show prompt for choice
        [Switch]$Raw
    )
    
    # $inputPrompt = $args -join ' '
    
    $shell = 'powershell, just code:'
    
    $promptComments = ', include comments'
    if (-not $IncludeComments) {
        $promptComments = ''
    }

    $prompt = "using {0} {1}: {2}`n" -f $shell, $promptComments, $inputPrompt
    $prompt += '```'

    $completion = Get-GPT3Completion -prompt $prompt -max_tokens $max_tokens -temperature $temperature -stop '```'
    $completion = $completion -split "`n"
    
    if ($completion[0] -ceq 'powershell') {
        $completion = $completion[1..($completion.Count - 1)]
    }

    if ($Raw) {
        return $completion
    }
    else {

        $result = @($inputPrompt)
        $result += ''
        $result += $completion

        $runnable = Get-Runnable -result $result
        
        if (Test-AifbScriptAnalyzerAvailable) {
            $runnable = Invoke-Formatter -ScriptDefinition $runnable -Verbose:$false
        }

        Write-Codeblock -Text $runnable -ShowLineNumbers -SyntaxHighlight

        $userInput = CustomReadHost
        
        switch ($userInput) {
            0 {
                (Get-Runnable -result $result) | Invoke-Expression
            }
            1 {
                explain -Value (Get-Runnable -result $result)
            }
            2 {
                Get-Runnable -result $result | Set-Clipboard
            }
            default { "Not running" }
        }
    }
}