Files
NekoMelody/example/speaker.ts
2024-08-18 15:25:34 +07:00

86 lines
2.4 KiB
TypeScript

import NekoMelody, { Player } from "../src";
import Speaker from "speaker";
import ffmpeg from "fluent-ffmpeg";
import { YtDlpProvider } from "../src/providers";
import { AudioInformation } from "../src/providers/base";
import dotenv from "dotenv";
dotenv.config();
const main = async () => {
const videoId = "l6gPxDSNbVk";
const videoId2 = "oM-JneFEdBk";
// Providers
const providers = [new YtDlpProvider()];
const player = NekoMelody.createPlayer(providers);
player.on("play", async (information: AudioInformation) => {
await playSpeaker(player);
player.startCurrentStream();
});
await player.enqueue(`https://www.youtube.com/watch?v=${videoId}`);
await player.enqueue(`https://www.youtube.com/watch?v=${videoId2}`);
console.log(player.getQueue());
};
let lastFFmpeg: ffmpeg.FfmpegCommand | null = null;
let lastSpeaker: Speaker | null = null;
const playSpeaker = async (player: Player) => {
if (!player.stream) {
console.error("No input stream");
return;
}
// A function that resolves when the speaker is closed and the ffmpeg process is killed
const closeSpeaker = () => {
return new Promise<void>((resolve) => {
if (lastSpeaker) {
lastSpeaker.on("close", () => {
resolve();
});
if (lastFFmpeg) lastFFmpeg.kill("SIGKILL");
lastSpeaker.close(true);
} else {
resolve();
}
});
};
await closeSpeaker();
// Create the Speaker instance
const speaker = new Speaker();
lastSpeaker = speaker;
// PCM data from stdin gets piped into the speaker
const ffmpegProcess = ffmpeg()
.input(player.stream)
.format("s16le") // Output format (PCM 16-bit little-endian)
.audioChannels(2)
.audioFrequency(44100)
// .on("end", async () => {
// await closeSpeaker();
// })
// @ts-expect-error - ffmpeg typings are incorrect
.on("error", (err) => {
console.error("[FFmpeg] > Error:", err.message);
});
speaker.on("unpipe", () => {
console.log("Speaker ended");
player.endCurrentStream();
});
// Pipe the ffmpeg output to the speaker
ffmpegProcess.pipe(speaker, { end: true });
lastFFmpeg = ffmpegProcess;
};
main();