support preloading streams

internal/streams/preload.go

@@ -0,0 +1,30 @@
+package streams
+
+import (
+	"net/url"
+	"strings"
+
+	"github.com/AlexxIT/go2rtc/pkg/preload"
+)
+
+func (s *Stream) Preload(query url.Values) error {
+	cons := preload.NewPreload(query)
+
+	if err := s.AddConsumer(cons); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func Preload(src string) {
+	name, rawQuery, _ := strings.Cut(src, "#")
+	query := ParseQuery(rawQuery)
+
+	if stream := Get(name); stream != nil {
+		if err := stream.Preload(query); err != nil {
+			log.Error().Err(err).Caller().Send()
+		}
+		return
+	}
+}

internal/streams/streams.go

@@ -16,6 +16,7 @@ func Init() {
 	var cfg struct {
 		Streams map[string]any `yaml:"streams"`
 		Publish map[string]any `yaml:"publish"`
+		Preload []string       `yaml:"preload"`
 	}
 
 	app.LoadConfig(&cfg)
@@ -29,14 +30,22 @@ func Init() {
 	api.HandleFunc("api/streams", apiStreams)
 	api.HandleFunc("api/streams.dot", apiStreamsDOT)
 
-	if cfg.Publish == nil {
+	if cfg.Publish == nil && cfg.Preload == nil {
 		return
 	}
 
 	time.AfterFunc(time.Second, func() {
-		for name, dst := range cfg.Publish {
-			if stream := Get(name); stream != nil {
-				Publish(stream, dst)
+		if cfg.Publish != nil {
+			for name, dst := range cfg.Publish {
+				if stream := Get(name); stream != nil {
+					Publish(stream, dst)
+				}
+			}
+		}
+
+		if cfg.Preload != nil {
+			for _, src := range cfg.Preload {
+				Preload(src)
 			}
 		}
 	})