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

package resource

import (
	"fmt"
	"strconv"

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

type (
	// ComposePage represents a ComposePage
	ComposePage struct {
		*base
		Res *types.Page

		RefNs     *Ref
		RefMod    *Ref
		RefParent *Ref

		WfRefs    RefSet
		ModRefs   RefSet
		RefCharts RefSet

		BlockRefs map[int]RefSet
	}
)

func NewComposePage(pg *types.Page, nsRef, modRef, parentRef *Ref) *ComposePage {
	r := &ComposePage{
		base:      &base{},
		WfRefs:    make(RefSet, 0, 10),
		ModRefs:   make(RefSet, 0, 10),
		RefCharts: make(RefSet, 0, 10),
		BlockRefs: make(map[int]RefSet),
	}
	r.SetResourceType(types.PageResourceType)
	r.Res = pg

	r.AddIdentifier(identifiers(pg.Handle, pg.Title, pg.ID)...)

	r.RefNs = r.addRef(nsRef)
	if modRef != nil {
		r.RefMod = r.addRef(modRef).Constraint(r.RefNs)
	}

	if parentRef != nil {
		r.RefParent = r.addRef(parentRef).Constraint(r.RefNs)
	}

	var ref *Ref
	for i, b := range pg.Blocks {
		switch b.Kind {
		case "RecordList":
			ref = r.pbRecordList(b.Options)
			if ref != nil {
				r.addRef(ref)
				r.BlockRefs[i] = append(r.BlockRefs[i], ref)
				r.ModRefs = append(r.ModRefs, ref)
			}

		case "Automation":
			bb, _ := b.Options["buttons"].([]interface{})
			for _, b := range bb {
				button, _ := b.(map[string]interface{})
				ref = r.pbAutomation(button)
				if ref != nil {
					r.addRef(ref)
					r.BlockRefs[i] = append(r.BlockRefs[i], ref)
					r.WfRefs = append(r.WfRefs, ref)
				}
			}

		case "RecordOrganizer":
			ref = r.pbRecordList(b.Options)
			if ref != nil {
				r.addRef(ref)
				r.BlockRefs[i] = append(r.BlockRefs[i], ref)
				r.ModRefs = append(r.ModRefs, ref)
			}

		case "Chart":
			ref = r.pbChart(b.Options)
			if ref != nil {
				r.addRef(ref)
				r.BlockRefs[i] = append(r.BlockRefs[i], ref)
				r.RefCharts = append(r.RefCharts, ref)
			}

		case "Calendar":
			ff, _ := b.Options["feeds"].([]interface{})
			for _, f := range ff {
				feed, _ := f.(map[string]interface{})
				fOpts, _ := (feed["options"]).(map[string]interface{})

				ref = r.pbCalendar(fOpts)
				if ref != nil {
					r.addRef(ref)
					r.BlockRefs[i] = append(r.BlockRefs[i], ref)
					r.ModRefs = append(r.ModRefs, ref)
				}
			}

		case "Metric":
			mm, _ := b.Options["metrics"].([]interface{})
			for _, m := range mm {
				mops, _ := m.(map[string]interface{})
				ref = r.pbMetric(mops)
				if ref != nil {
					r.addRef(ref)
					r.BlockRefs[i] = append(r.BlockRefs[i], ref)
					r.ModRefs = append(r.ModRefs, ref)
				}
			}

		case "Comment":
			ref = r.pbComment(b.Options)
			if ref != nil {
				r.addRef(ref.Constraint(r.RefNs))
				r.BlockRefs[i] = append(r.BlockRefs[i], ref)
				r.ModRefs = append(r.ModRefs, ref)
			}
		}
	}

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

	return r
}

func UnpackComposePage(p *types.Page) (*types.Page, *Ref, *Ref) {
	var modRef *Ref
	var parentRef *Ref

	if p.ModuleID != 0 {
		modRef = MakeModuleRef(p.ModuleID, "", "")
	}
	if p.SelfID != 0 {
		parentRef = MakePageRef(p.SelfID, "", "")
	}

	return p, modRef, parentRef
}

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

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

	for _, n := range new {
		switch n.ResourceType {
		case types.NamespaceResourceType:
			r.RefNs = MakeRef(types.NamespaceResourceType, n.Identifiers)
		case types.ModuleResourceType:
			r.RefMod = MakeRef(types.ModuleResourceType, n.Identifiers)
		case types.PageResourceType:
			r.RefParent = MakeRef(types.PageResourceType, n.Identifiers)
		}
	}
}

