src/search.psm1
|
$RepositoryFields = Get-Content -Raw "$PSScriptRoot/queries/RepositoryFields.graphql" $SearchQuery = (Get-Content -Raw "$PSScriptRoot/queries/Search.graphql") + $RepositoryFields function Search-Sourcegraph { <# .SYNOPSIS Get users on a Sourcegraph instance .PARAMETER Username Get only the user with the given username .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)] param( [Parameter(Mandatory, Position = 0)] [string] $Query, [Uri] $Endpoint = 'https://sourcegraph.com', [string] $Token ) $data = Invoke-SourcegraphApiRequest -Query $SearchQuery -Variables @{ query = $Query } -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 |