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

server/automation/rest/session.go

@@ -0,0 +1,161 @@
+package rest
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/cortezaproject/corteza/server/automation/rest/request"
+	"github.com/cortezaproject/corteza/server/automation/service"
+	"github.com/cortezaproject/corteza/server/automation/types"
+	"github.com/cortezaproject/corteza/server/compose/automation"
+	cmpService "github.com/cortezaproject/corteza/server/compose/service"
+	cmpTypes "github.com/cortezaproject/corteza/server/compose/types"
+	"github.com/cortezaproject/corteza/server/pkg/api"
+	"github.com/cortezaproject/corteza/server/pkg/auth"
+	"github.com/cortezaproject/corteza/server/pkg/expr"
+	"github.com/cortezaproject/corteza/server/pkg/filter"
+	"github.com/cortezaproject/corteza/server/pkg/wfexec"
+)
+
+type (
+	Session struct {
+		svc sessionService
+
+		// cross-link with compose service to load module on resolved records
+		svcModule interface {
+			FindByID(ctx context.Context, namespaceID, moduleID uint64) (*cmpTypes.Module, error)
+		}
+	}
+
+	sessionService interface {
+		Search(ctx context.Context, filter types.SessionFilter) (types.SessionSet, types.SessionFilter, error)
+		LookupByID(ctx context.Context, sessionID uint64) (*types.Session, error)
+		Resume(sessionID, stateID uint64, i auth.Identifiable, input *expr.Vars) error
+		PendingPrompts(context.Context) []*wfexec.PendingPrompt
+		Cancel(context.Context, uint64) error
+	}
+
+	sessionSetPayload struct {
+		Filter types.SessionFilter `json:"filter"`
+		Set    []*sessionSetItem   `json:"set"`
+	}
+
+	sessionSetItem struct {
+		*types.Session
+
+		// Make sure stacktrace is not included in the list
+		Stacktrace *struct{} `json:"stacktrace,omitempty"`
+		// Make sure output values are not included in the list
+		Output *struct{} `json:"output,omitempty"`
+		// Make sure input values are not included in the list
+		Input *struct{} `json:"input,omitempty"`
+	}
+)
+
+func (Session) New() *Session {
+	ctrl := &Session{}
+	ctrl.svc = service.DefaultSession
+	ctrl.svcModule = cmpService.DefaultModule
+	return ctrl
+}
+
+func (ctrl Session) List(ctx context.Context, r *request.SessionList) (interface{}, error) {
+	var (
+		err error
+		f   = types.SessionFilter{
+			WorkflowID:   r.WorkflowID,
+			SessionID:    r.SessionID,
+			CreatedBy:    r.CreatedBy,
+			EventType:    r.EventType,
+			ResourceType: r.ResourceType,
+			Completed:    filter.State(r.Completed),
+			Status:       r.Status,
+		}
+	)
+
+	if f.Paging, err = filter.NewPaging(r.Limit, r.PageCursor); err != nil {
+		return nil, err
+	}
+
+	// fixes issue with sorting of status column and pagination
+	// need to improve on cursor for this
+	if f.Paging.PageCursor != nil {
+		for _, status := range r.Status {
+			f.Paging.PageCursor.Set("status", status, false)
+		}
+	}
+
+	f.IncTotal = r.IncTotal
+
+	if f.Sorting, err = filter.NewSorting(r.Sort); err != nil {
+		return nil, err
+	}
+
+	set, filter, err := ctrl.svc.Search(ctx, f)
+	return ctrl.makeFilterPayload(ctx, set, filter, err)
+}
+
+func (ctrl Session) Read(ctx context.Context, r *request.SessionRead) (interface{}, error) {
+	return ctrl.svc.LookupByID(ctx, r.SessionID)
+}
+
+func (ctrl Session) Cancel(ctx context.Context, r *request.SessionCancel) (interface{}, error) {
+	return true, ctrl.svc.Cancel(ctx, r.SessionID)
+}
+
+func (ctrl Session) ListPrompts(ctx context.Context, r *request.SessionListPrompts) (interface{}, error) {
+	return struct {
+		Set []*wfexec.PendingPrompt `json:"set"`
+	}{
+		Set: ctrl.svc.PendingPrompts(ctx),
+	}, nil
+}
+
+func (ctrl Session) ResumeState(ctx context.Context, r *request.SessionResumeState) (_ interface{}, err error) {
+	if r.Input != nil {
+		if err = r.Input.ResolveTypes(service.Registry().Type); err != nil {
+			return nil, err
+		}
+	}
+
+	// Now when all types are resolved we have to load modules and link them to records
+	//
+	// Very naive approach for now.
+	//
+	// @note copied from https://github.com/cortezaproject/corteza/blob/2023.9.x/server/automation/rest/workflow.go#L189
+	//       copied to reduce the need for some dependency; should be good enough for now
+	r.Input.Each(func(k string, v expr.TypedValue) error {
+		switch c := v.(type) {
+		case *automation.ComposeRecord:
+			rec := c.GetValue()
+			if rec == nil {
+				return nil
+			}
+
+			mod, err := ctrl.svcModule.FindByID(ctx, rec.NamespaceID, rec.ModuleID)
+			if err != nil {
+				return fmt.Errorf("failed to resolve ComposeRecord type: %w", err)
+			}
+			c.GetValue().SetModule(mod)
+		}
+
+		return nil
+	})
+
+	return api.OK(), ctrl.svc.Resume(r.SessionID, r.StateID, auth.GetIdentityFromContext(ctx), r.Input)
+}
+
+func (ctrl Session) makeFilterPayload(ctx context.Context, ss types.SessionSet, f types.SessionFilter, err error) (*sessionSetPayload, error) {
+	if err != nil {
+		return nil, err
+	}
+
+	out := &sessionSetPayload{Filter: f, Set: make([]*sessionSetItem, len(ss))}
+
+	for i, s := range ss {
+		s.Stacktrace = nil
+		out.Set[i] = &sessionSetItem{Session: s}
+	}
+
+	return out, nil
+}