GT2/GT2-Android/node_modules/react-deep-force-update/src/index.js

89 lines
2.5 KiB
JavaScript

// Constant to identify a React Component. It's been extracted from ReactTypeOfWork
// (https://github.com/facebook/react/blob/master/src/shared/ReactTypeOfWork.js#L20)
const ReactClassComponent = 2;
function traverseRenderedChildren(internalInstance, callback, argument) {
callback(internalInstance, argument);
if (internalInstance._renderedComponent) {
traverseRenderedChildren(
internalInstance._renderedComponent,
callback,
argument
);
} else {
for (let key in internalInstance._renderedChildren) {
if (internalInstance._renderedChildren.hasOwnProperty(key)) {
traverseRenderedChildren(
internalInstance._renderedChildren[key],
callback,
argument
);
}
}
}
}
function setPendingForceUpdate(internalInstance) {
if (internalInstance._pendingForceUpdate === false) {
internalInstance._pendingForceUpdate = true;
}
}
function forceUpdateIfPending(internalInstance, React) {
if (internalInstance._pendingForceUpdate === true) {
const publicInstance = internalInstance._instance;
React.Component.prototype.forceUpdate.call(publicInstance);
}
}
function deepForceUpdateStack(instance, React) {
const internalInstance = instance._reactInternalInstance;
traverseRenderedChildren(internalInstance, setPendingForceUpdate);
traverseRenderedChildren(internalInstance, forceUpdateIfPending, React);
}
function deepForceUpdate(instance, React) {
const root = instance._reactInternalFiber || instance._reactInternalInstance;
if (typeof root.tag !== 'number') {
// Traverse stack-based React tree.
return deepForceUpdateStack(instance, React);
}
let node = root;
while (true) {
if (node.tag === ReactClassComponent) {
const publicInstance = node.stateNode;
const { updater } = publicInstance;
if (typeof publicInstance.forceUpdate === 'function') {
publicInstance.forceUpdate();
} else if (updater && typeof updater.enqueueForceUpdate === 'function') {
updater.enqueueForceUpdate(publicInstance);
}
}
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;
}
}
export default function getForceUpdate(React) {
return instance => {
deepForceUpdate(instance, React);
};
}