function.js

let docBlob; // Variable to store the docx blob
const btn_convert_1 = document.getElementById('btn_convert_1');
const btn_convert_2 = document.getElementById('btn_convert_2');
const btn_convert_3 = document.getElementById('btn_convert_3');
const btn_download = document.getElementById('btn_download');
const fileInput = document.getElementById('srtFile');
const result_div = document.getElementById('result_div');
const checkbox1 = document.getElementById('checkbox1');
let checkboxValue = false;
// copyrightYear function to set the current year in the #thisYear element
window.addEventListener('load', copyright, true);
function copyright() {
  const thisYear = new Date().getFullYear();
  document.getElementById('thisYear').innerHTML = thisYear;
}
// Enable the convert button once a file is selected
fileInput.addEventListener('change', function () {
  if (fileInput.files.length > 0) {
    btn_convert_1.disabled = false;
    btn_convert_2.disabled = false;
    btn_convert_3.disabled = false;
    checkbox1.disabled = false;
    const resultDiv = document.getElementById('result_div');
    resultDiv.innerHTML = ''; // Clear any previous content
  } else {
    btn_convert_1.disabled = true;
    btn_convert_2.disabled = true;
    btn_convert_3.disabled = true;
    checkbox1.disabled = true;
  }
});
// Start conversion process when the button is clicked
btn_convert_1.addEventListener('click', () => {
  convertSrtToDocx(1);
});
btn_convert_2.addEventListener('click', () => {
  convertSrtToDocx(2);
});
btn_convert_3.addEventListener('click', () => {
  convertSrtToDocx(3);
});
checkbox1.addEventListener('click', () => {
  const isValid = validateCheckbox();
  console.log(isValid);
  btn_convert_1.disabled = false;
  btn_convert_2.disabled = false;
  btn_convert_3.disabled = false;
});

// Check if the checkbox is checked
function validateCheckbox() {
  if (checkbox1.checked) {
    return true;
  } else {
    return false;
  }
}

