fix(badge-icons): some primary icons are not present in the svg-icons…

… or Icons
stakwork · Oct 22, 2024 · 636496c · 636496c
1 parent 23f6df5
commit 636496c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/components/common/TypeBadge/index.tsx b/src/components/common/TypeBadge/index.tsx
@@ -2,6 +2,7 @@ import styled from 'styled-components'
 import { useSchemaStore } from '~/stores/useSchemaStore'
 import { colors } from '~/utils/colors'
 import { Flex } from '../Flex'
+import { Icons } from '~/components/Icons'
 
 type Props = {
   type: string
@@ -21,7 +22,9 @@ export const TypeBadge = ({ type }: Props) => {
   const primaryColor = normalizedSchemasByType[type]?.primary_color
   const primaryIcon = normalizedSchemasByType[type]?.icon
 
-  const icon = primaryIcon ? `svg-icons/${primaryIcon}.svg` : null
+  const isIconPresent = primaryIcon && primaryIcon in Icons
+
+  const icon = isIconPresent ? `svg-icons/${primaryIcon}.svg` : null
 
   switch (nodeType) {
     case 'video':