122 lines
2.7 KiB
Go
122 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"path"
|
|
"strconv"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// Build-time injected values
|
|
var GitCommit string = "unknown"
|
|
var BuildVersion string = "0.0.0"
|
|
var BuildType string = "localdev"
|
|
|
|
var storage ChunkStorage
|
|
var configuration = getConfig()
|
|
var server *Server
|
|
var stats *Stats
|
|
|
|
var usingExternalStorage = false
|
|
|
|
func main() {
|
|
// logrus.SetReportCaller(true)
|
|
log.StandardLogger().Printf("Owncast v%s/%s (%s)", BuildVersion, BuildType, GitCommit)
|
|
|
|
checkConfig(configuration)
|
|
resetDirectories(configuration)
|
|
|
|
stats = getSavedStats()
|
|
stats.Setup()
|
|
|
|
if configuration.IPFS.Enabled {
|
|
storage = &IPFSStorage{}
|
|
usingExternalStorage = true
|
|
} else if configuration.S3.Enabled {
|
|
storage = &S3Storage{}
|
|
usingExternalStorage = true
|
|
}
|
|
|
|
if usingExternalStorage {
|
|
storage.Setup(configuration)
|
|
go monitorVideoContent(configuration.PrivateHLSPath, configuration, storage)
|
|
}
|
|
|
|
createInitialOfflineState()
|
|
go startRTMPService()
|
|
|
|
startWebServer()
|
|
}
|
|
|
|
func startWebServer() {
|
|
// websocket server
|
|
server = NewServer("/entry")
|
|
go server.Listen()
|
|
|
|
// static files
|
|
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
enableCors(&w)
|
|
http.ServeFile(w, r, path.Join("webroot", r.URL.Path))
|
|
|
|
if path.Ext(r.URL.Path) == ".m3u8" {
|
|
clientID := getClientIDFromRequest(r)
|
|
stats.SetClientActive(clientID)
|
|
disableCache(&w)
|
|
}
|
|
})
|
|
|
|
http.HandleFunc("/status", getStatus)
|
|
|
|
log.Printf("Starting public web server on port %d", configuration.WebServerPort)
|
|
|
|
log.Fatal(http.ListenAndServe(":"+strconv.Itoa(configuration.WebServerPort), nil))
|
|
}
|
|
|
|
func enableCors(w *http.ResponseWriter) {
|
|
(*w).Header().Set("Access-Control-Allow-Origin", "*")
|
|
}
|
|
|
|
func disableCache(w *http.ResponseWriter) {
|
|
(*w).Header().Set("Cache-Control", "no-cache, no-store, must-revalidate")
|
|
(*w).Header().Set("Expires", "0")
|
|
}
|
|
|
|
func getStatus(w http.ResponseWriter, r *http.Request) {
|
|
enableCors(&w)
|
|
|
|
status := Status{
|
|
Online: stats.IsStreamConnected(),
|
|
ViewerCount: stats.GetViewerCount(),
|
|
OverallMaxViewerCount: stats.GetOverallMaxViewerCount(),
|
|
SessionMaxViewerCount: stats.GetSessionMaxViewerCount(),
|
|
}
|
|
json.NewEncoder(w).Encode(status)
|
|
}
|
|
|
|
func streamConnected() {
|
|
stats.StreamConnected()
|
|
|
|
chunkPath := configuration.PublicHLSPath
|
|
if usingExternalStorage {
|
|
chunkPath = configuration.PrivateHLSPath
|
|
}
|
|
startThumbnailGenerator(chunkPath)
|
|
}
|
|
|
|
func streamDisconnected() {
|
|
stats.StreamDisconnected()
|
|
if configuration.EnableOfflineImage {
|
|
showStreamOfflineState(configuration)
|
|
}
|
|
}
|
|
|
|
func viewerAdded(clientID string) {
|
|
stats.SetClientActive(clientID)
|
|
}
|
|
|
|
func viewerRemoved(clientID string) {
|
|
stats.ViewerDisconnected(clientID)
|
|
}
|