upstream-corteza/server/pkg/corredor/util.go

package corredor

import (
	"context"
	"encoding/json"
	"fmt"
	"net/http"

	"github.com/cortezaproject/corteza/server/pkg/eventbus"
	"github.com/cortezaproject/corteza/server/pkg/slice"
)

type (
	automationListSetPayload struct {
		Filter Filter    `json:"filter"`
		Set    []*Script `json:"set"`
	}
)

// mapExplicitTriggers scans for explicit event types from the list of script's triggers
// and returns a hash map with resources from these explicit triggers
func mapExplicitTriggers(script *ServerScript) map[string]bool {
	var (
		hash = make(map[string]bool)
	)

	for i := range script.Triggers {
		// We're modifying trigger in the loop,
		// so let's make a copy we can play with
		trigger := script.Triggers[i]

		if len(slice.IntersectStrings(trigger.EventTypes, explicitEventTypes)) > 0 {
			for _, res := range trigger.ResourceTypes {
				hash[res] = true
			}
		}
	}

	return hash
}

// converts trigger's constraint to eventbus' constraint options
func triggerToHandlerOps(t *Trigger) (oo []eventbus.HandlerRegOp, err error) {
	if len(t.ResourceTypes) == 0 {
		return nil, fmt.Errorf("cannot generate event handler without at least one resource")
	}

	if len(t.EventTypes) == 0 {
		return nil, fmt.Errorf("cannot generate event handler without at least one events")
	}

	// Make a copy of event types slice so that we do not modify it
	types := slice.PluckString(t.EventTypes, onManualEventType)

	// If no other event types are left on the trigger,
	// no need for procede
	if len(types) == 0 && len(t.EventTypes) > 0 {
		return
	}

	oo = append(oo, eventbus.For(t.ResourceTypes...))
	oo = append(oo, eventbus.On(types...))

	if cc, err := constraintsToHandlerOps(t.Constraints); err != nil {
		return nil, err
	} else {
		oo = append(oo, cc...)
	}

	return
}

func constraintsToHandlerOps(cc []*TConstraint) (oo []eventbus.HandlerRegOp, err error) {
	for _, raw := range cc {
		if c, err := eventbus.ConstraintMaker(raw.Name, raw.Op, raw.Value...); err != nil {
			return nil, fmt.Errorf("cannot generate constraints: %w", err)
		} else {
			oo = append(oo, eventbus.Constraint(c))
		}
	}

	return
}

// encode adds entry (with json encoded value) to hash map
// used to prepare data for transmission
func encodeArguments(args map[string]string, key string, val interface{}) (err error) {
	var tmp []byte

	if tmp, err = json.Marshal(val); err != nil {
		return
	}

	args[key] = string(tmp)
	return
}

// GenericListHandler returns filtered list of scripts
func GenericListHandler(ctx context.Context, svc *service, f Filter, resourcePrefix string) (p *automationListSetPayload, err error) {
	if f.ExcludeInvalid {
		// resource prefix filtering is only applicable when we want to
		// exclude invalid scripts, because invalid scripts do not have
		// (usually, depends at what level error occurred) information
		// about triggers and resources
		f.procRTPrefixes(resourcePrefix)
	}

	p = &automationListSetPayload{}
	p.Set, p.Filter, err = svc.Find(ctx, f)
	return p, err
}

func GenericBundleHandler(ctx context.Context, svc *service, bundleName, bundleType, ext string) (interface{}, error) {
	return func(w http.ResponseWriter, req *http.Request) {
		// Serve bundle directly for now
		bundle := svc.GetBundle(ctx, bundleName, bundleType)
		if bundle == nil {
			w.WriteHeader(http.StatusNoContent)
			return
		}

		w.WriteHeader(http.StatusOK)
		w.Write([]byte(bundle.Code))
	}, nil
}