Adds keyframe API

cmd/api/api.go

@@ -39,6 +39,8 @@ func Init() {
 		HandleFunc("/", fileServerHandlder)
 	}
 
+	HandleFunc("/api/frame.mp4", frameHandler)
+	HandleFunc("/api/frame.raw", frameHandler)
 	HandleFunc("/api/stack", stackHandler)
 	HandleFunc("/api/stats", statsHandler)
 	HandleFunc("/api/ws", apiWS)

cmd/api/keyframe.go

@@ -0,0 +1,40 @@
+package api
+
+import (
+	"github.com/AlexxIT/go2rtc/cmd/streams"
+	"github.com/AlexxIT/go2rtc/pkg/keyframe"
+	"net/http"
+	"strings"
+)
+
+func frameHandler(w http.ResponseWriter, r *http.Request) {
+	url := r.URL.Query().Get("url")
+	stream := streams.Get(url)
+	if stream == nil {
+		return
+	}
+
+	var ch = make(chan []byte)
+
+	cons := new(keyframe.Consumer)
+	cons.IsMP4 = strings.HasSuffix(r.URL.Path, ".mp4")
+	cons.Listen(func(msg interface{}) {
+		switch msg.(type) {
+		case []byte:
+			ch <- msg.([]byte)
+		}
+	})
+
+	if err := stream.AddConsumer(cons); err != nil {
+		log.Warn().Err(err).Msg("[api.frame] add consumer")
+		return
+	}
+
+	data := <-ch
+
+	stream.RemoveConsumer(cons)
+
+	if _, err := w.Write(data); err != nil {
+		log.Error().Err(err).Msg("[api.frame] write")
+	}
+}