mirror of
https://github.com/YuzuZensai/NekoMelody.git
synced 2026-01-06 04:33:21 +00:00
86 lines
2.4 KiB
TypeScript
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();
|