Mihai Doarna
b2e1b257b3
* add legacy search (wip) * fix search field name * implement team search endpoint * generate openapi spec * generate endpoints for frontend * minor fixes * fix issues found while testing * add more fields to search result * add basic unit tests * add more unit tests * improve getColumns() func in legacy search * configure search endpoint in team.cue * add team search handler * add the searchTeams endpoint to manifest.cue * make gofmt * update openapi spec * generate frontend endpoints * remove unused field * move fields defiitions to separate builder * fix legacy search * fix unit tests * fix unit test * address feedback * fix unit test * update openapi specs * yarn generate-apis * add missing unit tests
83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
package search
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/grafana/grafana/apps/iam/pkg/apis/iam/v0alpha1"
|
|
"github.com/grafana/grafana/pkg/storage/unified/resource"
|
|
"github.com/grafana/grafana/pkg/storage/unified/resourcepb"
|
|
"github.com/grafana/grafana/pkg/storage/unified/search/builders"
|
|
)
|
|
|
|
func ParseResults(result *resourcepb.ResourceSearchResponse, offset int64) (v0alpha1.TeamSearchResults, error) {
|
|
if result == nil {
|
|
return v0alpha1.TeamSearchResults{}, nil
|
|
} else if result.Error != nil {
|
|
return v0alpha1.TeamSearchResults{}, fmt.Errorf("%d error searching: %s: %s", result.Error.Code, result.Error.Message, result.Error.Details)
|
|
} else if result.Results == nil {
|
|
return v0alpha1.TeamSearchResults{}, nil
|
|
}
|
|
|
|
titleIDX := -1
|
|
emailIDX := -1
|
|
provisionedIDX := -1
|
|
externalUIDIDX := -1
|
|
|
|
for i, v := range result.Results.Columns {
|
|
if v == nil {
|
|
continue
|
|
}
|
|
|
|
switch v.Name {
|
|
case resource.SEARCH_FIELD_TITLE:
|
|
titleIDX = i
|
|
case builders.TEAM_SEARCH_EMAIL:
|
|
emailIDX = i
|
|
case builders.TEAM_SEARCH_PROVISIONED:
|
|
provisionedIDX = i
|
|
case builders.TEAM_SEARCH_EXTERNAL_UID:
|
|
externalUIDIDX = i
|
|
}
|
|
}
|
|
|
|
sr := v0alpha1.TeamSearchResults{
|
|
Offset: offset,
|
|
TotalHits: result.TotalHits,
|
|
QueryCost: result.QueryCost,
|
|
MaxScore: result.MaxScore,
|
|
Hits: make([]v0alpha1.TeamHit, len(result.Results.Rows)),
|
|
}
|
|
|
|
for i, row := range result.Results.Rows {
|
|
if len(row.Cells) != len(result.Results.Columns) {
|
|
return v0alpha1.TeamSearchResults{}, fmt.Errorf("error parsing team search response: mismatch number of columns and cells")
|
|
}
|
|
|
|
hit := &v0alpha1.TeamHit{
|
|
Name: row.Key.Name,
|
|
}
|
|
|
|
if titleIDX >= 0 && row.Cells[titleIDX] != nil {
|
|
hit.Title = string(row.Cells[titleIDX])
|
|
} else {
|
|
hit.Title = "(no title)"
|
|
}
|
|
|
|
if emailIDX >= 0 && row.Cells[emailIDX] != nil {
|
|
hit.Email = string(row.Cells[emailIDX])
|
|
}
|
|
|
|
if provisionedIDX >= 0 && row.Cells[provisionedIDX] != nil {
|
|
hit.Provisioned = string(row.Cells[provisionedIDX]) == "true"
|
|
}
|
|
|
|
if externalUIDIDX >= 0 && row.Cells[externalUIDIDX] != nil {
|
|
hit.ExternalUID = string(row.Cells[externalUIDIDX])
|
|
}
|
|
|
|
sr.Hits[i] = *hit
|
|
}
|
|
|
|
return sr, nil
|
|
}
|