beautiful-react-hooks/useWillUnmount.js

A collection of beautiful (and hopefully useful) React hooks to speed-up your components and hooks development

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
var react_1 = require("react");
var createHandlerSetter_1 = __importDefault(require("./factory/createHandlerSetter"));
/**
 * Returns a callback setter for a callback to be performed when the component will unmount.
 */
var useWillUnmount = function (callback) {
    var mountRef = (0, react_1.useRef)(false);
    var _a = (0, createHandlerSetter_1.default)(callback), handler = _a[0], setHandler = _a[1];
    (0, react_1.useEffect)(function () {
        mountRef.current = true;
        return function () {
            if (handler && handler.current && typeof handler.current === 'function' && mountRef.current) {
                handler.current();
            }
        };
    }, []);
    return setHandler;
};
exports.default = useWillUnmount;