save error message if source error with pdf file

allenai · Oct 16, 2023 · 1b96a67 · 1b96a67
1 parent 2fb110b
commit 1b96a67
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ui/library/src/components/DocumentWrapper.tsx b/ui/library/src/components/DocumentWrapper.tsx
@@ -70,6 +70,11 @@ export const DocumentWrapper: React.FunctionComponent<Props> = ({
     }
   }, []);
 
+  const onSourceError = React.useCallback((error: unknown): void => {
+    setErrorMessage(getErrorMessage(error));
+    setIsLoading(false);
+  }, []);
+
   const onPdfLoadError = React.useCallback((error: unknown): void => {
     setErrorMessage(getErrorMessage(error));
     setIsLoading(false);
@@ -96,6 +101,7 @@ export const DocumentWrapper: React.FunctionComponent<Props> = ({
   return (
     <Document
       options={{ cMapUrl: 'cmaps/', cMapPacked: true }}
+      onSourceError={onSourceError}
       onLoadError={onPdfLoadError}
       onLoadSuccess={onPdfLoadSuccess}
       externalLinkTarget="_blank"