useMachine.js 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. "use strict";
  2. var __read = (this && this.__read) || function (o, n) {
  3. var m = typeof Symbol === "function" && o[Symbol.iterator];
  4. if (!m) return o;
  5. var i = m.call(o), r, ar = [], e;
  6. try {
  7. while ((n === void 0 || n-- > 0) && !(r = i.next()).done) ar.push(r.value);
  8. }
  9. catch (error) { e = { error: error }; }
  10. finally {
  11. try {
  12. if (r && !r.done && (m = i["return"])) m.call(i);
  13. }
  14. finally { if (e) throw e.error; }
  15. }
  16. return ar;
  17. };
  18. Object.defineProperty(exports, "__esModule", { value: true });
  19. exports.useMachine = void 0;
  20. var react_1 = require("react");
  21. var with_selector_1 = require("use-sync-external-store/shim/with-selector");
  22. var xstate_1 = require("xstate");
  23. var useInterpret_1 = require("./useInterpret");
  24. var utils_1 = require("./utils");
  25. function identity(a) {
  26. return a;
  27. }
  28. function useMachine(getMachine) {
  29. var _a = [];
  30. for (var _i = 1; _i < arguments.length; _i++) {
  31. _a[_i - 1] = arguments[_i];
  32. }
  33. var _b = __read(_a, 1), _c = _b[0], options = _c === void 0 ? {} : _c;
  34. // using `useIdleInterpreter` allows us to subscribe to the service *before* we start it
  35. // so we don't miss any notifications
  36. var service = (0, useInterpret_1.useIdleInterpreter)(getMachine, options);
  37. var getSnapshot = (0, react_1.useCallback)(function () {
  38. if (service.status === xstate_1.InterpreterStatus.NotStarted) {
  39. return (options.state
  40. ? xstate_1.State.create(options.state)
  41. : service.machine.initialState);
  42. }
  43. return service.getSnapshot();
  44. }, [service]);
  45. var isEqual = (0, react_1.useCallback)(function (prevState, nextState) {
  46. return (0, utils_1.isInterpreterStateEqual)(service, prevState, nextState);
  47. }, [service]);
  48. var subscribe = (0, react_1.useCallback)(function (handleStoreChange) {
  49. var unsubscribe = service.subscribe(handleStoreChange).unsubscribe;
  50. return unsubscribe;
  51. }, [service]);
  52. var storeSnapshot = (0, with_selector_1.useSyncExternalStoreWithSelector)(subscribe, getSnapshot, getSnapshot, identity, isEqual);
  53. (0, react_1.useEffect)(function () {
  54. var rehydratedState = options.state;
  55. service.start(rehydratedState ? xstate_1.State.create(rehydratedState) : undefined);
  56. return function () {
  57. service.stop();
  58. service.status = xstate_1.InterpreterStatus.NotStarted;
  59. };
  60. }, []);
  61. return [storeSnapshot, service.send, service];
  62. }
  63. exports.useMachine = useMachine;