Expose root fiber to DevTools

facebook · Jan 14, 2017 · ebdc1c1 · ebdc1c1
1 parent 6a48891
commit ebdc1c1
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/renderers/dom/fiber/ReactDOMFiber.js b/src/renderers/dom/fiber/ReactDOMFiber.js
@@ -318,6 +318,9 @@ function warnAboutUnstableUse() {
   warned = true;
 }
 
+// TODO
+let currentRoot;
+
 function renderSubtreeIntoContainer(parentComponent : ?ReactComponent<any, any, any>, children : ReactNodeList, containerNode : DOMContainerElement | Document, callback: ?Function) {
   validateContainer(containerNode);
 
@@ -331,6 +334,7 @@ function renderSubtreeIntoContainer(parentComponent : ?ReactComponent<any, any,
     }
     const newRoot = DOMRenderer.createContainer(container);
     root = container._reactRootContainer = newRoot;
+    currentRoot = root;
     // Initial mount should not be batched.
     DOMRenderer.unbatchedUpdates(() => {
       DOMRenderer.updateContainer(children, newRoot, parentComponent, callback);
@@ -385,4 +389,15 @@ var ReactDOM = {
 
 };
 
+/* globals __REACT_DEVTOOLS_GLOBAL_HOOK__*/
+if (
+  typeof __REACT_DEVTOOLS_GLOBAL_HOOK__ !== 'undefined' &&
+  typeof __REACT_DEVTOOLS_GLOBAL_HOOK__.inject === 'function') {
+  __REACT_DEVTOOLS_GLOBAL_HOOK__.inject({
+    getRoot() {
+      return currentRoot;
+    }
+  });
+}
+
 module.exports = ReactDOM;