diff --git a/src/components/hooks/useTimeUnit.ts b/src/components/hooks/useTimeUnit.ts new file mode 100644 index 000000000..e5f69cb07 --- /dev/null +++ b/src/components/hooks/useTimeUnit.ts @@ -0,0 +1,19 @@ +import { TimeUnit } from 'lib/types'; +import { useState } from 'react'; + +export function useTimeUnit(): { + currentTimeUnit: TimeUnit; + timeUnitOptions: TimeUnit[]; + saveTimeUnit: (value: TimeUnit) => void; +} { + const [timeUnit, setTimeUnit] = useState('hour'); + const timeUnitOptions = ['hour', 'day', 'week', 'month', 'year']; + + function saveTimeUnit(value: TimeUnit) { + setTimeUnit(value); + } + + return { currentTimeUnit: timeUnit, timeUnitOptions, saveTimeUnit }; +} + +export default useTimeUnit;