Peter Štibraný
81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
package builders
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
|
|
iamv0 "github.com/grafana/grafana/apps/iam/pkg/apis/iam/v0alpha1"
|
|
"github.com/grafana/grafana/pkg/apimachinery/utils"
|
|
"github.com/grafana/grafana/pkg/storage/unified/resource"
|
|
"github.com/grafana/grafana/pkg/storage/unified/resourcepb"
|
|
)
|
|
|
|
const (
|
|
USER_EMAIL = "email"
|
|
USER_LOGIN = "login"
|
|
)
|
|
|
|
var UserTableColumnDefinitions = map[string]*resourcepb.ResourceTableColumnDefinition{
|
|
USER_EMAIL: {
|
|
Name: USER_EMAIL,
|
|
Type: resourcepb.ResourceTableColumnDefinition_STRING,
|
|
Description: "The email address of the user",
|
|
Properties: &resourcepb.ResourceTableColumnDefinition_Properties{
|
|
UniqueValues: true,
|
|
Filterable: true,
|
|
},
|
|
},
|
|
USER_LOGIN: {
|
|
Name: USER_LOGIN,
|
|
Type: resourcepb.ResourceTableColumnDefinition_STRING,
|
|
Description: "The login of the user",
|
|
Properties: &resourcepb.ResourceTableColumnDefinition_Properties{
|
|
UniqueValues: true,
|
|
Filterable: true,
|
|
},
|
|
},
|
|
}
|
|
|
|
func GetUserBuilder() (resource.DocumentBuilderInfo, error) {
|
|
values := make([]*resourcepb.ResourceTableColumnDefinition, 0, len(UserTableColumnDefinitions))
|
|
for _, v := range UserTableColumnDefinitions {
|
|
values = append(values, v)
|
|
}
|
|
fields, err := resource.NewSearchableDocumentFields(values)
|
|
return resource.DocumentBuilderInfo{
|
|
GroupResource: iamv0.UserResourceInfo.GroupResource(),
|
|
Fields: fields,
|
|
Builder: new(userDocumentBuilder),
|
|
}, err
|
|
}
|
|
|
|
var _ resource.DocumentBuilder = new(userDocumentBuilder)
|
|
|
|
type userDocumentBuilder struct{}
|
|
|
|
func (u *userDocumentBuilder) BuildDocument(ctx context.Context, key *resourcepb.ResourceKey, rv int64, value []byte) (*resource.IndexableDocument, error) {
|
|
user := &iamv0.User{}
|
|
err := json.NewDecoder(bytes.NewReader(value)).Decode(user)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
obj, err := utils.MetaAccessor(user)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
doc := resource.NewIndexableDocument(key, rv, obj, nil)
|
|
|
|
doc.Fields = make(map[string]any)
|
|
if user.Spec.Email != "" {
|
|
doc.Fields[USER_EMAIL] = user.Spec.Email
|
|
}
|
|
if user.Spec.Login != "" {
|
|
doc.Fields[USER_LOGIN] = user.Spec.Login
|
|
}
|
|
|
|
return doc, nil
|
|
}
|