109 lines
3.5 KiB
TypeScript
109 lines
3.5 KiB
TypeScript
import { video_info } from '.';
|
|
import { LiveStreaming, Stream } from './classes/LiveStream';
|
|
|
|
export enum StreamType {
|
|
Arbitrary = 'arbitrary',
|
|
Raw = 'raw',
|
|
OggOpus = 'ogg/opus',
|
|
WebmOpus = 'webm/opus',
|
|
Opus = 'opus'
|
|
}
|
|
|
|
export interface StreamOptions {
|
|
quality?: number;
|
|
cookie?: string;
|
|
}
|
|
|
|
export interface InfoData {
|
|
LiveStreamData: {
|
|
isLive: boolean;
|
|
dashManifestUrl: string;
|
|
hlsManifestUrl: string;
|
|
};
|
|
html5player: string;
|
|
format: any[];
|
|
video_details: any;
|
|
}
|
|
|
|
function parseAudioFormats(formats: any[]) {
|
|
const result: any[] = [];
|
|
formats.forEach((format) => {
|
|
const type = format.mimeType as string;
|
|
if (type.startsWith('audio')) {
|
|
format.codec = type.split('codecs="')[1].split('"')[0];
|
|
format.container = type.split('audio/')[1].split(';')[0];
|
|
result.push(format);
|
|
}
|
|
});
|
|
return result;
|
|
}
|
|
|
|
export type YouTubeStream = Stream | LiveStreaming;
|
|
|
|
export async function stream(url: string, options: StreamOptions = {}): Promise<YouTubeStream> {
|
|
const info = await video_info(url, options.cookie);
|
|
const final: any[] = [];
|
|
if (
|
|
info.LiveStreamData.isLive === true &&
|
|
info.LiveStreamData.hlsManifestUrl !== null &&
|
|
info.video_details.durationInSec === 0
|
|
) {
|
|
return new LiveStreaming(
|
|
info.LiveStreamData.dashManifestUrl,
|
|
info.format[info.format.length - 1].targetDurationSec,
|
|
info.video_details.url
|
|
);
|
|
}
|
|
|
|
const audioFormat = parseAudioFormats(info.format);
|
|
if (typeof options.quality !== 'number') options.quality = audioFormat.length - 1;
|
|
else if (options.quality <= 0) options.quality = 0;
|
|
else if (options.quality >= audioFormat.length) options.quality = audioFormat.length - 1;
|
|
final.push(audioFormat[options.quality]);
|
|
let type: StreamType =
|
|
audioFormat[options.quality].codec === 'opus' && audioFormat[options.quality].container === 'webm'
|
|
? StreamType.WebmOpus
|
|
: StreamType.Arbitrary;
|
|
return new Stream(
|
|
final[0].url,
|
|
type,
|
|
info.video_details.durationInSec,
|
|
Number(final[0].contentLength),
|
|
info.video_details.url,
|
|
options.cookie as string
|
|
);
|
|
}
|
|
|
|
export async function stream_from_info(info: InfoData, options: StreamOptions = {}): Promise<YouTubeStream> {
|
|
const final: any[] = [];
|
|
if (
|
|
info.LiveStreamData.isLive === true &&
|
|
info.LiveStreamData.hlsManifestUrl !== null &&
|
|
info.video_details.durationInSec === '0'
|
|
) {
|
|
return new LiveStreaming(
|
|
info.LiveStreamData.dashManifestUrl,
|
|
info.format[info.format.length - 1].targetDurationSec,
|
|
info.video_details.url
|
|
);
|
|
}
|
|
|
|
const audioFormat = parseAudioFormats(info.format);
|
|
if (typeof options.quality !== 'number') options.quality = audioFormat.length - 1;
|
|
else if (options.quality <= 0) options.quality = 0;
|
|
else if (options.quality >= audioFormat.length) options.quality = audioFormat.length - 1;
|
|
final.push(audioFormat[options.quality]);
|
|
let type: StreamType =
|
|
audioFormat[options.quality].codec === 'opus' && audioFormat[options.quality].container === 'webm'
|
|
? StreamType.WebmOpus
|
|
: StreamType.Arbitrary;
|
|
return new Stream(
|
|
final[0].url,
|
|
type,
|
|
info.video_details.durationInSec,
|
|
Number(final[0].contentLength),
|
|
info.video_details.url,
|
|
options.cookie as string
|
|
);
|
|
}
|