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

server/pkg/apigw/registry/registry.go

@@ -0,0 +1,83 @@
+package registry
+
+import (
+	"fmt"
+	"net/http"
+
+	"github.com/cortezaproject/corteza/server/automation/service"
+	"github.com/cortezaproject/corteza/server/pkg/apigw/filter"
+	"github.com/cortezaproject/corteza/server/pkg/apigw/filter/proxy"
+	"github.com/cortezaproject/corteza/server/pkg/apigw/types"
+)
+
+type (
+	Registry struct {
+		cfg types.Config
+		h   map[string]types.Handler
+	}
+
+	secureStorageTodo struct{}
+)
+
+func NewRegistry(cfg types.Config) *Registry {
+	return &Registry{
+		h:   map[string]types.Handler{},
+		cfg: cfg,
+	}
+}
+
+func (r *Registry) Add(n string, h types.Handler) {
+	r.h[n] = h
+}
+
+func (r *Registry) Merge(h types.Handler, b []byte, cfg types.Config) (hh types.Handler, err error) {
+	hh, err = h.Merge(b, cfg)
+	return
+}
+
+func (r *Registry) Get(identifier string) (types.Handler, error) {
+	var (
+		ok bool
+		f  types.Handler
+	)
+
+	if f, ok = r.h[identifier]; !ok {
+		return nil, fmt.Errorf("could not get element from registry: %s", identifier)
+	}
+
+	return f.New(r.cfg), nil
+}
+
+func (r *Registry) All() (list types.FilterMetaList) {
+	for _, handler := range r.h {
+		if !handler.Enabled() {
+			continue
+		}
+
+		meta := handler.Meta()
+		list = append(list, &meta)
+	}
+
+	return
+}
+
+func (r *Registry) Preload() {
+	// prefilters
+	r.Add("queryParam", filter.NewQueryParam(r.cfg))
+	r.Add("header", filter.NewHeader(r.cfg))
+	r.Add("profiler", filter.NewProfiler(r.cfg))
+
+	// processers
+	r.Add("workflow", filter.NewWorkflow(r.cfg, NewWorkflow()))
+	r.Add("proxy", proxy.New(r.cfg, service.DefaultLogger, http.DefaultClient, secureStorageTodo{}))
+	r.Add("payload", filter.NewPayload(r.cfg, service.DefaultLogger))
+
+	// postfilters
+	r.Add("redirection", filter.NewRedirection(r.cfg))
+	r.Add("response", filter.NewResponse(r.cfg, service.Registry()))
+	r.Add("defaultJsonResponse", filter.NewDefaultJsonResponse(r.cfg))
+}
+
+func NewWorkflow() (wf filter.WfExecer) {
+	return service.DefaultWorkflow
+}

server/pkg/apigw/registry/registry_test.go

@@ -0,0 +1,101 @@
+package registry
+
+import (
+	"testing"
+
+	"github.com/cortezaproject/corteza/server/pkg/apigw/types"
+	"github.com/stretchr/testify/require"
+)
+
+func Test_registryAddGet(t *testing.T) {
+	var (
+		req = require.New(t)
+		r   = NewRegistry(types.Config{})
+	)
+
+	r.Add("mockHandler", types.MockHandler{})
+
+	h, err := r.Get("mockHandler")
+
+	req.NoError(err)
+	req.Len(r.h, 1)
+	req.IsType(types.MockHandler{}, h)
+}
+
+func Test_registryAddGetErr(t *testing.T) {
+	var (
+		req = require.New(t)
+		r   = NewRegistry(types.Config{})
+	)
+
+	r.Add("mockHandler", types.MockHandler{})
+
+	h, err := r.Get("foo")
+
+	req.EqualError(err, "could not get element from registry: foo")
+	req.Len(r.h, 1)
+	req.Nil(h)
+}
+
+func Test_registryMerge(t *testing.T) {
+	type (
+		tf struct {
+			name   string
+			err    string
+			params string
+			exp    string
+		}
+	)
+
+	var (
+		tcc = []tf{
+			{
+				name:   "set params",
+				params: `{"foo":"bar"}`,
+				exp:    "bar",
+			},
+			{
+				name:   "set invalid params",
+				params: `{"foo1":"bar"}`,
+				exp:    "",
+			},
+			{
+				name:   "set invalid params err",
+				params: `{"foo1":"bar"`,
+				exp:    "",
+				err:    "unexpected EOF",
+			},
+		}
+	)
+
+	for _, tc := range tcc {
+		var (
+			req = require.New(t)
+			r   = NewRegistry(types.Config{})
+		)
+
+		m, err := r.Merge(types.MockHandler{}, []byte(tc.params), types.Config{})
+
+		if tc.err != "" {
+			req.EqualError(err, tc.err)
+		} else {
+			req.Equal(m.(types.MockHandler).Foo, tc.exp)
+			req.NoError(err)
+		}
+	}
+
+}
+
+func Test_registryAll(t *testing.T) {
+	var (
+		req = require.New(t)
+		r   = NewRegistry(types.Config{})
+	)
+
+	r.Add("mockHandler", types.MockHandler{})
+
+	list := r.All()
+
+	req.Len(list, 1)
+	req.NotEmpty(list[0].Name)
+}