mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-13 11:34:26 +00:00
78ae64471a
* use instance id on update in projections * create index on domain in instance_domain projection * add missing instanceID filter to app queries
209 lines
7.0 KiB
Go
209 lines
7.0 KiB
Go
package projection
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/zitadel/zitadel/internal/errors"
|
|
"github.com/zitadel/zitadel/internal/eventstore"
|
|
"github.com/zitadel/zitadel/internal/eventstore/handler"
|
|
"github.com/zitadel/zitadel/internal/eventstore/handler/crdb"
|
|
"github.com/zitadel/zitadel/internal/repository/instance"
|
|
"github.com/zitadel/zitadel/internal/repository/org"
|
|
"github.com/zitadel/zitadel/internal/repository/user"
|
|
)
|
|
|
|
const (
|
|
IDPUserLinkTable = "projections.idp_user_links2"
|
|
IDPUserLinkIDPIDCol = "idp_id"
|
|
IDPUserLinkUserIDCol = "user_id"
|
|
IDPUserLinkExternalUserIDCol = "external_user_id"
|
|
IDPUserLinkCreationDateCol = "creation_date"
|
|
IDPUserLinkChangeDateCol = "change_date"
|
|
IDPUserLinkSequenceCol = "sequence"
|
|
IDPUserLinkResourceOwnerCol = "resource_owner"
|
|
IDPUserLinkInstanceIDCol = "instance_id"
|
|
IDPUserLinkDisplayNameCol = "display_name"
|
|
)
|
|
|
|
type idpUserLinkProjection struct {
|
|
crdb.StatementHandler
|
|
}
|
|
|
|
func newIDPUserLinkProjection(ctx context.Context, config crdb.StatementHandlerConfig) *idpUserLinkProjection {
|
|
p := new(idpUserLinkProjection)
|
|
config.ProjectionName = IDPUserLinkTable
|
|
config.Reducers = p.reducers()
|
|
config.InitCheck = crdb.NewTableCheck(
|
|
crdb.NewTable([]*crdb.Column{
|
|
crdb.NewColumn(IDPUserLinkIDPIDCol, crdb.ColumnTypeText),
|
|
crdb.NewColumn(IDPUserLinkUserIDCol, crdb.ColumnTypeText),
|
|
crdb.NewColumn(IDPUserLinkExternalUserIDCol, crdb.ColumnTypeText),
|
|
crdb.NewColumn(IDPUserLinkCreationDateCol, crdb.ColumnTypeTimestamp),
|
|
crdb.NewColumn(IDPUserLinkChangeDateCol, crdb.ColumnTypeTimestamp),
|
|
crdb.NewColumn(IDPUserLinkSequenceCol, crdb.ColumnTypeInt64),
|
|
crdb.NewColumn(IDPUserLinkResourceOwnerCol, crdb.ColumnTypeText),
|
|
crdb.NewColumn(IDPUserLinkInstanceIDCol, crdb.ColumnTypeText),
|
|
crdb.NewColumn(IDPUserLinkDisplayNameCol, crdb.ColumnTypeText),
|
|
},
|
|
crdb.NewPrimaryKey(IDPUserLinkInstanceIDCol, IDPUserLinkIDPIDCol, IDPUserLinkExternalUserIDCol),
|
|
crdb.WithIndex(crdb.NewIndex("idp_user_idx", []string{IDPUserLinkUserIDCol})),
|
|
),
|
|
)
|
|
p.StatementHandler = crdb.NewStatementHandler(ctx, config)
|
|
return p
|
|
}
|
|
|
|
func (p *idpUserLinkProjection) reducers() []handler.AggregateReducer {
|
|
return []handler.AggregateReducer{
|
|
{
|
|
Aggregate: user.AggregateType,
|
|
EventRedusers: []handler.EventReducer{
|
|
{
|
|
Event: user.UserIDPLinkAddedType,
|
|
Reduce: p.reduceAdded,
|
|
},
|
|
{
|
|
Event: user.UserIDPLinkCascadeRemovedType,
|
|
Reduce: p.reduceCascadeRemoved,
|
|
},
|
|
{
|
|
Event: user.UserIDPLinkRemovedType,
|
|
Reduce: p.reduceRemoved,
|
|
},
|
|
{
|
|
Event: user.UserRemovedType,
|
|
Reduce: p.reduceUserRemoved,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Aggregate: org.AggregateType,
|
|
EventRedusers: []handler.EventReducer{
|
|
{
|
|
Event: org.IDPConfigRemovedEventType,
|
|
Reduce: p.reduceIDPConfigRemoved,
|
|
},
|
|
{
|
|
Event: org.OrgRemovedEventType,
|
|
Reduce: p.reduceOrgRemoved,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Aggregate: instance.AggregateType,
|
|
EventRedusers: []handler.EventReducer{
|
|
{
|
|
Event: instance.IDPConfigRemovedEventType,
|
|
Reduce: p.reduceIDPConfigRemoved,
|
|
},
|
|
{
|
|
Event: instance.InstanceRemovedEventType,
|
|
Reduce: reduceInstanceRemovedHelper(IDPUserLinkInstanceIDCol),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (p *idpUserLinkProjection) reduceAdded(event eventstore.Event) (*handler.Statement, error) {
|
|
e, ok := event.(*user.UserIDPLinkAddedEvent)
|
|
if !ok {
|
|
return nil, errors.ThrowInvalidArgumentf(nil, "HANDL-DpmXq", "reduce.wrong.event.type %s", user.UserIDPLinkAddedType)
|
|
}
|
|
|
|
return crdb.NewCreateStatement(e,
|
|
[]handler.Column{
|
|
handler.NewCol(IDPUserLinkIDPIDCol, e.IDPConfigID),
|
|
handler.NewCol(IDPUserLinkUserIDCol, e.Aggregate().ID),
|
|
handler.NewCol(IDPUserLinkExternalUserIDCol, e.ExternalUserID),
|
|
handler.NewCol(IDPUserLinkCreationDateCol, e.CreationDate()),
|
|
handler.NewCol(IDPUserLinkChangeDateCol, e.CreationDate()),
|
|
handler.NewCol(IDPUserLinkSequenceCol, e.Sequence()),
|
|
handler.NewCol(IDPUserLinkResourceOwnerCol, e.Aggregate().ResourceOwner),
|
|
handler.NewCol(IDPUserLinkInstanceIDCol, e.Aggregate().InstanceID),
|
|
handler.NewCol(IDPUserLinkDisplayNameCol, e.DisplayName),
|
|
},
|
|
), nil
|
|
}
|
|
|
|
func (p *idpUserLinkProjection) reduceRemoved(event eventstore.Event) (*handler.Statement, error) {
|
|
e, ok := event.(*user.UserIDPLinkRemovedEvent)
|
|
if !ok {
|
|
return nil, errors.ThrowInvalidArgumentf(nil, "HANDL-AZmfJ", "reduce.wrong.event.type %s", user.UserIDPLinkRemovedType)
|
|
}
|
|
|
|
return crdb.NewDeleteStatement(e,
|
|
[]handler.Condition{
|
|
handler.NewCond(IDPUserLinkIDPIDCol, e.IDPConfigID),
|
|
handler.NewCond(IDPUserLinkUserIDCol, e.Aggregate().ID),
|
|
handler.NewCond(IDPUserLinkExternalUserIDCol, e.ExternalUserID),
|
|
handler.NewCond(IDPUserLinkInstanceIDCol, e.Aggregate().InstanceID),
|
|
},
|
|
), nil
|
|
}
|
|
|
|
func (p *idpUserLinkProjection) reduceCascadeRemoved(event eventstore.Event) (*handler.Statement, error) {
|
|
e, ok := event.(*user.UserIDPLinkCascadeRemovedEvent)
|
|
if !ok {
|
|
return nil, errors.ThrowInvalidArgumentf(nil, "HANDL-jQpv9", "reduce.wrong.event.type %s", user.UserIDPLinkCascadeRemovedType)
|
|
}
|
|
|
|
return crdb.NewDeleteStatement(e,
|
|
[]handler.Condition{
|
|
handler.NewCond(IDPUserLinkIDPIDCol, e.IDPConfigID),
|
|
handler.NewCond(IDPUserLinkUserIDCol, e.Aggregate().ID),
|
|
handler.NewCond(IDPUserLinkExternalUserIDCol, e.ExternalUserID),
|
|
handler.NewCond(IDPUserLinkInstanceIDCol, e.Aggregate().InstanceID),
|
|
},
|
|
), nil
|
|
}
|
|
|
|
func (p *idpUserLinkProjection) reduceOrgRemoved(event eventstore.Event) (*handler.Statement, error) {
|
|
e, ok := event.(*org.OrgRemovedEvent)
|
|
if !ok {
|
|
return nil, errors.ThrowInvalidArgumentf(nil, "HANDL-AZmfJ", "reduce.wrong.event.type %s", org.OrgRemovedEventType)
|
|
}
|
|
|
|
return crdb.NewDeleteStatement(e,
|
|
[]handler.Condition{
|
|
handler.NewCond(IDPUserLinkResourceOwnerCol, e.Aggregate().ID),
|
|
handler.NewCond(IDPUserLinkInstanceIDCol, e.Aggregate().InstanceID),
|
|
},
|
|
), nil
|
|
}
|
|
|
|
func (p *idpUserLinkProjection) reduceUserRemoved(event eventstore.Event) (*handler.Statement, error) {
|
|
e, ok := event.(*user.UserRemovedEvent)
|
|
if !ok {
|
|
return nil, errors.ThrowInvalidArgumentf(nil, "HANDL-uwlWE", "reduce.wrong.event.type %s", user.UserRemovedType)
|
|
}
|
|
|
|
return crdb.NewDeleteStatement(e,
|
|
[]handler.Condition{
|
|
handler.NewCond(IDPUserLinkUserIDCol, e.Aggregate().ID),
|
|
handler.NewCond(IDPUserLinkInstanceIDCol, e.Aggregate().InstanceID),
|
|
},
|
|
), nil
|
|
}
|
|
|
|
func (p *idpUserLinkProjection) reduceIDPConfigRemoved(event eventstore.Event) (*handler.Statement, error) {
|
|
var idpID string
|
|
|
|
switch e := event.(type) {
|
|
case *org.IDPConfigRemovedEvent:
|
|
idpID = e.ConfigID
|
|
case *instance.IDPConfigRemovedEvent:
|
|
idpID = e.ConfigID
|
|
default:
|
|
return nil, errors.ThrowInvalidArgumentf(nil, "HANDL-iCKSj", "reduce.wrong.event.type %v", []eventstore.EventType{org.IDPConfigRemovedEventType, instance.IDPConfigRemovedEventType})
|
|
}
|
|
|
|
return crdb.NewDeleteStatement(event,
|
|
[]handler.Condition{
|
|
handler.NewCond(IDPUserLinkIDPIDCol, idpID),
|
|
handler.NewCond(IDPUserLinkResourceOwnerCol, event.Aggregate().ResourceOwner),
|
|
handler.NewCond(IDPUserLinkInstanceIDCol, event.Aggregate().InstanceID),
|
|
},
|
|
), nil
|
|
}
|