Change from factory to provider, like react-redux

Also borrowed some code from them, hehe

index.js

@@ -1 +1,4 @@
-module.exports = require('./build/exothermic')
+module.exports = {
+  connect: require('./build/connect'),
+  Provider: require('./build/Provider'),
+}

package.json

@@ -9,6 +9,9 @@
     "eslint-plugin-react": "^3.5.1",
     "eslint": "^1.7.1"
   },
+  "peerDependencies": {
+    "react": "^0.14.0"
+  },
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1",
     "build": "./node_modules/.bin/babel --stage 0 source -d build",

source/Provider.js

@@ -0,0 +1,27 @@
+import {Component, PropTypes, Children} from 'react';
+import firebaseShape from './firebaseShape'
+
+export default class Provider extends Component {
+  static propTypes = {
+    firebase: firebaseShape.isRequired,
+    children: PropTypes.element.isRequired,
+  }
+
+  static childContextTypes = {
+    firebase: firebaseShape.isRequired,
+  }
+
+  getChildContext() {
+    return {firebase: this.firebase};
+  }
+
+  constructor(props, context) {
+    super(props, context);
+    this.firebase = props.firebase;
+  }
+
+  render() {
+    const {children} = this.props;
+    return Children.only(children);
+  }
+}

source/exothermic.js → source/connect.js

@@ -1,4 +1,5 @@
 import React from 'react'
+import firebaseShape from './firebaseShape'
 
 /*************/
 /*  HELPERS  */
@@ -39,6 +40,12 @@ const massUnbind = (keys, listeners) => {
   keys.forEach(key => listeners[key]())
 }
 
+const ensureCallable = maybeFn =>
+  typeof maybeFn === 'function' ? maybeFn : (_ => maybeFn)
+
+const getDisplayName = WrappedComponent =>
+  WrappedComponent.displayName || WrappedComponent.name || 'Component';
+
 /***************/
 /*  COMPONENT  */
 /***************/
@@ -47,28 +54,34 @@ const massUnbind = (keys, listeners) => {
 const initial = Symbol('Initial value')
 
 
-export default baseref => dataOrFn => Component => {
-  const linkFn = typeof dataOrFn === 'function' ? dataOrFn : (_ => dataOrFn)
+export default (dataOrFn = {}) => WrappedComponent => {
+  const linkFn = ensureCallable(dataOrFn)
 
   const getPaths = props =>
     props.endothermicLoaded === false
     ? {}
     : linkFn(props)
 
-  const listen = update => (path, key) =>
-    onValue(baseref.child(path), value => {
-      update(key, value)
-    })
-
   return class Endothermic extends React.Component {
-    static displayName = `Endothermic(${Component.displayName || Component.name})`
+    static displayName = `Endothermic(${getDisplayName(WrappedComponent)})`
+    static WrappedComponent = WrappedComponent
+    static contextTypes = {
+      firebase: firebaseShape,
+    }
+    static propTypes = {
+      firebase: firebaseShape,
+    }
 
-    constructor(props) {
-      super(props)
+    constructor(props, context) {
+      super(props, context)
+
+      this.firebase = props.firebase || context.firebase;
+
+      this.listen = (path, key) =>
+        onValue(this.firebase.child(path), value => {
+          this.setState({ [key]: value })
+        })
 
-      this.listen = listen( (key, value) => {
-        this.setState({ [key]: value })
-      })
       this.listeners = {}
       this.paths = getPaths(this.props)
       this.state = mapObject(this.paths, _ => initial)
@@ -118,9 +131,10 @@ export default baseref => dataOrFn => Component => {
         .every( ([key, value]) => value !== initial)
 
       return (
-        <Component
+        <WrappedComponent
           {...this.props}
           {...this.state}
+          firebase={this.firebase}
           endothermicLoaded={loaded}
         />
       )

source/firebaseShape.js

@@ -0,0 +1,7 @@
+import {PropTypes} from 'react';
+
+export default PropTypes.shape({
+  child: PropTypes.func.isRequired,
+  on: PropTypes.func.isRequired,
+  off: PropTypes.func.isRequired,
+});