118 lines
3.7 KiB
JavaScript
118 lines
3.7 KiB
JavaScript
|
import { spawn } from 'node:child_process';
|
||
|
|
import { Readable } from 'node:stream';
|
||
|
|
import { platform, versions } from 'node:process';
|
||
|
|
import { checkFileSupport } from "../internal/uploads.mjs";
|
||
|
|
const DEFAULT_SAMPLE_RATE = 24000;
|
||
|
|
const DEFAULT_CHANNELS = 1;
|
||
|
|
const isNode = Boolean(versions?.node);
|
||
|
|
const recordingProviders = {
|
||
|
|
win32: 'dshow',
|
||
|
|
darwin: 'avfoundation',
|
||
|
|
linux: 'alsa',
|
||
|
|
aix: 'alsa',
|
||
|
|
android: 'alsa',
|
||
|
|
freebsd: 'alsa',
|
||
|
|
haiku: 'alsa',
|
||
|
|
sunos: 'alsa',
|
||
|
|
netbsd: 'alsa',
|
||
|
|
openbsd: 'alsa',
|
||
|
|
cygwin: 'dshow',
|
||
|
|
};
|
||
|
|
function isResponse(stream) {
|
||
|
|
return typeof stream.body !== 'undefined';
|
||
|
|
}
|
||
|
|
function isFile(stream) {
|
||
|
|
checkFileSupport();
|
||
|
|
return stream instanceof File;
|
||
|
|
}
|
||
|
|
async function nodejsPlayAudio(stream) {
|
||
|
|
return new Promise((resolve, reject) => {
|
||
|
|
try {
|
||
|
|
const ffplay = spawn('ffplay', ['-autoexit', '-nodisp', '-i', 'pipe:0']);
|
||
|
|
if (isResponse(stream)) {
|
||
|
|
stream.body.pipe(ffplay.stdin);
|
||
|
|
}
|
||
|
|
else if (isFile(stream)) {
|
||
|
|
Readable.from(stream.stream()).pipe(ffplay.stdin);
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
stream.pipe(ffplay.stdin);
|
||
|
|
}
|
||
|
|
ffplay.on('close', (code) => {
|
||
|
|
if (code !== 0) {
|
||
|
|
reject(new Error(`ffplay process exited with code ${code}`));
|
||
|
|
}
|
||
|
|
resolve();
|
||
|
|
});
|
||
|
|
}
|
||
|
|
catch (error) {
|
||
|
|
reject(error);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
export async function playAudio(input) {
|
||
|
|
if (isNode) {
|
||
|
|
return nodejsPlayAudio(input);
|
||
|
|
}
|
||
|
|
throw new Error('Play audio is not supported in the browser yet. Check out https://npm.im/wavtools as an alternative.');
|
||
|
|
}
|
||
|
|
function nodejsRecordAudio({ signal, device, timeout } = {}) {
|
||
|
|
checkFileSupport();
|
||
|
|
return new Promise((resolve, reject) => {
|
||
|
|
const data = [];
|
||
|
|
const provider = recordingProviders[platform];
|
||
|
|
try {
|
||
|
|
const ffmpeg = spawn('ffmpeg', [
|
||
|
|
'-f',
|
||
|
|
provider,
|
||
|
|
'-i',
|
||
|
|
`:${device ?? 0}`, // default audio input device; adjust as needed
|
||
|
|
'-ar',
|
||
|
|
DEFAULT_SAMPLE_RATE.toString(),
|
||
|
|
'-ac',
|
||
|
|
DEFAULT_CHANNELS.toString(),
|
||
|
|
'-f',
|
||
|
|
'wav',
|
||
|
|
'pipe:1',
|
||
|
|
], {
|
||
|
|
stdio: ['ignore', 'pipe', 'pipe'],
|
||
|
|
});
|
||
|
|
ffmpeg.stdout.on('data', (chunk) => {
|
||
|
|
data.push(chunk);
|
||
|
|
});
|
||
|
|
ffmpeg.on('error', (error) => {
|
||
|
|
console.error(error);
|
||
|
|
reject(error);
|
||
|
|
});
|
||
|
|
ffmpeg.on('close', (code) => {
|
||
|
|
returnData();
|
||
|
|
});
|
||
|
|
function returnData() {
|
||
|
|
const audioBuffer = Buffer.concat(data);
|
||
|
|
const audioFile = new File([audioBuffer], 'audio.wav', { type: 'audio/wav' });
|
||
|
|
resolve(audioFile);
|
||
|
|
}
|
||
|
|
if (typeof timeout === 'number' && timeout > 0) {
|
||
|
|
const internalSignal = AbortSignal.timeout(timeout);
|
||
|
|
internalSignal.addEventListener('abort', () => {
|
||
|
|
ffmpeg.kill('SIGTERM');
|
||
|
|
});
|
||
|
|
}
|
||
|
|
if (signal) {
|
||
|
|
signal.addEventListener('abort', () => {
|
||
|
|
ffmpeg.kill('SIGTERM');
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
catch (error) {
|
||
|
|
reject(error);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
export async function recordAudio(options = {}) {
|
||
|
|
if (isNode) {
|
||
|
|
return nodejsRecordAudio(options);
|
||
|
|
}
|
||
|
|
throw new Error('Record audio is not supported in the browser. Check out https://npm.im/wavtools as an alternative.');
|
||
|
|
}
|
||
|
|
//# sourceMappingURL=audio.mjs.map
|