permanent-waves/discordeno-audio-plugin/src/audio-source/youtube.ts

37 lines
1.2 KiB
TypeScript

import { YouTube, ytDownload } from "../../deps.ts";
import { bufferIter } from "../../utils/mod.ts";
import { demux } from "../demux/mod.ts";
import { createAudioSource, empty } from "./audio-source.ts";
export async function getYoutubeSources(added_by?: string, ...queries: string[]) {
const sources = queries.map((query) => getYoutubeSource(query, added_by));
const awaitedSources = await Promise.all(sources);
return awaitedSources
.filter((source) => source !== undefined)
.map((source) => source!);
}
export async function getYoutubeSource(query: string, added_by?: string) {
try {
const results = await YouTube.search(query, { limit: 1, type: "video" });
if (results.length > 0) {
const { id, title } = results[0];
return createAudioSource(title!, async () => {
try {
const stream = await ytDownload(id!, {
mimeType: `audio/webm; codecs="opus"`,
});
return bufferIter(demux(stream));
} catch {
console.error("error");
console.log(`Failed to play ${title}\n Returning empty stream`);
return empty();
}
}, added_by);
}
} catch(err) {
console.error(err);
return undefined;
}
}