feat: add ref support in container

Description: Add ref support in container component VAN-1537
openedx · Jul 18, 2023 · ad33f55 · ad33f55
1 parent 1747665
commit ad33f55
Showing 1 changed file with 13 additions and 14 deletions.
diff --git a/src/Container/index.jsx b/src/Container/index.jsx
@@ -1,4 +1,4 @@
-import React from 'react';
+import React, { forwardRef } from 'react';
 import classNames from 'classnames';
 import RBContainer from 'react-bootstrap/Container';
 import PropTypes from 'prop-types';
@@ -11,19 +11,18 @@ const SIZE_CLASS_NAMES = {
   xl: 'container-mw-xl',
 };
 
-function Container({ size, children, ...props }) {
-  return (
-    <RBContainer
-      {...props}
-      className={classNames(
-        props.className,
-        SIZE_CLASS_NAMES[size],
-      )}
-    >
-      {children}
-    </RBContainer>
-  );
-}
+const Container = forwardRef(({ size, children, ...props }, ref) => (
+  <RBContainer
+    {...props}
+    ref={ref}
+    className={classNames(
+      props.className,
+      SIZE_CLASS_NAMES[size],
+    )}
+  >
+    {children}
+  </RBContainer>
+));
 
 Container.propTypes = {
   ...RBContainer.propTypes,