@ziplayer/plugin 0.1.51 → 0.2.1

package/src/index.ts

@@ -85,23 +85,6 @@ export * from "./SpotifyPlugin";
  */
 export * from "./TTSPlugin";
 
-/**
- * YTSR plugin for advanced YouTube search without streaming.
- *
- * Provides comprehensive YouTube search functionality including:
- * - Advanced video search with filters (duration, upload date, sort by)
- * - Playlist and channel search
- * - Multiple search types (video, playlist, channel, all)
- * - Metadata extraction without streaming
- * - Support for YouTube URLs
- *
- * @example
- * const ytsrPlugin = new YTSRPlugin();
- * const result = await ytsrPlugin.search("Never Gonna Give You Up", "user123");
- * const playlistResult = await ytsrPlugin.searchPlaylist("chill music", "user123");
- */
-export * from "./YTSRPlugin";
-
 /**
  * Attachments plugin for handling Discord attachment URLs and audio files.
  *

package/src/utils/sabr-stream-factory.ts

@@ -1,96 +1,282 @@
-import { createWriteStream } from "fs";
-import { join } from "path";
-import { tmpdir } from "os";
-
-// Import type declarations
-import type { SabrPlaybackOptions, StreamResult } from "../types/googlevideo";
-
-// Re-export types for external use
-export type { StreamResult, SabrPlaybackOptions };
-
-export interface OutputStream {
-	stream: NodeJS.WritableStream;
-	filePath: string;
-}
-
-/**
- * Creates a sabr stream for YouTube video download
- */
-export async function createSabrStream(videoId: string, options: SabrPlaybackOptions): Promise<{ streamResults: StreamResult }> {
-	try {
-		// Dynamic import to avoid build-time errors
-		const sabrModule = require("googlevideo/sabr-stream") as any;
-		const createSabrStreamImpl = sabrModule.createSabrStream;
-
-		const streamResults = await createSabrStreamImpl(videoId, options);
-
-		return { streamResults };
-	} catch (error) {
-		// Fallback implementation if sabr download is not available
-		throw new Error(`Sabr download not available: ${error}`);
-	}
-}
-
-/**
- * Creates an output stream for writing downloaded content
- */
-export function createOutputStream(videoTitle: string, mimeType: string): OutputStream {
-	const sanitizedTitle = videoTitle.replace(/[<>:"/\\|?*]/g, "_").substring(0, 100);
-	const extension = getExtensionFromMimeType(mimeType);
-	const fileName = `${sanitizedTitle}.${extension}`;
-	const filePath = join(tmpdir(), fileName);
-
-	const stream = createWriteStream(filePath);
-
-	return {
-		stream,
-		filePath,
-	};
-}
-
-/**
- * Creates a stream sink for piping data with progress tracking
- */
-export function createStreamSink(format: any, outputStream: NodeJS.WritableStream, progressBar: any) {
-	return new WritableStream({
-		start() {
-			// Initialize progress tracking
-		},
-		write(chunk) {
-			outputStream.write(chunk);
-			if (progressBar) {
-				progressBar.increment(chunk.length);
-			}
-		},
-		close() {
-			outputStream.end();
-		},
-	});
-}
-
-/**
- * Gets file extension from MIME type
- */
-function getExtensionFromMimeType(mimeType: string): string {
-	const mimeMap: { [key: string]: string } = {
-		"audio/mp4": "m4a",
-		"audio/webm": "webm",
-		"audio/ogg": "ogg",
-		"video/mp4": "mp4",
-		"video/webm": "webm",
-		"video/ogg": "ogv",
-	};
-
-	return mimeMap[mimeType] || "bin";
-}
-
-/**
- * Default sabr playback options
- */
-export const DEFAULT_SABR_OPTIONS: SabrPlaybackOptions = {
-	preferWebM: true,
-	preferOpus: true,
-	videoQuality: "720p",
-	audioQuality: "AUDIO_QUALITY_MEDIUM",
-	enabledTrackTypes: "VIDEO_AND_AUDIO",
-};
+import { createWriteStream } from "fs";
+import { join } from "path";
+import { tmpdir } from "os";
+import { Readable } from "stream";
+import { Constants, YTNodes, Platform } from "youtubei.js";
+import type Innertube from "youtubei.js";
+
+import { SabrStream } from "googlevideo/sabr-stream";
+import { buildSabrFormat } from "googlevideo/utils";
+
+import { BG } from "bgutils-js";
+import { JSDOM } from "jsdom";
+import { webStreamToNodeStream } from "./stream-converter";
+
+export interface OutputStream {
+	stream: NodeJS.WritableStream;
+	filePath: string;
+}
+
+export interface SabrAudioResult {
+	title: string;
+	stream: Readable;
+	format: {
+		mimeType: string;
+		itag: number;
+		contentLength: number;
+	};
+}
+
+export interface SabrPlaybackOptions {
+	preferWebM?: boolean;
+	preferOpus?: boolean;
+	videoQuality?: string;
+	audioQuality?: string;
+	enabledTrackTypes?: any;
+}
+/**
+ * Generates a web PoToken for YouTube authentication
+ * This is required for accessing restricted video content
+ */
+async function generateWebPoToken(contentBinding: string): Promise<{
+	visitorData: string;
+	placeholderPoToken: string;
+	poToken: string;
+}> {
+	try {
+		const requestKey = "O43z0dpjhgX20SCx4KAo";
+
+		if (!contentBinding) throw new Error("Could not get visitor data");
+
+		const dom = new JSDOM();
+
+		Object.assign(globalThis, {
+			window: dom.window,
+			document: dom.window.document,
+		});
+
+		const bgConfig = {
+			fetch: (input: any, init: any) => fetch(input, init),
+			globalObj: globalThis,
+			identifier: contentBinding,
+			requestKey,
+		};
+
+		const bgChallenge = await BG.Challenge.create(bgConfig);
+
+		if (!bgChallenge) throw new Error("Could not get challenge");
+
+		const interpreterJavascript = bgChallenge.interpreterJavascript.privateDoNotAccessOrElseSafeScriptWrappedValue;
+
+		if (interpreterJavascript) {
+			new Function(interpreterJavascript)();
+		} else throw new Error("Could not load VM");
+
+		const poTokenResult = await BG.PoToken.generate({
+			program: bgChallenge.program,
+			globalName: bgChallenge.globalName,
+			bgConfig,
+		});
+
+		const placeholderPoToken = BG.PoToken.generatePlaceholder(contentBinding);
+
+		return {
+			visitorData: contentBinding,
+			placeholderPoToken,
+			poToken: poTokenResult.poToken,
+		};
+	} catch (error) {
+		console.warn("PoToken generation failed, continuing without it:", error);
+		return {
+			visitorData: contentBinding,
+			placeholderPoToken: "",
+			poToken: "",
+		};
+	}
+}
+
+/**
+ * Makes a proper player request to YouTube API
+ */
+async function makePlayerRequest(innertube: Innertube, videoId: string, reloadPlaybackContext?: any): Promise<any> {
+	const watchEndpoint = new YTNodes.NavigationEndpoint({
+		watchEndpoint: { videoId },
+	});
+
+	const extraArgs: any = {
+		playbackContext: {
+			adPlaybackContext: { pyv: true },
+			contentPlaybackContext: {
+				vis: 0,
+				splay: false,
+				lactMilliseconds: "-1",
+				signatureTimestamp: innertube.session.player?.signature_timestamp,
+			},
+		},
+		contentCheckOk: true,
+		racyCheckOk: true,
+	};
+
+	if (reloadPlaybackContext) {
+		extraArgs.playbackContext.reloadPlaybackContext = reloadPlaybackContext;
+	}
+
+	return watchEndpoint.call(innertube.actions, {
+		...extraArgs,
+		parse: true,
+	});
+}
+
+/**
+ * YouTube VM shim
+ * This allows the SABR stream to execute YouTube's custom JavaScript for deciphering signatures and generating tokens
+ */
+Platform.shim.eval = async (data, env) => {
+	const properties = [];
+
+	if (env.n) properties.push(`n: exportedVars.nFunction("${env.n}")`);
+	if (env.sig) properties.push(`sig: exportedVars.sigFunction("${env.sig}")`);
+
+	const code = `${data.output}\nreturn { ${properties.join(", ")} }`;
+	return new Function(code)();
+};
+
+/**
+ * Creates a SABR audio stream for YouTube video download
+ * This provides better quality and more reliable streaming than standard methods
+ */
+export async function createSabrStream(
+	videoId: string,
+	innertube: Innertube,
+	options?: SabrPlaybackOptions,
+): Promise<SabrAudioResult> {
+	try {
+		// Generate PoToken for authentication
+		const webPo = await generateWebPoToken(videoId);
+
+		// Make initial player request
+		const player = await makePlayerRequest(innertube, videoId);
+
+		const title = player.video_details?.title || "unknown";
+
+		const serverAbrStreamingUrl = await innertube.session.player?.decipher(player.streaming_data?.server_abr_streaming_url);
+
+		const ustreamerConfig =
+			player.player_config?.media_common_config.media_ustreamer_request_config?.video_playback_ustreamer_config;
+
+		if (!serverAbrStreamingUrl || !ustreamerConfig) {
+			throw new Error("Missing SABR streaming config");
+		}
+
+		const sabrFormats = player.streaming_data?.adaptive_formats.map((f: any) => buildSabrFormat(f)) || [];
+
+		const sabr = new SabrStream({
+			formats: sabrFormats,
+			serverAbrStreamingUrl,
+			videoPlaybackUstreamerConfig: ustreamerConfig,
+			poToken: webPo.poToken,
+			clientInfo: {
+				clientName: parseInt(
+					Constants.CLIENT_NAME_IDS[innertube.session.context.client.clientName as keyof typeof Constants.CLIENT_NAME_IDS],
+				),
+				clientVersion: innertube.session.context.client.clientVersion,
+			},
+		});
+
+		// Handle player response reload events
+		sabr.on("reloadPlayerResponse", async (ctx: any) => {
+			try {
+				const pr = await makePlayerRequest(innertube, videoId, ctx);
+
+				const url = await innertube.session.player?.decipher(pr.streaming_data?.server_abr_streaming_url);
+
+				const config = pr.player_config?.media_common_config.media_ustreamer_request_config?.video_playback_ustreamer_config;
+
+				if (url && config) {
+					sabr.setStreamingURL(url);
+					sabr.setUstreamerConfig(config);
+				}
+			} catch (error) {
+				console.error("Failed to reload player response:", error);
+			}
+		});
+
+		// Start the stream with audio preference
+		const { audioStream, selectedFormats } = await sabr.start({
+			// audioQuality: options?.audioQuality || "high",
+		});
+
+		// Convert Web Stream to Node.js Readable stream
+		const nodeStream = webStreamToNodeStream(audioStream);
+
+		return {
+			title,
+			stream: nodeStream,
+			format: {
+				mimeType: selectedFormats.audioFormat.mimeType || "audio/webm",
+				itag: selectedFormats.audioFormat.itag || 0,
+				contentLength: selectedFormats.audioFormat.contentLength || 0,
+			},
+		};
+	} catch (error) {
+		throw new Error(`SABR stream creation failed: ${error instanceof Error ? error.message : String(error)}`);
+	}
+}
+
+/**
+ * Creates an output stream for writing downloaded content
+ */
+export function createOutputStream(videoTitle: string, mimeType: string): OutputStream {
+	const sanitizedTitle = videoTitle.replace(/[<>:"/\\|?*]/g, "_").substring(0, 100);
+	const extension = getExtensionFromMimeType(mimeType);
+	const fileName = `${sanitizedTitle}.${extension}`;
+	const filePath = join(tmpdir(), fileName);
+
+	const stream = createWriteStream(filePath);
+
+	return {
+		stream,
+		filePath,
+	};
+}
+
+/**
+ * Sanitizes a filename by removing invalid characters
+ */
+export function sanitizeFileName(name: string): string {
+	return name.replace(/[^\w\d]+/g, "_").slice(0, 128);
+}
+
+/**
+ * Converts bytes to megabytes
+ */
+export function bytesToMB(bytes: number): string {
+	return (bytes / 1024 / 1024).toFixed(2);
+}
+
+/**
+ * Gets file extension from MIME type
+ */
+function getExtensionFromMimeType(mimeType: string): string {
+	const mimeMap: { [key: string]: string } = {
+		"audio/mp4": "m4a",
+		"audio/webm": "webm",
+		"audio/ogg": "ogg",
+		"video/mp4": "mp4",
+		"video/webm": "webm",
+		"video/ogg": "ogv",
+	};
+
+	return mimeMap[mimeType] || "bin";
+}
+
+/**
+ * Default sabr playback options
+ */
+export const DEFAULT_SABR_OPTIONS: SabrPlaybackOptions = {
+	preferWebM: true,
+	preferOpus: true,
+	videoQuality: "720p",
+	audioQuality: "high",
+	enabledTrackTypes: "VIDEO_AND_AUDIO",
+};

package/src/utils/stream-converter.ts

@@ -34,46 +34,3 @@ export function webStreamToNodeStream(webStream: ReadableStream): Readable {
 
 	return nodeStream;
 }
-
-/**
- * Converts a Web ReadableStream to a Node.js Readable stream with progress tracking
- */
-export function webStreamToNodeStreamWithProgress(
-	webStream: ReadableStream,
-	progressCallback?: (bytesRead: number) => void,
-): Readable {
-	const nodeStream = new Readable({
-		read() {
-			// This will be handled by the Web Stream reader
-		},
-	});
-
-	let bytesRead = 0;
-	const reader = webStream.getReader();
-
-	const pump = async () => {
-		try {
-			while (true) {
-				const { done, value } = await reader.read();
-				if (done) {
-					nodeStream.push(null); // End the stream
-					break;
-				}
-
-				const buffer = Buffer.from(value);
-				nodeStream.push(buffer);
-
-				bytesRead += buffer.length;
-				if (progressCallback) {
-					progressCallback(bytesRead);
-				}
-			}
-		} catch (error) {
-			nodeStream.destroy(error as Error);
-		}
-	};
-
-	pump();
-
-	return nodeStream;
-}

package/tsconfig.json

@@ -16,8 +16,12 @@
 		"allowSyntheticDefaultImports": true,
 		"experimentalDecorators": true,
 		"emitDecoratorMetadata": true,
-		"resolveJsonModule": true
+		"resolveJsonModule": true,
+		"paths": {
+			"googlevideo/sabr-stream": ["./node_modules/googlevideo/dist/src/exports/sabr-stream"],
+			"googlevideo/utils": ["./node_modules/googlevideo/dist/src/exports/utils"]
+		}
 	},
-	"include": ["src/*", "src/types/*"],
+	"include": ["src/*"],
 	"exclude": ["node_modules", "dist", "examples"]
 }

package/dist/utils/progress-bar.d.ts

@@ -1,8 +0,0 @@
-interface MultiBar {
-    create(total: number, startValue: number, options: any): any;
-    stop(): void;
-}
-export declare function createMultiProgressBar(): MultiBar;
-export declare function setupProgressBar(multiBar: MultiBar, filename: string, total: number): any;
-export {};
-//# sourceMappingURL=progress-bar.d.ts.map

package/dist/utils/progress-bar.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"progress-bar.d.ts","sourceRoot":"","sources":["../../src/utils/progress-bar.ts"],"names":[],"mappings":"AACA,UAAU,QAAQ;IACjB,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,GAAG,GAAG,CAAC;IAC7D,IAAI,IAAI,IAAI,CAAC;CACb;AAOD,wBAAgB,sBAAsB,IAAI,QAAQ,CAcjD;AAED,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,OAEnF"}

package/dist/utils/progress-bar.js

@@ -1,24 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createMultiProgressBar = createMultiProgressBar;
-exports.setupProgressBar = setupProgressBar;
-// Mock implementation for now - will be replaced with actual cli-progress when available
-function createMultiProgressBar() {
-    return {
-        create(total, startValue, options) {
-            return {
-                increment(value) {
-                    // Mock progress bar implementation
-                    console.log(`Progress: ${value}/${total}`);
-                },
-            };
-        },
-        stop() {
-            console.log("Progress completed");
-        },
-    };
-}
-function setupProgressBar(multiBar, filename, total) {
-    return multiBar.create(total, 0, { filename });
-}
-//# sourceMappingURL=progress-bar.js.map

package/dist/utils/progress-bar.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"progress-bar.js","sourceRoot":"","sources":["../../src/utils/progress-bar.ts"],"names":[],"mappings":";;AAWA,wDAcC;AAED,4CAEC;AAnBD,yFAAyF;AACzF,SAAgB,sBAAsB;IACrC,OAAO;QACN,MAAM,CAAC,KAAa,EAAE,UAAkB,EAAE,OAAY;YACrD,OAAO;gBACN,SAAS,CAAC,KAAa;oBACtB,mCAAmC;oBACnC,OAAO,CAAC,GAAG,CAAC,aAAa,KAAK,IAAI,KAAK,EAAE,CAAC,CAAC;gBAC5C,CAAC;aACD,CAAC;QACH,CAAC;QACD,IAAI;YACH,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;QACnC,CAAC;KACD,CAAC;AACH,CAAC;AAED,SAAgB,gBAAgB,CAAC,QAAkB,EAAE,QAAgB,EAAE,KAAa;IACnF,OAAO,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;AAChD,CAAC"}