openai
Version:
The official TypeScript library for the OpenAI API
116 lines • 3.69 kB
JavaScript
import { File } from 'formdata-node';
import { spawn } from 'node:child_process';
import { Readable } from 'node:stream';
import { platform, versions } from 'node:process';
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) {
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 } = {}) {
return new Promise((resolve, reject) => {
const data = [];
const provider = recordingProviders[platform];
try {
const ffmpeg = spawn('ffmpeg', [
'-f',
provider,
'-i',
`:${device ?? 0}`,
'-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