From ed9211f67dffb05c521a6acbbea7f040773cb96e Mon Sep 17 00:00:00 2001
From: TPN <tpn@ainta.dev>
Date: Sun, 12 May 2024 14:32:52 +0530
Subject: [PATCH] added addMissingCaptions util

---
 src/utils/opensubtitles.ts | 46 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 src/utils/opensubtitles.ts

diff --git a/src/utils/opensubtitles.ts b/src/utils/opensubtitles.ts
new file mode 100644
index 0000000..643183d
--- /dev/null
+++ b/src/utils/opensubtitles.ts
@@ -0,0 +1,46 @@
+import { Caption, labelToLanguageCode, removeDuplicatedLanguages } from '@/providers/captions';
+import { IndividualEmbedRunnerOptions } from '@/runners/individualRunner';
+import { ProviderRunnerOptions } from '@/runners/runner';
+
+export async function addMissingCaptions(
+  captions: Caption[],
+  ops: ProviderRunnerOptions | IndividualEmbedRunnerOptions,
+  media: string,
+): Promise<Caption[]> {
+  try {
+    const [imdbId, season, episode] = atob(media)
+      .split('.')
+      .map((x, i) => (i === 0 ? x : Number(x) || null));
+    if (!imdbId) return captions;
+    const Res: { LanguageName: string; SubDownloadLink: string; SubFormat: 'srt' | 'vtt' }[] = await ops.proxiedFetcher(
+      `https://rest.opensubtitles.org/search/${season && episode ? `episode-${episode}/` : ''}imdbid-${(imdbId as string).slice(2)}${season && episode ? `/season-${season}` : ''}`,
+      {
+        headers: {
+          'X-User-Agent': 'VLSub 0.10.2',
+        },
+      },
+    );
+
+    const Captions: Caption[] = [];
+    for (const caption of Res) {
+      const url = caption.SubDownloadLink.replace('.gz', '').replace('download/', 'download/subencoding-utf8/');
+      const language = labelToLanguageCode(caption.LanguageName);
+      if (!url || !language) continue;
+
+      // check if the stream already has the language
+      const existingCaption = captions.find((x) => x.language === language);
+      if (existingCaption) Captions.push(existingCaption);
+      else
+        Captions.push({
+          id: url,
+          url,
+          type: caption.SubFormat || 'srt',
+          hasCorsRestrictions: false,
+          language,
+        });
+    }
+    return removeDuplicatedLanguages(Captions);
+  } catch {
+    return captions;
+  }
+}