fix?

main.go

@@ -1,11 +1,13 @@
 package main
 
 import (
+	"context"
 	"flag"
 	"fmt"
 	"net/http"
 	"os"
 	"os/signal"
+	"time"
 
 	"github.com/gorilla/sessions"
 	"github.com/nareix/joy4/format"
@@ -20,11 +22,12 @@ var (
 	sKey       string
 	stats      = newStreamStats()
 	sAdminPass string
+	confFile   string
 )
 
 func setupSettings() error {
 	var err error
-	settings, err = LoadSettings("settings.json")
+	settings, err = LoadSettings(confFile)
 	if err != nil {
 		return fmt.Errorf("Unable to load settings: %s", err)
 	}
@@ -53,6 +56,7 @@ func main() {
 	flag.StringVar(&sKey, "k", "", "Stream key, to protect your stream")
 	flag.StringVar(&sAdminPass, "a", "", "Set admin password.  Overrides configuration in settings.json.  This will not write the password to settings.json.")
 	flag.BoolVar(&pullEmotes, "e", false, "Pull emotes")
+	flag.StringVar(&confFile, "f", "./settings.json", "URI of the conf file")
 	flag.Parse()
 
 	format.RegisterAll()
@@ -109,47 +113,64 @@ func main() {
 	common.LogInfoln("RoomAccess: ", settings.RoomAccess)
 	common.LogInfoln("RoomAccessPin: ", settings.RoomAccessPin)
 
-	go startServer()
-	go startRmtpServer()
-
-	<-exit
-}
-
-func startRmtpServer() {
-	server := &rtmp.Server{
+	rtmpServer := &rtmp.Server{
 		HandlePlay:    handlePlay,
 		HandlePublish: handlePublish,
 		Addr:          rtmpAddr,
 	}
-	err := server.ListenAndServe()
-	if err != nil {
-		// If the server cannot start, don't pretend we can continue.
-		panic("Error trying to start rtmp server: " + err.Error())
+
+	router := http.NewServeMux()
+
+	router.HandleFunc("/ws", wsHandler) // Chat websocket
+	router.HandleFunc("/static/js/", wsStaticFiles)
+	router.HandleFunc("/static/css/", wsStaticFiles)
+	router.HandleFunc("/static/img/", wsImages)
+	router.HandleFunc("/static/main.wasm", wsWasmFile)
+	router.HandleFunc("/emotes/", wsEmotes)
+	router.HandleFunc("/favicon.ico", wsStaticFiles)
+	router.HandleFunc("/chat", handleIndexTemplate)
+	router.HandleFunc("/video", handleIndexTemplate)
+	router.HandleFunc("/help", handleHelpTemplate)
+	router.HandleFunc("/emotes", handleEmoteTemplate)
+
+	router.HandleFunc("/live", handleLive)
+	router.HandleFunc("/", handleDefault)
+
+	httpServer := &http.Server{
+		Addr:    addr,
+		Handler: router,
 	}
-}
 
-func startServer() {
-	// Chat websocket
-	http.HandleFunc("/ws", wsHandler)
-	http.HandleFunc("/static/js/", wsStaticFiles)
-	http.HandleFunc("/static/css/", wsStaticFiles)
-	http.HandleFunc("/static/img/", wsImages)
-	http.HandleFunc("/static/main.wasm", wsWasmFile)
-	http.HandleFunc("/emotes/", wsEmotes)
-	http.HandleFunc("/favicon.ico", wsStaticFiles)
-	http.HandleFunc("/chat", handleIndexTemplate)
-	http.HandleFunc("/video", handleIndexTemplate)
-	http.HandleFunc("/help", handleHelpTemplate)
-	http.HandleFunc("/pin", handlePin)
-	http.HandleFunc("/emotes", handleEmoteTemplate)
-
-	http.HandleFunc("/", handleDefault)
-
-	err := http.ListenAndServe(addr, nil)
-	if err != nil {
-		// If the server cannot start, don't pretend we can continue.
-		panic("Error trying to start chat/http server: " + err.Error())
+	// RTMP Server
+	go func() {
+		err := rtmpServer.ListenAndServe()
+		if err != nil {
+			// If the server cannot start, don't pretend we can continue.
+			panic("Error trying to start rtmp server: " + err.Error())
+		}
+	}()
+
+	// HTTP Server
+	go func() {
+		err := httpServer.ListenAndServe()
+		if err != nil && err != http.ErrServerClosed {
+			// If the server cannot start, don't pretend we can continue.
+			panic("Error trying to start chat/http server: " + err.Error())
+		}
+	}()
+
+	<-exit
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	if err := httpServer.Shutdown(ctx); err != nil && err != http.ErrServerClosed {
+		panic("Gracefull HTTP server shutdown failed: " + err.Error())
 	}
+
+	// I don't think the RTMP server can be shutdown cleanly.  Apparently the author
+	// of joy4 want's everyone to use joy5, but that one doesn't seem to allow clean
+	// shutdowns either? Idk, the documentation on joy4 and joy5 are non-existent.
 }
 
 func handleInterrupt(exit chan bool) {