103 lines
2.8 KiB
Go
103 lines
2.8 KiB
Go
package controllers
|
|
|
|
import (
|
|
"net/http"
|
|
"path"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/owncast/owncast/config"
|
|
"github.com/owncast/owncast/core"
|
|
"github.com/owncast/owncast/core/data"
|
|
"github.com/owncast/owncast/models"
|
|
"github.com/owncast/owncast/router/middleware"
|
|
"github.com/owncast/owncast/utils"
|
|
log "github.com/sirupsen/logrus"
|
|
cache "github.com/victorspringer/http-cache"
|
|
"github.com/victorspringer/http-cache/adapter/memory"
|
|
)
|
|
|
|
var (
|
|
hlsCacheAdapter *cache.Adapter
|
|
hlsResponseCache *cache.Client
|
|
)
|
|
|
|
type FileServerHandler struct {
|
|
HLSPath string
|
|
}
|
|
|
|
func (fsh *FileServerHandler) ServeHTTP(rw http.ResponseWriter, r *http.Request) {
|
|
http.ServeFile(rw, r, fsh.HLSPath)
|
|
}
|
|
|
|
// HandleHLSRequest will manage all requests to HLS content.
|
|
func HandleHLSRequest(w http.ResponseWriter, r *http.Request) {
|
|
// Sanity check to limit requests to HLS file types.
|
|
if filepath.Ext(r.URL.Path) != ".m3u8" && filepath.Ext(r.URL.Path) != ".ts" {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
if hlsCacheAdapter == nil {
|
|
ca, err := memory.NewAdapter(
|
|
memory.AdapterWithAlgorithm(memory.LRU),
|
|
memory.AdapterWithCapacity(50),
|
|
memory.AdapterWithStorageCapacity(104_857_600),
|
|
)
|
|
hlsCacheAdapter = &ca
|
|
if err != nil {
|
|
log.Warn("unable to create web cache", err)
|
|
}
|
|
}
|
|
|
|
// Since HLS segments cannot be changed once they're rendered, we can cache
|
|
// individual segments for a long time.
|
|
if hlsResponseCache == nil {
|
|
rc, err := cache.NewClient(
|
|
cache.ClientWithAdapter(*hlsCacheAdapter),
|
|
cache.ClientWithTTL(30*time.Second),
|
|
cache.ClientWithExpiresHeader(),
|
|
)
|
|
hlsResponseCache = rc
|
|
if err != nil {
|
|
log.Warn("unable to create web cache client", err)
|
|
}
|
|
}
|
|
|
|
requestedPath := r.URL.Path
|
|
relativePath := strings.Replace(requestedPath, "/hls/", "", 1)
|
|
fullPath := filepath.Join(config.HLSStoragePath, relativePath)
|
|
|
|
// If using external storage then only allow requests for the
|
|
// master playlist at stream.m3u8, no variants or segments.
|
|
if data.GetS3Config().Enabled && relativePath != "stream.m3u8" {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
// Handle playlists
|
|
if path.Ext(r.URL.Path) == ".m3u8" {
|
|
// Playlists should never be cached.
|
|
middleware.DisableCache(w)
|
|
|
|
// Force the correct content type
|
|
w.Header().Set("Content-Type", "application/x-mpegURL")
|
|
|
|
// Use this as an opportunity to mark this viewer as active.
|
|
viewer := models.GenerateViewerFromRequest(r)
|
|
core.SetViewerActive(&viewer)
|
|
} else {
|
|
cacheTime := utils.GetCacheDurationSecondsForPath(relativePath)
|
|
w.Header().Set("Cache-Control", "public, max-age="+strconv.Itoa(cacheTime))
|
|
|
|
fileServer := &FileServerHandler{HLSPath: fullPath}
|
|
hlsResponseCache.Middleware(fileServer).ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
middleware.EnableCors(w)
|
|
http.ServeFile(w, r, fullPath)
|
|
}
|