chore: compile

Exabyte-io · Mar 15, 2024 · 066a9cc · 066a9cc
1 parent 2434f65
commit 066a9cc
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 8 deletions.
diff --git a/dist/other/jupyterlite/JupyterLiteSession.d.ts b/dist/other/jupyterlite/JupyterLiteSession.d.ts
@@ -7,7 +7,8 @@ interface JupyterLiteSessionProps {
     messageHandler?: MessageHandler;
 }
 declare class JupyterLiteSession extends React.Component<JupyterLiteSessionProps> {
-    static defaultProps: Partial<JupyterLiteSessionProps>;
+    static defaultProps: JupyterLiteSessionProps;
+    constructor(props?: JupyterLiteSessionProps);
     componentDidMount(): void;
     componentWillUnmount(): void;
     render(): React.JSX.Element;

diff --git a/dist/other/jupyterlite/JupyterLiteSession.js b/dist/other/jupyterlite/JupyterLiteSession.js
@@ -6,9 +6,12 @@ const defaultProps = {
     frameId: "jupyter-lite-iframe",
 };
 class JupyterLiteSession extends React.Component {
+    constructor(props = defaultProps) {
+        super(props);
+    }
     componentDidMount() {
-        const { messageHandler, originURL } = this.props;
-        messageHandler === null || messageHandler === void 0 ? void 0 : messageHandler.init(originURL);
+        const { messageHandler, originURL, frameId } = this.props;
+        messageHandler === null || messageHandler === void 0 ? void 0 : messageHandler.init(originURL, frameId);
     }
     componentWillUnmount() {
         const { messageHandler } = this.props;

diff --git a/dist/other/jupyterlite/MessageHandler.d.ts b/dist/other/jupyterlite/MessageHandler.d.ts
@@ -1,8 +1,9 @@
-type HandlerFunction = (data?: any, variableName?: string) => void;
+type HandlerFunction = (...args: any[]) => void | any;
 declare class MessageHandler {
     private handlers;
     private originURL;
-    init(originURL: string): void;
+    private frameId;
+    init(originURL: string, frameId: string): void;
     destroy(): void;
     addHandlers(action: string, handlers: HandlerFunction[]): void;
     private receiveMessage;

diff --git a/dist/other/jupyterlite/MessageHandler.js b/dist/other/jupyterlite/MessageHandler.js
@@ -2,6 +2,7 @@ class MessageHandler {
     constructor() {
         this.handlers = {};
         this.originURL = "*";
+        this.frameId = "";
         this.receiveMessage = (event) => {
             if (this.originURL !== "*" && event.origin !== this.originURL) {
                 return;
@@ -12,16 +13,21 @@ class MessageHandler {
                 // @ts-ignore
                 if (this.handlers[action]) {
                     // @ts-ignore
-                    this.handlers[action].forEach((handler) => {
+                    this.handlers["set-data"].forEach((handler) => {
                         handler(event.data.payload.data, event.data.payload.variableName);
                     });
+                    this.handlers["get-data"].forEach((handler) => {
+                        const data = handler(event.data.payload.variableName);
+                        this.sendData(data, event.data.payload.variableName);
+                    });
                 }
             }
         };
     }
-    init(originURL) {
+    init(originURL, frameId) {
         window.addEventListener("message", this.receiveMessage);
         this.originURL = originURL;
+        this.frameId = frameId;
     }
     destroy() {
         window.removeEventListener("message", this.receiveMessage);
@@ -41,7 +47,11 @@ class MessageHandler {
                 variableName,
             },
         };
-        window.parent.postMessage(message, this.originURL);
+        const iframe = document.getElementById(this.frameId);
+        if (iframe) {
+            // @ts-ignore
+            iframe.contentWindow.postMessage(message, this.originURL);
+        }
     }
 }
 export default MessageHandler;