Merge branch 'feat/custom-filename' of https://github.com/aaquibnaved…

…/react-doc-viewer into aaquibnaved-feat/custom-filename
cyntler · Apr 15, 2022 · 8494603 · 8494603
2 parents 00ce4ea + 35b9fe1
commit 8494603
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 9 deletions.
diff --git a/src/components/FileName.tsx b/src/components/FileName.tsx
@@ -10,16 +10,22 @@ export const FileName: FC<{}> = () => {
 
   if (!currentDocument || config?.header?.disableFileName) return null;
 
-  let fileName = currentDocument.uri || "";
-  fileName = decodeURI(fileName);
+  let fileName = "";
 
-  if (!config?.header?.retainURLParams) {
-    fileName = fileName.split("?")[0];
-  }
+  if (currentDocument.fileName) {
+    fileName = currentDocument.fileName;
+  } else {
+    fileName = currentDocument.uri || "";
+    fileName = decodeURI(fileName);
+
+    if (!config?.header?.retainURLParams) {
+      fileName = fileName.split("?")[0];
+    }
 
-  const splitURL = fileName.split("/");
-  if (splitURL.length) {
-    fileName = splitURL[splitURL.length - 1];
+    const splitURL = fileName.split("/");
+    if (splitURL.length) {
+      fileName = splitURL[splitURL.length - 1];
+    }
   }
 
   return (

diff --git a/src/plugins/pdf/components/PDFControls.tsx b/src/plugins/pdf/components/PDFControls.tsx
@@ -30,7 +30,7 @@ const PDFControls: FC<{}> = () => {
         <DownloadButton
           id="pdf-download"
           href={currentDocument?.fileData as string}
-          download={currentDocument?.uri}
+          download={currentDocument?.fileName || currentDocument?.uri}
         >
           <DownloadPDFIcon color="#000" size="75%" />
         </DownloadButton>

diff --git a/src/types/index.ts b/src/types/index.ts
@@ -37,6 +37,7 @@ export interface IDocument {
   uri: string;
   fileType?: string;
   fileData?: string | ArrayBuffer;
+  fileName?: string;
 }
 
 export interface DocRendererProps {