Histmy/play-dl-fix
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Cleaning of some code and some fixes

Browse Source
This commit is contained in:
killer069 2021-09-03 17:55:21 +05:30
parent a854657338
commit 9baf0cea27
2 changed files with 41 additions and 141 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

162
play-dl/YouTube/classes/LiveStream.ts
View File

@ -2,6 +2,7 @@ import { PassThrough } from 'stream'
import got from 'got'
import { StreamType } from '../stream';
import Request from 'got/dist/source/core';
import { video_info } from '..';
export interface FormatInterface{
url : string;
@ -17,9 +18,11 @@ export class LiveStreaming{
private interval : number
private packet_count : number
private timer : NodeJS.Timer | null
private video_url : string
private dash_timer : NodeJS.Timer | null
private segments_urls : string[]
private request : Request | null
constructor(dash_url : string, target_interval : number){
constructor(dash_url : string, target_interval : number, video_url : string){
this.type = StreamType.Arbitrary
this.url = dash_url
this.base_url = ''
@ -28,13 +31,27 @@ export class LiveStreaming{
this.packet_count = 0
this.request = null
this.timer = null
this.video_url = video_url
this.interval = target_interval * 1000 || 0
this.dash_timer = setTimeout(() => {
this.dash_updater()
}, 1800000)
this.stream.on('close', () => {
this.cleanup()
});
this.start()
}
private async dash_updater(){
let info = await video_info(this.video_url)
if(info.LiveStreamData.isLive === true && info.LiveStreamData.hlsManifestUrl !== null && info.video_details.durationInSec === '0'){
this.url = info.LiveStreamData.dashManifestUrl
}
this.dash_timer = setTimeout(() => {
this.dash_updater()
}, 1800000)
}
private async dash_getter(){
let response = await got(this.url)
let audioFormat = response.body.split('<AdaptationSet id="0"')[1].split('</AdaptationSet>')[0].split('</Representation>')
@ -47,7 +64,10 @@ export class LiveStreaming{
private cleanup(){
clearTimeout(this.timer as NodeJS.Timer)
clearTimeout(this.dash_timer as NodeJS.Timer)
this.request?.destroy()
this.dash_timer = null
this.video_url = ''
this.request = null
this.timer = null
this.url = ''
@ -63,6 +83,7 @@ export class LiveStreaming{
return
}
await this.dash_getter()
if(this.segments_urls.length > 3) this.segments_urls.splice(0, this.segments_urls.length - 3)
if(this.packet_count === 0) this.packet_count = Number(this.segments_urls[0].split('sq/')[1].split('/')[0])
for await (let segment of this.segments_urls){
if(Number(segment.split('sq/')[1].split('/')[0]) !== this.packet_count){
@ -89,80 +110,6 @@ export class LiveStreaming{
}
}
export class LiveEnded{
type : StreamType
stream : PassThrough
private url : string;
private base_url : string;
private packet_count : number
private segments_urls : string[]
private request : Request | null
constructor(dash_url : string){
this.type = StreamType.Arbitrary
this.url = dash_url
this.base_url = ''
this.stream = new PassThrough({ highWaterMark : 10 * 1000 * 1000 })
this.segments_urls = []
this.request = null
this.packet_count = 0
this.stream.on('close', () => {
this.cleanup()
})
this.start()
}
private async dash_getter(){
let response = await got(this.url)
let audioFormat = response.body.split('<AdaptationSet id="0"')[1].split('</AdaptationSet>')[0].split('</Representation>')
if(audioFormat[audioFormat.length - 1] === '') audioFormat.pop()
this.base_url = audioFormat[audioFormat.length - 1].split('<BaseURL>')[1].split('</BaseURL>')[0]
let list = audioFormat[audioFormat.length - 1].split('<SegmentList>')[1].split('</SegmentList>')[0]
this.segments_urls = list.replace(new RegExp('<SegmentURL media="', 'g'), '').split('"/>')
if(this.segments_urls[this.segments_urls.length - 1] === '') this.segments_urls.pop()
}
private cleanup(){
this.request?.destroy()
this.request = null
this.url = ''
this.base_url = ''
this.segments_urls = []
this.packet_count = 0
}
private async start(){
if(this.stream.destroyed){
this.cleanup()
return
}
await this.dash_getter()
if(this.packet_count === 0) this.packet_count = Number(this.segments_urls[0].split('sq/')[1].split('/')[0])
for await (let segment of this.segments_urls){
if(this.stream.destroyed){
this.cleanup()
break
}
if(Number(segment.split('sq/')[1].split('/')[0]) !== this.packet_count){
continue
}
await (async () => {
return new Promise(async (resolve, reject) => {
let stream = got.stream(this.base_url + segment)
this.request = stream
stream.on('data', (chunk: any) => this.stream.write(chunk))
stream.on('end', () => {
this.packet_count++
resolve('')
})
stream.once('error', (err) => {
this.stream.emit('error', err)
})
})
})()
}
}
}
export class Stream {
type : StreamType
stream : PassThrough
@ -183,11 +130,12 @@ export class Stream {
this.stream.on('close', () => {
this.cleanup()
})
this.duration = duration;
(duration > 300) ? this.loop_start() : this.normal_start()
this.duration = duration
this.loop()
}
private cleanup(){
console.log('Ending Everything')
clearTimeout(this.timer as NodeJS.Timer)
this.request?.destroy()
this.request = null
@ -197,51 +145,6 @@ export class Stream {
this.per_sec_bytes = 0
}
private normal_start(){
if(this.stream.destroyed){
this.cleanup()
return
}
let stream = got.stream(this.url)
this.request = stream
stream.once('error', (err) => {
this.stream.emit('error', err)
})
stream.pipe(this.stream)
}
private loop_start(){
if(this.stream.destroyed){
this.cleanup()
return
}
let stream = got.stream(this.url)
this.request = stream
stream.once('data', () => {
this.per_sec_bytes = Math.ceil((stream.downloadProgress.total as number)/this.duration)
})
stream.once('error', (err) => {
this.stream.emit('error', err)
})
stream.on('data', (chunk: any) => {
this.bytes_count += chunk.length
this.stream.write(chunk)
})
stream.on('data', () => {
if(this.bytes_count > (this.per_sec_bytes * 300)){
stream.destroy()
}
})
this.timer = setTimeout(() => {
this.loop()
}, 280 * 1000)
}
private loop(){
if(this.stream.destroyed){
this.cleanup()
@ -254,18 +157,21 @@ export class Stream {
}
})
this.request = stream
stream.on('data', (chunk: any) => {
absolute_bytes += chunk.length
this.bytes_count += chunk.length
this.stream.write(chunk)
stream.once('data', () => {
if(this.per_sec_bytes === 0){
this.per_sec_bytes = Math.ceil((stream.downloadProgress.total as number)/this.duration)
}
})
stream.once('error', (err) => {
this.stream.emit('error', err)
})
stream.on('data', () => {
if(absolute_bytes > (this.per_sec_bytes * 300)){
stream.on('data', (chunk: any) => {
absolute_bytes += chunk.length
this.bytes_count += chunk.length
this.stream.write(chunk)
if(absolute_bytes > (this.per_sec_bytes * 300) && this.per_sec_bytes !== 0){
stream.destroy()
}
})

20
play-dl/YouTube/stream.ts
View File

@ -1,6 +1,6 @@
import got from "got/dist/source"
import { video_info } from "."
import { LiveEnded, LiveStreaming, Stream } from "./classes/LiveStream"
import { LiveStreaming, Stream } from "./classes/LiveStream"
export enum StreamType{
Arbitrary = 'arbitrary',
@ -35,11 +35,11 @@ function parseAudioFormats(formats : any[]){
return result
}
export async function stream(url : string, cookie? : string): Promise<Stream | LiveStreaming | LiveEnded>{
export async function stream(url : string, cookie? : string): Promise<Stream | LiveStreaming>{
let info = await video_info(url, cookie)
let final: any[] = [];
let type : StreamType;
if(info.LiveStreamData.isLive === true && info.LiveStreamData.hlsManifestUrl !== null) {
if(info.LiveStreamData.isLive === true && info.LiveStreamData.hlsManifestUrl !== null && info.video_details.durationInSec === '0') {
return live_stream(info as InfoData)
}
@ -72,10 +72,10 @@ export async function stream(url : string, cookie? : string): Promise<Stream | L
return new Stream(final[0].url, type, info.video_details.durationInSec)
}
export async function stream_from_info(info : InfoData): Promise<Stream | LiveStreaming | LiveEnded>{
export async function stream_from_info(info : InfoData): Promise<Stream | LiveStreaming>{
let final: any[] = [];
let type : StreamType;
if(info.LiveStreamData.isLive === true && info.LiveStreamData.hlsManifestUrl !== null) {
if(info.LiveStreamData.isLive === true && info.LiveStreamData.hlsManifestUrl !== null && info.video_details.durationInSec === '0') {
return live_stream(info as InfoData)
}
@ -116,13 +116,7 @@ function filterFormat(formats : any[], codec : string){
return result
}
function live_stream(info : InfoData): LiveStreaming | LiveEnded{
let stream : LiveStreaming | LiveEnded
if(info.video_details.durationInSec === '0') {
stream = new LiveStreaming(info.LiveStreamData.dashManifestUrl, info.format[info.format.length - 1].targetDurationSec)
}
else {
stream = new LiveEnded(info.LiveStreamData.dashManifestUrl)
}
function live_stream(info : InfoData): LiveStreaming{
let stream = new LiveStreaming(info.LiveStreamData.dashManifestUrl, info.format[info.format.length - 1].targetDurationSec, info.video_details.url)
return stream
}