Feature/SOF-7291-1

dist/other/iframe-messaging/IframeToFromHostMessageHandler.js

@@ -15,12 +15,17 @@ class IframeToFromHostMessageHandler {
             if (event.data.type === "from-iframe-to-host") {
                 const { action, payload } = event.data;
                 if (this.handlers[action]) {
-                    this.handlers["set-data"].forEach((handler) => {
-                        handler(payload);
-                    });
-                    this.handlers["get-data"].forEach((handler) => {
-                        const data = handler();
-                        this.sendData(data);
+                    this.handlers[action].forEach((handler) => {
+                        Promise.resolve(handler(payload))
+                            .then((data) => {
+                            // If the handler returns data, send it to the iframe
+                            if (data !== undefined) {
+                                this.sendData(data);
+                            }
+                        })
+                            .catch((error) => {
+                            console.error(`Error in handler for ${action}:`, error);
+                        });
                     });
                 }
             }

dist/other/jupyterlite/JupyterLiteSession.d.ts

@@ -1,16 +1,23 @@
+import { IframeMessageSchema } from "@mat3ra/esse/dist/js/types";
 import React from "react";
 import IframeToFromHostMessageHandler from "../iframe-messaging/IframeToFromHostMessageHandler";
-interface JupyterLiteSessionProps {
+export interface IMessageHandlerConfigItem {
+    action: IframeMessageSchema["action"];
+    handlers: ((...args: any[]) => void)[];
+}
+export interface JupyterLiteSessionProps {
     originURL: string;
     defaultNotebookPath?: string;
     iframeId: string;
-    messageHandler?: IframeToFromHostMessageHandler;
+    messageHandlerConfigs?: IMessageHandlerConfigItem[];
 }
 declare class JupyterLiteSession extends React.Component<JupyterLiteSessionProps> {
     static defaultProps: JupyterLiteSessionProps;
+    messageHandler: IframeToFromHostMessageHandler;
     constructor(props?: JupyterLiteSessionProps);
     componentDidMount(): void;
     componentWillUnmount(): void;
+    sendData: (data: any) => void;
     render(): React.JSX.Element;
 }
 export default JupyterLiteSession;

dist/other/jupyterlite/JupyterLiteSession.js

@@ -1,4 +1,5 @@
 import React from "react";
+import IframeToFromHostMessageHandler from "../iframe-messaging/IframeToFromHostMessageHandler";
 const defaultProps = {
     // eslint-disable-next-line react/default-props-match-prop-types
     originURL: "https://jupyterlite.mat3ra.com",
@@ -8,14 +9,21 @@ const defaultProps = {
 class JupyterLiteSession extends React.Component {
     constructor(props = defaultProps) {
         super(props);
+        this.messageHandler = new IframeToFromHostMessageHandler();
+        // eslint-disable-next-line react/no-unused-class-component-methods
+        this.sendData = (data) => {
+            this.messageHandler.sendData(data);
+        };
     }
     componentDidMount() {
-        const { messageHandler, originURL, iframeId } = this.props;
-        messageHandler === null || messageHandler === void 0 ? void 0 : messageHandler.init(originURL, iframeId);
+        const { originURL, iframeId, messageHandlerConfigs } = this.props;
+        this.messageHandler.init(originURL, iframeId);
+        messageHandlerConfigs === null || messageHandlerConfigs === void 0 ? void 0 : messageHandlerConfigs.forEach((config) => {
+            this.messageHandler.addHandlers(config.action, config.handlers);
+        });
     }
     componentWillUnmount() {
-        const { messageHandler } = this.props;
-        messageHandler === null || messageHandler === void 0 ? void 0 : messageHandler.destroy();
+        this.messageHandler.destroy();
     }
     render() {
         const { defaultNotebookPath, originURL, iframeId } = this.props;