upstream-corteza/server/pkg/envoy/resource/compose_chart.go

package resource

import (
	"fmt"
	"strconv"

	"github.com/cortezaproject/corteza/server/compose/types"
)

type (
	// ComposeChart represents a ComposeChart
	ComposeChart struct {
		*base
		Res *types.Chart

		// Might keep track of related namespace
		RefNs   *Ref
		RefMods RefSet
	}
)

func NewComposeChart(res *types.Chart, nsRef *Ref, mmRef RefSet) *ComposeChart {
	r := &ComposeChart{
		base:    &base{},
		RefMods: make(RefSet, len(mmRef)),
	}
	r.SetResourceType(types.ChartResourceType)
	r.Res = res

	r.AddIdentifier(identifiers(res.Handle, res.Name, res.ID)...)

	r.RefNs = r.addRef(nsRef)
	for i, mRef := range mmRef {
		r.RefMods[i] = r.addRef(mRef).Constraint(r.RefNs)
	}

	// Initial timestamps
	r.SetTimestamps(MakeTimestampsCUDA(&res.CreatedAt, res.UpdatedAt, res.DeletedAt, nil))

	return r
}

func (r *ComposeChart) Resource() interface{} {
	return r.Res
}

func (r *ComposeChart) ReRef(old RefSet, new RefSet) {
	r.base.ReRef(old, new)

	// Additional references
	for _, n := range new {
		if n.ResourceType == types.NamespaceResourceType {
			r.RefNs = MakeRef(types.NamespaceResourceType, n.Identifiers)
		}
	}

	for i, o := range old {
		if o.ResourceType == types.ModuleResourceType {
			r.RefMods = r.RefMods.replaceRef(o, new[i])
		}
	}
}

func (r *ComposeChart) SysID() uint64 {
	return r.Res.ID
}

func (r *ComposeChart) resourceParts(tpl string) (resource string, ref *Ref, path []*Ref) {
	ref = r.Ref().Constraint(r.RefNs)
	path = []*Ref{r.RefNs}
	resource = fmt.Sprintf(types.ChartRbacResourceTpl(), types.ChartResourceType, r.RefNs.Identifiers.First(), firstOkString(strconv.FormatUint(r.Res.ID, 10), r.Res.Handle))

	return
}

func (r *ComposeChart) RBACParts() (resource string, ref *Ref, path []*Ref) {
	return r.resourceParts(types.ChartRbacResourceTpl())
}

func (r *ComposeChart) ResourceTranslationParts() (resource string, ref *Ref, path []*Ref) {
	return r.resourceParts(types.ChartResourceTranslationTpl())
}

func (r *ComposeChart) encodeTranslations() ([]*ResourceTranslation, error) {
	return nil, nil
}

// FindComposeChart looks for the chart in the resources
func FindComposeChart(rr InterfaceSet, ii Identifiers) (ch *types.Chart) {
	var chRes *ComposeChart

	rr.Walk(func(r Interface) error {
		cr, ok := r.(*ComposeChart)
		if !ok {
			return nil
		}

		if cr.Identifiers().HasAny(ii) {
			chRes = cr
		}
		return nil
	})

	// Found it
	if chRes != nil {
		return chRes.Res
	}

	return nil
}

func ComposeChartErrUnresolved(ii Identifiers) error {
	return fmt.Errorf("compose chart unresolved %v", ii.StringSlice())
}