zitadel/internal/org/repository/eventsourcing/model/domain.go

package model

import (
	"encoding/json"
	"github.com/caos/zitadel/internal/errors"
	es_models "github.com/caos/zitadel/internal/eventstore/models"
	"github.com/caos/zitadel/internal/org/model"
)

type OrgDomain struct {
	es_models.ObjectRoot `json:"-"`

	Domain   string `json:"domain"`
	Verified bool   `json:"-"`
	Primary  bool   `json:"-"`
}

func GetDomain(domains []*OrgDomain, domain string) (int, *OrgDomain) {
	for i, d := range domains {
		if d.Domain == domain {
			return i, d
		}
	}
	return -1, nil
}

func (o *Org) appendAddDomainEvent(event *es_models.Event) error {
	domain := new(OrgDomain)
	err := domain.SetData(event)
	if err != nil {
		return err
	}
	domain.ObjectRoot.CreationDate = event.CreationDate
	o.Domains = append(o.Domains, domain)
	return nil
}

func (o *Org) appendRemoveDomainEvent(event *es_models.Event) error {
	domain := new(OrgDomain)
	err := domain.SetData(event)
	if err != nil {
		return err
	}
	if i, r := GetDomain(o.Domains, domain.Domain); r != nil {
		o.Domains[i] = o.Domains[len(o.Domains)-1]
		o.Domains[len(o.Domains)-1] = nil
		o.Domains = o.Domains[:len(o.Domains)-1]
	}
	return nil
}

func (o *Org) appendVerifyDomainEvent(event *es_models.Event) error {
	domain := new(OrgDomain)
	err := domain.SetData(event)
	if err != nil {
		return err
	}
	if i, d := GetDomain(o.Domains, domain.Domain); d != nil {
		d.Verified = true
		o.Domains[i] = d
	}
	return nil
}

func (o *Org) appendPrimaryDomainEvent(event *es_models.Event) error {
	domain := new(OrgDomain)
	err := domain.SetData(event)
	if err != nil {
		return err
	}
	for _, d := range o.Domains {
		d.Primary = false
		if d.Domain == domain.Domain {
			d.Primary = true
		}
	}
	return nil
}

func (m *OrgDomain) SetData(event *es_models.Event) error {
	err := json.Unmarshal(event.Data, m)
	if err != nil {
		return errors.ThrowInternal(err, "EVENT-Hz7Mb", "unable to unmarshal data")
	}
	return nil
}

func OrgDomainsFromModel(domains []*model.OrgDomain) []*OrgDomain {
	convertedDomainss := make([]*OrgDomain, len(domains))
	for i, m := range domains {
		convertedDomainss[i] = OrgDomainFromModel(m)
	}
	return convertedDomainss
}

func OrgDomainFromModel(domain *model.OrgDomain) *OrgDomain {
	return &OrgDomain{
		ObjectRoot: domain.ObjectRoot,
		Domain:     domain.Domain,
		Verified:   domain.Verified,
		Primary:    domain.Primary,
	}
}

func OrgDomainsToModel(domains []*OrgDomain) []*model.OrgDomain {
	convertedDomains := make([]*model.OrgDomain, len(domains))
	for i, m := range domains {
		convertedDomains[i] = OrgDomainToModel(m)
	}
	return convertedDomains
}

func OrgDomainToModel(domain *OrgDomain) *model.OrgDomain {
	return &model.OrgDomain{
		ObjectRoot: domain.ObjectRoot,
		Domain:     domain.Domain,
		Verified:   domain.Verified,
		Primary:    domain.Primary,
	}
}