webrtc: move JavaScript reader and publisher into portable classes (#…

…3921)
bluenviron · Oct 28, 2024 · 20ad88f · 20ad88f
1 parent 51b21c8
commit 20ad88f
Show file tree

Hide file tree

Showing 5 changed files with 1,157 additions and 1,065 deletions.
diff --git a/internal/servers/webrtc/http_server.go b/internal/servers/webrtc/http_server.go
@@ -26,9 +26,15 @@ import (
 //go:embed publish_index.html
 var publishIndex []byte
 
+//go:embed publisher.js
+var publisherJS []byte
+
 //go:embed read_index.html
 var readIndex []byte
 
+//go:embed reader.js
+var readerJS []byte
+
 var (
 	reWHIPWHEPNoID   = regexp.MustCompile("^/(.+?)/(whip|whep)$")
 	reWHIPWHEPWithID = regexp.MustCompile("^/(.+?)/(whip|whep)/(.+?)$")
@@ -323,6 +329,22 @@ func (s *httpServer) middlewareOrigin(ctx *gin.Context) {
 }
 
 func (s *httpServer) onRequest(ctx *gin.Context) {
+	if strings.HasSuffix(ctx.Request.URL.Path, "/publisher.js") {
+		ctx.Header("Cache-Control", "max-age=3600")
+		ctx.Header("Content-Type", "application/javascript")
+		ctx.Writer.WriteHeader(http.StatusOK)
+		ctx.Writer.Write(publisherJS)
+		return
+	}
+
+	if strings.HasSuffix(ctx.Request.URL.Path, "/reader.js") {
+		ctx.Header("Cache-Control", "max-age=3600")
+		ctx.Header("Content-Type", "application/javascript")
+		ctx.Writer.WriteHeader(http.StatusOK)
+		ctx.Writer.Write(readerJS)
+		return
+	}
+
 	// WHIP/WHEP, outside session
 	if m := reWHIPWHEPNoID.FindStringSubmatch(ctx.Request.URL.Path); m != nil {
 		switch ctx.Request.Method {