vtt

Author: Moebytes

.DS_Store

@@ -25,8 +25,8 @@ import coverButton from "../assets/icons/cover.png"
 import coverButtonHover from "../assets/icons/cover-hover.png"
 import renameButton from "../assets/icons/rename.png"
 import renameButtonHover from "../assets/icons/rename-hover.png"
-import mkvButton from "../assets/icons/mkv.png"
-import mkvButtonHover from "../assets/icons/mkv-hover.png"
+import vttButton from "../assets/icons/vtt.png"
+import vttButtonHover from "../assets/icons/vtt-hover.png"
 import "../styles/titlebar.less"
 
 const TitleBar: React.FunctionComponent = (props) => {
@@ -41,7 +41,7 @@ const TitleBar: React.FunctionComponent = (props) => {
     const [hoverPDF, setHoverPDF] = useState(false)
     const [hoverCover, setHoverCover] = useState(false)
     const [hoverRename, setHoverRename] = useState(false)
-    const [hoverMKV, setHoverMKV] = useState(false)
+    const [hoverVTT, setHoverVTT] = useState(false)
     const [theme, setTheme] = useState("light")
 
     useEffect(() => {
@@ -98,7 +98,7 @@ const TitleBar: React.FunctionComponent = (props) => {
         if (files?.[0]) ipcRenderer.invoke("rename", files)
     }
 
-    const mkv = async () => {
+    const vtt = async () => {
         const files = await ipcRenderer.invoke("multi-open", "subs")
         if (files?.[0]) ipcRenderer.invoke("extract-subtitles", files)
     }
@@ -143,7 +143,7 @@ const TitleBar: React.FunctionComponent = (props) => {
                     </div>
                     <div className="title-bar-buttons">
                     <img src={hoverTheme ? (theme === "light" ? darkButtonHover : lightButtonHover) : (theme === "light" ? darkButton : lightButton)} height="20" width="20" className="title-bar-button theme-button" onClick={() => changeTheme()} onMouseEnter={() => setHoverTheme(true)} onMouseLeave={() => setHoverTheme(false)}/>
-                        <img src={hoverMKV ? mkvButtonHover : mkvButton} height="20" width="20" className="title-bar-button mkv-button" onClick={mkv} onMouseEnter={() => setHoverMKV(true)} onMouseLeave={() => setHoverMKV(false)}/>
+                        <img src={hoverVTT ? vttButtonHover : vttButton} height="20" width="20" className="title-bar-button mkv-button" onClick={vtt} onMouseEnter={() => setHoverVTT(true)} onMouseLeave={() => setHoverVTT(false)}/>
                         <img src={hoverRename ? renameButtonHover : renameButton} height="20" width="20" className="title-bar-button rename-button" onClick={rename} onMouseEnter={() => setHoverRename(true)} onMouseLeave={() => setHoverRename(false)}/>
                         <img src={hoverCover ? coverButtonHover : coverButton} height="20" width="20" className="title-bar-button cover-button" onClick={cover} onMouseEnter={() => setHoverCover(true)} onMouseLeave={() => setHoverCover(false)}/>
                         <img src={hoverPDF ? pdfButtonHover : pdfButton} height="20" width="20" className="title-bar-button pdf-button" onClick={pdf} onMouseEnter={() => setHoverPDF(true)} onMouseLeave={() => setHoverPDF(false)}/>

assets/.DS_Store

@@ -10,6 +10,7 @@ declare module "sharp-phash/distance"
 declare module "image-pixels"
 declare module "bing-translate-api"
 declare module "mkv-subtitle-extractor"
+declare module "srt-to-vtt"
 
 type OpenImageProps = {
     label: string

assets/icons/vtt-hover.png

@@ -22,6 +22,7 @@ import Helvetica from "pdfkit/js/data/Helvetica.afm"
 import PDFDocument from "pdfkit"
 import child_process from "child_process"
 import mkvExtractor from "mkv-subtitle-extractor"
+import srt2vtt from "srt-to-vtt"
 
 import util from "util"
 
@@ -45,6 +46,26 @@ const queue: Array<{started: boolean, info: any}> = []
 if (!fs.existsSync(path.join(__dirname, "data"))) fs.mkdirSync(path.join(__dirname, "data"))
 fs.writeFileSync(path.join(__dirname, "data/Helvetica.afm"), Helvetica)
 
+const srtToVtt = async (subtitles: string[]) => {
+  for (let i = 0; i < subtitles.length; i++) {
+    await new Promise<void>((resolve, reject) => {
+      const readStream =  fs.createReadStream(subtitles[i]).pipe(srt2vtt())
+      const writeStream = fs.createWriteStream(`${path.dirname(subtitles[i])}/${path.basename(subtitles[i], path.extname(subtitles[i]))}.vtt`)
+      readStream.pipe(writeStream)
+          .on("end", () => resolve())
+          .on("finish", () => resolve())
+    })
+  }
+
+  const promiseArray: any[] = []
+  for (let i = 0; i < subtitles.length; i++) {
+    promiseArray.push(new Promise<void>((resolve) => {
+      fs.unlink(subtitles[i], () => resolve())
+    }))
+  }
+  await Promise.all(promiseArray)
+}
+
 const extractSubtitles = async (videos: string[]) => {
   for (let i = 0; i < videos.length; i++) {
     await mkvExtractor(videos[i])
@@ -63,14 +84,24 @@ const extractSubtitles = async (videos: string[]) => {
 ipcMain.handle("extract-subtitles", async (event, files: string[]) => {
   const directories = files.filter((f) => fs.lstatSync(f).isDirectory())
   const videos = files.filter((f) => path.extname(f).toLowerCase() === ".mkv")
+  const subtitles = files.filter((f) => path.extname(f).toLowerCase() === ".srt")
 
   let openDir = ""
 
   for (let i = 0; i < directories.length; i++) {
     const dir = directories[i]
-    let videos = fs.readdirSync(dir).map((i) => path.join(dir, i))
-    videos = videos.filter((f) => path.extname(f).toLowerCase() === ".mkv")
-    await extractSubtitles(videos)
+    let files = fs.readdirSync(dir).map((i) => path.join(dir, i))
+    let videos = files.filter((f) => path.extname(f).toLowerCase() === ".mkv")
+    let subs = files.filter((f) => path.extname(f).toLowerCase() === ".srt")
+    if (videos.length) {
+      await extractSubtitles(videos)
+      let files = fs.readdirSync(dir).map((i) => path.join(dir, i))
+      let subs = files.filter((f) => path.extname(f).toLowerCase() === ".srt")
+      await srtToVtt(subs)
+    }
+    if (subs.length) {
+      await srtToVtt(subs)
+    }
     try {
       fs.rmdirSync(dir)
     } catch (e) {
@@ -81,8 +112,16 @@ ipcMain.handle("extract-subtitles", async (event, files: string[]) => {
 
   if (videos.length) {
     await extractSubtitles(videos)
+    let subs = fs.readdirSync(path.dirname(videos[0])).map((i) => path.join(path.dirname(videos[0]), i))
+    subs = subs.filter((f) => path.extname(f).toLowerCase() === ".srt")
+    await srtToVtt(subs)
     if (!openDir) openDir = videos[0]
   }
+
+  if (subtitles.length) {
+    await srtToVtt(subtitles)
+    if (!openDir) openDir = subtitles[0]
+  }
   shell.openPath(path.dirname(openDir))
 })