zitadel/internal/usergrant/repository/eventsourcing/model/user_grant.go

package model

import (
	"encoding/json"
	"github.com/caos/logging"
	caos_errs "github.com/caos/zitadel/internal/errors"
	es_models "github.com/caos/zitadel/internal/eventstore/models"
	"github.com/caos/zitadel/internal/usergrant/model"
	"reflect"
)

const (
	UserGrantVersion = "v1"
)

type UserGrant struct {
	es_models.ObjectRoot

	State     int32    `json:"-"`
	UserID    string   `json:"userId,omitempty"`
	ProjectID string   `json:"projectId,omitempty"`
	RoleKeys  []string `json:"roleKeys,omitempty"`
}

type UserGrantID struct {
	es_models.ObjectRoot
	GrantID string `json:"grantId"`
}

func (g *UserGrant) Changes(changed *UserGrant) map[string]interface{} {
	changes := make(map[string]interface{}, 1)
	if !reflect.DeepEqual(g.RoleKeys, changed.RoleKeys) {
		changes["roleKeys"] = changed.RoleKeys
	}
	return changes
}

func UserGrantFromModel(grant *model.UserGrant) *UserGrant {
	return &UserGrant{
		ObjectRoot: grant.ObjectRoot,
		UserID:     grant.UserID,
		ProjectID:  grant.ProjectID,
		State:      int32(grant.State),
		RoleKeys:   grant.RoleKeys,
	}
}

func UserGrantToModel(grant *UserGrant) *model.UserGrant {
	return &model.UserGrant{
		ObjectRoot: grant.ObjectRoot,
		UserID:     grant.UserID,
		ProjectID:  grant.ProjectID,
		State:      model.UserGrantState(grant.State),
		RoleKeys:   grant.RoleKeys,
	}
}

func (g *UserGrant) AppendEvents(events ...*es_models.Event) error {
	for _, event := range events {
		if err := g.AppendEvent(event); err != nil {
			return err
		}
	}
	return nil
}

func (g *UserGrant) AppendEvent(event *es_models.Event) error {
	g.ObjectRoot.AppendEvent(event)
	switch event.Type {
	case UserGrantAdded,
		UserGrantChanged:
		return g.setData(event)
	case UserGrantDeactivated:
		g.appendGrantStateEvent(model.USERGRANTSTATE_INACTIVE)
	case UserGrantReactivated:
		g.appendGrantStateEvent(model.USERGRANTSTATE_ACTIVE)
	case UserGrantRemoved:
		g.appendGrantStateEvent(model.USERGRANTSTATE_REMOVED)
	}
	return nil
}

func (g *UserGrant) appendGrantStateEvent(state model.UserGrantState) {
	g.State = int32(state)
}

func (g *UserGrant) setData(event *es_models.Event) error {
	if err := json.Unmarshal(event.Data, g); err != nil {
		logging.Log("EVEN-lso9x").WithError(err).Error("could not unmarshal event data")
		return caos_errs.ThrowInternal(err, "MODEL-o0se3", "could not unmarshal event")
	}
	return nil
}