From e5954712e263bbf4f3ae2a44eb4aabc834b20140 Mon Sep 17 00:00:00 2001
From: Zoe Roux <zoe.roux@zoriya.dev>
Date: Sun, 14 Jan 2024 15:36:10 +0100
Subject: [PATCH] Save processed segments in the stream

---
 transcoder/src/stream.go | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/transcoder/src/stream.go b/transcoder/src/stream.go
index a2be4587..d09a3cc0 100644
--- a/transcoder/src/stream.go
+++ b/transcoder/src/stream.go
@@ -1,6 +1,7 @@
 package src
 
 import (
+	"bufio"
 	"context"
 	"fmt"
 	"log"
@@ -76,6 +77,38 @@ func (ts *Stream) run(start int32, end int32) error {
 	)
 	log.Printf("Running %s", strings.Join(cmd.Args, " "))
 
+	stdout, err := cmd.StdoutPipe()
+	if err != nil {
+		return err
+	}
+	var stderr strings.Builder
+	cmd.Stderr = &stderr
+
+	go func() {
+		scanner := bufio.NewScanner(stdout)
+		for scanner.Scan() {
+			var segment int32
+			_, _ = fmt.Sscanf(scanner.Text(), "segment-%d.ts", &segment)
+
+			ts.lock.Lock()
+			ts.segments[segment] = true
+			ts.lock.Unlock()
+		}
+
+		if err := scanner.Err(); err != nil {
+			log.Println("Error reading stdout of ffmpeg", err)
+		}
+	}()
+
+	go func() {
+		err := cmd.Wait()
+		if err != nil {
+			log.Println("ffmpeg occured an error", err, stderr.String())
+		} else {
+			log.Println("ffmpeg finished successfully")
+		}
+	}()
+
 	return nil
 }