chaos
81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
package common
|
|
|
|
import (
|
|
"embed"
|
|
"io/fs"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/gin-contrib/static"
|
|
)
|
|
|
|
// Credit: https://github.com/gin-contrib/static/issues/19
|
|
|
|
type embedFileSystem struct {
|
|
http.FileSystem
|
|
}
|
|
|
|
func (e *embedFileSystem) Exists(prefix string, path string) bool {
|
|
// gin-contrib/static passes the raw URL path (e.g. "/image-gen/assets/x.js")
|
|
// together with the URL prefix we registered (e.g. "/image-gen"). The
|
|
// underlying fs.Sub FS only knows about the sub-tree (no prefix), so we
|
|
// must strip the prefix before asking it whether the file exists. An
|
|
// empty prefix means "served at /" — nothing to strip.
|
|
p := strings.TrimPrefix(path, prefix)
|
|
if p == path {
|
|
// prefix didn't match — definitely not in this FS
|
|
return false
|
|
}
|
|
_, err := e.Open(p)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (e *embedFileSystem) Open(name string) (http.File, error) {
|
|
if name == "/" {
|
|
// This will make sure the index page goes to NoRouter handler,
|
|
// which will use the replaced index bytes with analytic codes.
|
|
return nil, os.ErrNotExist
|
|
}
|
|
return e.FileSystem.Open(name)
|
|
}
|
|
|
|
func EmbedFolder(fsEmbed embed.FS, targetPath string) static.ServeFileSystem {
|
|
efs, err := fs.Sub(fsEmbed, targetPath)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return &embedFileSystem{
|
|
FileSystem: http.FS(efs),
|
|
}
|
|
}
|
|
|
|
// themeAwareFileSystem delegates to the appropriate embedded FS based on
|
|
// the current theme (via GetTheme). This enables runtime theme switching
|
|
// without restarting the server.
|
|
type themeAwareFileSystem struct {
|
|
defaultFS static.ServeFileSystem
|
|
classicFS static.ServeFileSystem
|
|
}
|
|
|
|
func (t *themeAwareFileSystem) Exists(prefix string, path string) bool {
|
|
if GetTheme() == "classic" {
|
|
return t.classicFS.Exists(prefix, path)
|
|
}
|
|
return t.defaultFS.Exists(prefix, path)
|
|
}
|
|
|
|
func (t *themeAwareFileSystem) Open(name string) (http.File, error) {
|
|
if GetTheme() == "classic" {
|
|
return t.classicFS.Open(name)
|
|
}
|
|
return t.defaultFS.Open(name)
|
|
}
|
|
|
|
func NewThemeAwareFS(defaultFS, classicFS static.ServeFileSystem) static.ServeFileSystem {
|
|
return &themeAwareFileSystem{defaultFS: defaultFS, classicFS: classicFS}
|
|
}
|