// Converts an uploaded SRT file to DOCX format
function convertSrtToDocx(convertMode) {
  result_div.classList.add('glow'); // Add glow effect to the result div
  setTimeout(() => {
    result_div.classList.remove('glow');
  }, 400);
  const file = fileInput.files[0];
  if (!file) {
    alert('Please upload a .srt file first.');
    return;
  }
  const reader = new FileReader();
  reader.onload = function (event) {
    const srtText = event.target.result;
    let result;
    if (convertMode === 1) {
      result = parseSrtM1(srtText);
      btn_convert_1.disabled = true;
      btn_convert_2.disabled = false;
      btn_convert_3.disabled = false;
    } else if (convertMode === 2) {
      result = parseSrtM2(srtText);
      btn_convert_1.disabled = false;
      btn_convert_2.disabled = true;
      btn_convert_3.disabled = false;
    } else if (convertMode === 3) {
      result = parseSrtM3(srtText);
      btn_convert_1.disabled = false;
      btn_convert_2.disabled = false;
      btn_convert_3.disabled = true;
    }
    displayResult(result);
    // Scroll to the first <p> inside result_div
    const firstParagraph = result_div.querySelector('p');
    if (firstParagraph) {
      firstParagraph.scrollIntoView({ behavior: 'smooth' });
    }
  };
  reader.readAsText(file);
  btn_download.disabled = false;
}
// Parses the SRT text into an array of sentences and counts the dialogues
// Mode 1: Preserves original line breaks only
function parseSrtM1(srtContent) {
  // Split the SRT content by lines
  const lines = srtContent.split('\n');
  // Initialize an empty array to store the trimmed dialogue
  const trimmedDialogue = [];
  // Initialize a variable to hold the current dialogue
  let currentDialogue = [];
  for (let i = 0; i < lines.length; i++) {
    const line = lines[i].trim();
    // Skip the line number and timestamp lines
    if (line === '' || /^\d+$/.test(line) || /^(\d{2}):(\d{2}):(\d{2}),(\d{3}) --> (\d{2}):(\d{2}):(\d{2}),(\d{3})$/.test(line)) {
      // If we have accumulated dialogue, push it to the trimmedDialogue array
      if (currentDialogue.length > 0) {
        trimmedDialogue.push(currentDialogue.join('\n'));
        currentDialogue = []; // Reset currentDialogue for the next dialogue
      }
      continue; // Skip to the next iteration
    }
    // If the line is part of the dialogue, add it to the currentDialogue
    currentDialogue.push(line);
  }
  // After the loop, check if there's any remaining dialogue to push
  if (currentDialogue.length > 0) {
    trimmedDialogue.push(currentDialogue.join('\n'));
  }
  let parsedSentences;
  // Check do we need line breaks after punctuation marks
  if (checkbox1.checked) {
    // Join all dialogues with an empty line between each
    parsedSentences = trimmedDialogue.join('\n\n\n');
  } else {
    parsedSentences = trimmedDialogue.join('\n');
  }
  const dialogueCount = trimmedDialogue.length;
  return { parsedSentences, dialogueCount };
}
// Mode 2: Strips out original line breaks and add line breaks after punctuation marks for each dialogue
function parseSrtM2(srtContent) {
  // Split the SRT content by lines
  const lines = srtContent.split('\n');
  // Initialize an empty array to store the trimmed dialogue
  const trimmedDialogue = [];
  // Initialize a variable to hold the current dialogue
  let currentDialogue = [];
  for (let i = 0; i < lines.length; i++) {
    const line = lines[i].trim();
    // Skip the line number and timestamp lines
    if (line === '' || /^\d+$/.test(line) || /^(\d{2}):(\d{2}):(\d{2}),(\d{3}) --> (\d{2}):(\d{2}):(\d{2}),(\d{3})$/.test(line)) {
      // If we have accumulated dialogue, push it to the trimmedDialogue array
      if (currentDialogue.length > 0) {
        let joinedDialogue = currentDialogue.join(' '); // Join with a space
        joinedDialogue = joinedDialogue.replace(/("|\.{3}|\.|!|\?)[^\S\r\n]|(……|⋯⋯|。|！|？)(?![\n”"」』…⋯。！？]|$)/g, '$1$2\n'); // Add line breaks
        trimmedDialogue.push(joinedDialogue);
        currentDialogue = []; // Reset currentDialogue for the next dialogue
      }
      continue; // Skip to the next iteration
    }
    // If the line is part of the dialogue, add it to the currentDialogue
    currentDialogue.push(line);
  }
  // After the loop, check if there's any remaining dialogue to push
  if (currentDialogue.length > 0) {
    let joinedDialogue = currentDialogue.join(' '); // Join with a space
    joinedDialogue = joinedDialogue.replace(/("|\.{3}|\.|!|\?)[^\S\r\n]|(……|⋯⋯|。|！|？)(?![\n”"」』…⋯。！？]|$)/g, '$1$2\n'); // Add line breaks
    trimmedDialogue.push(joinedDialogue);
  }
  let parsedSentences;
  // Check do we need line breaks after punctuation marks
  if (checkbox1.checked) {
    // Join all dialogues with an empty line between each
    parsedSentences = trimmedDialogue.join('\n\n\n');
  } else {
    parsedSentences = trimmedDialogue.join('\n');
  }
  const dialogueCount = trimmedDialogue.length;
  return { parsedSentences, dialogueCount };
}
// Mode 3: Merges all dialogue and group them into paragraphs by adding line breaks after punctuation marks
function parseSrtM3(srtContent) {
  // Split the SRT content by lines
  const lines = srtContent.split('\n');
  // Initialize an empty array to store the trimmed dialogue
  const trimmedDialogue = [];
  // Initialize a variable to hold the current dialogue
  let currentDialogue = [];
  for (let i = 0; i < lines.length; i++) {
    const line = lines[i].trim();
    // Skip the line number and timestamp lines
    if (line === '' || /^\d+$/.test(line) || /^(\d{2}):(\d{2}):(\d{2}),(\d{3}) --> (\d{2}):(\d{2}):(\d{2}),(\d{3})$/.test(line)) {
      // If we have accumulated dialogue, push it to the trimmedDialogue array
      if (currentDialogue.length > 0) {
        trimmedDialogue.push(currentDialogue.join(' '));
        currentDialogue = []; // Reset currentDialogue for the next dialogue
      }
      continue; // Skip to the next iteration
    }
    // If the line is part of the dialogue, add it to the currentDialogue
    currentDialogue.push(line);
  }
  // After the loop, check if there's any remaining dialogue to push
  if (currentDialogue.length > 0) {
    trimmedDialogue.push(currentDialogue.join(' '));
  }
  // Join all dialogues with space between each
  let parsedSentences = trimmedDialogue.join(' ');
  // Check do we need line breaks after punctuation marks
  if (checkbox1.checked) {
    // Use the regular expression to insert line breaks
    parsedSentences = parsedSentences.replace(/("|\.{3}|\.|!|\?)[^\S\r\n]|(……|⋯⋯|。|！|？)(?![\n”"」』…⋯。！？]|$)/g, '$1$2\n\n\n');
  } else {
    parsedSentences = parsedSentences.replace(/("|\.{3}|\.|!|\?)[^\S\r\n]|(……|⋯⋯|。|！|？)(?![\n”"」』…⋯。！？]|$)/g, '$1$2\n');
  }
  const dialogueCount = trimmedDialogue.length;
  return { parsedSentences, dialogueCount };
}
// Displays the parsed sentences in the result div
function displayResult(result) {
  const resultDiv = document.getElementById('result_div');
  resultDiv.innerHTML = ''; // Clear previous results
  // Display the count of dialogues
  const countParagraph = document.createElement('p');
  countParagraph.textContent = `*** Original Dialogues: ${result.dialogueCount} ***`;
  resultDiv.appendChild(countParagraph);
  // Add a space line
  const spaceLine = document.createElement('p');
  spaceLine.innerHTML = '<br>';
  resultDiv.appendChild(spaceLine);
  // Ensure parsedSentences is a string
  if (typeof result.parsedSentences === 'string') {
    result.parsedSentences.split('\n\n').forEach((sentence) => {
      const p = document.createElement('p');
      // Replace newline characters with <br> tags
      p.innerHTML = sentence.replace(/\n/g, '<br>');
      // If the sentence is empty, add a <br> tag
      if (sentence === '') {
        p.innerHTML = '<br>';
      }
      resultDiv.appendChild(p);
    });
  } else {
    console.error('Expected a string but got:', result.parsedSentences);
  }
  generateDocxFile(); // Prepare the docx file for download
}
// Generates a DOCX file from the parsed sentences
function generateDocxFile() {
  const resultDiv = document.getElementById('result_div');
  const paragraphs = resultDiv.querySelectorAll('p');
  // Split each paragraph by <br> and create separate Paragraph objects
  const children = Array.from(paragraphs).flatMap((p) => {
    return p.innerHTML.split(/<br\s*\/?>/gi).map((line) => new docx.Paragraph(line.trim()));
  });
  const doc = new docx.Document({
    creator: 'SRT to DOCX Converter',
    title: 'Subtitle Document',
    sections: [
      {
        children: children,
      },
    ],
  });
  docx.Packer.toBlob(doc).then((blob) => {
    docBlob = blob; // Store the blob for later download
  });
}
// Downloads the generated DOCX file
function downloadDocx() {
  if (!docBlob) {
    alert('Please convert a file before downloading.');
    return;
  }
  const fileName = fileInput.files[0].name.replace('.srt', '');
  const url = URL.createObjectURL(docBlob);
  const link = document.createElement('a');
  link.href = url;
  link.download = `${fileName}.docx`;
  link.click();
  // Reset after download
  fileInput.value = '';
  const resultDiv = document.getElementById('result_div');
  resultDiv.innerHTML = '';
  btn_convert_1.disabled = true;
  btn_convert_2.disabled = true;
  btn_convert_3.disabled = true;
  btn_download.disabled = true;
  checkbox1.disabled = true;
  docBlob = null;
}