useStepQueue.js 3.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. "use strict";
  2. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
  3. var _typeof = require("@babel/runtime/helpers/typeof");
  4. Object.defineProperty(exports, "__esModule", {
  5. value: true
  6. });
  7. exports.default = exports.SkipStep = exports.DoStep = void 0;
  8. exports.isActive = isActive;
  9. var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
  10. var _useState3 = _interopRequireDefault(require("rc-util/lib/hooks/useState"));
  11. var React = _interopRequireWildcard(require("react"));
  12. var _interface = require("../interface");
  13. var _useIsomorphicLayoutEffect = _interopRequireDefault(require("./useIsomorphicLayoutEffect"));
  14. var _useNextFrame3 = _interopRequireDefault(require("./useNextFrame"));
  15. function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(e) { return e ? t : r; })(e); }
  16. function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != _typeof(e) && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
  17. var FULL_STEP_QUEUE = [_interface.STEP_PREPARE, _interface.STEP_START, _interface.STEP_ACTIVE, _interface.STEP_ACTIVATED];
  18. var SIMPLE_STEP_QUEUE = [_interface.STEP_PREPARE, _interface.STEP_PREPARED];
  19. /** Skip current step */
  20. var SkipStep = exports.SkipStep = false;
  21. /** Current step should be update in */
  22. var DoStep = exports.DoStep = true;
  23. function isActive(step) {
  24. return step === _interface.STEP_ACTIVE || step === _interface.STEP_ACTIVATED;
  25. }
  26. var _default = exports.default = function _default(status, prepareOnly, callback) {
  27. var _useState = (0, _useState3.default)(_interface.STEP_NONE),
  28. _useState2 = (0, _slicedToArray2.default)(_useState, 2),
  29. step = _useState2[0],
  30. setStep = _useState2[1];
  31. var _useNextFrame = (0, _useNextFrame3.default)(),
  32. _useNextFrame2 = (0, _slicedToArray2.default)(_useNextFrame, 2),
  33. nextFrame = _useNextFrame2[0],
  34. cancelNextFrame = _useNextFrame2[1];
  35. function startQueue() {
  36. setStep(_interface.STEP_PREPARE, true);
  37. }
  38. var STEP_QUEUE = prepareOnly ? SIMPLE_STEP_QUEUE : FULL_STEP_QUEUE;
  39. (0, _useIsomorphicLayoutEffect.default)(function () {
  40. if (step !== _interface.STEP_NONE && step !== _interface.STEP_ACTIVATED) {
  41. var index = STEP_QUEUE.indexOf(step);
  42. var nextStep = STEP_QUEUE[index + 1];
  43. var result = callback(step);
  44. if (result === SkipStep) {
  45. // Skip when no needed
  46. setStep(nextStep, true);
  47. } else if (nextStep) {
  48. // Do as frame for step update
  49. nextFrame(function (info) {
  50. function doNext() {
  51. // Skip since current queue is ood
  52. if (info.isCanceled()) return;
  53. setStep(nextStep, true);
  54. }
  55. if (result === true) {
  56. doNext();
  57. } else {
  58. // Only promise should be async
  59. Promise.resolve(result).then(doNext);
  60. }
  61. });
  62. }
  63. }
  64. }, [status, step]);
  65. React.useEffect(function () {
  66. return function () {
  67. cancelNextFrame();
  68. };
  69. }, []);
  70. return [startQueue, step];
  71. };