mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-13 03:24:26 +00:00
4980cd6a0c
* define roles and permissions * support system user memberships * don't limit system users * cleanup permissions * restrict memberships to aggregates * default to SYSTEM_OWNER * update unit tests * test: system user token test (#6778) * update unit tests * refactor: make authz testable * move session constants * cleanup * comment * comment * decode member type string to enum (#6780) * decode member type string to enum * handle all membership types * decode enums where necessary * decode member type in steps config * update system api docs * add technical advisory * tweak docs a bit * comment in comment * lint * extract token from Bearer header prefix * review changes * fix tests * fix: add fix for activityhandler * add isSystemUser * remove IsSystemUser from activity info * fix: add fix for activityhandler --------- Co-authored-by: Stefan Benz <stefan@caos.ch>
91 lines
2.7 KiB
Go
91 lines
2.7 KiB
Go
package eventstore
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/zitadel/zitadel/internal/api/authz"
|
|
"github.com/zitadel/zitadel/internal/query"
|
|
"github.com/zitadel/zitadel/internal/telemetry/tracing"
|
|
)
|
|
|
|
type UserMembershipRepo struct {
|
|
Queries *query.Queries
|
|
}
|
|
|
|
func (repo *UserMembershipRepo) SearchMyMemberships(ctx context.Context, orgID string, shouldTriggerBulk bool) (_ []*authz.Membership, err error) {
|
|
ctx, span := tracing.NewSpan(ctx)
|
|
defer func() { span.EndWithError(err) }()
|
|
|
|
memberships, err := repo.searchUserMemberships(ctx, orgID, shouldTriggerBulk)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return userMembershipsToMemberships(memberships), nil
|
|
}
|
|
|
|
func (repo *UserMembershipRepo) searchUserMemberships(ctx context.Context, orgID string, shouldTriggerBulk bool) (_ []*query.Membership, err error) {
|
|
ctx, span := tracing.NewSpan(ctx)
|
|
defer func() { span.EndWithError(err) }()
|
|
ctxData := authz.GetCtxData(ctx)
|
|
userIDQuery, err := query.NewMembershipUserIDQuery(ctxData.UserID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
orgIDsQuery, err := query.NewMembershipResourceOwnersSearchQuery(orgID, authz.GetInstance(ctx).InstanceID())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
grantedIDQuery, err := query.NewMembershipGrantedOrgIDSearchQuery(orgID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
memberships, err := repo.Queries.Memberships(ctx, &query.MembershipSearchQuery{
|
|
Queries: []query.SearchQuery{userIDQuery, query.Or(orgIDsQuery, grantedIDQuery)},
|
|
}, false, shouldTriggerBulk)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return memberships.Memberships, nil
|
|
}
|
|
|
|
func userMembershipToMembership(membership *query.Membership) *authz.Membership {
|
|
if membership.IAM != nil {
|
|
return &authz.Membership{
|
|
MemberType: authz.MemberTypeIAM,
|
|
AggregateID: membership.IAM.IAMID,
|
|
ObjectID: membership.IAM.IAMID,
|
|
Roles: membership.Roles,
|
|
}
|
|
}
|
|
if membership.Org != nil {
|
|
return &authz.Membership{
|
|
MemberType: authz.MemberTypeOrganization,
|
|
AggregateID: membership.Org.OrgID,
|
|
ObjectID: membership.Org.OrgID,
|
|
Roles: membership.Roles,
|
|
}
|
|
}
|
|
if membership.Project != nil {
|
|
return &authz.Membership{
|
|
MemberType: authz.MemberTypeProject,
|
|
AggregateID: membership.Project.ProjectID,
|
|
ObjectID: membership.Project.ProjectID,
|
|
Roles: membership.Roles,
|
|
}
|
|
}
|
|
return &authz.Membership{
|
|
MemberType: authz.MemberTypeProjectGrant,
|
|
AggregateID: membership.ProjectGrant.ProjectID,
|
|
ObjectID: membership.ProjectGrant.GrantID,
|
|
Roles: membership.Roles,
|
|
}
|
|
}
|
|
|
|
func userMembershipsToMemberships(memberships []*query.Membership) []*authz.Membership {
|
|
result := make([]*authz.Membership, len(memberships))
|
|
for i, m := range memberships {
|
|
result[i] = userMembershipToMembership(m)
|
|
}
|
|
return result
|
|
}
|