import { DeviceType } from '@/types';
import {
    Activity,
    Droplet,
    FlaskConical,
    Gauge,
    HeartPulse,
    Router,
    Scale,
    Thermometer,
    Watch,
    Wind,
} from 'lucide-react';
import { ReactElement } from 'react';

const typeIcons: Record<DeviceType, ReactElement> = {
    watch: <Watch />,
    balance: <Scale />,
    thermometer: <Thermometer />,
    blood_pressure: <Gauge />,
    oximeter: <HeartPulse />,
    ecg: <Activity />,
    glucometer: <Droplet />,
    spirometer: <Wind />,
    lab_analyzer: <FlaskConical />,
    gateway: <Router />,
};

export default function DeviceTypeIcon({ type }: { type: DeviceType }) {
    return typeIcons[type] ?? null;
}