"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.deepForceUpdate = deepForceUpdate;
exports.default = void 0;

/**
 * Constant to identify a React Component. It's been extracted from ReactTypeOfWork
 *
 * @see https://github.com/facebook/react/blob/master/src/shared/ReactTypeOfWork.js#L20
 * @author https://github.com/gaearon/react-deep-force-update
 */
const ReactClassComponent = 2;

function traverseRenderedChildren(internalInstance, callback, argument) {
  callback(internalInstance, argument);

  if (internalInstance._renderedComponent) {
    traverseRenderedChildren(internalInstance._renderedComponent, callback, argument);
  } else {
    for (const key in internalInstance._renderedChildren) {
      if (internalInstance._renderedChildren.hasOwnProperty(key)) {
        traverseRenderedChildren(internalInstance._renderedChildren[key], callback, argument);
      }
    }
  }
}

function setPendingForceUpdate(internalInstance, shouldUpdate) {
  if (internalInstance._pendingForceUpdate === false && shouldUpdate(internalInstance)) {
    internalInstance._pendingForceUpdate = true;
  }
}

function forceUpdateIfPending(internalInstance, onUpdate) {
  if (internalInstance._pendingForceUpdate === true) {
    const publicInstance = internalInstance._instance;
    const updater = publicInstance.updater;

    if (typeof publicInstance.forceUpdate === 'function') {
      publicInstance.forceUpdate();
    } else if (updater && typeof updater.enqueueForceUpdate === 'function') {
      updater.enqueueForceUpdate(publicInstance);
    }

    onUpdate(internalInstance);
  }
}

function deepForceUpdateStack(instance, shouldUpdate, onUpdate) {
  const internalInstance = instance._reactInternalInstance;
  traverseRenderedChildren(internalInstance, setPendingForceUpdate, shouldUpdate);
  traverseRenderedChildren(internalInstance, forceUpdateIfPending, onUpdate);
} // eslint-disable-next-line


function deepForceUpdate(instance, shouldUpdate = () => true, onUpdate = () => {}) {
  const _root = instance._reactInternalFiber || instance._reactInternalInstance;

  if (typeof _root.tag !== 'number') {
    // Traverse stack-based React tree.
    return deepForceUpdateStack(instance, shouldUpdate, onUpdate);
  }

  let node = _root;

  while (true) {
    if (node.tag === ReactClassComponent && shouldUpdate(node)) {
      const publicInstance = node.stateNode;
      const updater = publicInstance.updater;

      if (typeof publicInstance.forceUpdate === 'function') {
        publicInstance.forceUpdate();
      } else if (updater && typeof updater.enqueueForceUpdate === 'function') {
        updater.enqueueForceUpdate(publicInstance);
      }

      onUpdate(node);
    }

    if (node.child) {
      node.child.return = node;
      node = node.child;
      continue;
    }

    if (node === _root) {
      return undefined;
    }

    while (!node.sibling) {
      if (!node.return || node.return === _root) {
        return undefined;
      }

      node = node.return;
    }

    node.sibling.return = node.return;
    node = node.sibling;
  }
} // process.env.NODE_ENV === 'development'
// if (typeof window === 'object') {
//   window.deepForceUpdate = deepForceUpdate
// }


var _default = deepForceUpdate;
exports.default = _default;