From da3420ac40968979618ffa0e3c6024cb0c14221d Mon Sep 17 00:00:00 2001
From: Chocobozzz <me@florianbigard.com>
Date: Thu, 8 Aug 2024 15:22:55 +0200
Subject: [PATCH] Avoid too many errors in client on fast-restream

---
 .../shared/p2p-media-loader/hls-plugin.ts     |  4 ++++
 .../p2p-media-loader-plugin.ts                | 20 +++++++++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/client/src/assets/player/shared/p2p-media-loader/hls-plugin.ts b/client/src/assets/player/shared/p2p-media-loader/hls-plugin.ts
index ca4637f49..e158d650c 100644
--- a/client/src/assets/player/shared/p2p-media-loader/hls-plugin.ts
+++ b/client/src/assets/player/shared/p2p-media-loader/hls-plugin.ts
@@ -411,6 +411,10 @@ export class Html5Hlsjs {
         this.edgeMargin = this.hlsjsConfig.liveSyncDurationCount * data.details.targetduration
       }
 
+      if (this.isLive && !data.details.live) {
+        this.player.trigger('hlsjs-live-ended')
+      }
+
       this.isLive = data.details.live
       this.dvrDuration = data.details.totalduration
 
diff --git a/client/src/assets/player/shared/p2p-media-loader/p2p-media-loader-plugin.ts b/client/src/assets/player/shared/p2p-media-loader/p2p-media-loader-plugin.ts
index 591497df5..e217d06da 100644
--- a/client/src/assets/player/shared/p2p-media-loader/p2p-media-loader-plugin.ts
+++ b/client/src/assets/player/shared/p2p-media-loader/p2p-media-loader-plugin.ts
@@ -31,6 +31,8 @@ class P2pMediaLoaderPlugin extends Plugin {
 
   private networkInfoInterval: any
 
+  private liveEnded = false
+
   constructor (player: videojs.Player, options?: P2PMediaLoaderPluginOptions) {
     super(player)
 
@@ -75,8 +77,21 @@ class P2pMediaLoaderPlugin extends Plugin {
 
       player.on('hlsjs-initialized', onHLSJSInitialized)
 
+      // ---------------------------------------------------------------------------
+
+      const onHLSJSLiveEnded = () => {
+        debugLogger('hls.js says the live is ended')
+
+        this.liveEnded = true
+      }
+
+      player.on('hlsjs-live-ended', onHLSJSLiveEnded)
+
+      // ---------------------------------------------------------------------------
+
       this.on('dispose', () => {
         this.player.off('hlsjs-initialized', onHLSJSInitialized)
+        this.player.off('hlsjs-live-ended', onHLSJSLiveEnded)
       })
     }
 
@@ -127,6 +142,11 @@ class P2pMediaLoaderPlugin extends Plugin {
 
     this.p2pEngine.on(Events.SegmentError, (segment: Segment, err) => {
       if (navigator.onLine === false) return
+      // We may have errors if the live ended because of a fast-restream in the same permanent live
+      if (this.liveEnded) {
+        (this.player as any).handleTechEnded_()
+        return
+      }
 
       logger.error(`Segment ${segment.id} error.`, err)