package web import ( "html/template" "io/ioutil" "log" "net/http" "os" "regexp" "strings" "github.com/markbates/pkger" "github.com/pion/webrtc/v3" ) // Options holds web package configuration type Options struct { Favicon string Hostname string ListenAddress string Name string OneStreamPerDomain bool SRTServerPort string STUNServers []string ViewersCounterRefreshPeriod int WidgetURL string } var ( cfg *Options // WebRTC session description channels remoteSdpChan chan struct { StreamID string RemoteDescription webrtc.SessionDescription } localSdpChan chan webrtc.SessionDescription // Preload templates templates *template.Template // Precompile regex validPath = regexp.MustCompile("^\\/[a-z0-9_-]*\\/?$") ) // Load templates with pkger // templates will be packed in the compiled binary func loadTemplates() error { templates = template.New("") return pkger.Walk("/web/template", func(path string, info os.FileInfo, err error) error { if err != nil { return err } // Skip non-templates if info.IsDir() || !strings.HasSuffix(path, ".html") { return nil } // Open file with pkger f, err := pkger.Open(path) if err != nil { return err } // Read and parse template temp, err := ioutil.ReadAll(f) if err != nil { return err } templates, err = templates.Parse(string(temp)) return err }) } // Serve HTTP server func Serve(rSdpChan chan struct { StreamID string RemoteDescription webrtc.SessionDescription }, lSdpChan chan webrtc.SessionDescription, c *Options) { remoteSdpChan = rSdpChan localSdpChan = lSdpChan cfg = c // Load templates if err := loadTemplates(); err != nil { log.Fatalln("Failed to load templates:", err) } // Set up HTTP router and server mux := http.NewServeMux() mux.HandleFunc("/", viewerHandler) mux.Handle("/static/", staticHandler()) mux.HandleFunc("/_stats/", statisticsHandler) log.Printf("HTTP server listening on %s", cfg.ListenAddress) log.Fatal(http.ListenAndServe(cfg.ListenAddress, mux)) }