Ability to supply a baseUrl (#4)

* Ability to supply a baseUrl * Update handleEvent.ts --------- Co-authored-by: Marc Veens <veens.marc@gmail.com>
marcveens · Mar 14, 2024 · 230e9cf · 230e9cf
1 parent de84d84
commit 230e9cf
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 3 deletions.
diff --git a/src/DrawIoEmbed.tsx b/src/DrawIoEmbed.tsx
@@ -14,6 +14,7 @@ import React from 'react';
 export const DrawIoEmbed = forwardRef<DrawIoEmbedRef, DrawIoEmbedProps>(
   (props, ref) => {
     const {
+      baseUrl,
       urlParameters,
       configuration,
       xml,
@@ -31,7 +32,7 @@ export const DrawIoEmbed = forwardRef<DrawIoEmbedRef, DrawIoEmbedProps>(
 
     const iframeRef = useRef<HTMLIFrameElement>(null);
     const action = useActions(iframeRef);
-    const iframeUrl = getEmbedUrl(urlParameters, !!configuration);
+    const iframeUrl = getEmbedUrl(baseUrl, urlParameters, !!configuration);
     const [isInitialized, setIsInitialized] = useState(false);
 
     const messageHandler = (evt: MessageEvent) => {

diff --git a/src/types.ts b/src/types.ts
@@ -1,6 +1,10 @@
 import { useActions } from './hooks/useActions';
 
 export type DrawIoEmbedProps = {
+  /**
+   * Base URL for draw.io embed URL. Defaults to https://embed.diagrams.net
+   */
+  baseUrl?: string;
   /**
    * Parameters documented at https://www.drawio.com/doc/faq/embed-mode
    */

diff --git a/src/utils/getEmbedUrl.ts b/src/utils/getEmbedUrl.ts
@@ -1,10 +1,11 @@
 import { UrlParameters } from '../types';
 
 export const getEmbedUrl = (
+  baseUrl?: string,
   urlParameters?: UrlParameters,
   addConfiguration?: boolean
 ) => {
-  const url = new URL('/', 'https://embed.diagrams.net');
+  const url = new URL('/', baseUrl ?? 'https://embed.diagrams.net');
   const urlSearchParams = new URLSearchParams();
 
   urlSearchParams.append('proto', 'json');

diff --git a/src/utils/handleEvent.ts b/src/utils/handleEvent.ts
@@ -5,7 +5,7 @@ type EventHandler = {
 };
 
 export function handleEvent(event: MessageEvent, handlers: EventHandler) {
-  if (!event.origin.includes('embed.diagrams.net')) {
+  if (!event.origin.includes('embed.diagrams.net') && !event.origin.includes(baseUrl)) {
     return;
   }