Initial commit

cmd/mse/mse.go

@@ -0,0 +1,42 @@
+package mse
+
+import (
+	"github.com/AlexxIT/go2rtc/cmd/api"
+	"github.com/AlexxIT/go2rtc/cmd/streams"
+	"github.com/AlexxIT/go2rtc/pkg/mse"
+	"github.com/AlexxIT/go2rtc/pkg/streamer"
+	"github.com/rs/zerolog/log"
+)
+
+func Init() {
+	api.HandleWS("mse", handler)
+}
+
+func handler(ctx *api.Context, msg *streamer.Message) {
+	url := ctx.Request.URL.Query().Get("url")
+	stream := streams.Get(url)
+	if stream == nil {
+		return
+	}
+
+	cons := new(mse.Consumer)
+	cons.UserAgent = ctx.Request.UserAgent()
+	cons.RemoteAddr = ctx.Request.RemoteAddr
+	cons.Listen(func(msg interface{}) {
+		switch msg.(type) {
+		case *streamer.Message, []byte:
+			ctx.Write(msg)
+		}
+	})
+	if err := stream.AddConsumer(cons); err != nil {
+		log.Warn().Err(err).Msg("[api.mse] Add consumer")
+		ctx.Error(err)
+		return
+	}
+
+	ctx.OnClose(func() {
+		stream.RemoveConsumer(cons)
+	})
+
+	cons.Init()
+}