Add nullable and not-null icon to TableNode

liam-hq · Dec 3, 2024 · 43fcb83 · 43fcb83
1 parent b2d57dc
commit 43fcb83
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/frontend/packages/erd-core/src/components/ERDRenderer/ERDContent/TableNode/TableNode.tsx b/frontend/packages/erd-core/src/components/ERDRenderer/ERDContent/TableNode/TableNode.tsx
@@ -1,5 +1,7 @@
 import type { Table } from '@liam-hq/db-structure'
 import {
+  DiamondFillIcon,
+  DiamondIcon,
   DrawerClose,
   DrawerContent,
   DrawerPortal,
@@ -39,6 +41,25 @@ export const TableNode: FC<Props> = ({ data: { table } }) => {
                       />
                     )}
                   </span>
+                  {!column.primary && (
+                    <span>
+                      {column.notNull ? (
+                        <DiamondFillIcon
+                          width={16}
+                          height={16}
+                          role="img"
+                          aria-label="Not Null"
+                        />
+                      ) : (
+                        <DiamondIcon
+                          width={16}
+                          height={16}
+                          role="img"
+                          aria-label="Nullable"
+                        />
+                      )}
+                    </span>
+                  )}
                   <span> </span>
                   <span>{column.name}</span>
                   <span> </span>