Ability to supply a baseUrl (#4)

* Ability to supply a baseUrl

* Update handleEvent.ts

---------

Co-authored-by: Marc Veens <[email protected]>

Author: sabinayakc

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) => {

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
    */

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');

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;
   }