NeoAnd/evolution-api
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/evolution-api.git synced 2025-07-23 04:22:02 -06:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: audio convertion with stream to mp4

Browse Source
This commit is contained in:
Davidson Gomes 2024-06-10 14:01:07 -03:00
parent 9cd50833cf
commit 6d6f7293f9
1 changed files with 41 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
src/api/services/channels/whatsapp.baileys.service.ts
View File

@ -38,7 +38,6 @@ import makeWASocket, {
import { Label } from '@whiskeysockets/baileys/lib/Types/Label'; import { Label } from '@whiskeysockets/baileys/lib/Types/Label';
import { LabelAssociation } from '@whiskeysockets/baileys/lib/Types/LabelAssociation'; import { LabelAssociation } from '@whiskeysockets/baileys/lib/Types/LabelAssociation';
import axios from 'axios'; import axios from 'axios';
import { exec } from 'child_process';
import { isBase64, isURL } from 'class-validator'; import { isBase64, isURL } from 'class-validator';
import EventEmitter2 from 'eventemitter2'; import EventEmitter2 from 'eventemitter2';
// import { exec } from 'child_process'; // import { exec } from 'child_process';
@ -2118,13 +2117,11 @@ export class BaileysStartupService extends ChannelStartupService {
} }
if (status.type === 'audio') { if (status.type === 'audio') {
const convert = await this.processAudioMp4(status.content, 'status@broadcast'); const convert = await this.processAudioMp4(status.content);
if (typeof convert === 'string') { if (Buffer.isBuffer(convert)) {
const audio = fs.readFileSync(convert).toString('base64');
const result = { const result = {
content: { content: {
audio: Buffer.from(audio, 'base64'), audio: convert,
ptt: true, ptt: true,
mimetype: 'audio/ogg; codecs=opus', mimetype: 'audio/ogg; codecs=opus',
}, },
@ -2133,8 +2130,6 @@ export class BaileysStartupService extends ChannelStartupService {
}, },
}; };
fs.unlinkSync(convert);
return result; return result;
} else { } else {
throw new InternalServerErrorException(convert); throw new InternalServerErrorException(convert);
@ -2326,39 +2321,54 @@ export class BaileysStartupService extends ChannelStartupService {
); );
} }
public async processAudioMp4(audio: string, number: string) { public async processAudioMp4(audio: string) {
let tempAudioPath: string; let inputAudioStream: PassThrough;
let outputAudio: string;
number = number.replace(/\D/g, '');
const hash = `${number}-${new Date().getTime()}`;
if (isURL(audio)) { if (isURL(audio)) {
outputAudio = `${join(this.storePath, 'temp', `${hash}.mp4`)}`;
tempAudioPath = `${join(this.storePath, 'temp', `temp-${hash}.mp3`)}`;
const timestamp = new Date().getTime(); const timestamp = new Date().getTime();
const url = `${audio}?timestamp=${timestamp}`; const url = `${audio}?timestamp=${timestamp}`;
const response = await axios.get(url, { responseType: 'arraybuffer' }); const config: any = {
responseType: 'stream',
};
fs.writeFileSync(tempAudioPath, response.data); const response = await axios.get(url, config);
inputAudioStream = response.data.pipe(new PassThrough());
} else { } else {
outputAudio = `${join(this.storePath, 'temp', `${hash}.mp4`)}`;
tempAudioPath = `${join(this.storePath, 'temp', `temp-${hash}.mp3`)}`;
const audioBuffer = Buffer.from(audio, 'base64'); const audioBuffer = Buffer.from(audio, 'base64');
fs.writeFileSync(tempAudioPath, audioBuffer); inputAudioStream = new PassThrough();
inputAudioStream.end(audioBuffer);
} }
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
exec(`${ffmpegPath.path} -i ${tempAudioPath} -vn -ab 128k -ar 44100 -f ipod ${outputAudio} -y`, (error) => { const outputAudioStream = new PassThrough();
fs.unlinkSync(tempAudioPath); const chunks: Buffer[] = [];
if (error) reject(error); outputAudioStream.on('data', (chunk) => chunks.push(chunk));
outputAudioStream.on('end', () => {
resolve(outputAudio); const outputBuffer = Buffer.concat(chunks);
resolve(outputBuffer);
}); });
outputAudioStream.on('error', (error) => {
console.log('error', error);
reject(error);
});
ffmpeg.setFfmpegPath(ffmpegPath.path);
ffmpeg(inputAudioStream)
.outputFormat('mp4')
.noVideo()
.audioCodec('aac')
.audioBitrate('128k')
.audioFrequency(44100)
.addOutputOptions('-f ipod')
.pipe(outputAudioStream, { end: true })
.on('error', function (error) {
console.log('error', error);
reject(error);
});
}); });
} }
@ -2853,12 +2863,9 @@ export class BaileysStartupService extends ChannelStartupService {
const typeMessage = getContentType(msg.message); const typeMessage = getContentType(msg.message);
if (convertToMp4 && typeMessage === 'audioMessage') { if (convertToMp4 && typeMessage === 'audioMessage') {
const number = msg.key.remoteJid.split('@')[0]; const convert = await this.processAudioMp4(buffer.toString('base64'));
const convert = await this.processAudioMp4(buffer.toString('base64'), number);
if (typeof convert === 'string') {
const audio = fs.readFileSync(convert).toString('base64');
if (Buffer.isBuffer(convert)) {
const result = { const result = {
mediaType, mediaType,
fileName: mediaMessage['fileName'], fileName: mediaMessage['fileName'],
@ -2869,11 +2876,9 @@ export class BaileysStartupService extends ChannelStartupService {
width: mediaMessage['width'], width: mediaMessage['width'],
}, },
mimetype: 'audio/mp4', mimetype: 'audio/mp4',
base64: Buffer.from(audio, 'base64').toString('base64'), base64: convert,
}; };
fs.unlinkSync(convert);
return result; return result;
} }
} }