From e1d36be2d7939be78fbcc00ad6218b12e1585a9e Mon Sep 17 00:00:00 2001
From: Daniel Haller <zerotwofa@proton.me>
Date: Wed, 24 Apr 2024 03:22:29 +0200
Subject: [PATCH] added more adn stuff

---
 components/ADN/ListAnimes.ts              | 65 +++++++++++++++--
 components/ADN/ListEpisodes.ts            |  4 +-
 components/ADN/Types.ts                   |  3 +-
 components/Search/Types.ts                | 70 +++++++++---------
 pages/addanime.vue                        | 22 ++++--
 src/api/routes/adn/adn.service.ts         |  6 +-
 src/api/routes/service/service.service.ts | 36 ++++++++--
 src/api/services/folder.ts                | 21 +++++-
 src/api/services/subs.ts                  | 88 +++++++++++++++++++++--
 9 files changed, 248 insertions(+), 67 deletions(-)

diff --git a/components/ADN/ListAnimes.ts b/components/ADN/ListAnimes.ts
index 50fdae6..02f63ce 100644
--- a/components/ADN/ListAnimes.ts
+++ b/components/ADN/ListAnimes.ts
@@ -1,12 +1,12 @@
 import type { ADNSearchFetch } from "./Types";
 
 export async function searchADN(q: string) {
-    const { data, error } = await useFetch<ADNSearchFetch>(
+    const { data: deData, error: deError } = await useFetch<ADNSearchFetch>(
         `https://gw.api.animationdigitalnetwork.fr/show/catalog`,
         {
             method: "GET",
             headers: {
-                "x-target-distribution": "de",
+                "x-target-distribution": 'de',
             },
             query: {
                 "maxAgeCategory": "18",
@@ -15,11 +15,64 @@ export async function searchADN(q: string) {
         }
     );
 
-    if (error.value) {
-        throw new Error(error.value?.data.message as string)
+    if (deError.value) {
+        throw new Error(deError.value?.data.message as string)
     }
 
-    if (!data.value) return
+    if (!deData.value) return
 
-    return data.value.shows
+    const { data: frData, error: frError } = await useFetch<ADNSearchFetch>(
+        `https://gw.api.animationdigitalnetwork.fr/show/catalog`,
+        {
+            method: "GET",
+            headers: {
+                "x-target-distribution": 'fr',
+            },
+            query: {
+                "maxAgeCategory": "18",
+                "search": q
+            }
+        }
+    );
+
+    if (frError.value) {
+        throw new Error(frError.value?.data.message as string)
+    }
+
+    if (!frData.value) return
+
+    const deShows = deData.value.shows;
+    const frShows = frData.value.shows;
+
+    const mergeLanguagesOfDuplicates = (shows: {
+        id: number
+        url: string
+        title: string
+        image2x: string
+        episodeCount: number,
+        languages: Array<string>
+      }[]) => {
+        shows.forEach(show => {
+            const existingShow = shows.find(s => s.id === show.id);
+            if (existingShow) {
+                const existingShowIndex = shows.findIndex(s=> s === existingShow);
+                const rawLanguages = [...show.languages, ...existingShow.languages];
+                const languages: Array<string> = []
+
+                for (const l of rawLanguages) {
+                    if (!languages.includes(l)) {
+                        languages.push(l)
+                    }
+                }
+                show.languages = languages
+            }
+        });
+        return shows;
+    };
+
+    const allShows = mergeLanguagesOfDuplicates([...deShows, ...frShows]);
+
+    const unique = [...new Map(allShows.map((s) => [s.id, s])).values()];
+
+    return unique
 }
diff --git a/components/ADN/ListEpisodes.ts b/components/ADN/ListEpisodes.ts
index feef9af..171cacb 100644
--- a/components/ADN/ListEpisodes.ts
+++ b/components/ADN/ListEpisodes.ts
@@ -1,10 +1,10 @@
 import type { ADNEpisodes, ADNEpisodesFetch } from './Types'
 
-export async function getEpisodesWithShowIdADN(id: number) {
+export async function getEpisodesWithShowIdADN(id: number, lang: 'de' | 'fr') {
   const { data, error } = await useFetch<ADNEpisodesFetch>(`https://gw.api.animationdigitalnetwork.fr/video/show/${id}?offset=0&limit=-1&order=asc`, {
     method: 'GET',
     headers: {
-        "x-target-distribution": "de",
+        "x-target-distribution": lang,
     },
   })
 
diff --git a/components/ADN/Types.ts b/components/ADN/Types.ts
index 3b4630f..53621b4 100644
--- a/components/ADN/Types.ts
+++ b/components/ADN/Types.ts
@@ -4,7 +4,8 @@ export interface ADNSearchFetch {
     url: string
     title: string
     image2x: string
-    episodeCount: number
+    episodeCount: number,
+    languages: Array<string>
   }>
 }
 
diff --git a/components/Search/Types.ts b/components/Search/Types.ts
index c7e61ea..ab24023 100644
--- a/components/Search/Types.ts
+++ b/components/Search/Types.ts
@@ -1,43 +1,43 @@
 export interface CrunchyrollSearchResult {
-    ID: string;
-    Url: string;
-    Title: string;
-    Description: string;
-    Dubs: Array<string>;
-    Subs: Array<string>;
-    Episodes: number;
-    Seasons: number;
-    PEGI: Array<string>;
-    Year: number;
-    Images: {
-        poster_tall: Array<
-            Array<{
-                height: number;
-                source: string;
-                type: string;
-                width: number;
-            }>
-        >;
-        poster_wide: Array<
-            Array<{
-                height: number;
-                source: string;
-                type: string;
-                width: number;
-            }>
-        >;
-    };
+  ID: string
+  Url: string
+  Title: string
+  Description: string
+  Dubs: Array<string>
+  Subs: Array<string>
+  Episodes: number
+  Seasons: number
+  PEGI: Array<string>
+  Year: number
+  Images: {
+    poster_tall: Array<
+      Array<{
+        height: number
+        source: string
+        type: string
+        width: number
+      }>
+    >
+    poster_wide: Array<
+      Array<{
+        height: number
+        source: string
+        type: string
+        width: number
+      }>
+    >
+  }
 }
 
-export interface CrunchyrollSearchResults
-    extends Array<CrunchyrollSearchResult> {}
+export interface CrunchyrollSearchResults extends Array<CrunchyrollSearchResult> {}
 
 export interface ADNSearchResult {
-    id: number;
-    url: string;
-    title: string;
-    image2x: string;
-    episodeCount: number;
+  id: number
+  url: string
+  title: string
+  image2x: string
+  episodeCount: number
+  languages: Array<string>
 }
 
 export interface ADNSearchResults extends Array<ADNSearchResult> {}
diff --git a/pages/addanime.vue b/pages/addanime.vue
index a006da1..c92e7ba 100644
--- a/pages/addanime.vue
+++ b/pages/addanime.vue
@@ -19,9 +19,6 @@
           <option value="adn">ADN</option>
         </select>
       </div>
-      <div v-if="service === 'adn'" class="text-sm text-center">
-        ADN downloader is still in beta and can only download ADN Germany
-      </div>
       <div v-if="(isLoggedInCR && service === 'crunchyroll') || (isLoggedInADN && service === 'adn')" class="relative flex flex-col">
         <input
           v-model="search"
@@ -188,6 +185,12 @@
           </button>
         </div>
       </div>
+      <div v-if="service === 'adn'" class="relative flex flex-col select-none">
+        <div class="bg-[#5c5b5b] focus:outline-none px-3 py-2 rounded-xl text-sm text-center cursor-pointer">
+          Dub:
+          {{ selectedDubs.map((t) => t.name).join(', ') }}
+        </div>
+      </div>
       <div v-if="service === 'crunchyroll'" class="relative flex flex-col select-none">
         <div @click="selectSub ? (selectSub = false) : (selectSub = true)" class="bg-[#5c5b5b] focus:outline-none px-3 py-2 rounded-xl text-sm text-center cursor-pointer">
           Subs:
@@ -206,6 +209,12 @@
           </button>
         </div>
       </div>
+      <div v-if="service === 'adn'" class="relative flex flex-col select-none">
+        <div class="bg-[#5c5b5b] focus:outline-none px-3 py-2 rounded-xl text-sm text-center cursor-pointer">
+          Sub:
+          {{ selectedSubs.length !== 0 ? selectedSubs.map((t) => t.name).join(', ') : 'No Subs selected' }}
+        </div>
+      </div>
       <div class="flex flex-row gap-3">
         <div v-if="service === 'crunchyroll'" class="relative flex flex-col w-full">
           <select v-model="hardsub" name="episode" class="bg-[#5c5b5b] focus:outline-none px-3 py-2 rounded-xl text-sm text-center cursor-pointer" :disabled="isHardsubDisabled">
@@ -557,16 +566,15 @@ const switchToSeason = async () => {
   }
 
   if (ADNselectedShow.value) {
-    episodesADN.value = await getEpisodesWithShowIdADN(ADNselectedShow.value.id)
+    episodesADN.value = await getEpisodesWithShowIdADN(ADNselectedShow.value.id, 'de')
     if (!episodesADN.value) {
       isFetchingSeasons.value--
       return
-    }
+    };
     selectedStartEpisodeADN.value = episodesADN.value[0]
     selectedEndEpisodeADN.value = episodesADN.value[0]
     tab.value = 2
-    selectedDubs.value = [{ locale: 'ja-JP', name: 'JP' }],
-    selectedSubs.value = [{ locale: 'de-DE', name: 'DE' }]
+    selectedDubs.value = [{ locale: 'ja-JP', name: 'JP' }];
   }
 
   if (CRselectedShow.value) {
diff --git a/src/api/routes/adn/adn.service.ts b/src/api/routes/adn/adn.service.ts
index 7c49293..c6bcc68 100644
--- a/src/api/routes/adn/adn.service.ts
+++ b/src/api/routes/adn/adn.service.ts
@@ -217,9 +217,11 @@ export async function adnGetPlaylist(animeid: number) {
       const data: ADNLink = await JSON.parse(await response.text())
       return { data: data, secret: token.random }
     } else {
-      const data: ADNLink = JSON.parse(await response.text())
+      const data: {  message: string, code: string, statusCode: string} = JSON.parse(await response.text())
 
-      return { data: data, secret: token.random }
+      messageBox('error', ['Cancel'], 2, 'Failed to fetch Playlist', 'Failed to fetch ADN Playlist', `${data.message} - ${data.code}`)
+
+      return null
     }
   } catch (e) {
     throw new Error(e as string)
diff --git a/src/api/routes/service/service.service.ts b/src/api/routes/service/service.service.ts
index 7ed7ecf..a26bc53 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 } from '../../services/folder'
+import { createFolder, createFolderName, deleteFolder, deleteTemporaryFolders } from '../../services/folder'
 import { downloadADNSub, downloadCRSub } from '../../services/subs'
 import { CrunchyEpisode } from '../../types/crunchyroll'
 import { crunchyGetPlaylist, crunchyGetPlaylistMPD } from '../crunchyroll/crunchyroll.service'
@@ -44,6 +44,16 @@ export async function getPlaylist() {
   return episodes
 }
 
+
+// Delete Playlist and TEMP folders After Start
+async function deletePlaylistandTMP() {
+  await Playlist.truncate()
+
+  deleteTemporaryFolders()
+}
+
+deletePlaylistandTMP()
+
 // Update Playlist Item
 export async function updatePlaylistByID(id: number, status: 'waiting' | 'preparing' | 'downloading' | 'completed' | 'merging' | 'failed') {
   await Playlist.update({ status: status }, { where: { id: id } })
@@ -177,7 +187,10 @@ export async function downloadADNPlaylist(
   const seasonFolder = await createFolderName(`${name.replace(/[/\\?%*:|"<>]/g, '')} Season ${season}`, downloadPath)
 
   const subDownload = async () => {
-    if (!playlist) return
+    if (!playlist) {
+      await updatePlaylistByID(downloadID, 'failed')
+      return
+    }
     const sbs: Array<string> = []
     const name = await downloadADNSub(playlist.data.links.subtitles.all, subFolder, playlist.secret)
     sbs.push(name)
@@ -185,9 +198,10 @@ export async function downloadADNPlaylist(
   }
 
   const downloadVideo = async () => {
-    var code
-
-    if (!playlist) return
+    if (!playlist) {
+      await updatePlaylistByID(downloadID, 'failed')
+      return
+    }
 
     var link: string = '';
 
@@ -222,9 +236,17 @@ export async function downloadADNPlaylist(
 
   const [subss, file] = await Promise.all([subDownload(), downloadVideo()])
 
-  if (!subss) return
+  if (!subss) {
+    await updatePlaylistByID(downloadID, 'failed')
+    return
+  }
 
-  await mergeVideoFile(file as string, [], [], String(playlist?.data.video.guid), seasonFolder, `${name.replace(/[/\\?%*:|"<>]/g, '')} Season ${season} Episode ${episode}`, format)
+  if (!file) {
+    await updatePlaylistByID(downloadID, 'failed')
+    return
+  }
+
+  await mergeVideoFile(file as string, [], subss, String(playlist?.data.video.guid), seasonFolder, `${name.replace(/[/\\?%*:|"<>]/g, '')} Season ${season} Episode ${episode}`, format)
 
   await updatePlaylistByID(downloadID, 'completed')
 
diff --git a/src/api/services/folder.ts b/src/api/services/folder.ts
index 57f738d..ea6a0cd 100644
--- a/src/api/services/folder.ts
+++ b/src/api/services/folder.ts
@@ -3,7 +3,7 @@ import { app } from 'electron'
 import fs from 'fs'
 
 export async function createFolder() {
-  const tempFolderPath = path.join(app.getPath('documents'), (Math.random() + 1).toString(36).substring(2))
+  const tempFolderPath = path.join(app.getPath('documents'), `crd-tmp-${(Math.random() + 1).toString(36).substring(2)}`)
   try {
     await fs.promises.mkdir(tempFolderPath, { recursive: true })
     return tempFolderPath
@@ -52,3 +52,22 @@ export async function createFolderName(name: string, dir: string) {
 export async function deleteFolder(folderPath: string) {
   fs.rmSync(folderPath, { recursive: true, force: true })
 }
+
+export async function deleteTemporaryFolders() {
+  const documentsPath = app.getPath('documents');
+  const folderPrefix = 'crd-tmp-';
+
+  try {
+    const files = await fs.promises.readdir(documentsPath);
+    const tempFolders = files.filter(file => file.startsWith(folderPrefix));
+
+    for (const folder of tempFolders) {
+      const folderPath = path.join(documentsPath, folder);
+      await deleteFolder(folderPath);
+      console.log(`Temporary folder ${folder} deleted.`);
+    }
+  } catch (error) {
+    console.error('Error deleting temporary folders:', error);
+    throw error;
+  }
+}
diff --git a/src/api/services/subs.ts b/src/api/services/subs.ts
index c9578e8..8541b1e 100644
--- a/src/api/services/subs.ts
+++ b/src/api/services/subs.ts
@@ -39,11 +39,32 @@ export async function downloadCRSub(
   return path
 }
 
-export async function downloadADNSub(
-  link: string,
-  dir: string,
-  secret: string
-) {
+export async function downloadADNSub(link: string, dir: string, secret: string) {
+  var templateASS = `[Script Info]
+; Script generated by Aegisub 3.2.2
+; http://www.aegisub.org/
+Title: Deutsch
+Original Script: crd  [https://github.com/stratuma/]
+Original Translation: 
+Original Editing: 
+Original Timing: 
+Synch Point: 
+Script Updated By: 
+Update Details: 
+ScriptType: v4.00+
+PlayResX: 1920
+PlayResY: 1080
+Timer: 0.0000
+WrapStyle: 0
+
+[Aegisub Project Garbage]
+
+[V4+ Styles]
+Format: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour, BackColour, Bold, Italic, Underline, StrikeOut, ScaleX, ScaleY, Spacing, Angle, BorderStyle, Outline, Shadow, Alignment, MarginL, MarginR, MarginV, Encoding
+Style: Default,Arial,56,&H00FFFFFF,&H000000FF,&H00000000,&H00000000,-1,0,0,0,100,100,0,0,1,4,0,2,0,0,20,1
+
+[Events]
+Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text\n`
   const path = `${dir}/de-DE.ass`
 
   const stream = fs.createWriteStream(path)
@@ -57,6 +78,41 @@ export async function downloadADNSub(
 
   const subs = await ADNparseSub(rawSubs, secret)
 
+  const parsedSubs: {
+    vde: Array<{
+      startTime: number
+      endTime: number
+      positionAligh: string
+      lineAlign: string
+      text: string
+    }>,
+    vostde: Array<{
+      startTime: number
+      endTime: number
+      positionAligh: string
+      lineAlign: string
+      text: string
+    }>,
+  } = await JSON.parse(subs)
+
+  if (parsedSubs.vde) {
+    for (const s of parsedSubs.vde) {
+      const convertedStart = convertToTimeFormat(s.startTime)
+      const convertedEnd = convertToTimeFormat(s.endTime)
+  
+      templateASS = templateASS + `Dialogue: 0,${convertedStart},${convertedEnd},Default,,0,0,0,,${s.text.replace('\n', '\\N').replace('<i>', '{\\i1}').replace('</i>', '{\\i0}')}\n`
+    }
+  }
+
+  if (parsedSubs.vostde) {
+    for (const s of parsedSubs.vostde) {
+      const convertedStart = convertToTimeFormat(s.startTime)
+      const convertedEnd = convertToTimeFormat(s.endTime)
+  
+      templateASS = templateASS + `Dialogue: 0,${convertedStart},${convertedEnd},Default,,0,0,0,,${s.text.replace('\n', '\\N').replace('<i>', '{\\i1}').replace('</i>', '{\\i0}')}\n`
+    }
+  }
+
   // Disabling Changing ASS because still broken in vlc
 
   // parsedASS.info.PlayResX = "1920";
@@ -68,7 +124,7 @@ export async function downloadADNSub(
 
   // const fixed = stringify(parsedASS)
 
-  const readableStream = Readable.from([subs])
+  const readableStream = Readable.from([templateASS])
 
   await finished(readableStream.pipe(stream))
   console.log(`Sub downloaded`)
@@ -76,6 +132,26 @@ export async function downloadADNSub(
   return path
 }
 
+function convertToTimeFormat(time: number) {
+  var seconds: number | string = Math.floor(time);
+  var milliseconds = Math.round((time - seconds) * 1000);
+
+  var hours: number | string = Math.floor(seconds / 3600);
+  var minutes: number | string = Math.floor((seconds % 3600) / 60);
+  seconds = seconds % 60;
+
+  hours = String(hours).padStart(2, '0');
+  minutes = String(minutes).padStart(2, '0');
+  seconds = String(seconds).padStart(2, '0');
+
+  milliseconds = Math.round(milliseconds / 10);
+
+  var formattedMilliseconds = milliseconds < 10 ? '0' + milliseconds : milliseconds;
+
+  var formattedTime = hours + ':' + minutes + ':' + seconds + '.' + formattedMilliseconds;
+  return formattedTime;
+}
+
 export async function ADNparseSub(raw: string, secret: string) {
   var key = secret + '7fac1178830cfe0c'