Make root instance provider injectable. Fixes #53

DefaultRootInstanceProvider.js

@@ -0,0 +1,8 @@
+'use strict';
+
+module.exports = {
+  getRootInstances: function () {
+    var ReactMount = require('react/lib/ReactMount');
+    return ReactMount._instancesByReactRootID || ReactMount._instancesByContainerID;
+  }
+};

Injection.js

@@ -0,0 +1,7 @@
+'use strict';
+
+var RootInstanceProvider = require('./RootInstanceProvider');
+
+module.exports = {
+  RootInstanceProvider: RootInstanceProvider.injection
+};

RootInstanceProvider.js

@@ -0,0 +1,21 @@
+'use strict';
+
+var DefaultRootInstanceProvider = require('./DefaultRootInstanceProvider');
+
+var injectedProvider = null;
+
+var RootInstanceProvider = {
+  injection: {
+    injectProvider: function (provider) {
+      injectedProvider = provider;
+    }
+  },
+
+  getRootInstances: function () {
+    return injectedProvider.getRootInstances();
+  }
+};
+
+RootInstanceProvider.injection.injectProvider(DefaultRootInstanceProvider);
+
+module.exports = RootInstanceProvider;

index.js

@@ -29,11 +29,11 @@ module.exports = function (source, map) {
     '/* REACT HOT LOADER */',
     'if (module.hot) {',
       '(function () {',
-        'var reactHotApi = require(' + JSON.stringify(require.resolve('react-hot-api')) + '),',
-        '    getRootInstances = require(' + JSON.stringify(require.resolve('./getRootInstances')) + ');',
+        'var ReactHotAPI = require(' + JSON.stringify(require.resolve('react-hot-api')) + '),',
+        '    RootInstanceProvider = require(' + JSON.stringify(require.resolve('./RootInstanceProvider')) + ');',
         '',
-        'if (typeof reactHotApi === "function" && typeof getRootInstances === "function") {',
-          'module.makeHot = module.hot.data ? module.hot.data.makeHot : reactHotApi(getRootInstances);',
+        'if (typeof ReactHotAPI === "function" && typeof RootInstanceProvider === "object") {',
+          'module.makeHot = module.hot.data ? module.hot.data.makeHot : ReactHotAPI(RootInstanceProvider.getRootInstances);',
         '}',
       '})();',
     '}'