From 0dec051afbdceec3495a49ce2f50713fb27a484c Mon Sep 17 00:00:00 2001
From: stratuma <zerotwofa@proton.me>
Date: Sat, 1 Jun 2024 22:53:27 +0200
Subject: [PATCH] added better drm key checks

---
 src/api/routes/service/service.service.ts | 38 +++++++++++++++++++++--
 src/api/services/folder.ts                | 11 +++++++
 2 files changed, 46 insertions(+), 3 deletions(-)

diff --git a/src/api/routes/service/service.service.ts b/src/api/routes/service/service.service.ts
index 41c4dc8..77b3199 100644
--- a/src/api/routes/service/service.service.ts
+++ b/src/api/routes/service/service.service.ts
@@ -1,7 +1,7 @@
 import { Account, Playlist } from '../../db/database'
 import { downloadMPDAudio } from '../../services/audio'
 import { concatenateTSFiles } from '../../services/concatenate'
-import { createFolder, createFolderName, deleteFolder, deleteTemporaryFolders } from '../../services/folder'
+import { checkFileExistence, createFolder, createFolderName, deleteFolder, deleteTemporaryFolders } from '../../services/folder'
 import { downloadADNSub, downloadCRSub } from '../../services/subs'
 import { CrunchyEpisode } from '../../types/crunchyroll'
 import { checkAccountMaxStreams, crunchyGetPlaylist, crunchyGetPlaylistMPD } from '../crunchyroll/crunchyroll.service'
@@ -541,8 +541,40 @@ export async function downloadCrunchyrollPlaylist(
 
     await updatePlaylistByID(downloadID, undefined, undefined, seasonFolder)
 
-    const drmL3blob = await settings.get('l3blob')
-    const drmL3key = await settings.get('l3key')
+    const drmL3blob = (await settings.get('l3blob')) as string
+    const drmL3key = (await settings.get('l3key')) as string
+
+    if (drmL3blob) {
+        const found = await checkFileExistence(drmL3blob)
+
+        if (!found) {
+            messageBox('error', ['Cancel'], 2, 'Widevine Key path is invalid', 'Widevine Key path is invalid', 'Widevine Key path is invalid, downloading without drm decryption')
+            server.logger.log({
+                level: 'error',
+                message: 'Widevine Key path is invalid, downloading without drm decryption',
+                timestamp: new Date().toISOString(),
+                section: 'crunchyrollCheckDRMPath'
+            })
+            await settings.set('CREndpoint', 1)
+            await settings.set('l3blob', null)
+        }
+    }
+
+    if (drmL3key) {
+        const found = await checkFileExistence(drmL3key)
+
+        if (!found) {
+            messageBox('error', ['Cancel'], 2, 'Widevine Key path is invalid', 'Widevine Key path is invalid', 'Widevine Key path is invalid, downloading without drm decryption')
+            server.logger.log({
+                level: 'error',
+                message: 'Widevine Key path is invalid, downloading without drm decryption',
+                timestamp: new Date().toISOString(),
+                section: 'crunchyrollCheckDRMPath'
+            })
+            await settings.set('CREndpoint', 1)
+            await settings.set('l3key', null)
+        }
+    }
 
     const dubDownloadList: Array<{
         audio_locale: string
diff --git a/src/api/services/folder.ts b/src/api/services/folder.ts
index 0918982..97c9767 100644
--- a/src/api/services/folder.ts
+++ b/src/api/services/folder.ts
@@ -31,6 +31,17 @@ export async function checkDirectoryExistence(dir: string) {
     }
 }
 
+export async function checkFileExistence(file: string) {
+    try {
+        await fs.promises.access(file)
+        console.log(`File ${file} exists.`)
+        return true
+    } catch (error) {
+        console.log(`File ${file} does not exist.`)
+        return false
+    }
+}
+
 export async function createFolderName(name: string, dir: string) {
     var tempPath = (await settings.get('tempPath')) as string