bloom-layout
Version:
layout components used in bloom packages
86 lines (75 loc) • 2.7 kB
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)
;
exports.__esModule = true;
exports['default'] = deepForceUpdate;
var ReactClassComponent = 2;
function traverseRenderedChildren(internalInstance, callback, argument) {
callback(internalInstance, argument);
if (internalInstance._renderedComponent) {
traverseRenderedChildren(internalInstance._renderedComponent, callback, argument);
} else {
for (var 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) {
if (internalInstance._pendingForceUpdate === true) {
var publicInstance = internalInstance._instance;
var updater = publicInstance.updater;
if (typeof publicInstance.forceUpdate === 'function') {
publicInstance.forceUpdate();
} else if (updater && typeof updater.enqueueForceUpdate === 'function') {
updater.enqueueForceUpdate(publicInstance);
}
}
}
function deepForceUpdateStack(instance) {
var internalInstance = instance._reactInternalInstance;
traverseRenderedChildren(internalInstance, setPendingForceUpdate);
traverseRenderedChildren(internalInstance, forceUpdateIfPending);
}
function deepForceUpdate(instance) {
var root = instance._reactInternalFiber || instance._reactInternalInstance;
if (typeof root.tag !== 'number') {
// Traverse stack-based React tree.
return deepForceUpdateStack(instance);
}
var node = root;
while (true) {
if (node.tag === ReactClassComponent) {
var publicInstance = node.stateNode;
var updater = publicInstance.updater;
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;
}
}
module.exports = exports['default'];