89 lines
2.5 KiB
JavaScript
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);
|
||
|
};
|
||
|
}
|