feat: add new message "streamlit-app-print" for triggering print dialog

cognitedata · Jun 12, 2024 · 4fec73c · 4fec73c
2 parents c17ea25 + f67a1dc
commit 4fec73c
Showing 1 changed file with 15 additions and 21 deletions.
diff --git a/packages/kernel/src/kernel.ts b/packages/kernel/src/kernel.ts
@@ -438,28 +438,22 @@ const initTokenStorageAndAuthHandler = (worker: StliteWorker) => {
       if (
         typeof event.data === "object" &&
         "type" in event.data &&
-        "data" in event.data &&
-        event.data.type === "streamlit-app-generate-screenshot"
+        "data" in event.data
       ) {
-        const appScreenshot = await generateAppScreenshot();
-        console.log("GeneratedApp screenshot", appScreenshot);
-
-        // communicate if in iframe to parent (top)
-        postMessageToFusion({
-          type: "streamlit-app-generate-screenshot",
-          data: appScreenshot,
-        });
-      }
-
-      // StreamLit app main thread, forward the message to the worker
-      // so that the kernel can process the request
-      if (
-        typeof event.data === "object" &&
-        "type" in event.data &&
-        "data" in event.data &&
-        event.data.type.startsWith("language-server:")
-      ) {
-        worker.postMessage(event.data);
+        if (event.data.type === "streamlit-app-generate-screenshot") {
+          const appScreenshot = await generateAppScreenshot();
+          // send generated screenshot if in iframe to parent (top)
+          postMessageToFusion({
+            type: "streamlit-app-generate-screenshot",
+            data: appScreenshot,
+          });
+        } else if (event.data.type === "streamlit-app-print") {
+          window.print();
+        } else if (event.data.type.startsWith("language-server:")) {
+          // StreamLit app main thread, forward the message to the worker
+          // so that the kernel can process the request
+          worker.postMessage(event.data);
+        }
       }
     },
     false