mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-22 15:57:34 +00:00
94 lines
2.3 KiB
Go
94 lines
2.3 KiB
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
|
||
|
}
|