@livelybone/use-count-down
Version:
A react hook for seconds count down
102 lines (86 loc) • 2.44 kB
JavaScript
/**
* 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;