From 853bc26587793e9ee6a360388e3f30312e9d0149 Mon Sep 17 00:00:00 2001
From: wukko <me@wukko.me>
Date: Mon, 9 Sep 2024 02:30:03 +0600
Subject: [PATCH] web/download: support downloading and sharing raw files

---
 web/src/components/dialog/PickerItem.svelte   | 12 ++--
 web/src/components/dialog/SavingDialog.svelte | 59 ++++++++++++------
 .../save/buttons/DownloadButton.svelte        | 34 +++++-----
 web/src/lib/download.ts                       | 62 +++++++++++++++----
 web/src/lib/types/dialog.ts                   |  3 +-
 5 files changed, 120 insertions(+), 50 deletions(-)

diff --git a/web/src/components/dialog/PickerItem.svelte b/web/src/components/dialog/PickerItem.svelte
index f5071885..7edf8c68 100644
--- a/web/src/components/dialog/PickerItem.svelte
+++ b/web/src/components/dialog/PickerItem.svelte
@@ -18,7 +18,13 @@
     $: itemType = item.type ?? "photo";
 </script>
 
-<button class="picker-item" on:click={() => downloadFile(item.url)}>
+<button
+    class="picker-item"
+    on:click={() =>
+        downloadFile({
+            url: item.url,
+        })}
+>
     <div class="picker-type">
         {#if itemType === "video"}
             <IconMovie />
@@ -32,11 +38,9 @@
     <img
         class="picker-image"
         src={item.thumb ?? item.url}
-
         class:loading={!imageLoaded}
         class:video-thumbnail={["video", "gif"].includes(itemType)}
-        on:load={() => imageLoaded = true}
-
+        on:load={() => (imageLoaded = true)}
         alt="{$t(`a11y.dialog.picker.item.${itemType}`)} {number}"
     />
     <Skeleton class="picker-image elevated" hidden={imageLoaded} />
diff --git a/web/src/components/dialog/SavingDialog.svelte b/web/src/components/dialog/SavingDialog.svelte
index 16c57142..f71e2b9f 100644
--- a/web/src/components/dialog/SavingDialog.svelte
+++ b/web/src/components/dialog/SavingDialog.svelte
@@ -2,7 +2,13 @@
     import { t } from "$lib/i18n/translations";
 
     import { device } from "$lib/device";
-    import { copyURL, openURL, shareURL } from "$lib/download";
+    import {
+        copyURL,
+        openURL,
+        shareURL,
+        openFile,
+        shareFile,
+    } from "$lib/download";
 
     import DialogContainer from "$components/dialog/DialogContainer.svelte";
 
@@ -17,9 +23,11 @@
 
     import CopyIcon from "$components/misc/CopyIcon.svelte";
     export let id: string;
-    export let url: string;
-    export let bodyText: string = "";
     export let dismissable = true;
+    export let bodyText: string = "";
+
+    export let url: string = "";
+    export let file: File | undefined = undefined;
 
     let close: () => void;
 
@@ -45,13 +53,20 @@
                     {$t("dialog.saving.title")}
                 </h2>
             </div>
+
             <div class="action-buttons">
                 {#if device.supports.directDownload}
                     <VerticalActionButton
                         id="save-download"
                         fill
                         elevated
-                        click={() => openURL(url)}
+                        click={() => {
+                            if (file) {
+                                return openFile(file);
+                            } else if (url) {
+                                return openURL(url);
+                            }
+                        }}
                     >
                         <IconDownload />
                         {$t("button.download")}
@@ -63,26 +78,34 @@
                         id="save-share"
                         fill
                         elevated
-                        click={async () => await shareURL(url)}
+                        click={async () => {
+                            if (file) {
+                                return await shareFile(file);
+                            } else if (url) {
+                                return await shareURL(url);
+                            }
+                        }}
                     >
                         <IconShare2 />
                         {$t("button.share")}
                     </VerticalActionButton>
                 {/if}
 
-                <VerticalActionButton
-                    id="save-copy"
-                    fill
-                    elevated
-                    click={async () => {
-                        copyURL(url);
-                        copied = true;
-                    }}
-                    ariaLabel={copied ? $t("button.copied") : ""}
-                >
-                    <CopyIcon check={copied} />
-                    {$t("button.copy")}
-                </VerticalActionButton>
+                {#if !file}
+                    <VerticalActionButton
+                        id="save-copy"
+                        fill
+                        elevated
+                        click={async () => {
+                            copyURL(url);
+                            copied = true;
+                        }}
+                        ariaLabel={copied ? $t("button.copied") : ""}
+                    >
+                        <CopyIcon check={copied} />
+                        {$t("button.copy")}
+                    </VerticalActionButton>
+                {/if}
             </div>
 
             {#if device.is.iOS}
diff --git a/web/src/components/save/buttons/DownloadButton.svelte b/web/src/components/save/buttons/DownloadButton.svelte
index c8424494..08800e88 100644
--- a/web/src/components/save/buttons/DownloadButton.svelte
+++ b/web/src/components/save/buttons/DownloadButton.svelte
@@ -30,29 +30,29 @@
     type DownloadButtonState = "idle" | "think" | "check" | "done" | "error";
 
     const changeDownloadButton = (state: DownloadButtonState) => {
-        disabled = state !== 'idle';
+        disabled = state !== "idle";
 
-        buttonText = ({
-             idle: ">>",
+        buttonText = {
+            idle: ">>",
             think: "...",
             check: "..?",
-             done: ">>>",
-            error: "!!"
-        })[state];
+            done: ">>>",
+            error: "!!",
+        }[state];
 
         buttonAltText = $t(
-            ({
-                 idle: "a11y.save.download",
+            {
+                idle: "a11y.save.download",
                 think: "a11y.save.download.think",
                 check: "a11y.save.download.check",
-                 done: "a11y.save.download.done",
+                done: "a11y.save.download.done",
                 error: "a11y.save.download.error",
-            })[state]
+            }[state]
         );
 
         // states that don't wait for anything, and thus can
         // transition back to idle after some period of time.
-        const final: DownloadButtonState[] = ['done', 'error'];
+        const final: DownloadButtonState[] = ["done", "error"];
         if (final.includes(state)) {
             setTimeout(() => changeDownloadButton("idle"), 1500);
         }
@@ -84,7 +84,9 @@
         if (response.status === "redirect") {
             changeDownloadButton("done");
 
-            return downloadFile(response.url);
+            return downloadFile({
+                url: response.url,
+            });
         }
 
         if (response.status === "tunnel") {
@@ -95,7 +97,9 @@
             if (probeResult === 200) {
                 changeDownloadButton("done");
 
-                return downloadFile(response.url);
+                return downloadFile({
+                    url: response.url,
+                });
             } else {
                 changeDownloadButton("error");
 
@@ -122,7 +126,9 @@
                     text: $t("button.download.audio"),
                     main: false,
                     action: () => {
-                        downloadFile(pickerAudio);
+                        downloadFile({
+                            url: pickerAudio,
+                        });
                     },
                 });
             }
diff --git a/web/src/lib/download.ts b/web/src/lib/download.ts
index fa09a5cc..9ed1042e 100644
--- a/web/src/lib/download.ts
+++ b/web/src/lib/download.ts
@@ -8,23 +8,47 @@ import { t } from "$lib/i18n/translations";
 import { createDialog } from "$lib/dialogs";
 import type { DialogInfo } from "$lib/types/dialog";
 
-export const openSavingDialog = (url: string, body: string | void) => {
+const openSavingDialog = ({ url, file, body }: { url?: string, file?: File, body?: string }) => {
     const dialogData: DialogInfo = {
         type: "saving",
         id: "saving",
-        url
+        file,
+        url,
     }
     if (body) dialogData.bodyText = body;
 
     createDialog(dialogData)
 }
 
+export const openFile = async (file: File) => {
+    const a = document.createElement("a");
+    const url = URL.createObjectURL(file);
+
+    a.href = url;
+    a.download = file.name;
+    a.click();
+    URL.revokeObjectURL(url);
+}
+
+export const shareFile = async (file: File) => {
+    return await navigator?.share({
+        files: [
+            new File([file], file.name, {
+                type: file.type,
+            }),
+        ],
+    });
+}
+
 export const openURL = (url: string) => {
     const open = window.open(url, "_blank");
 
     /* if new tab got blocked by user agent, show a saving dialog */
     if (!open) {
-        return openSavingDialog(url, get(t)("dialog.saving.blocked"));
+        return openSavingDialog({
+            url,
+            body: get(t)("dialog.saving.blocked")
+        });
     }
 }
 
@@ -36,7 +60,9 @@ export const copyURL = async (url: string) => {
     return await navigator?.clipboard?.writeText(url);
 }
 
-export const downloadFile = (url: string) => {
+export const downloadFile = ({ url, file }: { url?: string, file?: File }) => {
+    if (!url && !file) throw new Error("attempted to download void");
+
     const pref = get(settings).save.savingMethod;
 
     /*
@@ -50,18 +76,28 @@ export const downloadFile = (url: string) => {
     */
 
     if (pref === "ask" || !navigator.userActivation.isActive) {
-        return openSavingDialog(url);
+        return openSavingDialog({ url, file });
     }
 
     try {
-        if (pref === "share" && device.supports.share) {
-            return shareURL(url);
-        } else if (pref === "download" && device.supports.directDownload) {
-            return openURL(url);
-        } else if (pref === "copy") {
-            return copyURL(url);
+        if (file) {
+            if (pref === "share" && device.supports.share) {
+                return shareFile(file);
+            } else if (pref === "download" && device.supports.directDownload) {
+                return openFile(file);
+            }
         }
-    } catch {}
 
-    return openSavingDialog(url);
+        if (url) {
+            if (pref === "share" && device.supports.share) {
+                return shareURL(url);
+            } else if (pref === "download" && device.supports.directDownload) {
+                return openURL(url);
+            } else if (pref === "copy" && !file) {
+                return copyURL(url);
+            }
+        }
+    } catch { /* catch & ignore */ }
+
+    return openSavingDialog({ url, file });
 }
diff --git a/web/src/lib/types/dialog.ts b/web/src/lib/types/dialog.ts
index e6791899..fa0df5fe 100644
--- a/web/src/lib/types/dialog.ts
+++ b/web/src/lib/types/dialog.ts
@@ -40,8 +40,9 @@ type PickerDialog = Dialog & {
 
 type SavingDialog = Dialog & {
     type: "saving",
-    url: string,
     bodyText?: string,
+    url?: string,
+    file?: File,
 };
 
 export type DialogInfo = SmallDialog | PickerDialog | SavingDialog;