From 3349450cb929277e0dae72b29a6ffd95ba2463ce Mon Sep 17 00:00:00 2001
From: Sabda Yagra <msabdayagra@gmail.com>
Date: Wed, 26 Nov 2025 12:12:19 +0700
Subject: [PATCH] fix: type file in download audio and doc

---
 components/main/audio-detail.tsx    | 15 ++++++++++++---
 components/main/document-detail.tsx | 14 +++++++++++++-
 2 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/components/main/audio-detail.tsx b/components/main/audio-detail.tsx
index 437bfd29..517a663a 100644
--- a/components/main/audio-detail.tsx
+++ b/components/main/audio-detail.tsx
@@ -247,7 +247,16 @@ const DetailAudio = () => {
       if (xhr.readyState === 4 && xhr.status === 200) {
         const contentType =
           xhr.getResponseHeader("content-type") || "application/octet-stream";
-        const extension = contentType.split("/")[1];
+        let extension = contentType.split("/")[1];
+
+        if (selectedSize === "MP3") {
+          extension = "mp3";
+        }
+
+        if (selectedSize === "WAV") {
+          extension = "wav";
+        }
+
         const filename = `${name}.${extension}`;
 
         const blob = new Blob([xhr.response], {
@@ -403,7 +412,7 @@ const DetailAudio = () => {
         const { default: WaveSurfer } = await import("wavesurfer.js");
 
         if (wavesurfer.current) {
-          wavesurfer.current.destroy(); 
+          wavesurfer.current.destroy();
         }
 
         setPlaying(false);
@@ -443,7 +452,7 @@ const DetailAudio = () => {
 
       return () => {
         if (wavesurfer.current) {
-          wavesurfer.current.destroy(); 
+          wavesurfer.current.destroy();
         }
       };
     }
diff --git a/components/main/document-detail.tsx b/components/main/document-detail.tsx
index 96020b32..607dc2dd 100644
--- a/components/main/document-detail.tsx
+++ b/components/main/document-detail.tsx
@@ -247,7 +247,19 @@ const DetailDocument = () => {
       if (xhr.readyState === 4 && xhr.status === 200) {
         const contentType =
           xhr.getResponseHeader("content-type") || "application/octet-stream";
-        const extension = contentType.split("/")[1];
+       let extension = contentType.split("/")[1];
+
+        if (selectedSize === "DOC") {
+          extension = "doc";
+        }
+
+        if (selectedSize === "PPT") {
+          extension = "ppt";
+        }
+
+        if (selectedSize === "PDF") {
+          extension = "pdf";
+        }
         const filename = `${name}.${extension}`;
 
         const blob = new Blob([xhr.response], {