Add hhit color

server-plugins/generate-captions.js

@@ -63,9 +63,16 @@ export class CaptionPlugin extends Plugin {
       fontSize = 72,
       wordsPerGroup = 4,
       videoWidth = 1920,
-      videoHeight = 1080
+      videoHeight = 1080,
+      fontColor = '#FFFFFF',
+      fontHighlightColor = '#00FF00'
     } = options;
 
+    const assFontColor = hexToASSColor(fontColor);
+    const assHighlightColor = hexToASSColor(fontHighlightColor);
+    const assHighlightColorInline = `${assHighlightColor}&`;
+    const assFontColorInline = `${assFontColor}&`;
+
     // Read and parse JSON file
     const jsonData = JSON.parse(readFileSync(captionFilePath, 'utf-8'));
     const transcript = jsonData.transcript || '';
@@ -86,7 +93,19 @@ export class CaptionPlugin extends Plugin {
     const output = createWriteStream(outputFilePath);
 
     // Write header with calculated margins
-    output.write(createASSHeader(videoWidth, videoHeight, fontName, fontSize, translateY, sideMargin, sideMargin));
+    output.write(
+      createASSHeader(
+        videoWidth,
+        videoHeight,
+        fontName,
+        fontSize,
+        translateY,
+        sideMargin,
+        sideMargin,
+        assFontColor,
+        assHighlightColor
+      )
+    );
 
     // Process words in groups respecting sentence boundaries
     let i = 0;
@@ -135,7 +154,7 @@ export class CaptionPlugin extends Plugin {
         const captionParts = wordGroup.map((w, idx) => {
           if (idx === wordIdx) {
             // Current word - highlighted in green
-            return `{\\c&H00FF00&}${w.word}{\\c&HFFFFFF&}`;
+            return `{\\c${assHighlightColorInline}}${w.word}{\\c${assFontColorInline}}`;
           } else {
             // Other words - white
             return w.word;
@@ -234,7 +253,17 @@ function assignSentenceToWords(words, transcript) {
  * @param {number} marginV 
  * @returns {string}
  */
-function createASSHeader(videoWidth = 1920, videoHeight = 1080, fontName = 'Impact', fontSize = 72, marginV = 200, marginL = 10, marginR = 10) {
+function createASSHeader(
+  videoWidth = 1920,
+  videoHeight = 1080,
+  fontName = 'Impact',
+  fontSize = 72,
+  marginV = 200,
+  marginL = 10,
+  marginR = 10,
+  primaryColor = '&H00FFFFFF',
+  highlightColor = '&H0000FF00'
+) {
   return `[Script Info]
 Title: Word-by-Word Captions
 ScriptType: v4.00+
@@ -245,11 +274,27 @@ ScaledBorderAndShadow: yes
 
 [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,${fontName},${fontSize},&H00FFFFFF,&H000000FF,&H00000000,&H80000000,-1,0,0,0,100,100,0,0,1,3,2,2,${marginL},${marginR},${marginV},1
-Style: Highlight,${fontName},${fontSize},&H0000FF00,&H000000FF,&H00000000,&H80000000,-1,0,0,0,100,100,0,0,1,3,2,2,${marginL},${marginR},${marginV},1
+Style: Default,${fontName},${fontSize},${primaryColor},&H000000FF,&H00000000,&H80000000,-1,0,0,0,100,100,0,0,1,3,2,2,${marginL},${marginR},${marginV},1
+Style: Highlight,${fontName},${fontSize},${highlightColor},&H000000FF,&H00000000,&H80000000,-1,0,0,0,100,100,0,0,1,3,2,2,${marginL},${marginR},${marginV},1
 
 [Events]
 Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text
 `;
 }
 
+function hexToASSColor(hexValue) {
+  if (typeof hexValue !== 'string') {
+    throw new Error('fontColor values must be hex strings like #RRGGBB');
+  }
+
+  const normalized = hexValue.trim().replace('#', '');
+  if (!/^[0-9a-fA-F]{6}$/.test(normalized)) {
+    throw new Error(`Invalid hex color provided: ${hexValue}`);
+  }
+
+  const r = normalized.slice(0, 2);
+  const g = normalized.slice(2, 4);
+  const b = normalized.slice(4, 6);
+  return `&H00${b}${g}${r}`.toUpperCase();
+}
+