policies implemented

internal/v2/repository/member/event_added.go

@@ -21,11 +21,16 @@ func (e *MemberAddedEvent) Data() interface{} {
 	return e
 }
 
-func NewMemberAddedEvent(ctx context.Context, eventType eventstore.EventType, service, userID string, roles ...string) *MemberAddedEvent {
+func NewMemberAddedEvent(
+	ctx context.Context,
+	eventType eventstore.EventType,
+	userID string,
+	roles ...string,
+) *MemberAddedEvent {
+
 	return &MemberAddedEvent{
 		BaseEvent: *eventstore.NewBaseEventForPush(
 			ctx,
-			service,
 			eventType,
 		),
 		Roles:  roles,

internal/v2/repository/member/event_changed.go

@@ -0,0 +1,39 @@
+package member
+
+import (
+	"context"
+
+	"github.com/caos/zitadel/internal/eventstore/v2"
+)
+
+type MemberChangedEvent struct {
+	eventstore.BaseEvent `json:"-"`
+
+	Roles  []string `json:"roles"`
+	UserID string   `json:"userId"`
+}
+
+func (e *MemberChangedEvent) CheckPrevious() bool {
+	return true
+}
+
+func (e *MemberChangedEvent) Data() interface{} {
+	return e
+}
+
+func NewMemberChangedEvent(
+	ctx context.Context,
+	eventType eventstore.EventType,
+	userID string,
+	roles ...string,
+) *MemberChangedEvent {
+
+	return &MemberChangedEvent{
+		BaseEvent: *eventstore.NewBaseEventForPush(
+			ctx,
+			eventType,
+		),
+		Roles:  roles,
+		UserID: userID,
+	}
+}

internal/v2/repository/member/event_removed.go

@@ -20,11 +20,15 @@ func (e *MemberRemovedEvent) Data() interface{} {
 	return e
 }
 
-func NewMemberRemovedEvent(ctx context.Context, eventType eventstore.EventType, service, userID string) *MemberRemovedEvent {
+func NewMemberRemovedEvent(
+	ctx context.Context,
+	eventType eventstore.EventType,
+	userID string,
+) *MemberRemovedEvent {
+
 	return &MemberRemovedEvent{
 		BaseEvent: *eventstore.NewBaseEventForPush(
 			ctx,
-			service,
 			eventType,
 		),
 		UserID: userID,

internal/v2/repository/member/read_models.go

@@ -0,0 +1,84 @@
+package member
+
+import "github.com/caos/zitadel/internal/eventstore/v2"
+
+type MemberReadModel struct {
+	eventstore.ReadModel
+
+	UserID string
+	Roles  []string
+}
+
+func NewMemberReadModel(userID string) *MemberReadModel {
+	return &MemberReadModel{
+		ReadModel: *eventstore.NewReadModel(),
+	}
+}
+
+func (rm *MemberReadModel) Reduce() error {
+	for _, event := range rm.Events {
+		switch e := event.(type) {
+		case *MemberAddedEvent:
+			rm.UserID = e.UserID
+			rm.Roles = e.Roles
+		case *MemberChangedEvent:
+			rm.UserID = e.UserID
+			rm.Roles = e.Roles
+		}
+	}
+	return rm.ReadModel.Reduce()
+}
+
+type MembersReadModel struct {
+	eventstore.ReadModel
+	Members []*MemberReadModel
+}
+
+func NewMembersReadModel() *MembersReadModel {
+	return &MembersReadModel{
+		ReadModel: *eventstore.NewReadModel(),
+		Members:   []*MemberReadModel{},
+	}
+}
+
+func (rm *MembersReadModel) MemberByUserID(id string) (idx int, member *MemberReadModel) {
+	for idx, member = range rm.Members {
+		if member.UserID == id {
+			return idx, member
+		}
+	}
+	return -1, nil
+}
+
+func (rm *MembersReadModel) AppendEvents(events ...eventstore.EventReader) (err error) {
+	for _, event := range events {
+		switch e := event.(type) {
+		case *MemberAddedEvent:
+			member := NewMemberReadModel(e.UserID)
+			rm.Members = append(rm.Members, member)
+			member.AppendEvents(e)
+		case *MemberChangedEvent:
+			_, member := rm.MemberByUserID(e.UserID)
+			member.AppendEvents(e)
+		case *MemberRemovedEvent:
+			idx, _ := rm.MemberByUserID(e.UserID)
+			if idx < 0 {
+				return nil
+			}
+			copy(rm.Members[idx:], rm.Members[idx+1:])
+			rm.Members[len(rm.Members)-1] = nil
+			rm.Members = rm.Members[:len(rm.Members)-1]
+		}
+	}
+	return err
+}
+
+func (rm *MembersReadModel) Reduce() (err error) {
+	for _, member := range rm.Members {
+		err = member.Reduce()
+		if err != nil {
+			return err
+		}
+	}
+	return rm.ReadModel.Reduce()
+}