add events to hls.

Author: m1k1o

hls/manager.go

@@ -37,6 +37,11 @@ type ManagerCtx struct {
 	mu         sync.Mutex
 	cmdFactory func() *exec.Cmd
 	active     bool
+	events     struct {
+		onStart  func()
+		onCmdLog func(message string)
+		onStop   func()
+	}
 
 	cmd         *exec.Cmd
 	tempdir     string
@@ -77,7 +82,12 @@ func (m *ManagerCtx) Start() error {
 
 	m.cmd = m.cmdFactory()
 	m.cmd.Dir = m.tempdir
-	m.cmd.Stderr = utils.LogWriter(m.logger)
+
+	if m.events.onStart != nil {
+		m.cmd.Stderr = utils.LogEvent(m.events.onCmdLog)
+	} else {
+		m.cmd.Stderr = utils.LogWriter(m.logger)
+	}
 
 	read, write := io.Pipe()
 	m.cmd.Stdout = write
@@ -138,6 +148,10 @@ func (m *ManagerCtx) Start() error {
 		}
 	}()
 
+	if m.events.onStart != nil {
+		m.events.onStart()
+	}
+
 	return m.cmd.Start()
 }
 
@@ -162,6 +176,10 @@ func (m *ManagerCtx) Stop() {
 		err := os.RemoveAll(m.tempdir)
 		m.logger.Err(err).Msg("removing tempdir")
 	})
+
+	if m.events.onStop != nil {
+		m.events.onStop()
+	}
 }
 
 func (m *ManagerCtx) Cleanup() {
@@ -239,3 +257,15 @@ func (m *ManagerCtx) ServeMedia(w http.ResponseWriter, r *http.Request) {
 	w.Header().Set("Cache-Control", "no-cache")
 	http.ServeFile(w, r, path)
 }
+
+func (m *ManagerCtx) OnStart(event func()) {
+	m.events.onStart = event
+}
+
+func (m *ManagerCtx) OnCmdLog(event func(message string)) {
+	m.events.onCmdLog = event
+}
+
+func (m *ManagerCtx) OnStop(event func()) {
+	m.events.onStop = event
+}

hls/types.go

@@ -9,4 +9,8 @@ type Manager interface {
 
 	ServePlaylist(w http.ResponseWriter, r *http.Request)
 	ServeMedia(w http.ResponseWriter, r *http.Request)
+
+	OnStart(event func())
+	OnCmdLog(event func(message string))
+	OnStop(event func())
 }

internal/utils/logevent.go

@@ -0,0 +1,18 @@
+package utils
+
+import "strings"
+
+type LogEventCtx struct {
+	event func(message string)
+}
+
+func LogEvent(event func(message string)) *LogEventCtx {
+	return &LogEventCtx{
+		event: event,
+	}
+}
+
+func (l LogEventCtx) Write(p []byte) (n int, err error) {
+	l.event(strings.TrimSpace(string(p)))
+	return len(p), nil
+}