UNPKG

@livelybone/use-count-down

Version:
102 lines (86 loc) 2.44 kB
/** * Bundle of @livelybone/use-count-down * Generated: 2019-10-23 * Version: 1.0.0 * License: MIT * Author: 2631541504@qq.com */ import { useRef, useState, useEffect, useMemo } from 'react'; function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; } function _iterableToArrayLimit(arr, i) { if (!(Symbol.iterator in Object(arr) || Object.prototype.toString.call(arr) === "[object Arguments]")) { return; } var _arr = []; var _n = true; var _d = false; var _e = undefined; try { for (var _i = arr[Symbol.iterator](), _s; !(_n = (_s = _i.next()).done); _n = true) { _arr.push(_s.value); if (i && _arr.length === i) break; } } catch (err) { _d = true; _e = err; } finally { try { if (!_n && _i["return"] != null) _i["return"](); } finally { if (_d) throw _e; } } return _arr; } function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance"); } function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _nonIterableRest(); } function useCountDown() { var maxTime = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : 60; var timer = useRef(null); var _useState = useState(maxTime), _useState2 = _slicedToArray(_useState, 2), count = _useState2[0], setCount = _useState2[1]; var _useState3 = useState(false), _useState4 = _slicedToArray(_useState3, 2), isRunning = _useState4[0], setRunning = _useState4[1]; useEffect(function () { return function () { return clearInterval(timer.current); }; }, []); return useMemo(function () { return { count: count, isRunning: isRunning, start: function start(cb) { clearInterval(timer.current); setCount(maxTime); setRunning(true); timer.current = setInterval(function () { setCount(function (preCount) { if (preCount <= 1) { clearInterval(timer.current); timer.current = null; setRunning(false); if (typeof cb === 'function') cb(); } return preCount - 1; }); }, 1000); }, stop: function stop() { clearInterval(timer.current); setRunning(false); } }; }, [count, isRunning, maxTime]); } export default useCountDown;