-
Notifications
You must be signed in to change notification settings - Fork 15
/
index.js
116 lines (109 loc) · 3.29 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
const fs = require("fs");
//this extracts data from a json file
function extractData(filename) {
const data = fs.readFileSync(filename);
return JSON.parse(data);
}
function msToSrt(timeInMs) {
const convertMs = Math.floor(timeInMs / 1000);
const ms = convertMs % 1000;
const totalSeconds = (convertMs - ms) / 1000;
const seconds = totalSeconds % 60;
const totalMinutes = (totalSeconds - seconds) / 60;
const minutes = totalMinutes % 60;
const hour = (totalMinutes - minutes) / 60;
return `${hour < 10 ? "0" + hour : hour}:${
minutes < 10 ? "0" + minutes : minutes
}:${seconds < 10 ? "0" + seconds : seconds},${ms}`;
}
var draftFileName = "draft_content.json";
let os = process.env.os;
switch (os) {
case "Windows_NT":
draftFileName = "draft_content.json";
break;
case "Darwin":
draftFileName = "draft_info.json";
break;
case "Linux":
draftFileName = "draft_info.json";
}
console.log("file: " + draftFileName);
const data = extractData(draftFileName);
const { materials, tracks } = data;
// let regexOptions = [ /\<size.*?\>((.|\n)*?)\<\/size.*?\>/, /\[((.|\n)*?)\]/ ]
let subTrackNumber = 1;
let subTiming = tracks[subTrackNumber].segments;
var subtitlesInfo = materials.texts.map((i) => {
// var matches = regexOptions.map(r => i.content.match(r))
let content = i.content
.replace(/\<.*?\>/g, "")
.replace(/\<\/.*?\>/g, "")
.replace(/\[|\]/g, "");
try {
let contentV3 = JSON.parse(i.content);
if (contentV3) {
content = contentV3.text;
}
} catch (error) {}
// console.log(matches)
// var content = matches.reduce((acc, m) => {
// if (m){
// // console.log(i.id, m[1])
// if (m[1].substring(0,1) === '<') {
// return acc}
// else {
// // console.log(acc+m[1])
// return acc + m[1]
// }
// } else {
// return acc
// }
// }
// , '')
return {
content,
id: i.id,
};
});
// console.log(subtitlesInfo)
subtitlesInfo = subtitlesInfo.map((s, i) => {
let segment = subTiming.find((i) => i.material_id === s.id);
while (!segment) {
subTrackNumber++;
subTiming = tracks[subTrackNumber].segments;
segment = subTiming.find((i) => i.material_id === s.id);
}
s.start = segment.target_timerange.start;
s.end = s.start + segment.target_timerange.duration;
s.srtStart = msToSrt(s.start);
s.srtEnd = msToSrt(s.end);
s.subNumber = i + 1;
s.srtTiming = s.srtStart + " --> " + s.srtEnd;
return s;
});
// console.log(subtitlesInfo)
const srtOut = subtitlesInfo.reduce((srt, i) => {
const subtitle = `${i.subNumber}\n${i.srtTiming}\n${i.content}\n\n`;
return srt + subtitle;
}, "");
const copyOut = subtitlesInfo.reduce((srt, i) => {
const subtitle = `${i.content}\n`;
return srt + subtitle;
}, "");
//this function writes the string to a file
function writeToFile(data, filename) {
console.log("Saving subtitles to file...");
fs.writeFileSync(filename, data);
console.log("Done!");
// fs.writeFileSync('subtitles.json', JSON.stringify(subtitlesInfo));
}
if (process.argv[2] === "--txt") {
console.log("Copy extraction is a courtesy of @dellucanil");
writeToFile(copyOut, "copy.txt");
} else {
writeToFile(srtOut, "subtitles.srt");
console.log(
'Run "node index.js --txt" to get a copy version of the subtitles, courtesy of @dellucanil'
);
}