feat: use initial_value in lookup strategy

ory · Oct 19, 2021 · efe272f · efe272f
1 parent 572a131
commit efe272f
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 9 deletions.
diff --git a/selfservice/strategy/lookup/nodes.go b/selfservice/strategy/lookup/nodes.go
@@ -6,25 +6,31 @@ import (
 )
 
 func NewRevealLookupNode() *node.Node {
-	return node.NewInputField(node.LookupReveal, "true", node.LookupGroup,
-		node.InputAttributeTypeSubmit).
+	return node.NewInputField(node.LookupReveal, "true", node.LookupGroup, node.InputAttributeTypeSubmit,
+		node.WithInputAttributes(func(a *node.InputAttributes) {
+			a.InitialValue = "false"
+		})).
 		WithMetaLabel(text.NewInfoSelfServiceSettingsRevealLookup())
 }
 
 func NewRegenerateLookupNode() *node.Node {
-	return node.NewInputField(node.LookupRegenerate, "true", node.LookupGroup,
-		node.InputAttributeTypeSubmit).
+	return node.NewInputField(
+		node.LookupRegenerate, "true", node.LookupGroup, node.InputAttributeTypeSubmit, node.WithInputAttributes(func(a *node.InputAttributes) {
+			a.InitialValue = "false"
+		})).
 		WithMetaLabel(text.NewInfoSelfServiceSettingsRegenerateLookup())
 }
 
 func NewDisableLookupNode() *node.Node {
-	return node.NewInputField(node.LookupDisable, "true", node.LookupGroup,
-		node.InputAttributeTypeSubmit).
+	return node.NewInputField(node.LookupDisable, "true", node.LookupGroup, node.InputAttributeTypeSubmit, node.WithInputAttributes(func(a *node.InputAttributes) {
+		a.InitialValue = "false"
+	})).
 		WithMetaLabel(text.NewInfoSelfServiceSettingsDisableLookup())
 }
 
 func NewConfirmLookupNode() *node.Node {
-	return node.NewInputField(node.LookupConfirm, "true", node.LookupGroup,
-		node.InputAttributeTypeSubmit).
+	return node.NewInputField(node.LookupConfirm, "true", node.LookupGroup, node.InputAttributeTypeSubmit, node.WithInputAttributes(func(a *node.InputAttributes) {
+		a.InitialValue = "false"
+	})).
 		WithMetaLabel(text.NewInfoSelfServiceSettingsLookupConfirm())
 }
diff --git a/selfservice/strategy/lookup/settings.go b/selfservice/strategy/lookup/settings.go
@@ -170,6 +170,23 @@ func (s *Strategy) continueSettingsFlowDisable(w http.ResponseWriter, r *http.Re
 
 	i.DeleteCredentialsType(s.ID())
 
+	ctxUpdate.Flow.UI.Nodes.Remove(node.LookupCodes)
+	ctxUpdate.Flow.UI.Nodes.Remove(node.LookupReveal)
+	ctxUpdate.Flow.UI.Nodes.Remove(node.LookupDisable)
+	ctxUpdate.Flow.UI.Nodes.Upsert(NewRegenerateLookupNode())
+	ctxUpdate.Flow.InternalContext, err = sjson.SetBytes(ctxUpdate.Flow.InternalContext, flow.PrefixInternalContextKey(s.ID(), internalContextKeyRevealed), false)
+	if err != nil {
+		return err
+	}
+	ctxUpdate.Flow.InternalContext, err = sjson.SetRawBytes(ctxUpdate.Flow.InternalContext, flow.PrefixInternalContextKey(s.ID(), InternalContextKeyRegenerated), []byte("{}"))
+	if err != nil {
+		return err
+	}
+
+	if err := s.d.SettingsFlowPersister().UpdateSettingsFlow(r.Context(), ctxUpdate.Flow); err != nil {
+		return err
+	}
+
 	ctxUpdate.UpdateIdentity(i)
 	return nil
 }
@@ -196,7 +213,7 @@ func (s *Strategy) continueSettingsFlowReveal(w http.ResponseWriter, r *http.Req
 
 	ctxUpdate.Flow.UI.Nodes.Upsert(creds.ToNode())
 	ctxUpdate.Flow.UI.Nodes.Remove(node.LookupReveal)
-	ctxUpdate.Flow.UI.Nodes.Remove(node.LookupDisable)
+	ctxUpdate.Flow.UI.Nodes.Upsert(NewDisableLookupNode())
 	ctxUpdate.Flow.UI.Nodes.Upsert(NewRegenerateLookupNode())
 	ctxUpdate.Flow.InternalContext, err = sjson.SetBytes(ctxUpdate.Flow.InternalContext, flow.PrefixInternalContextKey(s.ID(), internalContextKeyRevealed), true)
 	if err != nil {