packages/pass/hooks/useCountdown.ts

   1 import { useEffect, useMemo, useRef, useState } from 'react';
   2
   3 import type { Maybe } from '@proton/pass/types';
   4
   5 export const useCountdown = (initial?: number) => {
   6     const interval = useRef<Maybe<NodeJS.Timeout>>();
   7     const [value, setValue] = useState<number>(0);
   8
   9     const handle = useRef({
  10         start: (value: number) => {
  11             clearInterval(interval.current);
  12             setValue(value);
  13             interval.current = setInterval(
  14                 () =>
  15                     setValue((prev) => {
  16                         if (prev === 0) clearInterval(interval.current);
  17                         return Math.max(0, prev - 1);
  18                     }),
  19                 1_000
  20             );
  21         },
  22         cancel: () => {
  23             clearInterval(interval.current);
  24             setValue(0);
  25         },
  26     });
  27
  28     useEffect(() => {
  29         if (initial) handle.current.start(initial);
  30         return handle.current.cancel;
  31     }, []);
  32
  33     return useMemo(() => [value, handle.current] as const, [value]);
  34 };