124 lines
3.5 KiB
TypeScript
124 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'
|
|
}
|
|
|
|
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 async function stream(url: string, cookie?: string): Promise<Stream | LiveStreaming> {
|
|
const info = await video_info(url, cookie);
|
|
const final: any[] = [];
|
|
let type: StreamType;
|
|
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);
|
|
const opusFormats = filterFormat(audioFormat, 'opus');
|
|
|
|
if (opusFormats.length === 0) {
|
|
type = StreamType.Arbitrary;
|
|
if (audioFormat.length === 0) {
|
|
final.push(info.format[info.format.length - 1]);
|
|
} else {
|
|
final.push(audioFormat[audioFormat.length - 1]);
|
|
}
|
|
} else {
|
|
type = StreamType.WebmOpus;
|
|
final.push(opusFormats[opusFormats.length - 1]);
|
|
}
|
|
|
|
return new Stream(
|
|
final[0].url,
|
|
type,
|
|
info.video_details.durationInSec,
|
|
Number(final[0].contentLength),
|
|
info.video_details.url,
|
|
cookie as string
|
|
);
|
|
}
|
|
|
|
export async function stream_from_info(info: InfoData, cookie?: string): Promise<Stream | LiveStreaming> {
|
|
const final: any[] = [];
|
|
let type: StreamType;
|
|
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);
|
|
const opusFormats = filterFormat(audioFormat, 'opus');
|
|
|
|
if (opusFormats.length === 0) {
|
|
type = StreamType.Arbitrary;
|
|
if (audioFormat.length === 0) {
|
|
final.push(info.format[info.format.length - 1]);
|
|
} else {
|
|
final.push(audioFormat[audioFormat.length - 1]);
|
|
}
|
|
} else {
|
|
type = StreamType.WebmOpus;
|
|
final.push(opusFormats[opusFormats.length - 1]);
|
|
}
|
|
|
|
return new Stream(
|
|
final[0].url,
|
|
type,
|
|
info.video_details.durationInSec,
|
|
Number(final[0].contentLength),
|
|
info.video_details.url,
|
|
cookie as string
|
|
);
|
|
}
|
|
|
|
function filterFormat(formats: any[], codec: string) {
|
|
const result: any[] = [];
|
|
formats.forEach((format) => {
|
|
if (format.codec === codec) result.push(format);
|
|
});
|
|
return result;
|
|
}
|