From 3a964a6ac3a78e8f4091effc88bd531de24badc2 Mon Sep 17 00:00:00 2001
From: ArabCoders <admin@arabcoders.org>
Date: Thu, 27 Feb 2025 19:19:32 +0300
Subject: [PATCH] add mimetype to source list

---
 .vscode/settings.json         |  3 +++
 app/library/HttpAPI.py        | 12 ++++++---
 app/library/Utils.py          | 50 +++++++++++++++++++++++++++++++++++
 ui/components/VideoPlayer.vue |  4 ++-
 4 files changed, 64 insertions(+), 5 deletions(-)

diff --git a/.vscode/settings.json b/.vscode/settings.json
index f21e8522..ac6857c9 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -23,13 +23,16 @@
     "httpx",
     "libcurl",
     "libx",
+    "matroska",
     "mpegts",
+    "msvideo",
     "muxdelay",
     "nodesc",
     "noprogress",
     "postprocessor",
     "preferredcodec",
     "preferredquality",
+    "quicktime",
     "tmpfilename",
     "vcodec",
     "vconvert",
diff --git a/app/library/HttpAPI.py b/app/library/HttpAPI.py
index 4ffef7b1..c8ac7162 100644
--- a/app/library/HttpAPI.py
+++ b/app/library/HttpAPI.py
@@ -38,6 +38,7 @@ from .Utils import (
     StreamingError,
     arg_converter,
     calc_download_path,
+    get_mime_type,
     get_sidecar_subtitles,
     get_video_info,
     validate_url,
@@ -1204,15 +1205,18 @@ class HttpAPI(Common):
 
             realFile = Path(realFile)
 
+            ff_info = await ffprobe(realFile)
+
             response = {
-                "ffprobe": await ffprobe(realFile),
+                "ffprobe": ff_info,
+                "mimetype": get_mime_type(ff_info.get("metadata", {}), realFile),
                 "sidecar": get_sidecar_subtitles(realFile),
             }
 
             for i, f in enumerate(response["sidecar"]):
-                response["sidecar"][i]["file"] = str(Path(realFile).with_name(f["file"].name)).replace(
-                    self.config.download_path, ""
-                ).strip("/")
+                response["sidecar"][i]["file"] = (
+                    str(Path(realFile).with_name(f["file"].name)).replace(self.config.download_path, "").strip("/")
+                )
 
             return web.json_response(data=response, status=web.HTTPOk.status_code, dumps=self.encoder.encode)
         except Exception as e:
diff --git a/app/library/Utils.py b/app/library/Utils.py
index 45c26df1..7a986130 100644
--- a/app/library/Utils.py
+++ b/app/library/Utils.py
@@ -554,3 +554,53 @@ def get_sidecar_subtitles(file: pathlib.Path) -> list[dict]:
         files.append({"file": f, "lang": lang, "name": f"{f.suffix[1:].upper()} ({i}) - {lang}"})
 
     return files
+
+
+def get_mime_type(metadata: dict, file_path: pathlib.Path) -> str:
+    """
+    Determine the correct MIME type for a video file based on ffprobe metadata.
+
+    Args:
+        metadata (dict): Parsed JSON output from ffprobe.
+        file_path (str): The path to the video file for fallback detection.
+
+    Returns:
+        str: MIME type compatible with HTML5 <video> tag.
+
+    """
+    # Extract format name from ffprobe
+    format_name = metadata.get("format_name", "")
+
+    # Define mappings for HTML5-compatible video types
+    format_to_mime = {
+        "matroska": "video/x-matroska",  # Default for MKV
+        "webm": "video/webm",  # MKV can also be WebM
+        "mp4": "video/mp4",
+        "mov": "video/quicktime",
+        "avi": "video/x-msvideo",
+        "flv": "video/x-flv",
+        "ogg": "video/ogg",
+        "mpegts": "video/mp2t",
+        "3gp": "video/3gpp",
+    }
+
+    # Check format_name against known formats
+    if format_name:
+        selected = None
+        for fmt in format_name.split(","):
+            fmt = fmt.strip().lower()
+            if fmt in format_to_mime:
+                selected = format_to_mime[fmt]
+
+        if selected:
+            return selected
+
+    # Fallback: Use Python's mimetypes module
+    import mimetypes
+
+    mime_type, _ = mimetypes.guess_type(str(file_path))
+    if mime_type:
+        return mime_type
+
+    # Final fallback: Return generic binary type
+    return "application/octet-stream"
diff --git a/ui/components/VideoPlayer.vue b/ui/components/VideoPlayer.vue
index e27df2f5..c1244291 100644
--- a/ui/components/VideoPlayer.vue
+++ b/ui/components/VideoPlayer.vue
@@ -19,7 +19,8 @@
 <template>
   <div>
     <video id="player" ref="video" :poster="thumbnail" :title="title" playsinline>
-      <source v-for="source in sources" :key="source.src" :src="source.src" @error="source.onerror" />
+      <source v-for="source in sources" :key="source.src" :src="source.src" @error="source.onerror"
+        :type="source.type" />
       <track v-for="track in tracks" :key="track.file" :kind="track.kind" :label="track.label" :srclang="track.lang"
         :src="track.file" default />
     </video>
@@ -70,6 +71,7 @@ onMounted(async () => {
 
   sources.value.push({
     src: makeDownload(config, props.item, 'api/download'),
+    type: response.mimetype,
     onerror: e => src_error(e),
   })