owncast/controllers/admin/index.go

package admin

import (
	"bytes"
	"net/http"
	"os"
	"path/filepath"
	"strings"

	"github.com/owncast/owncast/router/middleware"
	"github.com/owncast/owncast/static"
	log "github.com/sirupsen/logrus"
)

// ServeAdmin will return admin web assets.
func ServeAdmin(w http.ResponseWriter, r *http.Request) {
	adminFiles := static.GetAdmin()

	// Determine if the requested path is a directory.
	// If so, append index.html to the request.
	path := strings.TrimPrefix(r.URL.Path, "/")
	if strings.HasSuffix(path, "/") {
		path = filepath.Join(path, "index.html")
	}

	f, err := adminFiles.Open(path)
	if os.IsNotExist(err) {
		w.WriteHeader(http.StatusNotFound)
		return
	}

	info, err := f.Stat()

	if os.IsNotExist(err) {
		w.WriteHeader(http.StatusNotFound)
		return
	}

	// Set a cache control max-age header
	middleware.SetCachingHeaders(w, r)

	d, err := adminFiles.ReadFile(path)
	if err != nil {
		log.Errorln(err)
		w.WriteHeader(http.StatusInternalServerError)
		return
	}

	http.ServeContent(w, r, info.Name(), info.ModTime(), bytes.NewReader(d))
}