diff --git a/web/src/components/Timeline/Timeline.tsx b/web/src/components/Timeline/Timeline.tsx index 890fb15aa..564467f0c 100644 --- a/web/src/components/Timeline/Timeline.tsx +++ b/web/src/components/Timeline/Timeline.tsx @@ -9,7 +9,7 @@ export interface TimelineEvent { startTime: Date; endTime: Date; id: string; - label: string; + label: 'car' | 'person' | 'dog'; } export interface TimelineEventBlock extends TimelineEvent { diff --git a/web/src/components/Timeline/TimelineBlockView.tsx b/web/src/components/Timeline/TimelineBlockView.tsx index 1611d663d..18d5d6233 100644 --- a/web/src/components/Timeline/TimelineBlockView.tsx +++ b/web/src/components/Timeline/TimelineBlockView.tsx @@ -9,11 +9,22 @@ interface TimelineBlockViewProps { export const TimelineBlockView = ({ block, onClick }: TimelineBlockViewProps) => { const onClickHandler = () => onClick(block); + const getColor = (eventBlock: TimelineEventBlock) => { + if (eventBlock.label === 'car') { + return 'red'; + } else if (eventBlock.label === 'person') { + return 'blue'; + } else if (eventBlock.label === 'dog') { + return 'green'; + } + return 'gray'; + }; + return (