mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-12 11:04:25 +00:00
81 lines
2.5 KiB
Go
81 lines
2.5 KiB
Go
package user
|
|
|
|
import (
|
|
"github.com/zitadel/zitadel/internal/api/grpc/object"
|
|
"github.com/zitadel/zitadel/internal/errors"
|
|
"github.com/zitadel/zitadel/internal/query"
|
|
user_pb "github.com/zitadel/zitadel/pkg/grpc/user"
|
|
)
|
|
|
|
func MembershipQueriesToQuery(queries []*user_pb.MembershipQuery) (_ []query.SearchQuery, err error) {
|
|
q := make([]query.SearchQuery, 0)
|
|
for _, query := range queries {
|
|
qs, err := MembershipQueryToQuery(query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
q = append(q, qs)
|
|
}
|
|
return q, nil
|
|
}
|
|
|
|
func MembershipQueryToQuery(req *user_pb.MembershipQuery) (query.SearchQuery, error) {
|
|
switch q := req.Query.(type) {
|
|
case *user_pb.MembershipQuery_OrgQuery:
|
|
return query.NewMembershipOrgIDQuery(q.OrgQuery.OrgId)
|
|
case *user_pb.MembershipQuery_ProjectQuery:
|
|
return query.NewMembershipProjectIDQuery(q.ProjectQuery.ProjectId)
|
|
case *user_pb.MembershipQuery_ProjectGrantQuery:
|
|
return query.NewMembershipProjectGrantIDQuery(q.ProjectGrantQuery.ProjectGrantId)
|
|
case *user_pb.MembershipQuery_IamQuery:
|
|
return query.NewMembershipIsIAMQuery()
|
|
default:
|
|
return nil, errors.ThrowInvalidArgument(nil, "USER-dsg3z", "Errors.List.Query.Invalid")
|
|
}
|
|
}
|
|
|
|
func MembershipsToMembershipsPb(memberships []*query.Membership) []*user_pb.Membership {
|
|
converted := make([]*user_pb.Membership, len(memberships))
|
|
for i, membership := range memberships {
|
|
converted[i] = MembershipToMembershipPb(membership)
|
|
}
|
|
return converted
|
|
}
|
|
|
|
func MembershipToMembershipPb(membership *query.Membership) *user_pb.Membership {
|
|
typ, name := memberTypeToPb(membership)
|
|
return &user_pb.Membership{
|
|
UserId: membership.UserID,
|
|
Type: typ,
|
|
DisplayName: name,
|
|
Roles: membership.Roles,
|
|
Details: object.ToViewDetailsPb(
|
|
membership.Sequence,
|
|
membership.CreationDate,
|
|
membership.ChangeDate,
|
|
membership.ResourceOwner,
|
|
),
|
|
}
|
|
}
|
|
|
|
func memberTypeToPb(membership *query.Membership) (user_pb.MembershipType, string) {
|
|
if membership.Org != nil {
|
|
return &user_pb.Membership_OrgId{
|
|
OrgId: membership.Org.OrgID,
|
|
}, membership.Org.Name
|
|
} else if membership.Project != nil {
|
|
return &user_pb.Membership_ProjectId{
|
|
ProjectId: membership.Project.ProjectID,
|
|
}, membership.Project.Name
|
|
} else if membership.ProjectGrant != nil {
|
|
return &user_pb.Membership_ProjectGrantId{
|
|
ProjectGrantId: membership.ProjectGrant.GrantID,
|
|
}, membership.ProjectGrant.ProjectName
|
|
} else if membership.IAM != nil {
|
|
return &user_pb.Membership_Iam{
|
|
Iam: true,
|
|
}, membership.IAM.Name
|
|
}
|
|
return nil, ""
|
|
}
|