Merge pull request #5 from 4Catalyzer/additions

feat: add injectContextAsProp

Author: jquense

src/forwardRef.js

@@ -2,10 +2,21 @@ import React from 'react';
 
 export default function forwardRef(
   renderFn,
-  { displayName, propTypes, defaultProps },
+  { displayName, propTypes, defaultProps, allowFallback = false },
 ) {
   const render = (...args) => renderFn(...args);
 
-  Object.assign(render, { displayName, propTypes, defaultProps });
-  return React.forwardRef(render);
+  Object.assign(render, { displayName });
+
+  if (React.forwardRef || !allowFallback)
+    return Object.assign(React.forwardRef(render), {
+      propTypes,
+      defaultProps,
+    });
+
+  return Object.assign(props => render(props, null), {
+    displayName,
+    propTypes,
+    defaultProps,
+  });
 }

src/injectContextAsProp.js

@@ -0,0 +1,4 @@
+import mapContextToProps from './mapContextToProps';
+
+export default (Context, prop, Component) =>
+  mapContextToProps(Context, context => ({ [prop]: context }), Component);

test/injectAsProp.test.js

@@ -0,0 +1,20 @@
+import React from 'react';
+import { mount } from 'enzyme';
+
+import injectContextAsProp from '../src/injectContextAsProp';
+
+describe('injectContextAsProp', () => {
+  it('should inject context', () => {
+    const Mapper = injectContextAsProp(
+      React.createContext('foo'),
+      'foo',
+      props => <div>{props.foo}</div>,
+    );
+
+    expect(
+      mount(<Mapper />)
+        .find('div')
+        .text(),
+    ).toEqual('foo');
+  });
+});