From 7ffcbca830eb6ac00e103af887c0c3711c6c99e7 Mon Sep 17 00:00:00 2001
From: killer069 <65385476+killer069@users.noreply.github.com>
Date: Wed, 15 Sep 2021 12:24:56 +0530
Subject: [PATCH] Stream errors handled

---
 play-dl/YouTube/classes/LiveStream.ts | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/play-dl/YouTube/classes/LiveStream.ts b/play-dl/YouTube/classes/LiveStream.ts
index 003ff4c..5ce601a 100644
--- a/play-dl/YouTube/classes/LiveStream.ts
+++ b/play-dl/YouTube/classes/LiveStream.ts
@@ -91,7 +91,8 @@ export class LiveStreaming{
                 continue
             }
             await new Promise(async(resolve, reject) => {
-                let stream = await request_stream(this.base_url + segment)
+                let stream = await request_stream(this.base_url + segment).catch((err) => {this.stream.emit('error', err); return null})
+                if(!stream) return
                 this.request = stream
                 stream.pipe(this.stream, { end : false })
                 stream.on('end', () => {
@@ -180,7 +181,14 @@ export class Stream {
             headers : {
                 "range" : `bytes=${this.bytes_count}-${end >= this.content_length ? '' : end}`
             }
-        })
+        }).catch((err) => {this.stream.emit('error', err); return null})
+        if(!stream) {
+            this.data_ended = true
+            this.bytes_count = 0
+            this.per_sec_bytes = 0
+            this.cleanup()
+            return
+        }
         if(Number(stream.statusCode) >= 400){
             this.cleanup()
             await this.retry()