From 5978edfb640752d4988e6dda1a4e1559ca8e3f31 Mon Sep 17 00:00:00 2001
From: Marc Veens <marcveens@hotmail.com>
Date: Thu, 14 Mar 2024 20:56:41 +0100
Subject: [PATCH] Fix

---
 src/DrawIoEmbed.tsx      | 2 +-
 src/utils/handleEvent.ts | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/DrawIoEmbed.tsx b/src/DrawIoEmbed.tsx
index fcb426a..445f790 100644
--- 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
index 78e711d..a2a0537 100644
--- 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;
   }