func (r *ComposePage) Prune(ref *Ref) {
	var auxRef *Ref

outer:
	for i := len(r.Res.Blocks) - 1; i >= 0; i-- {
		b := r.Res.Blocks[i]

		switch b.Kind {
		// Implement the rest when support is needed
		case "Automation":
			if b.Options["buttons"] == nil {
				// In case the block isn't connected to a workflow (placeholder, script)
				if auxRef == nil {
					r.removeBlock(i)
					continue outer
				}
			} else {
				bb, _ := b.Options["buttons"].([]interface{})
				for _, b := range bb {
					button, _ := b.(map[string]interface{})
					auxRef = r.pbAutomation(button)

					// In case the block isn't connected to a workflow (placeholder, script)
					if auxRef == nil {
						r.removeBlock(i)
						continue outer
					}

					// In case we are removing it
					if auxRef.equals(ref) {
						r.ReplaceRef(ref, nil)
						r.WfRefs = r.WfRefs.replaceRef(ref, nil)
						r.removeBlock(i)
						continue outer
					}
				}
			}
		}
	}
}

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

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

	return
}

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

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

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

// FindComposePage looks for the page in the resources
func FindComposePage(rr InterfaceSet, ii Identifiers) (pg *types.Page) {
	var pgRes *ComposePage

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

		if pr.Identifiers().HasAny(ii) {
			pgRes = pr
		}
		return nil
	})

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

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

// Page block utilities
func (r *ComposePage) removeBlock(i int) {
	// do the swap remove thing
	r.Res.Blocks[i] = r.Res.Blocks[len(r.Res.Blocks)-1]
	r.Res.Blocks = r.Res.Blocks[:len(r.Res.Blocks)-1]

	// correct block refs.
	// +1 here because it's already cut
	r.BlockRefs[i] = r.BlockRefs[len(r.Res.Blocks)]
	delete(r.BlockRefs, len(r.Res.Blocks))
}

func (r *ComposePage) optString(opt map[string]interface{}, kk ...string) string {
	for _, k := range kk {
		if vr, has := opt[k]; has {
			v, _ := vr.(string)
			return v
		}
	}
	return ""
}

func (r *ComposePage) pbRecordList(opt map[string]interface{}) (out *Ref) {
	id := r.optString(opt, "module", "moduleID")
	if id == "" || id == "0" {
		return
	}

	return MakeRef(types.ModuleResourceType, MakeIdentifiers(id)).Constraint(r.RefNs)
}

func (r *ComposePage) pbComment(opt map[string]interface{}) (out *Ref) {
	id := r.optString(opt, "module", "moduleID")
	if id == "" || id == "0" {
		return
	}

	return MakeRef(types.ModuleResourceType, MakeIdentifiers(id)).Constraint(r.RefNs)
}

func (r *ComposePage) pbAutomation(opt map[string]interface{}) (out *Ref) {
	id := r.optString(opt, "workflow", "workflowID")
	if id == "" || id == "0" {
		return
	}

	return MakeRef(automationTypes.WorkflowResourceType, MakeIdentifiers(id))
}

func (r *ComposePage) pbRecordOrganizer(opt map[string]interface{}) (out *Ref) {
	id := r.optString(opt, "module", "moduleID")
	if id == "" || id == "0" {
		return
	}

	return MakeRef(types.ModuleResourceType, MakeIdentifiers(id)).Constraint(r.RefNs)
}

func (r *ComposePage) pbChart(opt map[string]interface{}) (out *Ref) {
	id := r.optString(opt, "chart", "chartID")
	if id == "" || id == "0" {
		return
	}

	return MakeRef(types.ChartResourceType, MakeIdentifiers(id)).Constraint(r.RefNs)
}

func (r *ComposePage) pbCalendar(opt map[string]interface{}) (out *Ref) {
	id := r.optString(opt, "module", "moduleID")
	if id == "" || id == "0" {
		return
	}

	return MakeRef(types.ModuleResourceType, MakeIdentifiers(id)).Constraint(r.RefNs)
}

func (r *ComposePage) pbMetric(opt map[string]interface{}) (out *Ref) {
	id := r.optString(opt, "module", "moduleID")
	if id == "" || id == "0" {
		return
	}

	return MakeRef(types.ModuleResourceType, MakeIdentifiers(id)).Constraint(r.RefNs)
}