CardMedia: refactor away from the createComponent function (#34915)

WordPress · Sep 19, 2021 · 55ed542 · 55ed542
1 parent fcc32ee
commit 55ed542
Showing 1 changed file with 14 additions and 7 deletions.
diff --git a/packages/components/src/card/card-media/component.js b/packages/components/src/card/card-media/component.js
@@ -1,9 +1,20 @@
 /**
  * Internal dependencies
  */
-import { createComponent } from '../../ui/utils';
+import { contextConnect } from '../../ui/context';
+import { View } from '../../view';
 import { useCardMedia } from './hook';
 
+/**
+ * @param {import('../../ui/context').WordPressComponentProps<{ children: import('react').ReactNode }, 'div'>} props
+ * @param {import('react').Ref<any>}                                                                           forwardedRef
+ */
+function CardMedia( props, forwardedRef ) {
+	const cardMediaProps = useCardMedia( props );
+
+	return <View { ...cardMediaProps } ref={ forwardedRef } />;
+}
+
 /**
  * `CardMedia` provides a container for media elements within a `Card`.
  *
@@ -21,10 +32,6 @@ import { useCardMedia } from './hook';
  * );
  * ```
  */
-const CardMedia = createComponent( {
-	as: 'div',
-	useHook: useCardMedia,
-	name: 'CardMedia',
-} );
+const ConnectedCardMedia = contextConnect( CardMedia, 'CardMedia' );
 
-export default CardMedia;
+export default ConnectedCardMedia;