From 57be0dfcf248aa2a8241a7c99e46f41d6c230254 Mon Sep 17 00:00:00 2001
From: Hosmel Quintana <hosmelq@gmail.com>
Date: Wed, 26 Sep 2018 14:42:28 -0600
Subject: [PATCH] forward ref

---
 packages/djs-editor/src/Editor.js          | 4 +++-
 packages/djs-editor/src/EditorContainer.js | 8 +++++---
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/packages/djs-editor/src/Editor.js b/packages/djs-editor/src/Editor.js
index a4d00b4..59b16c0 100644
--- a/packages/djs-editor/src/Editor.js
+++ b/packages/djs-editor/src/Editor.js
@@ -2,4 +2,6 @@ import React from 'react'
 import Draft from 'draft-js'
 import { withConsumer } from './EditorContainer'
 
-export default withConsumer(({ editorProps }) => <Draft.Editor {...editorProps}/>)
+export default withConsumer(
+  React.forwardRef(({ editorProps }, ref) => <Draft.Editor {...editorProps} ref={ref} />)
+)
diff --git a/packages/djs-editor/src/EditorContainer.js b/packages/djs-editor/src/EditorContainer.js
index 75a41f4..5f419a0 100644
--- a/packages/djs-editor/src/EditorContainer.js
+++ b/packages/djs-editor/src/EditorContainer.js
@@ -9,9 +9,11 @@ const { CompositeDecorator, EditorState, getDefaultKeyBinding, DefaultDraftBlock
 
 export const Context = createContext({})
 
-export const withConsumer = Comp => props => (
-  <Context.Consumer>{contextProps => <Comp {...props} {...contextProps} />}</Context.Consumer>
-)
+export const withConsumer = Comp => React.forwardRef((props, ref) => (
+  <Context.Consumer>
+    {contextProps => <Comp {...props} {...contextProps} ref={ref} />}
+  </Context.Consumer>
+))
 
 const resolveDecorator = plugins => new CompositeDecorator(
   Array.from(plugins.values()).reduce((acc, plugin) => (