rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove duplicate membership method

Browse Source 
Signed-off-by: Émile Ré <emile@getprobo.com>
This commit is contained in:
Émile Ré
2026-02-16 11:14:20 +04:00
parent 56ec0ab3c3
commit 7794563e94
6 changed files with 29 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/iam/account_service.go
View File

@@ -268,7 +268,7 @@ func (s *AccountService) AcceptInvitation(
}
existingMembership := &coredata.Membership{}
if err := existingMembership.LoadByIdentityAndOrg(
if err := existingMembership.LoadByIdentityIDAndOrganizationID(
ctx,
tx,
scope,
@@ -775,7 +775,13 @@ func (s AccountService) GetMembershipForOrganization(
return fmt.Errorf("cannot load identity %q: %w", identityID, err)
}
if err := membership.LoadByIdentityInOrganization(ctx, tx, identityID, organizationID); err != nil {
if err := membership.LoadByIdentityIDAndOrganizationID(
ctx,
tx,
coredata.NewScopeFromObjectID(organizationID),
identityID,
organizationID,
); err != nil {
return fmt.Errorf("cannot load membership: %w", err)
}

2
pkg/iam/organization_service.go
View File

@@ -281,7 +281,7 @@ func (s *OrganizationService) RemoveMember(
}
membership := &coredata.Membership{}
if err := membership.LoadByIdentityAndOrg(ctx, tx, scope, profile.IdentityID, profile.OrganizationID); err != nil {
if err := membership.LoadByIdentityIDAndOrganizationID(ctx, tx, scope, profile.IdentityID, profile.OrganizationID); err != nil {
return fmt.Errorf("cannot load membership: %w", err)
}

2
pkg/iam/saml/service.go
View File

@@ -328,7 +328,7 @@ func (s *Service) HandleAssertion(
}
}
if err := membership.LoadByIdentityAndOrg(
if err := membership.LoadByIdentityIDAndOrganizationID(
ctx,
tx,
scope,

6
pkg/iam/scim/service.go
View File

@@ -213,7 +213,7 @@ func (s *Service) CreateUser(
// Check if membership exists
membership = &coredata.Membership{}
if err := membership.LoadByIdentityAndOrg(
if err := membership.LoadByIdentityIDAndOrganizationID(
ctx,
tx,
scope,
@@ -294,7 +294,7 @@ func (s *Service) GetUser(
}
membership = &coredata.Membership{}
if err := membership.LoadByIdentityAndOrg(
if err := membership.LoadByIdentityIDAndOrganizationID(
ctx,
conn,
scope,
@@ -437,7 +437,7 @@ func (s *Service) updateUser(
}
membership = &coredata.Membership{}
if err := membership.LoadByIdentityAndOrg(ctx, tx, scope, profile.IdentityID, profile.OrganizationID); err != nil {
if err := membership.LoadByIdentityIDAndOrganizationID(ctx, tx, scope, profile.IdentityID, profile.OrganizationID); err != nil {
return fmt.Errorf("cannot load membership: %w", err)
}

6
pkg/iam/session_service.go
View File

@@ -366,7 +366,7 @@ func (s SessionService) OpenPasswordChildSessionForOrganization(
return NewUserInactiveError(profile.ID)
}
err = membership.LoadByIdentityInOrganization(ctx, tx, rootSession.IdentityID, organizationID)
err = membership.LoadByIdentityIDAndOrganizationID(ctx, tx, scope, rootSession.IdentityID, organizationID)
if err != nil {
if err == coredata.ErrResourceNotFound {
return NewMembershipNotFoundError(organizationID)
@@ -468,7 +468,7 @@ func (s SessionService) OpenSAMLChildSessionForOrganization(
return NewUserInactiveError(profile.ID)
}
err = membership.LoadByIdentityInOrganization(ctx, tx, rootSession.IdentityID, organizationID)
err = membership.LoadByIdentityIDAndOrganizationID(ctx, tx, scope, rootSession.IdentityID, organizationID)
if err != nil {
if err == coredata.ErrResourceNotFound {
return NewMembershipNotFoundError(organizationID)
@@ -554,7 +554,7 @@ func (s SessionService) AssumeOrganizationSession(
return NewUserInactiveError(profile.ID)
}
if err := membership.LoadByIdentityInOrganization(ctx, tx, rootSession.IdentityID, organizationID); err != nil {
if err := membership.LoadByIdentityIDAndOrganizationID(ctx, tx, scope, rootSession.IdentityID, organizationID); err != nil {
if err == coredata.ErrResourceNotFound {
return NewMembershipNotFoundError(organizationID)
}