fix: sort ui nodes when code is added (PS-144)

monta-app · Nov 22, 2023 · 9a120fc · 9a120fc
1 parent 71ba520
commit 9a120fc
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/selfservice/strategy/code/strategy_registration.go b/selfservice/strategy/code/strategy_registration.go
@@ -129,6 +129,13 @@ func (s *Strategy) Register(w http.ResponseWriter, r *http.Request, f *registrat
 			return s.handleRegistrationError(w, r, f, &p, err)
 		}
 		f.UI.Nodes.Upsert(node.NewInputField("code", "", node.CodeGroup, node.InputAttributeTypeText))
+		ds, err := s.deps.Config().DefaultIdentityTraitsSchemaURL(r.Context())
+		if err != nil {
+			return s.handleRegistrationError(w, r, f, &p, err)
+		}
+		if err := registration.SortNodes(r.Context(), f.UI.Nodes, ds.String()); err != nil {
+			return s.handleRegistrationError(w, r, f, &p, err)
+		}
 		return s.handleRegistrationError(w, r, f, &p, NewCodeSentError())
 	} else {
 		code, err := s.deps.CodeAuthenticationService().VerifyCode(r.Context(), f, p.Code)