refactor(123): refactor the color selection based on connection and r…

…untime status

hivemq-edge/src/frontend/src/modules/EdgeVisualisation/utils/status-utils.ts

@@ -5,9 +5,22 @@ import { Dict } from '@chakra-ui/utils'
 import { Adapter, Bridge, Status } from '@/api/__generated__'
 import { NodeTypes } from '@/modules/EdgeVisualisation/types.ts'
 
+/**
+ * @param theme
+ * @param status
+ *
+ * TODO[NVL] Unify the styling with ConnectionStatusBadge
+ * @see ConnectionStatusBadge
+ */
 export const getThemeForStatus = (theme: Partial<WithCSSVar<Dict>>, status: Status | undefined) => {
-  const isConnected = status?.connection === Status.connection.CONNECTED
-  return isConnected ? theme.colors.status.connected[500] : theme.colors.status.disconnected[500]
+  if (status?.connection === Status.connection.CONNECTED) return theme.colors.status.connected[500]
+  if (status?.connection === Status.connection.DISCONNECTED) return theme.colors.status.disconnected[500]
+  if (status?.connection === Status.connection.STATELESS) return theme.colors.status.stateless[500]
+
+  // if (status?.runtime === Status.runtime.STOPPED) return theme.colors.status.error[500]
+  // if (status?.connection === Status.connection.ERROR) return theme.colors.status.error[500]
+  // if (status?.connection === Status.connection.UNKNOWN) return theme.colors.status.error[500]
+  return theme.colors.status.error[500]
 }
 
 export const updateNodeStatus = (currentNodes: Node[], updates: Status[]) => {