From 1ae97c9688c4392892892a05a192e5c97475fd9b Mon Sep 17 00:00:00 2001
From: Jonas Hungershausen <jonas.hungershausen@ory.sh>
Date: Fri, 15 Nov 2024 11:32:07 +0100
Subject: [PATCH] fix: improve card header texts

---
 .../default/utils/constructCardHeader.ts      | 96 ++++++++++++++++++-
 1 file changed, 94 insertions(+), 2 deletions(-)

diff --git a/packages/elements-react/src/theme/default/utils/constructCardHeader.ts b/packages/elements-react/src/theme/default/utils/constructCardHeader.ts
index 94059f367..fe59c7809 100644
--- a/packages/elements-react/src/theme/default/utils/constructCardHeader.ts
+++ b/packages/elements-react/src/theme/default/utils/constructCardHeader.ts
@@ -2,7 +2,7 @@
 // SPDX-License-Identifier: Apache-2.0
 
 import { FlowType, isUiNodeInputAttributes, UiNode } from "@ory/client-fetch"
-import { useIntl } from "react-intl"
+import { IntlShape, useIntl } from "react-intl"
 
 function joinWithCommaOr(list: string[], orText = "or"): string {
   if (list.length === 0) {
@@ -54,6 +54,21 @@ export function useCardHeaderText(
   const intl = useIntl()
   switch (opts.flowType) {
     case FlowType.Recovery:
+      if (
+        nodes.find(
+          (node) =>
+            "name" in node.attributes && node.attributes.name === "code",
+        )
+      ) {
+        return {
+          title: intl.formatMessage({
+            id: "recovery.title",
+          }),
+          description: intl.formatMessage({
+            id: "identities.messages.1060003",
+          }),
+        }
+      }
       return {
         title: intl.formatMessage({
           id: "recovery.title",
@@ -72,6 +87,21 @@ export function useCardHeaderText(
         }),
       }
     case FlowType.Verification:
+      if (
+        nodes.find(
+          (node) =>
+            "name" in node.attributes && node.attributes.name === "code",
+        )
+      ) {
+        return {
+          title: intl.formatMessage({
+            id: "verification.title",
+          }),
+          description: intl.formatMessage({
+            id: "identities.messages.1080003",
+          }),
+        }
+      }
       return {
         title: intl.formatMessage({
           id: "verification.title",
@@ -141,11 +171,25 @@ export function useCardHeaderText(
             id: "card.header.parts.identifier-first",
           },
           {
-            identifierLabel: identifier.meta.label?.text,
+            identifierLabel: identifier.meta.label?.text.toLocaleLowerCase(),
           },
         ),
       )
     }
+  } else if (isFirstStepRegistration(nodes)) {
+    parts.push(
+      intl.formatMessage(
+        {
+          id: "card.header.parts.identifier-first",
+        },
+        {
+          identifierLabel: joinWithCommaOr(
+            collectRegistrationTraits(nodes, intl),
+            intl.formatMessage({ id: "misc.and" }),
+          ),
+        },
+      ),
+    )
   }
 
   switch (opts.flowType) {
@@ -212,3 +256,51 @@ export function useCardHeaderText(
     description: "An error occurred",
   }
 }
+
+function collectRegistrationTraits(nodes: UiNode[], intl: IntlShape) {
+  return nodes.reduce((acc, node) => {
+    if (!isUiNodeInputAttributes(node.attributes)) {
+      return acc
+    }
+    if (!node.attributes.name.startsWith("traits.")) {
+      return acc
+    }
+
+    if (node.attributes.type === "hidden") {
+      return acc
+    }
+
+    if (!node.attributes.required) {
+      return acc
+    }
+    if (
+      !["text", "email", "tel"].includes(node.attributes.type.toLowerCase())
+    ) {
+      return acc
+    }
+
+    acc.push(
+      translateTrait(node.attributes.name, intl) ??
+        node.meta.label?.text ??
+        node.attributes.name,
+    )
+
+    return acc
+  }, [] as string[])
+}
+function isFirstStepRegistration(nodes: UiNode[]) {
+  return nodes.find(
+    (node) =>
+      node.group === "profile" &&
+      "name" in node.attributes &&
+      node.attributes.type === "submit",
+  )
+}
+
+function translateTrait(trait: string, intl: IntlShape): string | undefined {
+  if (intl.messages[`identity-schema.${trait}`]) {
+    return intl.formatMessage({
+      id: `identity-schema.${trait}`,
+    })
+  }
+}