Fix

marcveens · Mar 14, 2024 · 5978edf · 5978edf
1 parent aaab92d
commit 5978edf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/DrawIoEmbed.tsx b/src/DrawIoEmbed.tsx
@@ -106,7 +106,7 @@ export const DrawIoEmbed = forwardRef<DrawIoEmbedRef, DrawIoEmbedProps>(
             onTemplate(data);
           }
         }
-      });
+      }, baseUrl);
     };
 
     useImperativeHandle(

diff --git a/src/utils/handleEvent.ts b/src/utils/handleEvent.ts
@@ -4,8 +4,8 @@ type EventHandler = {
   [key in EmbedEvents['event']]?: (data: Extract<EmbedEvents, { event: key }>) => void;
 };
 
-export function handleEvent(event: MessageEvent, handlers: EventHandler) {
-  if (!event.origin.includes('embed.diagrams.net') && !event.origin.includes(baseUrl)) {
+export function handleEvent(event: MessageEvent, handlers: EventHandler, baseUrl?: string) {
+  if (!event.origin.includes('embed.diagrams.net') && (baseUrl && !event.origin.includes(baseUrl))) {
     return;
   }
-Original file line number
+Diff line change
@@ Expand Up @@
                 onTemplate(data);
               }
             }
-          });
+          }, baseUrl);
         };
         useImperativeHandle(
@@ Expand Down @@