@marimo-team/islands 0.19.7-dev42 → 0.19.7-dev43

package/dist/main.js

@@ -73168,7 +73168,7 @@ Image URL: ${r.imageUrl}`)), contextToXml({
       return Logger.warn("Failed to get version from mount config"), null;
     }
   }
-  const marimoVersionAtom = atom(getVersionFromMountConfig() || "0.19.7-dev42"), showCodeInRunModeAtom = atom(true);
+  const marimoVersionAtom = atom(getVersionFromMountConfig() || "0.19.7-dev43"), showCodeInRunModeAtom = atom(true);
   atom(null);
   var import_compiler_runtime$88 = require_compiler_runtime();
   function useKeydownOnElement(e, r) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@marimo-team/islands",
-  "version": "0.19.7-dev42",
+  "version": "0.19.7-dev43",
   "main": "dist/main.js",
   "types": "dist/index.d.ts",
   "type": "module",

package/src/utils/__tests__/download.test.tsx

@@ -401,13 +401,30 @@ describe("downloadCellOutputAsImage", () => {
     vi.restoreAllMocks();
   });
 
-  it("should return early if element not found", async () => {
+  it("should show error toast if element not found", async () => {
     await downloadCellOutputAsImage("nonexistent" as CellId, "test");
 
     expect(toPng).not.toHaveBeenCalled();
     expect(Logger.error).toHaveBeenCalledWith(
       "Output element not found for cell nonexistent",
     );
+    expect(toast).toHaveBeenCalledWith({
+      title: "Failed to download PNG",
+      description: expect.any(String),
+      variant: "danger",
+    });
+  });
+
+  it("should show error toast if toPng fails", async () => {
+    vi.mocked(toPng).mockRejectedValue(new Error("Screenshot failed"));
+
+    await downloadCellOutputAsImage("cell-1" as CellId, "result");
+
+    expect(toast).toHaveBeenCalledWith({
+      title: "Failed to download PNG",
+      description: expect.stringContaining("Screenshot failed"),
+      variant: "danger",
+    });
   });
 
   it("should download cell output as image", async () => {

package/src/utils/download.ts

@@ -100,11 +100,19 @@ export async function downloadCellOutputAsImage(
   cellId: CellId,
   filename: string,
 ) {
-  const dataUrl = await getImageDataUrlForCell(cellId);
-  if (!dataUrl) {
-    return;
+  try {
+    const dataUrl = await getImageDataUrlForCell(cellId);
+    if (!dataUrl) {
+      throw new Error("Failed to get image data URL");
+    }
+    downloadByURL(dataUrl, Filenames.toPNG(filename));
+  } catch (error) {
+    toast({
+      title: "Failed to download PNG",
+      description: prettyError(error),
+      variant: "danger",
+    });
   }
-  downloadByURL(dataUrl, Filenames.toPNG(filename));
 }
 
 export const ADD_PRINTING_CLASS = (): (() => void) => {