-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils.js
197 lines (145 loc) · 5.21 KB
/
utils.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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
import fs from 'node:fs';
import got from 'got';
import { createFFmpeg, fetchFile } from '@ffmpeg/ffmpeg';
import sdk from "microsoft-cognitiveservices-speech-sdk";
const ffmpegInstance = createFFmpeg({ log: false });
let ffmpegLoadingPromise = ffmpegInstance.load();
export async function getFFmpeg() {
if (ffmpegLoadingPromise) {
await ffmpegLoadingPromise;
ffmpegLoadingPromise = undefined;
}
return ffmpegInstance;
}
export async function getFileFromVoice(bot, voice){
const fileId = voice.file_id
// Retrieve the file path using the file ID
const fileInfo = await bot.telegram.getFile(fileId);
console.log('fileInfo', fileInfo);
const fileLink = `https://api.telegram.org/file/bot${process.env.BOT_TOKEN}/${fileInfo.file_path}`;
//const filePath = `/tmp/${fileId}.oga`;
//await downloadFile(fileLink, filePath);
return fileLink;
}
export async function downloadFile(fileUrl, filePath) {
return new Promise((resolve, reject) => {
try {
got.stream(fileUrl)
.pipe(fs.createWriteStream(filePath))
.on('close', function () {
//console.log('File written!', filePath);
resolve();
});
} catch (error) {
console.error(`Error downloading file: ${error.message}`);
reject();
}
});
}
export async function fetchFileToAB(fileUrl){
return await fetchFile(fileUrl);
}
export async function ogaBuffer2Mp3Buffer(buffer){
const ffmpeg = await getFFmpeg();
const inputFileName = `input.oga`;
const outputFileName = `output.mp3`;
let outputData = null;
ffmpeg.FS('writeFile', inputFileName, new Uint8Array(buffer));
await ffmpeg.run(
'-i', inputFileName,
'-acodec', 'libmp3lame',
outputFileName
);
outputData = ffmpeg.FS('readFile', outputFileName);
//console.log(outputData);
ffmpeg.FS('unlink', inputFileName);
ffmpeg.FS('unlink', outputFileName);
return outputData;
}
export async function mp3Buffer2OgaBuffer(buffer){
const ffmpeg = await getFFmpeg();
const inputFileName = `input.mp3`;
const outputFileName = `output.oga`;
let outputData = null;
ffmpeg.FS('writeFile', inputFileName, new Uint8Array(buffer));
await ffmpeg.run(
'-i', inputFileName,
'-c:a libvorbis',
'-q:a 4',
outputFileName
);
outputData = ffmpeg.FS('readFile', outputFileName);
ffmpeg.FS('unlink', inputFileName);
ffmpeg.FS('unlink', outputFileName);
return outputData;
}
export async function abToFile(ab, file){
const buffer = Buffer.from(ab);
return new Promise((resolve, reject) => {
fs.writeFile(file, buffer, (err) => {
if(err){
console.log('write file error', error);
return reject();
}
resolve(file);
});
});
}
export async function fileToab(file){
return new Promise((resolve, reject) => {
fs.readFile(file, (err, data) => {
if(err){
console.log('read file error', error);
return reject();
}
resolve(data);
});
});
}
export async function getFileFromVoiceAndConvertToMp3(bot, voice){
const fileLink = await getFileFromVoice(bot, voice);
const filePath = `/tmp/${voice.file_unique_id}.oga`;
await downloadFile(fileLink, filePath);
const ogaab = await fileToab(filePath);
const mp3ab = await ogaBuffer2Mp3Buffer(ogaab);
const mp3file = await abToFile(mp3ab, `/tmp/${voice.file_unique_id}.mp3`);
return mp3file;
}
export async function convertMp3ToOga(mp3file){
const mp3ab = await fileToab(mp3file);
const ogaab = await mp3Buffer2OgaBuffer(mp3ab);
const ogaFileName = mp3file.replace('.mp3', '.oga');
const ogafile = await abToFile(ogaab, ogaFileName);
return ogafile;
}
export async function tts(text, lang, audioFile){
const speechConfig = sdk.SpeechConfig.fromSubscription(process.env.SPEECH_KEY, process.env.SPEECH_REGION);
const audioConfig = sdk.AudioConfig.fromAudioFileOutput(audioFile);
if(lang == 'zh'){
speechConfig.speechSynthesisLanguage = "zh-CN";
speechConfig.speechSynthesisVoiceName = "zh-CN-XiaoxiaoNeural";
}else{
speechConfig.speechSynthesisLanguage = "en-US";
speechConfig.speechSynthesisVoiceName = "en-US-JennyNeural";
}
var synthesizer = new sdk.SpeechSynthesizer(speechConfig, audioConfig);
console.log("Now synthesizing to: " + audioFile);
return new Promise((resolve, reject) => {
synthesizer.speakTextAsync(text, function (result) {
if (result.reason === sdk.ResultReason.SynthesizingAudioCompleted) {
console.log("synthesis finished.");
resolve();
} else {
console.error("Speech synthesis canceled", result.errorDetails);
reject();
}
synthesizer.close();
synthesizer = null;
}, function (err) {
console.trace("err - " + err);
synthesizer.close();
synthesizer = null;
reject();
});
});
}