mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-15 04:18:01 +00:00
710652ef24
* feat: project role remove * feat: search queries * feat: search queries * feat: cascade remove/change project role * fix: comment in project grant * fix: remove projecr grant * fix: only search usergrants of my org * fix: delete usergrants * fix: delete usergrants * fix: check if role exists on project grant * feat: bulk add project role * fix: tests * fix: update user grants on project update * fix: return roles * feat: add resourceowner name on project grants * fix: migration number * fix: tests * fix: generate protos * fix: some unnecessary code
151 lines
3.8 KiB
Go
151 lines
3.8 KiB
Go
package model
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/caos/logging"
|
|
es_models "github.com/caos/zitadel/internal/eventstore/models"
|
|
"github.com/caos/zitadel/internal/project/model"
|
|
"reflect"
|
|
)
|
|
|
|
type ProjectGrant struct {
|
|
es_models.ObjectRoot
|
|
State int32 `json:"-"`
|
|
GrantID string `json:"grantId,omitempty"`
|
|
GrantedOrgID string `json:"grantedOrgId,omitempty"`
|
|
RoleKeys []string `json:"roleKeys,omitempty"`
|
|
Members []*ProjectGrantMember `json:"-"`
|
|
}
|
|
|
|
type ProjectGrantID struct {
|
|
es_models.ObjectRoot
|
|
GrantID string `json:"grantId"`
|
|
}
|
|
|
|
func GetProjectGrant(grants []*ProjectGrant, id string) (int, *ProjectGrant) {
|
|
for i, g := range grants {
|
|
if g.GrantID == id {
|
|
return i, g
|
|
}
|
|
}
|
|
return -1, nil
|
|
}
|
|
|
|
func GetProjectGrantByOrgID(grants []*ProjectGrant, resourceOwner string) (int, *ProjectGrant) {
|
|
for i, g := range grants {
|
|
if g.GrantedOrgID == resourceOwner {
|
|
return i, g
|
|
}
|
|
}
|
|
return -1, nil
|
|
}
|
|
|
|
func (g *ProjectGrant) Changes(changed *ProjectGrant) map[string]interface{} {
|
|
changes := make(map[string]interface{}, 1)
|
|
changes["grantId"] = g.GrantID
|
|
if !reflect.DeepEqual(g.RoleKeys, changed.RoleKeys) {
|
|
changes["roleKeys"] = changed.RoleKeys
|
|
}
|
|
return changes
|
|
}
|
|
|
|
func GrantsToModel(grants []*ProjectGrant) []*model.ProjectGrant {
|
|
convertedGrants := make([]*model.ProjectGrant, len(grants))
|
|
for i, g := range grants {
|
|
convertedGrants[i] = GrantToModel(g)
|
|
}
|
|
return convertedGrants
|
|
}
|
|
|
|
func GrantsFromModel(grants []*model.ProjectGrant) []*ProjectGrant {
|
|
convertedGrants := make([]*ProjectGrant, len(grants))
|
|
for i, g := range grants {
|
|
convertedGrants[i] = GrantFromModel(g)
|
|
}
|
|
return convertedGrants
|
|
}
|
|
|
|
func GrantFromModel(grant *model.ProjectGrant) *ProjectGrant {
|
|
members := GrantMembersFromModel(grant.Members)
|
|
return &ProjectGrant{
|
|
ObjectRoot: grant.ObjectRoot,
|
|
GrantID: grant.GrantID,
|
|
GrantedOrgID: grant.GrantedOrgID,
|
|
State: int32(grant.State),
|
|
RoleKeys: grant.RoleKeys,
|
|
Members: members,
|
|
}
|
|
}
|
|
|
|
func GrantToModel(grant *ProjectGrant) *model.ProjectGrant {
|
|
members := GrantMembersToModel(grant.Members)
|
|
return &model.ProjectGrant{
|
|
ObjectRoot: grant.ObjectRoot,
|
|
GrantID: grant.GrantID,
|
|
GrantedOrgID: grant.GrantedOrgID,
|
|
State: model.ProjectGrantState(grant.State),
|
|
RoleKeys: grant.RoleKeys,
|
|
Members: members,
|
|
}
|
|
}
|
|
|
|
func (p *Project) appendAddGrantEvent(event *es_models.Event) error {
|
|
grant := new(ProjectGrant)
|
|
err := grant.getData(event)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
grant.ObjectRoot.CreationDate = event.CreationDate
|
|
p.Grants = append(p.Grants, grant)
|
|
return nil
|
|
}
|
|
|
|
func (p *Project) appendChangeGrantEvent(event *es_models.Event) error {
|
|
grant := new(ProjectGrant)
|
|
err := grant.getData(event)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if i, g := GetProjectGrant(p.Grants, grant.GrantID); g != nil {
|
|
p.Grants[i].getData(event)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *Project) appendGrantStateEvent(event *es_models.Event, state model.ProjectGrantState) error {
|
|
grant := new(ProjectGrant)
|
|
err := grant.getData(event)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if i, g := GetProjectGrant(p.Grants, grant.GrantID); g != nil {
|
|
g.State = int32(state)
|
|
p.Grants[i] = g
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *Project) appendRemoveGrantEvent(event *es_models.Event) error {
|
|
grant := new(ProjectGrant)
|
|
err := grant.getData(event)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if i, g := GetProjectGrant(p.Grants, grant.GrantID); g != nil {
|
|
p.Grants[i] = p.Grants[len(p.Grants)-1]
|
|
p.Grants[len(p.Grants)-1] = nil
|
|
p.Grants = p.Grants[:len(p.Grants)-1]
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (g *ProjectGrant) getData(event *es_models.Event) error {
|
|
g.ObjectRoot.AppendEvent(event)
|
|
if err := json.Unmarshal(event.Data, g); err != nil {
|
|
logging.Log("EVEN-4h6gd").WithError(err).Error("could not unmarshal event data")
|
|
return err
|
|
}
|
|
return nil
|
|
}
|