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

Socket issue tried to fix

Browse Source
This commit is contained in:
killer069 2021-08-30 15:58:23 +05:30
parent 5bef17528a
commit 7b1f082c27
1 changed files with 22 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,7 +1,7 @@
import { PassThrough } from 'stream'
import got from 'got'
import Request from 'got/dist/source/core';
import { StreamType } from '../stream';
import { Socket } from 'net'
export interface FormatInterface{
url : string;
@ -17,6 +17,7 @@ export class LiveStreaming{
private interval : number
private packet_count : number
private timer : NodeJS.Timer | null
private socket : Socket | null
private segments_urls : string[]
constructor(dash_url : string, target_interval : number){
this.type = StreamType.Arbitrary
@ -26,6 +27,7 @@ export class LiveStreaming{
this.segments_urls = []
this.packet_count = 0
this.timer = null
this.socket = null
this.interval = target_interval * 1000 || 0
this.stream.on('close', () => {
this.cleanup()
@ -45,6 +47,8 @@ export class LiveStreaming{
private cleanup(){
clearTimeout(this.timer as NodeJS.Timer)
this.socket?.destroy()
this.socket = null
this.timer = null
this.url = ''
this.base_url = ''
@ -88,6 +92,7 @@ export class LiveEnded{
private base_url : string;
private packet_count : number
private segments_urls : string[]
private socket : Socket | null
constructor(dash_url : string){
this.type = StreamType.Arbitrary
this.url = dash_url
@ -95,6 +100,7 @@ export class LiveEnded{
this.stream = new PassThrough({ highWaterMark : 10 * 1000 * 1000 })
this.segments_urls = []
this.packet_count = 0
this.socket = null
this.stream.on('close', () => {
this.cleanup()
})
@ -112,6 +118,8 @@ export class LiveEnded{
}
private cleanup(){
this.socket?.destroy()
this.socket = null
this.url = ''
this.base_url = ''
this.segments_urls = []
@ -126,6 +134,10 @@ export class LiveEnded{
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
}
@ -151,6 +163,7 @@ export class Stream {
private per_sec_bytes : number;
private duration : number;
private timer : NodeJS.Timer | null
private socket : Socket | null
constructor(url : string, type : StreamType, duration : number){
this.url = url
this.type = type
@ -158,12 +171,14 @@ export class Stream {
this.bytes_count = 0
this.per_sec_bytes = 0
this.timer = null
this.duration = duration;
this.duration = duration
this.socket = null;
(duration > 300) ? this.loop_start() : this.normal_start()
}
private cleanup(){
clearTimeout(this.timer as NodeJS.Timer)
this.socket?.destroy()
this.timer = null
this.url = ''
this.bytes_count = 0
@ -176,6 +191,7 @@ export class Stream {
return
}
let stream = got.stream(this.url)
this.socket = stream.socket as Socket
stream.pipe(this.stream)
}
@ -193,9 +209,10 @@ export class Stream {
this.bytes_count += chunk.length
this.stream.write(chunk)
})
this.socket = stream.socket as Socket
stream.on('data', () => {
if(this.bytes_count > (this.per_sec_bytes * 300)){
this.socket?.destroy()
stream.destroy()
}
})
@ -222,9 +239,10 @@ export class Stream {
this.bytes_count += chunk.length
this.stream.write(chunk)
})
this.socket = stream.socket as Socket
stream.on('data', () => {
if(absolute_bytes > (this.per_sec_bytes * 300)){
this.socket?.destroy()
stream.destroy()
}
})