From 9d18a1bb61d969b05f4034565f7b64cdb747e847 Mon Sep 17 00:00:00 2001
From: Tim <tymmesyde@gmail.com>
Date: Thu, 6 Feb 2025 18:10:34 +0100
Subject: [PATCH 1/2] fix: drop srt file with shell

---
 src/common/CONSTANTS.js                 |  6 ++++++
 src/common/FileDrop/FileDrop.tsx        | 16 ++++++++++------
 src/common/FileDrop/utils.ts            | 19 +++++++++++++++++++
 src/routes/Player/Player.js             |  5 ++---
 src/services/DragAndDrop/DragAndDrop.js |  2 ++
 5 files changed, 39 insertions(+), 9 deletions(-)
 create mode 100644 src/common/FileDrop/utils.ts

diff --git a/src/common/CONSTANTS.js b/src/common/CONSTANTS.js
index ea88b0114..8e4e3efdc 100644
--- a/src/common/CONSTANTS.js
+++ b/src/common/CONSTANTS.js
@@ -41,6 +41,11 @@ const ICON_FOR_TYPE = new Map([
     ['other', 'movies'],
 ]);
 
+const MIME_SIGNATURES = {
+    'application/x-subrip': ['310D0A', '310A'],
+    'text/vtt': ['574542565454'],
+};
+
 const SUPPORTED_LOCAL_SUBTITLES = [
     'application/x-subrip',
     'text/vtt',
@@ -118,6 +123,7 @@ module.exports = {
     WRITERS_LINK_CATEGORY,
     TYPE_PRIORITIES,
     ICON_FOR_TYPE,
+    MIME_SIGNATURES,
     SUPPORTED_LOCAL_SUBTITLES,
     EXTERNAL_PLAYERS,
     WHITELISTED_HOSTS,
diff --git a/src/common/FileDrop/FileDrop.tsx b/src/common/FileDrop/FileDrop.tsx
index 9117a4c7a..aae4e146b 100644
--- a/src/common/FileDrop/FileDrop.tsx
+++ b/src/common/FileDrop/FileDrop.tsx
@@ -1,8 +1,9 @@
 import React, { createContext, useCallback, useContext, useEffect, useState } from 'react';
 import classNames from 'classnames';
+import { isFileType } from './utils';
 
 export type FileType = string;
-export type FileDropListener = (file: File) => void;
+export type FileDropListener = (filename: string, buffer: ArrayBuffer) => void;
 
 type FileDropContext = {
     on: (type: FileType, listener: FileDropListener) => void,
@@ -31,15 +32,18 @@ const FileDropProvider = ({ className, children }: Props) => {
 
     const onDrop = useCallback((event: DragEvent) => {
         event.preventDefault();
-
         const { dataTransfer } = event;
 
-        if (dataTransfer && dataTransfer.files.length > 0) {
+        if (dataTransfer && dataTransfer?.files.length > 0) {
             const file = dataTransfer.files[0];
 
-            listeners
-                .filter(([type]) => type === file.type)
-                .forEach(([, listerner]) => listerner(file));
+            file
+                .arrayBuffer()
+                .then((buffer) => {
+                    listeners
+                        .filter(([type]) => file.type ? type === file.type : isFileType(buffer, type))
+                        .forEach(([, listerner]) => listerner(file.name, buffer));
+                });
         }
 
         setActive(false);
diff --git a/src/common/FileDrop/utils.ts b/src/common/FileDrop/utils.ts
new file mode 100644
index 000000000..b5427ecfb
--- /dev/null
+++ b/src/common/FileDrop/utils.ts
@@ -0,0 +1,19 @@
+import { MIME_SIGNATURES } from 'stremio/common/CONSTANTS';
+
+const SIGNATURES = MIME_SIGNATURES as Record<string, string[]>;
+
+const isFileType = (buffer: ArrayBuffer, type: string) => {
+    const signatures = SIGNATURES[type];
+
+    return signatures.some((signature) => {
+        const array = new Uint8Array(buffer);
+        const signatureBuffer = Buffer.from(signature, 'hex');
+        const bufferToCompare = array.subarray(0, signatureBuffer.length);
+
+        return Buffer.compare(signatureBuffer, bufferToCompare) === 0;
+    });
+};
+
+export {
+    isFileType,
+};
\ No newline at end of file
diff --git a/src/routes/Player/Player.js b/src/routes/Player/Player.js
index 5ea8975f0..4dbc12295 100644
--- a/src/routes/Player/Player.js
+++ b/src/routes/Player/Player.js
@@ -277,9 +277,8 @@ const Player = ({ urlParams, queryParams }) => {
         event.nativeEvent.immersePrevented = true;
     }, []);
 
-    onFileDrop(CONSTANTS.SUPPORTED_LOCAL_SUBTITLES, async (file) => {
-        const buffer = await file.arrayBuffer();
-        video.addLocalSubtitles(file.name, buffer);
+    onFileDrop(CONSTANTS.SUPPORTED_LOCAL_SUBTITLES, async (filename, buffer) => {
+        video.addLocalSubtitles(filename, buffer);
     });
 
     React.useEffect(() => {
diff --git a/src/services/DragAndDrop/DragAndDrop.js b/src/services/DragAndDrop/DragAndDrop.js
index 6e20a840e..1538c328e 100644
--- a/src/services/DragAndDrop/DragAndDrop.js
+++ b/src/services/DragAndDrop/DragAndDrop.js
@@ -40,6 +40,8 @@ function DragAndDrop({ core }) {
                     break;
                 case 'text/vtt':
                     break;
+                case '':
+                    break;
                 default: {
                     events.emit('error', {
                         message: 'Unsupported file',

From 330fac3f3987d484a04d1eab5e63844fc7ff4d63 Mon Sep 17 00:00:00 2001
From: Tim <tymmesyde@gmail.com>
Date: Thu, 6 Feb 2025 18:29:15 +0100
Subject: [PATCH 2/2] style: lint

---
 src/common/FileDrop/utils.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/common/FileDrop/utils.ts b/src/common/FileDrop/utils.ts
index b5427ecfb..f9996aed3 100644
--- a/src/common/FileDrop/utils.ts
+++ b/src/common/FileDrop/utils.ts
@@ -16,4 +16,4 @@ const isFileType = (buffer: ArrayBuffer, type: string) => {
 
 export {
     isFileType,
-};
\ No newline at end of file
+};