snapshot of https://github.com/cortezaproject/corteza.git (2024.9.x, 2026-09-18; blobs > 131072B omitted)

server/pkg/corredor/util.go

@@ -0,0 +1,125 @@
+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
+}