Add minimal web interface

2021-08-23 02:45:44 +02:00
parent 72c84dff1e
commit 1d6e4270ba
8 changed files with 325 additions and 0 deletions
--- a/server/templates/index.html
+++ b/server/templates/index.html
@@ -0,0 +1,8 @@
+<DOCTYPE html>
+<html>
+    <ol>
+    {{ range .Streams }}
+    <li>{{ .URL }}</li>
+    {{ end }}
+    </ol>
+</html>
--- a/server/web.go
+++ b/server/web.go
@@ -0,0 +1,68 @@
+package server
+
+import (
+	"context"
+	"fmt"
+	"html/template"
+	"net/http"
+	"time"
+
+	"go.uber.org/zap"
+)
+
+type WebServer struct {
+	Logger     *zap.SugaredLogger
+	ctx        context.Context
+	rtmpServer *RTMPServer
+	httpServer *http.Server
+}
+
+func NewWebServer(ctx context.Context, rs *RTMPServer) *WebServer {
+	return &WebServer{
+		ctx:        ctx,
+		rtmpServer: rs,
+		Logger:     zap.NewNop().Sugar(),
+	}
+}
+
+func (ws *WebServer) Serve() error {
+	ws.httpServer = &http.Server{
+		Addr:    ":8077",
+		Handler: http.HandlerFunc(ws.IndexHandler),
+	}
+
+	go func() {
+		<-ws.ctx.Done()
+		shutdownCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+		defer cancel()
+		_ = ws.httpServer.Shutdown(shutdownCtx)
+	}()
+
+	return ws.httpServer.ListenAndServe()
+}
+
+func (ws *WebServer) IndexHandler(w http.ResponseWriter, r *http.Request) {
+	data := struct {
+		Streams []struct {
+			Name string
+			URL  template.URL
+		}
+	}{}
+
+	for _, s := range ws.rtmpServer.streams {
+		stream := struct {
+			Name string
+			URL  template.URL
+		}{
+			Name: s.Name,
+			URL:  template.URL(fmt.Sprintf("rtmp://localhost:5566/view/%s", s.Name)),
+		}
+		data.Streams = append(data.Streams, stream)
+	}
+
+	tmpl := template.Must(template.ParseFiles("server/templates/index.html"))
+
+	w.Header().Add("Content-Type", "text/html")
+
+	tmpl.Execute(w, data)
+}