From 3f71bbbee94571a771331d2af4a688cc42dca548 Mon Sep 17 00:00:00 2001
From: Allan Lasser <lasser.allan@gmail.com>
Date: Thu, 12 Dec 2024 17:22:52 -0500
Subject: [PATCH 1/2] Simulate 403 error in Viewer storybook

---
 .../viewer/stories/Viewer.stories.svelte        | 17 +++++++++++++++++
 src/test/handlers/viewer.ts                     |  6 ++++++
 2 files changed, 23 insertions(+)
 create mode 100644 src/test/handlers/viewer.ts

diff --git a/src/lib/components/viewer/stories/Viewer.stories.svelte b/src/lib/components/viewer/stories/Viewer.stories.svelte
index a3b7d2451..80e67022c 100644
--- a/src/lib/components/viewer/stories/Viewer.stories.svelte
+++ b/src/lib/components/viewer/stories/Viewer.stories.svelte
@@ -8,6 +8,8 @@
   import doc from "@/test/fixtures/documents/document-expanded.json";
   import txt from "@/test/fixtures/documents/document.txt.json";
   import { searchWithin } from "@/test/handlers/documents";
+  import { simulatePDF403Error } from "@/test/handlers/viewer";
+  import { pdfUrl } from "@/lib/api/documents";
 
   const document = doc as Document;
 
@@ -102,6 +104,21 @@
   }}
 />
 
+<Story
+  name="With 403 Error"
+  parameters={{
+    msw: { handlers: [simulatePDF403Error(pdfUrl(document).href)] },
+  }}
+  args={{
+    ...args,
+    mode: "document",
+    document: {
+      ...document,
+      access: "private",
+    },
+  }}
+/>
+
 <style>
   .vh {
     height: 100vh;
diff --git a/src/test/handlers/viewer.ts b/src/test/handlers/viewer.ts
new file mode 100644
index 000000000..84ac288e2
--- /dev/null
+++ b/src/test/handlers/viewer.ts
@@ -0,0 +1,6 @@
+import { rest } from "msw";
+
+export const simulatePDF403Error = (asset_url: string) =>
+  rest.get(asset_url, (req, res, ctx) =>
+    res(ctx.status(403, "Not Found"), ctx.json({ detail: "Not found." })),
+  );

From 41ebce73e65b640d019e61246d26ea5f8af7329b Mon Sep 17 00:00:00 2001
From: Allan Lasser <lasser.allan@gmail.com>
Date: Thu, 12 Dec 2024 17:23:09 -0500
Subject: [PATCH 2/2] Attempt to retry load using a fresh assetUrl

---
 .../components/viewer/ViewerContext.svelte    | 32 +++++++++++++++----
 1 file changed, 25 insertions(+), 7 deletions(-)

diff --git a/src/lib/components/viewer/ViewerContext.svelte b/src/lib/components/viewer/ViewerContext.svelte
index bf4ae6c84..15ad6f48e 100644
--- a/src/lib/components/viewer/ViewerContext.svelte
+++ b/src/lib/components/viewer/ViewerContext.svelte
@@ -21,7 +21,12 @@ layouts, stories, and tests.
   import { getContext, onMount, setContext } from "svelte";
   import { type Writable, writable } from "svelte/store";
 
-  import { pageFromHash, pdfUrl, shouldPaginate } from "$lib/api/documents";
+  import {
+    pageFromHash,
+    pdfUrl,
+    shouldPaginate,
+    assetUrl,
+  } from "$lib/api/documents";
 
   import * as pdfjs from "pdfjs-dist/legacy/build/pdf.mjs";
   if (!pdfjs.GlobalWorkerOptions.workerSrc) {
@@ -162,8 +167,8 @@ layouts, stories, and tests.
     }
   }
 
-  onMount(() => {
-    // we might move this to a load function
+  let retriesOn403Error = 0;
+  function loadPDF(asset_url: URL) {
     if (!task) {
       task = pdfjs.getDocument({ url: asset_url });
       $pdf = task.promise;
@@ -172,12 +177,25 @@ layouts, stories, and tests.
         $progress = p;
       };
 
-      task.promise.catch((error) => {
-        console.error(error);
-        $currentErrors = [...$currentErrors, error];
-        throw error;
+      task.promise.catch(async (error) => {
+        if (error.status === 403 && retriesOn403Error < 5) {
+          // try to load the document again using a fresh asset_url
+          const fresh_asset_url = await assetUrl(document);
+          task = null;
+          retriesOn403Error++;
+          loadPDF(fresh_asset_url);
+        } else {
+          console.error(error);
+          $currentErrors = [...$currentErrors, error];
+          throw error;
+        }
       });
     }
+  }
+
+  onMount(() => {
+    // we might move this to a load function
+    loadPDF(asset_url);
   });
 
   afterNavigate(() => {