From cf1bd9e435e8b43d84ec4aa1031b7447e2459e76 Mon Sep 17 00:00:00 2001
From: Aaryan Khandelwal <aaryankhandu123@gmail.com>
Date: Tue, 24 Sep 2024 14:01:50 +0530
Subject: [PATCH] fix: pass update image size

---
 .../core/extensions/custom-image/components/image-block.tsx   | 4 ++--
 .../custom-image/components/toolbar/full-screen.tsx           | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/packages/editor/src/core/extensions/custom-image/components/image-block.tsx b/packages/editor/src/core/extensions/custom-image/components/image-block.tsx
index 89ace94a57f..c1003cbd8c2 100644
--- a/packages/editor/src/core/extensions/custom-image/components/image-block.tsx
+++ b/packages/editor/src/core/extensions/custom-image/components/image-block.tsx
@@ -158,8 +158,8 @@ export const CustomImageBlock: React.FC<CustomImageNodeViewProps> = (props) => {
         containerClassName="absolute top-1 right-1 z-20 bg-black/40 rounded opacity-0 pointer-events-none group-hover/image-component:opacity-100 group-hover/image-component:pointer-events-auto transition-opacity"
         image={{
           src,
-          height,
-          width,
+          height: size.height,
+          width: size.width,
         }}
       />
       {editor.isEditable && selected && <div className="absolute inset-0 size-full bg-custom-primary-500/30" />}
diff --git a/packages/editor/src/core/extensions/custom-image/components/toolbar/full-screen.tsx b/packages/editor/src/core/extensions/custom-image/components/toolbar/full-screen.tsx
index 9333fb21c58..93019c60680 100644
--- a/packages/editor/src/core/extensions/custom-image/components/toolbar/full-screen.tsx
+++ b/packages/editor/src/core/extensions/custom-image/components/toolbar/full-screen.tsx
@@ -21,8 +21,8 @@ export const ImageFullScreenAction: React.FC<Props> = (props) => {
   // states
   const [magnification, setMagnification] = useState(1);
   // derived values
-  const widthInNumber = useMemo(() => Number(width.replace("px", "")), [width]);
-  const heightInNumber = useMemo(() => Number(height.replace("px", "")), [height]);
+  const widthInNumber = useMemo(() => Number(width?.replace("px", "")), [width]);
+  const heightInNumber = useMemo(() => Number(height?.replace("px", "")), [height]);
   const aspectRatio = useMemo(() => widthInNumber / heightInNumber, [heightInNumber, widthInNumber]);
   // close handler
   const handleClose = useCallback(() => {