fix: pass context to registration flow

ory · Feb 14, 2022 · c8d55b3 · c8d55b3
1 parent a1d3254
commit c8d55b3
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/selfservice/flow/registration/error.go b/selfservice/flow/registration/error.go
@@ -104,7 +104,7 @@ func (s *ErrorHandler) WriteFlowError(
 		return
 	}
 
-	if err := SortNodes(f.UI.Nodes, ds.String()); err != nil {
+	if err := SortNodes(r.Context(), f.UI.Nodes, ds.String()); err != nil {
 		s.forward(w, r, f, err)
 		return
 	}

diff --git a/selfservice/flow/registration/handler.go b/selfservice/flow/registration/handler.go
@@ -113,7 +113,7 @@ func (h *Handler) NewRegistrationFlow(w http.ResponseWriter, r *http.Request, ft
 		return nil, err
 	}
 
-	if err := SortNodes(f.UI.Nodes, ds.String()); err != nil {
+	if err := SortNodes(r.Context(), f.UI.Nodes, ds.String()); err != nil {
 		return nil, err
 	}
 

diff --git a/selfservice/flow/registration/sort.go b/selfservice/flow/registration/sort.go
@@ -6,9 +6,7 @@ import (
 	"github.com/ory/kratos/ui/node"
 )
 
-var ctx = context.Background()
-
-func SortNodes(n node.Nodes, schemaRef string) error {
+func SortNodes(ctx context.Context, n node.Nodes, schemaRef string) error {
 	return n.SortBySchema(ctx,
 		node.SortBySchema(schemaRef),
 		node.SortByGroups([]node.Group{