Add forwardRef to Datefield

material-table-core · Jan 22, 2021 · fddb850 · fddb850
1 parent a41c19a
commit fddb850
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/components/MTableEditField/DateField.js b/src/components/MTableEditField/DateField.js
@@ -2,11 +2,12 @@ import React from 'react';
 import DateFnsUtils from '@date-io/date-fns';
 import { MuiPickersUtilsProvider, DatePicker } from '@material-ui/pickers';
 
-export default function DateField({
+function DateField({
   columnDef,
   value,
   onChange,
   locale,
+  forwardedRef,
   ...rest
 }) {
   const getProps = () => {
@@ -33,6 +34,7 @@ export default function DateField({
     <MuiPickersUtilsProvider utils={DateFnsUtils} locale={locale}>
       <DatePicker
         {...datePickerProps}
+        ref={forwardedRef}
         format={dateFormat}
         value={value || null}
         onChange={onChange}
@@ -49,3 +51,7 @@ export default function DateField({
     </MuiPickersUtilsProvider>
   );
 }
+
+export default React.forwardRef(function DateFieldRef(props, ref) {
+  return <DateField {...props} forwardedRef={ref} />;
+});