fix: use type-guards

ory · Oct 11, 2023 · e642b4f · e642b4f
1 parent 09963c0
commit e642b4f
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/src/react-components/ory/helpers/filter-flow-nodes.tsx b/src/react-components/ory/helpers/filter-flow-nodes.tsx
@@ -3,7 +3,11 @@ import {
   FilterNodesByGroups,
   filterNodesByGroups,
   getNodeInputType,
+  isUiNodeAnchorAttributes,
+  isUiNodeImageAttributes,
   isUiNodeInputAttributes,
+  isUiNodeScriptAttributes,
+  isUiNodeTextAttributes,
 } from "@ory/integrations/ui"
 import { JSX } from "react"
 
@@ -31,14 +35,13 @@ export const FilterFlowNodes = ({
         <Node
           node={node}
           key={
-            // input node
-            "name" in node.attributes
+            isUiNodeInputAttributes(node.attributes)
               ? node.attributes.name
-              : // image node
-              "src" in node.attributes
+              : isUiNodeImageAttributes(node.attributes)
               ? node.attributes.src
-              : // anchor, text & script node
-              "id" in node.attributes
+              : isUiNodeAnchorAttributes(node.attributes) ||
+                isUiNodeTextAttributes(node.attributes) ||
+                isUiNodeScriptAttributes(node.attributes)
               ? node.attributes.id
               : k
           }