From 8f04e0163aa2e712dcb7be48de06d36f8fdedfc2 Mon Sep 17 00:00:00 2001
From: sawka <mike.sawka@gmail.com>
Date: Mon, 3 Jun 2024 13:13:41 -0700
Subject: [PATCH] add pdf preview

---
 frontend/app/view/preview.tsx | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/frontend/app/view/preview.tsx b/frontend/app/view/preview.tsx
index e896baf8e..fed7a833b 100644
--- a/frontend/app/view/preview.tsx
+++ b/frontend/app/view/preview.tsx
@@ -26,6 +26,13 @@ function MarkdownPreview({ contentAtom }: { contentAtom: jotai.Atom<Promise<stri
 function StreamingPreview({ fileInfo }: { fileInfo: FileInfo }) {
     const filePath = fileInfo.path;
     const streamingUrl = "/wave/stream-file?path=" + encodeURIComponent(filePath);
+    if (fileInfo.mimetype == "application/pdf") {
+        return (
+            <div className="view-preview view-preview-pdf">
+                <iframe src={streamingUrl} width="100%" height="100%" name="pdfview" />
+            </div>
+        );
+    }
     if (fileInfo.mimetype.startsWith("video/")) {
         return (
             <div className="view-preview view-preview-video">
@@ -116,7 +123,12 @@ function PreviewView({ blockId }: { blockId: string }) {
     const fileInfo = jotai.useAtomValue(statFileAtom);
 
     // handle streaming files here
-    if (mimeType.startsWith("video/") || mimeType.startsWith("audio/") || mimeType.startsWith("image/")) {
+    if (
+        mimeType == "application/pdf" ||
+        mimeType.startsWith("video/") ||
+        mimeType.startsWith("audio/") ||
+        mimeType.startsWith("image/")
+    ) {
         return <StreamingPreview fileInfo={fileInfo} />;
     }
     if (fileInfo == null) {