Merge pull request #3253 from hakimLyon/hakimLyon-patch-1

Fix: useMediaState
udecode · Jun 9, 2024 · 60ea69f · 60ea69f
2 parents 0eeecf6 + 88dea35
commit 60ea69f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/.changeset/weak-games-relax.md b/.changeset/weak-games-relax.md
@@ -0,0 +1,5 @@
+---
+"@udecode/plate-media": patch
+---
+
+Fix: useMediaState
diff --git a/packages/media/src/media/useMediaState.ts b/packages/media/src/media/useMediaState.ts
@@ -4,9 +4,8 @@ import { useElement } from '@udecode/plate-common';
 import { useFocused, useReadOnly, useSelected } from 'slate-react';
 
 import type { TMediaElement } from './types';
-
-import { VIDEO_PROVIDERS } from '../media-embed';
 import { ELEMENT_VIDEO } from '../video';
+import { VIDEO_PROVIDERS, ELEMENT_MEDIA_EMBED} from '../media-embed';
 
 export type EmbedUrlData = {
   id?: string;
@@ -29,7 +28,7 @@ export const useMediaState = ({
   const { align = 'left', id, isUpload, name, type, url } = element;
 
   const embed = React.useMemo(() => {
-    if (!urlParsers || type !== ELEMENT_VIDEO) return;
+    if (!urlParsers || (type !== ELEMENT_VIDEO && type !== ELEMENT_MEDIA_EMBED)) return;
 
     for (const parser of urlParsers) {
       const data = parser(url);