yt-dlx
Version:
Effortless Audio-Video Downloader And Streamer!
162 lines • 6.39 kB
JavaScript
import * as fs from "fs";
import colors from "colors";
import * as path from "path";
import { z, ZodError } from "zod";
import ffmpeg from "fluent-ffmpeg";
import { EventEmitter } from "events";
import { encore } from "yt-dlx-encore";
import ytdlx from "../../../base/Agent";
var ZodSchema = z.object({
query: z.string().min(2),
output: z.string().optional(),
useTor: z.boolean().optional(),
stream: z.boolean().optional(),
verbose: z.boolean().optional(),
metadata: z.boolean().optional(),
resolution: z.enum([
"144p",
"240p",
"360p",
"480p",
"720p",
"1080p",
"1440p",
"2160p",
"3072p",
"4320p",
"6480p",
"8640p",
"12000p",
]),
filter: z
.enum([
"invert",
"rotate90",
"rotate270",
"grayscale",
"rotate180",
"flipVertical",
"flipHorizontal",
])
.optional(),
});
/**
* Downloads a YouTube video with custom resolution and optional video filter.
*
* @param query - The YouTube video URL or ID or name.
* @param resolution - The desired resolution of the video.
* @param stream - (optional) Whether to return the FfmpegCommand instead of downloading the video.
* @param verbose - (optional) Whether to log verbose output or not.
* @param useTor - (optional) Whether to use Tor for the download or not.
* @param output - (optional) The output directory for the processed files.
* @param metadata - (optional) If true, the function returns the extracted metadata and filename without processing the audio.
* @param filter - (optional) The video filter to apply.
*
* @returns An EventEmitter instance to handle events.
*/
export default function VideoCustom({ query, stream, useTor, filter, output, verbose, metadata, resolution, }) {
var emitter = new EventEmitter();
(async () => {
try {
ZodSchema.parse({
query,
stream,
useTor,
filter,
output,
verbose,
metadata,
resolution,
});
var engineData = await ytdlx({
query,
verbose,
useTor,
});
if (!engineData) {
throw new Error(`${colors.red("@error:")} unable to get response!`);
}
var title = engineData.metaData.title.replace(/[^a-zA-Z0-9_]+/g, "_");
var folder = output ? output : __dirname;
if (!fs.existsSync(folder))
fs.mkdirSync(folder, { recursive: true });
var ff = ffmpeg()
.setFfmpegPath((await encore().then(fp => fp.ffmpeg)).toString())
.setFfprobePath((await encore().then(fp => fp.ffprobe)).toString())
.addInput(engineData.AudioHighF.url)
.withOutputFormat("matroska")
.videoCodec("copy")
.addOption("-headers", `X-Forwarded-For: ${engineData.ipAddress}`);
var filenameBase = `yt-dlx_(VideoCustom_${resolution}_`;
let filename = `${filenameBase}${filter ? filter + ")_" : ")_"}${title}.mkv`;
var vdata = engineData.ManifestHigh.find(i => i.format.includes(resolution.replace("p", "").toString()));
if (vdata)
ff.addInput(vdata.url.toString());
else {
throw new Error(`${colors.red("@error:")} no video data found. Use list_formats() maybe?`);
}
var filterMap = {
grayscale: ["colorchannelmixer=.3:.4:.3:0:.3:.4:.3:0:.3:.4:.3"],
invert: ["negate"],
rotate90: ["rotate=PI/2"],
rotate180: ["rotate=PI"],
rotate270: ["rotate=3*PI/2"],
flipHorizontal: ["hflip"],
flipVertical: ["vflip"],
};
if (filter && filterMap[filter])
ff.withVideoFilter(filterMap[filter]);
ff.addOption("-headers", `X-Forwarded-For: ${engineData.ipAddress}`);
ff.on("start", comd => {
if (verbose)
emitter.emit("log", colors.green("@comd:"), comd);
emitter.emit("start", comd);
})
.on("progress", progress => emitter.emit("progress", progress))
.on("error", error => emitter.emit("error", error.message))
.on("end", () => emitter.emit("end", filename));
switch (true) {
case stream:
emitter.emit("ready", {
ffmpeg: ff,
filename: path.join(folder, filename),
});
break;
case metadata:
emitter.emit("metadata", {
filename,
metaData: engineData.metaData,
ipAddress: engineData.ipAddress,
VideoLowF: engineData.VideoLowF,
VideoHighF: engineData.VideoHighF,
VideoLowHDR: engineData.VideoLowHDR,
VideoHighHDR: engineData.VideoHighHDR,
ManifestLow: engineData.ManifestLow,
ManifestHigh: engineData.ManifestHigh,
});
break;
default:
ff.output(path.join(folder, filename))
.on("end", () => emitter.emit("end", filename))
.on("error", error => emitter.emit("error", error.message))
.run();
break;
}
}
catch (error) {
switch (true) {
case error instanceof ZodError:
emitter.emit("error", colors.red("@zod-error:"), error.errors);
break;
default:
emitter.emit("error", colors.red("@error:"), error.message);
break;
}
}
finally {
emitter.emit("info", colors.green("@info:"), "❣️ Thank you for using yt-dlx. Consider 🌟starring the GitHub repo https://github.com/yt-dlx.");
}
})();
return emitter;
}
//# sourceMappingURL=VideoCustom.js.map