@ziplayer/plugin 0.1.2 → 0.1.40

package/src/types/googlevideo.d.ts

@@ -0,0 +1,45 @@
+export interface SabrPlaybackOptions {
+	preferWebM?: boolean;
+	preferOpus?: boolean;
+	videoQuality?: string;
+	audioQuality?: string;
+	enabledTrackTypes?: any;
+}
+
+export interface StreamResult {
+	videoStream: ReadableStream;
+	audioStream: ReadableStream;
+	selectedFormats: {
+		videoFormat: any;
+		audioFormat: any;
+	};
+	videoTitle: string;
+}
+
+declare module "googlevideo/sabr-stream" {
+	export interface SabrPlaybackOptions {
+		preferWebM?: boolean;
+		preferOpus?: boolean;
+		videoQuality?: string;
+		audioQuality?: string;
+		enabledTrackTypes?: any;
+	}
+
+	export interface StreamResult {
+		videoStream: ReadableStream;
+		audioStream: ReadableStream;
+		selectedFormats: {
+			videoFormat: any;
+			audioFormat: any;
+		};
+		videoTitle: string;
+	}
+
+	export function createSabrStream(videoId: string, options: SabrPlaybackOptions): Promise<{ streamResults: StreamResult }>;
+}
+
+declare module "googlevideo/utils" {
+	export enum EnabledTrackTypes {
+		VIDEO_AND_AUDIO = "VIDEO_AND_AUDIO",
+	}
+}

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

@@ -0,0 +1,96 @@
+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",
+};

package/src/utils/stream-converter.ts

@@ -0,0 +1,87 @@
+import { Readable } from "stream";
+
+/**
+ * Converts a Web ReadableStream to a Node.js Readable stream
+ */
+export function webStreamToNodeStream(webStream: ReadableStream): Readable {
+	console.log("🔄 webStreamToNodeStream: Starting conversion");
+
+	const nodeStream = new Readable({
+		read() {
+			// This will be handled by the Web Stream reader
+		},
+	});
+
+	// Create a reader from the Web Stream
+	const reader = webStream.getReader();
+	console.log("🔄 webStreamToNodeStream: Got reader");
+
+	// Read chunks and push to Node.js stream
+	const pump = async () => {
+		try {
+			console.log("🔄 webStreamToNodeStream: Starting pump");
+			while (true) {
+				const { done, value } = await reader.read();
+				if (done) {
+					console.log("🔄 webStreamToNodeStream: Stream ended");
+					nodeStream.push(null); // End the stream
+					break;
+				}
+				console.log(`🔄 webStreamToNodeStream: Pushing chunk of ${value.length} bytes`);
+				nodeStream.push(Buffer.from(value));
+			}
+		} catch (error) {
+			console.error("🔄 webStreamToNodeStream: Error in pump:", error);
+			nodeStream.destroy(error as Error);
+		}
+	};
+
+	// Start pumping data
+	pump();
+
+	console.log("🔄 webStreamToNodeStream: Conversion complete, returning Node.js stream");
+	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

@@ -18,6 +18,6 @@
 		"emitDecoratorMetadata": true,
 		"resolveJsonModule": true
 	},
-	"include": ["src/*"],
+	"include": ["src/*", "src/types/*"],
 	"exclude": ["node_modules", "dist", "examples"]
 }