From 6bf7e9721f0e14810364c8a8ff9e70b7158c037c Mon Sep 17 00:00:00 2001
From: IONI0 <soubsplus@gmail.com>
Date: Tue, 2 Apr 2024 16:45:29 +1100
Subject: [PATCH] Fixes/enchancements for vtt2ass with new Hidive Q styles

---
 modules/module.vtt2ass.ts | 91 ++++++++++++++++++++++++++++-----------
 1 file changed, 67 insertions(+), 24 deletions(-)

diff --git a/modules/module.vtt2ass.ts b/modules/module.vtt2ass.ts
index 30104f8..f790eac 100644
--- a/modules/module.vtt2ass.ts
+++ b/modules/module.vtt2ass.ts
@@ -239,6 +239,42 @@ function timestampToCentiseconds(timestamp: string) {
   return 360000 * hour + 6000 * minute + 100 * second + centisecond;
 }
 
+function combineLines(events: string[]): string[] {
+  // This function is for combining adjacent lines with same information
+  const newLines: string[] = [];
+  for (const currentLine of events) {
+    let hasCombined: boolean = false;
+    // Check previous 7 elements, arbritary lookback amount 
+    for (let j = 1; j < 8 && j < newLines.length; j++) {
+      const checkLine = newLines[newLines.length - j];
+      let checkLineSplit = checkLine.split(',');
+      let currentLineSplit = currentLine.split(',');
+      // 1 = start, 2 = end, 3 = style, 9+ = text
+      if (checkLineSplit.slice(9).join(',') == currentLineSplit.slice(9).join(',') &&
+          checkLineSplit[3] == currentLineSplit[3] &&
+          checkLineSplit[2] == currentLineSplit[1]
+      ) {
+          checkLineSplit[2] = currentLineSplit[2];
+          newLines[newLines.length - j] = checkLineSplit.join(',');
+          hasCombined = true;
+          break;
+      } 
+    }
+    if (!hasCombined) {
+      newLines.push(currentLine);
+    }
+  }
+  return newLines;
+}
+
+function pushBuffer(buffer: ReturnType<typeof convertLine>[], events: string[]) {
+  buffer.reverse();
+  const bufferStrings: string[] = buffer.map(line => 
+    `Dialogue: 1,${line.start},${line.end},${line.style},,0,0,0,,${line.text}`);
+  events.push(...bufferStrings);
+  buffer.splice(0,buffer.length);
+}
+
 function convert(css: Css, vtt: Vtt[]) {
   const stylesMap: Record<string, string> = {};
   let ass = [
@@ -274,8 +310,8 @@ function convert(css: Css, vtt: Vtt[]) {
     song_cap: [],
   };
   const linesMap: Record<string, number> = {};
-  let previousLine: ReturnType<typeof convertLine> | undefined = undefined;
-  const buffer: string[] = [];
+  const buffer: ReturnType<typeof convertLine>[] = [];
+  const captionsBuffer: string[] = [];
   for (const l in vtt) {
     const x = convertLine(stylesMap, vtt[l]);
     if (x.ind !== '' && linesMap[x.ind] !== undefined) {
@@ -295,34 +331,41 @@ function convert(css: Css, vtt: Vtt[]) {
     }
     /**
      * What cursed code have I brought upon this land?
-     * This checks if a subtitle should be multi-line, and if it is, pops the just inserted 
-     * subtitle and the previous subtitle, and merges them into a single subtitle.
+     * This handles making lines multi-line when neccesary and reverses
+     * order of subtitles so that they display correctly
      */
-    if (previousLine) {
-      const previousStart = timestampToCentiseconds(previousLine.start);
+    if (x.type != 'subtitle') {
+      // Do nothing
+    } else if (x.text.includes('\\pos')) {
+      events['subtitle'].pop();
+      captionsBuffer.push(x.res);
+    } else if (buffer.length > 0) {
+      const previousBufferLine = buffer[buffer.length - 1];
+      const previousStart = timestampToCentiseconds(previousBufferLine.start);
       const currentStart = timestampToCentiseconds(x.start);
-      if (
-        (currentStart - previousStart) <= 2 &&
-        previousLine.type == x.type && 
-        previousLine.style == x.style && 
-        !previousLine.text.includes('\\pos') && 
-        !x.text.includes('\\pos')
-      ) {
-        events[x.type as keyof typeof events].pop();
-        const previousLinePop = events[x.type as keyof typeof events].pop();
-        events[x.type as keyof typeof events].push(previousLinePop + '\\N'+x.text);
-      } else if ((currentStart - previousStart) <= 3) {
-        const currentLinePop = events[x.type as keyof typeof events].pop();
-        const previousLinePop = events[previousLine.type as keyof typeof events].pop();
-        buffer.push(previousLinePop as string, currentLinePop as string);
-      } else if ((currentStart - previousStart) >= 3 && buffer.length > 0) {
-        events[x.type as keyof typeof events].push(...buffer.reverse());
-        buffer.splice(0,buffer.length);
+      events['subtitle'].pop();
+      if ((currentStart - previousStart) <= 2) {
+        if (previousBufferLine.style == x.style) {
+          buffer.pop();
+          x.text = previousBufferLine.text + '\\N' + x.text;
+        } else {
+          x.start = previousBufferLine.start;
+        }
+      } else {
+        pushBuffer(buffer, events['subtitle']);
       }
+      buffer.push(x);
+    }
+    else {
+      events['subtitle'].pop();
+      buffer.push(x);
     }
-    previousLine = x;
   }
 
+  pushBuffer(buffer, events['subtitle']);
+  events['subtitle'].push(...captionsBuffer);
+  events['subtitle'] = combineLines(events['subtitle']);
+
   if (events.subtitle.length > 0) {
     ass = ass.concat(
       //`Comment: 0,0:00:00.00,0:00:00.00,${defaultStyleName},,0,0,0,,** Subtitles **`,