src/search.psm1


$RepositoryFields = Get-Content -Raw "$PSScriptRoot/queries/RepositoryFields.graphql"
$SearchQuery = (Get-Content -Raw "$PSScriptRoot/queries/Search.graphql") + $RepositoryFields
$SuggestionsQuery = (Get-Content -Raw "$PSScriptRoot/queries/Suggestions.graphql")

function Search-Sourcegraph {
    <#
    .SYNOPSIS
        Get users on a Sourcegraph instance
    .PARAMETER Query
        The search query.
    .PARAMETER CaseSensitive
        Match the query case-sensitive. Only for regexp and literal search.
    .PARAMETER Structural
        Interpret the query as a structural search query.
    .PARAMETER RegularExpression
        Interpret the query as a regular expression.
    .PARAMETER Endpoint
        The endpoint URL of the Sourcegraph instance (default https://sourcegraph.com)
    .PARAMETER Token
        The authentication token (if needed). Go to the settings page on Sourcegraph to generate one.
    #>

    [CmdletBinding(SupportsPaging, DefaultParameterSetName = 'literal')]
    param(
        [Parameter(Mandatory, Position = 0)]
        [string] $Query,

        [Parameter(ParameterSetName = 'regexp')]
        [Parameter(ParameterSetName = 'literal')]
        [switch] $CaseSensitive,

        [Parameter(ParameterSetName = 'regexp', Mandatory)]
        [Alias('Regexp')]
        [switch] $RegularExpression,

        [Parameter(ParameterSetName = 'structural', Mandatory)]
        [switch] $Structural,

        [Uri] $Endpoint = 'https://sourcegraph.com',
        [string] $Token
    )

    process {
        if ($CaseSensitive) {
            $Query += ' case:yes'
        }
        $variables = @{
            query = $Query
            patternType = $PSCmdlet.ParameterSetName
        }
        $data = Invoke-SourcegraphApiRequest -Query $SearchQuery -Variables $variables -Endpoint $Endpoint -Token $Token
        if ($data.search.results.cloning.Count -gt 0) {
            Write-Warning "Cloning:"
            $data.search.results.cloning.name | Write-Warning
        }
        if ($data.search.results.missing.Count -gt 0) {
            Write-Warning "Missing:"
            $data.search.results.missing.name | Write-Warning
        }
        if ($data.search.results.timedout.Count -gt 0) {
            Write-Warning "Timed out:"
            $data.search.results.timedout.name | Write-Warning
        }
        if ($PSCmdlet.PagingParameters.IncludeTotalCount) {
            $PSCmdlet.PagingParameters.NewTotalCount($data.search.results.resultCount, 1)
        }
        if ($data.search.results.limitHit) {
            Write-Warning "Result limit hit"
        }

        foreach ($result in $data.search.results.results) {
            $result.PSObject.TypeNames.Insert(0, 'Sourcegraph.' + $result.__typename)
            # Make the metadata accessible from the match objects
            Add-Member -InputObject $result -MemberType NoteProperty -Name 'SearchResults' -Value $data.search.results
            if ($result.__typename -eq 'FileMatch') {
                # Make URL absolute
                $result.File.Url = [Uri]::new($Endpoint, $result.File.Url)
                $result.Repository.Url = [Uri]::new($Endpoint, $result.Repository.Url)

                if ($result.LineMatches -or $result.Symbols) {
                    # Instead of nesting LineMatches and Symbols in FileMatches, we flat out the list and let PowerShell formatting do the grouping
                    foreach ($lineMatch in $result.LineMatches) {
                        $lineMatch.PSObject.TypeNames.Insert(0, 'Sourcegraph.LineMatch')
                        Add-Member -InputObject $lineMatch -MemberType NoteProperty -Name 'FileMatch' -Value $result
                        $lineMatch
                    }
                    foreach ($symbol in $result.Symbols) {
                        $symbol.PSObject.TypeNames.Insert(0, 'Sourcegraph.Symbol')
                        Add-Member -InputObject $symbol -MemberType NoteProperty -Name 'FileMatch' -Value $result
                        $symbol
                    }
                } else {
                    # The FileMatch has no line or symbol matches, which means the file name matched, so add the FileMatch itself as a result
                    $result
                }
            } else {
                $result
            }
        }
    }
}
Set-Alias Search-Src Search-Sourcegraph

function Get-SourcegraphSearchSuggestions {
    [CmdletBinding(DefaultParameterSetName = 'literal')]
    param (
        [Parameter(Mandatory)]
        [string] $Query,

        [Parameter(ParameterSetName = 'regexp')]
        [Parameter(ParameterSetName = 'literal')]
        [switch] $CaseSensitive,

        [Parameter(ParameterSetName = 'regexp', Mandatory)]
        [Alias('Regexp')]
        [switch] $RegularExpression,

        [Parameter(ParameterSetName = 'structural', Mandatory)]
        [switch] $Structural,

        [Uri] $Endpoint = 'https://sourcegraph.com',
        [string] $Token
    )

    process {
        if ($CaseSensitive) {
            $Query += ' case:yes'
        }
        $vars = @{
            query = $Query
            first = 10
            patternType = $PSCmdlet.ParameterSetName
        }
        Invoke-SourcegraphApiRequest -Query $SuggestionsQuery -Variables $vars -Endpoint $Endpoint -Token $Token |
            ForEach-Object { $_.search.suggestions }
    }
}

# Merges two potentially overlapping strings
function Merge-Strings([string] $a, [string] $b) {
    for ($i = $b.Length; $i -gt 0; $i--) {
        if ($a.EndsWith($b.Substring(0, $i))) {
            return $a.Substring(0, $a.Length - $i) + $b
        }
    }
    # replace last word
    return ($a -replace "\b\S*$", "") + $b
}

Register-ArgumentCompleter -CommandName Search-Sourcegraph -ParameterName Query -ScriptBlock {
    [CmdletBinding()]
    param([string]$command, [string]$parameter, [string]$wordToComplete, [CommandAst]$commandAst, [Hashtable]$params)

    $suggestionParams = @{}
    if ($params.ContainsKey('Token')) {
        $suggestionParams.Token = $params.Token
    }
    if ($params.ContainsKey('Endpoint')) {
        $suggestionParams.Endpoint = $params.Endpoint
    }
    if ($params.ContainsKey('CaseSensitive')) {
        $suggestionParams.CaseSensitive = $params.CaseSensitive
    }
    if ($params.ContainsKey('RegularExpression')) {
        $suggestionParams.RegularExpression = $params.RegularExpression
    }
    if ($params.ContainsKey('Structural')) {
        $suggestionParams.Structural = $params.Structural
    }

    Get-SourcegraphSearchSuggestions @suggestionParams -Query $wordToComplete.Trim(@("'", '"')) |
        ForEach-Object {
            $suggestion = $_
            $insertText, $tooltip = switch ($suggestion.__typename) {
                'Repository' {
                    'repo:' + $suggestion.Name
                    "Repository $($suggestion.Name)"
                }
                'File' {
                    'file:' + $suggestion.Path
                    "File $($suggestion.Name)"
                }
                'Symbol' {
                    $suggestion.Name
                    $kind = $suggestion.Kind[0] + $suggestion.Kind.Substring(1).ToLower()
                    "$kind symbol $($suggestion.Name)`n$($suggestion.Location.Resource.Path)"
                }
            }
            $replaceText = Merge-Strings $wordToComplete.TrimEnd(@('"', "'")) $insertText
            [CompletionResult]::new($replaceText, $suggestion.Name, [CompletionResultType]::ParameterValue, $tooltip)
        }
}