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

package model

import (
	"encoding/json"
	"time"

	"github.com/caos/logging"
	caos_errs "github.com/caos/zitadel/internal/errors"
	"github.com/caos/zitadel/internal/eventstore/models"
	"github.com/caos/zitadel/internal/usergrant/model"
	es_model "github.com/caos/zitadel/internal/usergrant/repository/eventsourcing/model"
	"github.com/lib/pq"
)

const (
	UserGrantKeyID            = "id"
	UserGrantKeyUserID        = "user_id"
	UserGrantKeyProjectID     = "project_id"
	UserGrantKeyResourceOwner = "resource_owner"
	UserGrantKeyState         = "state"
)

type UserGrantView struct {
	ID            string         `json:"-" gorm:"column:id;primary_key"`
	ResourceOwner string         `json:"-" gorm:"resource_owner"`
	UserID        string         `json:"userId" gorm:"user_id"`
	ProjectID     string         `json:"projectId" gorm:"column:project_id"`
	UserName      string         `json:"-" gorm:"column:user_name"`
	FirstName     string         `json:"-" gorm:"column:first_name"`
	LastName      string         `json:"-" gorm:"column:last_name"`
	Email         string         `json:"-" gorm:"column:email"`
	ProjectName   string         `json:"-" gorm:"column:project_name"`
	OrgName       string         `json:"-" gorm:"column:org_name"`
	OrgDomain     string         `json:"-" gorm:"column:org_domain"`
	RoleKeys      pq.StringArray `json:"roleKeys" gorm:"column:role_keys"`

	CreationDate time.Time `json:"-" gorm:"column:creation_date"`
	ChangeDate   time.Time `json:"-" gorm:"column:change_date"`
	State        int32     `json:"-" gorm:"column:grant_state"`

	Sequence uint64 `json:"-" gorm:"column:sequence"`
}

func UserGrantFromModel(grant *model.UserGrantView) *UserGrantView {
	return &UserGrantView{
		ID:            grant.ID,
		ResourceOwner: grant.ResourceOwner,
		UserID:        grant.UserID,
		ProjectID:     grant.ProjectID,
		ChangeDate:    grant.ChangeDate,
		CreationDate:  grant.CreationDate,
		State:         int32(grant.State),
		UserName:      grant.UserName,
		FirstName:     grant.FirstName,
		LastName:      grant.LastName,
		Email:         grant.Email,
		ProjectName:   grant.ProjectName,
		OrgName:       grant.OrgName,
		OrgDomain:     grant.OrgDomain,
		RoleKeys:      grant.RoleKeys,
		Sequence:      grant.Sequence,
	}
}

func UserGrantToModel(grant *UserGrantView) *model.UserGrantView {
	return &model.UserGrantView{
		ID:            grant.ID,
		ResourceOwner: grant.ResourceOwner,
		UserID:        grant.UserID,
		ProjectID:     grant.ProjectID,
		ChangeDate:    grant.ChangeDate,
		CreationDate:  grant.CreationDate,
		State:         model.UserGrantState(grant.State),
		UserName:      grant.UserName,
		FirstName:     grant.FirstName,
		LastName:      grant.LastName,
		Email:         grant.Email,
		ProjectName:   grant.ProjectName,
		OrgName:       grant.OrgName,
		OrgDomain:     grant.OrgDomain,
		RoleKeys:      grant.RoleKeys,
		Sequence:      grant.Sequence,
	}
}

func UserGrantsToModel(grants []*UserGrantView) []*model.UserGrantView {
	result := make([]*model.UserGrantView, len(grants))
	for i, g := range grants {
		result[i] = UserGrantToModel(g)
	}
	return result
}

func (g *UserGrantView) AppendEvent(event *models.Event) (err error) {
	g.ChangeDate = event.CreationDate
	g.Sequence = event.Sequence
	switch event.Type {
	case es_model.UserGrantAdded:
		g.State = int32(model.USERGRANTSTATE_ACTIVE)
		g.CreationDate = event.CreationDate
		g.setRootData(event)
		err = g.setData(event)
	case es_model.UserGrantChanged:
		err = g.setData(event)
	case es_model.UserGrantDeactivated:
		g.State = int32(model.USERGRANTSTATE_INACTIVE)
	case es_model.UserGrantReactivated:
		g.State = int32(model.USERGRANTSTATE_ACTIVE)
	}
	return err
}

func (u *UserGrantView) setRootData(event *models.Event) {
	u.ID = event.AggregateID
	u.ResourceOwner = event.ResourceOwner
}

func (u *UserGrantView) setData(event *models.Event) error {
	if err := json.Unmarshal(event.Data, u); err != nil {
		logging.Log("EVEN-l9sw4").WithError(err).Error("could not unmarshal event data")
		return caos_errs.ThrowInternal(nil, "MODEL-7xhke", "could not unmarshal data")
	}
	return nil
}