fix m3u8 streams

proxy/buffer.go

@@ -33,13 +33,15 @@ func (b *Buffer) Write(data []byte) {
 	b.cond.Broadcast() // Notify all waiting clients that new data is available
 }
 
-func (b *Buffer) ReadChunk(size int) ([]byte, bool) {
+func (b *Buffer) ReadChunk(size int, force bool) ([]byte, bool) {
 	b.mu.Lock()
 	defer b.mu.Unlock()
 
 	// Wait for buffer to have enough data
-	for len(b.data) < size {
-		b.cond.Wait()
+	if !force {
+		for len(b.data) < size {
+			b.cond.Wait()
+		}
 	}
 
 	chunk := b.data[:size]

proxy/stream_handler.go

@@ -184,7 +184,7 @@ func (instance *StreamInstance) BufferStream(ctx context.Context, m3uIndex int,
 	}
 }
 
-func (instance *StreamInstance) StreamBuffer(ctx context.Context, w http.ResponseWriter) {
+func (instance *StreamInstance) StreamBuffer(ctx context.Context, resp *http.Response, r *http.Request, w http.ResponseWriter) {
 	for {
 		select {
 		case <-ctx.Done(): // handle context cancellation
@@ -196,9 +196,13 @@ func (instance *StreamInstance) StreamBuffer(ctx context.Context, w http.Respons
 				bufferSize = bufferMbInt * 1024 * 1024
 			}
 
-			chunk, ok := instance.Buffer.ReadChunk(bufferSize)
+			forceRead := r.Method != http.MethodGet || utils.EOFIsExpected(resp)
+
+			chunk, ok := instance.Buffer.ReadChunk(bufferSize, forceRead)
 			if !ok {
-				time.Sleep(100 * time.Millisecond) // Wait a bit before checking buffer again
+				if !forceRead {
+					time.Sleep(100 * time.Millisecond) // Wait a bit before checking buffer again
+				}
 				continue
 			}
 
@@ -304,7 +308,7 @@ func Handler(w http.ResponseWriter, r *http.Request) {
 					}
 				}
 			}()
-			go stream.StreamBuffer(ctx, w)
+			go stream.StreamBuffer(ctx, resp, r, w)
 
 			stream.Buffer.testedIndexes = append(stream.Buffer.testedIndexes, selectedIndex)